fix: 过滤cohere模型列表非对话模型

TBXark · Nov 12, 2024 · 568c4ee · 568c4ee
1 parent be5574a
commit 568c4ee
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 9 deletions.
diff --git a/dist/buildinfo.json b/dist/buildinfo.json
diff --git a/dist/index.js b/dist/index.js
diff --git a/src/agent/cohere.ts b/src/agent/cohere.ts
@@ -49,7 +49,7 @@ export class Cohere implements ChatAgent {
             const data = await fetch(url, {
                 headers: { Authorization: `Bearer ${context.COHERE_API_KEY}` },
             }).then(res => res.json());
-            return data.models?.map((model: any) => model.name) || [];
+            return data.models?.filter((model: any) => model.endpoints?.includes('chat')).map((model: any) => model.name) || [];
         });
     };
 }
diff --git a/src/config/config.ts b/src/config/config.ts
@@ -149,7 +149,7 @@ export class WorkersConfig {
     // Text-to-Image Model
     WORKERS_IMAGE_MODEL = '@cf/stabilityai/stable-diffusion-xl-base-1.0';
     // Workers Chat Models List
-    WORKERS_CHAT_MODELS_LIST = `["@cf/google/gemma-7b-it-lora", "@cf/meta/llama-2-7b-chat-fp16", "@cf/qwen/qwen1.5-14b-chat-awq"]`;
+    WORKERS_CHAT_MODELS_LIST = `["@cf/google/gemma-7b-it-lora", "@cf/meta/llama-2-7b-chat-fp16", "@cf/qwen/qwen1.5-14b-chat-awq", "@cf/mistral/mistral-7b-instruct-v0.1"]`;
 }
 
 // -- Gemini 配置 --

diff --git a/src/config/version.ts b/src/config/version.ts
@@ -1,2 +1,2 @@
-export const BUILD_TIMESTAMP = 1731381631;
-export const BUILD_VERSION = '41e8df5';
+export const BUILD_TIMESTAMP = 1731382216;
+export const BUILD_VERSION = 'be5574a';