Merge pull request #315 from yz778/gemini-models

fix: alternate handler for retrieving Gemini models

Author: n4ze3m

src/libs/openai.ts

@@ -24,6 +24,20 @@ export const getAllOpenAIModels = async (baseUrl: string, apiKey?: string) => {
 
     clearTimeout(timeoutId)
 
+    // if Google API fails to return models, try another approach
+    if (res.status === 401 && res.url == 'https://generativelanguage.googleapis.com/v1beta/openai/models') {
+      const urlGoogle = `https://generativelanguage.googleapis.com/v1beta/models?key=${apiKey}`
+      const resGoogle = await fetch(urlGoogle, {
+        signal: controller.signal
+      })
+
+      const data = await resGoogle.json()
+      return data.models.map(model => ({
+        id: model.name.replace(/^models\//, ""),
+        name: model.name.replace(/^models\//, ""),
+      })) as Model[]
+    }
+
     if (!res.ok) {
       return []
     }