From 54204dab0c6c245e32557ce9227345249d11e241 Mon Sep 17 00:00:00 2001
From: Kujtim Hoxha <kujtimii.h@gmail.com>
Date: Thu, 4 Dec 2025 16:36:48 +0100
Subject: [PATCH] chore: improve zai tool calls (#1551)

---
 internal/agent/coordinator.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/internal/agent/coordinator.go b/internal/agent/coordinator.go
index 40dc818a55edf0eee005cc6d984622a5253b2151..91463fe4c24be90b743bcdb654f865ce60ecf2af 100644
--- a/internal/agent/coordinator.go
+++ b/internal/agent/coordinator.go
@@ -694,6 +694,12 @@ func (c *coordinator) buildProvider(providerCfg config.ProviderConfig, model con
 	case "google-vertex":
 		return c.buildGoogleVertexProvider(headers, providerCfg.ExtraParams)
 	case openaicompat.Name:
+		if providerCfg.ID == string(catwalk.InferenceProviderZAI) {
+			if providerCfg.ExtraBody == nil {
+				providerCfg.ExtraBody = map[string]any{}
+			}
+			providerCfg.ExtraBody["tool_stream"] = true
+		}
 		return c.buildOpenaiCompatProvider(baseURL, apiKey, headers, providerCfg.ExtraBody)
 	default:
 		return nil, fmt.Errorf("provider type not supported: %q", providerCfg.Type)