Ver código fonte

tweak: setting chat_template_args in transform

Aiden Cline 2 meses atrás
pai
commit
342595e0f7
1 arquivos alterados com 4 adições e 1 exclusões
  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 }
       result["chat_template_args"] = { enable_thinking: true }
     }
     }