fix: Fix models api base url

Amery2010 · Oct 15, 2024 · 42bd03d · 42bd03d
1 parent c6ab6d6
commit 42bd03d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/app/api/models/route.ts b/app/api/models/route.ts
@@ -23,7 +23,8 @@ export async function GET(req: NextRequest) {
   }
 
   try {
-    const response = await fetch(`${geminiApiBaseUrl}/v1beta/models?key=${geminiApiKey}`)
+    const apiBaseUrl = geminiApiBaseUrl || 'https://generativelanguage.googleapis.com'
+    const response = await fetch(`${apiBaseUrl}/v1beta/models?key=${geminiApiKey}`)
     return new NextResponse(response.body)
   } catch (error) {
     if (error instanceof Error) {