fixbug: gemini 模型参数错误

Bistutu · May 25, 2024 · 94c79a8 · 94c79a8
1 parent d39b206
commit 94c79a8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/entrypoints/service/gemini.ts b/entrypoints/service/gemini.ts
@@ -10,7 +10,7 @@ async function gemini(config: Config, message: any) {
 
     // 判断是否使用代理
     let url: string = config.proxy[config.service] ?
-        config.proxy[config.service] : `https://generativelanguage.googleapis.com/v1beta/models/${model}:generateContent?key=${config.token[services.gemini]}`;
+        config.proxy[config.service] : `https://generativelanguage.googleapis.com/v1beta/models/${model}:generateContent?key=${config.token[config.service]}`;
 
     const resp = await fetch(url, {
         method: method.POST,

diff --git a/entrypoints/utils/option.ts b/entrypoints/utils/option.ts
@@ -49,7 +49,7 @@ export const services = {
 export const customModelString = "自定义模型"
 export const models = new Map<string, Array<string>>([
     [services.openai, ["gpt-3.5-turbo", "gpt-4o", "gpt-4", "gpt-4-turbo", customModelString]],
-    [services.gemini, ["gemini-pro", "gemini-1.5-pro", "gemini-flash-1.5", customModelString]],
+    [services.gemini, ["gemini-pro", "gemini-1.5-pro", "gemini-1.5-flash", customModelString]],
     [services.yiyan, ["ERNIE-Bot 4.0", "ERNIE-Bot","ERNIE-Speed-8K"]],  // 因文心一言模式不同，暂不支持自定义模型（还需根据model获取最终的url请求参数）
     [services.tongyi, ["qwen-turbo", "qwen-plus", "qwen-max", "qwen-max-longcontext", customModelString]],
     [services.zhipu, ["glm-4", "glm-4v", "glm-3-turbo", customModelString]],