Merge pull request #315 from yz778/gemini-models
fix: alternate handler for retrieving Gemini models
This commit is contained in:
commit
2d1e465582
@ -24,6 +24,20 @@ export const getAllOpenAIModels = async (baseUrl: string, apiKey?: string) => {
|
|||||||
|
|
||||||
clearTimeout(timeoutId)
|
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) {
|
if (!res.ok) {
|
||||||
return []
|
return []
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user