fix: fix gemini model not found (#2157)

This commit is contained in:
Yingjie He
2025-04-11 10:41:48 -07:00
committed by GitHub
parent 0b800e05b7
commit 9bd27cebcb
3 changed files with 24 additions and 21 deletions

View File

@@ -14,6 +14,7 @@ export const google_models = [
'gemini-2.0-flash-lite-preview-02-05',
'gemini-2.0-flash-thinking-exp-01-21',
'gemini-2.0-pro-exp-02-05',
'gemini-2.5-pro-exp-03-25',
];
export const groq_models = ['llama-3.3-70b-versatile'];
@@ -49,7 +50,7 @@ export const default_models = {
};
export function getDefaultModel(key: string): string | undefined {
return default_models[key] || undefined;
return default_models[key as keyof typeof default_models] || undefined;
}
export const short_list = ['gpt-4o', 'claude-3-5-sonnet-latest'];