修复非预期模型名处理

LLM-Red-Team · Dec 31, 2024 · 2ce738b · 2ce738b
1 parent 719e3b6
commit 2ce738b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
@@ -191,7 +191,7 @@ async function createCompletion(
     // 如果引用对话ID不正确则重置引用
     if (!/[0-9a-zA-Z]{24}/.test(refConvId)) refConvId = "";
 
-    let assistantId = /^[a-z0-9]{24,}$/.test(model) ? model : undefined;
+    let assistantId = /^[a-z0-9]{24,}$/.test(model) ? model : DEFAULT_ASSISTANT_ID;
 
     if(model.indexOf('think') != -1 || model.indexOf('zero') != -1) {
       assistantId = ZERO_ASSISTANT_ID;
@@ -302,7 +302,7 @@ async function createCompletionStream(
     // 如果引用对话ID不正确则重置引用
     if (!/[0-9a-zA-Z]{24}/.test(refConvId)) refConvId = "";
 
-    let assistantId = /^[a-z0-9]{24,}$/.test(model) ? model : undefined;
+    let assistantId = /^[a-z0-9]{24,}$/.test(model) ? model : DEFAULT_ASSISTANT_ID;
 
     if(model.indexOf('think') != -1 || model.indexOf('zero') != -1) {
       assistantId = ZERO_ASSISTANT_ID;