fix model select ollama bug

jollytoad · Jan 20, 2024 · b4ec5a0 · b4ec5a0
1 parent 4ff64b7
commit b4ec5a0
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/components/models/model-select.tsx b/components/models/model-select.tsx
@@ -93,10 +93,12 @@ export const ModelSelect: FC<ModelSelectProps> = ({
     model => model.modelId === selectedModelId
   )
 
-  if (!SELECTED_MODEL || !profile) return null
+  if (!profile) return null
 
   const usingLocalModels = availableLocalModels.length > 0
 
+  console.log("test")
+
   return (
     <DropdownMenu
       open={isOpen}
@@ -119,22 +121,22 @@ export const ModelSelect: FC<ModelSelectProps> = ({
               <WithTooltip
                 display={
                   <div>
-                    Save {SELECTED_MODEL.provider} API key in profile settings
+                    Save {SELECTED_MODEL?.provider} API key in profile settings
                     to unlock.
                   </div>
                 }
                 trigger={<IconLock className="mr-2" size={26} />}
               />
             ) : (
               <ModelIcon
-                modelId={SELECTED_MODEL.modelId as LLMID}
+                modelId={SELECTED_MODEL?.modelId as LLMID}
                 width={26}
                 height={26}
               />
             )}
 
             <div className="ml-2 flex items-center">
-              {SELECTED_MODEL.modelName}
+              {SELECTED_MODEL?.modelName}
             </div>
           </div>