on main help sign in
chore: improve zai tool calls (#1551)
commit54204dab0c6c245e32557ce9227345249d11e241 authorKujtim Hoxha <kujtimii.h@gmail.com> dateThu 4 Dec 2025, 16:36 (1 year ago) parent5c96ab4
1 file changed+6
internal/agent/coordinator.go+6
internal/agent/coordinator.go+6
@@ -694,6 +694,12 @@ func (c *coordinator) buildProvider(providerCfg config.ProviderConfig, model con
694694 case "google-vertex":
695695 return c.buildGoogleVertexProvider(headers, providerCfg.ExtraParams)
696696 case openaicompat.Name:
697+ if providerCfg.ID == string(catwalk.InferenceProviderZAI) {
698+ if providerCfg.ExtraBody == nil {
699+ providerCfg.ExtraBody = map[string]any{}
700+ }
701+ providerCfg.ExtraBody["tool_stream"] = true
702+ }
697703 return c.buildOpenaiCompatProvider(baseURL, apiKey, headers, providerCfg.ExtraBody)
698704 default:
699705 return nil, fmt.Errorf("provider type not supported: %q", providerCfg.Type)