Просмотр исходного кода

tweak: setting chat_template_args in transform

Aiden Cline 2 месяцев назад
Родитель
Сommit
342595e0f7
1 измененных файлов с 4 добавлено и 1 удалено
  1. 4 1
      packages/opencode/src/provider/transform.ts

+ 4 - 1
packages/opencode/src/provider/transform.ts

@@ -226,7 +226,10 @@ export namespace ProviderTransform {
       }
     }
 
-    if (model.providerID === "baseten") {
+    if (
+      model.providerID === "baseten" ||
+      (model.providerID === "opencode" && ["kimi-k2-thinking", "glm-4.6"].includes(model.api.id))
+    ) {
       result["chat_template_args"] = { enable_thinking: true }
     }