iod/page-assist
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #315 from yz778/gemini-models

Browse Source 
fix: alternate handler for retrieving Gemini models
This commit is contained in:
Muhammed Nazeem 2025-02-01 10:19:20 +05:30 committed by GitHub
commit 2d1e465582
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/libs/openai.ts
View File

@ -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 []
}