model fix on cloud version

OpenHealthForAll · Feb 19, 2025 · c0ce9cf · c0ce9cf
1 parent a4da67b
commit c0ce9cf
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/src/app/api/llm-providers/[id]/models/route.ts b/src/app/api/llm-providers/[id]/models/route.ts
@@ -38,8 +38,10 @@ export async function GET(
         for await (const models of modelsPage.iterPages()) {
             const modelList = models.data;
             results.push(...modelList.filter((model) => {
-                return !(currentDeploymentEnv === 'cloud' && model.id.startsWith('ft:'));
-
+                if (currentDeploymentEnv === 'cloud') {
+                    return ['gpt-4o', 'o3-mini'].includes(model.id);
+                }
+                return !model.id.startsWith('ft:');
             }).map(model => ({id: model.id, name: model.id})))
         }
         return NextResponse.json<LLMProviderModelListResponse>({
@@ -52,7 +54,12 @@ export async function GET(
         const modelsPage = await anthropic.models.list();
         for await (const models of modelsPage.iterPages()) {
             const modelList = models.data;
-            results.push(...modelList.map(
+            results.push(...modelList.filter((model) => {
+                if (currentDeploymentEnv === 'cloud') {
+                    return model.id.startsWith('claude-3-5');
+                }
+                return true;
+            }).map(
                 model => ({id: model.id, name: model.id})
             ))
         }
@@ -67,7 +74,12 @@ export async function GET(
         const response = await fetch(url.toString())
         const {models} = await response.json()
         return NextResponse.json<LLMProviderModelListResponse>({
-            llmProviderModels: models.map(
+            llmProviderModels: models.filter(({displayName}: { displayName: string }) => {
+                if (currentDeploymentEnv === 'cloud') {
+                    return displayName === 'Gemini 2.0 Flash';
+                }
+                return true;
+            }).map(
                 ({name, displayName}: { name: string, displayName: string }) => ({
                     id: name,
                     name: displayName

diff --git a/src/lib/health-data/parser/vision/openai.ts b/src/lib/health-data/parser/vision/openai.ts
@@ -19,12 +19,17 @@ export class OpenAIVisionParser extends BaseVisionParser {
     }
 
     async models(): Promise<VisionParserModel[]> {
+        if (currentDeploymentEnv === 'cloud') {
+            return [
+                {id: 'gpt-4o', name: 'gpt-4o'},
+            ];
+        }
         return [
             {id: 'gpt-4o-mini', name: 'gpt-4o-mini'},
             {id: 'gpt-4o', name: 'gpt-4o'},
             {id: 'o1', name: 'o1'},
             {id: 'o1-mini', name: 'o1-mini'},
-        ]
+        ];
     }
 
     async parse(options: VisionParseOptions) {