Update index.ts

Signed-off-by: InternetBugs <77239633+InternetBugs@users.noreply.github.com>
InternetBugs · Apr 27, 2024 · 01a5cc9 · 01a5cc9
1 parent 5a33729
commit 01a5cc9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/service/src/chatgpt/index.ts b/service/src/chatgpt/index.ts
@@ -91,6 +91,10 @@ export async function initApi(key: KeyConfig, chatModel: string, maxContextCount
 		else if (model.toLowerCase().includes('claude')) {
       options.maxModelTokens = 2000000
       options.maxResponseTokens = 4096
+    }
+		else if (model.toLowerCase().includes('qwen-1.5')) {
+      options.maxModelTokens = 32000
+      options.maxResponseTokens = 8000
     }
     else {
       options.maxModelTokens = 4096