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 }), }); }