From 4dcbff0c06e3af3682cc4b786f79184e0d9e355e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?T=C3=82CHES?= <afromanguy@me.com>
Date: Sat, 14 Mar 2026 13:34:47 -0600
Subject: [PATCH] fix: increase timeout for z.ai provider to handle slow API
 spikes (#379) (#396)

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 packages/pi-ai/src/providers/openai-completions.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/packages/pi-ai/src/providers/openai-completions.ts b/packages/pi-ai/src/providers/openai-completions.ts
index 2d28bf4e1..28c8d17f6 100644
--- a/packages/pi-ai/src/providers/openai-completions.ts
+++ b/packages/pi-ai/src/providers/openai-completions.ts
@@ -357,11 +357,14 @@ function createClient(
 		Object.assign(headers, optionsHeaders);
 	}
 
+	const isZai = model.provider === "zai" || model.baseUrl.includes("api.z.ai");
+
 	return new OpenAI({
 		apiKey,
 		baseURL: model.baseUrl,
 		dangerouslyAllowBrowser: true,
 		defaultHeaders: headers,
+		...(isZai && { timeout: 100_000, maxRetries: 4 }),
 	});
 }