@ank1015/providers 0.0.35 → 0.0.37

package/dist/providers/zai/stream.js

@@ -0,0 +1,321 @@
+import { createClient, buildParams, mapStopReason, getMockZaiMessage } from "./utils.js";
+import { AssistantMessageEventStream } from "../../utils/event-stream.js";
+import { parseStreamingJson } from "../../utils/json-parse.js";
+import { validateToolArguments } from "../../utils/validation.js";
+import { calculateCost } from "../../models.js";
+export const streamZai = (model, context, options, id) => {
+    const stream = new AssistantMessageEventStream();
+    (async () => {
+        const startTimestamp = Date.now();
+        const output = {
+            role: "assistant",
+            api: model.api,
+            model: model,
+            id,
+            content: [],
+            usage: {
+                input: 0,
+                output: 0,
+                cacheRead: 0,
+                cacheWrite: 0,
+                totalTokens: 0,
+                cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+            },
+            stopReason: "stop",
+            timestamp: startTimestamp,
+            duration: 0
+        };
+        // Build final ChatCompletion from streamed chunks
+        let finalResponse = getMockZaiMessage();
+        let accumulatedContent = '';
+        let accumulatedReasoningContent = '';
+        const accumulatedToolCalls = new Map();
+        try {
+            const client = createClient(model, options?.apiKey);
+            const params = buildParams(model, context, options);
+            const zaiStream = await client.chat.completions.create({ ...params, stream: true }, { signal: options?.signal });
+            stream.push({ type: "start", message: { ...output, timestamp: Date.now() } });
+            let currentBlock = null;
+            const blocks = output.content;
+            const blockIndex = () => blocks.length - 1;
+            // Track tool call states for streaming
+            const toolCallBlocks = new Map();
+            for await (const chunk of zaiStream) {
+                const choice = chunk.choices[0];
+                if (!choice)
+                    continue;
+                const delta = choice.delta;
+                // Handle reasoning/thinking content (Z.AI-specific)
+                if (delta.reasoning_content) {
+                    accumulatedReasoningContent += delta.reasoning_content;
+                    // Start thinking block if needed
+                    if (!currentBlock || currentBlock.type !== 'thinking') {
+                        // End previous block if exists
+                        if (currentBlock) {
+                            if (currentBlock.type === 'response') {
+                                stream.push({
+                                    type: "text_end",
+                                    contentIndex: blockIndex(),
+                                    content: currentBlock.content,
+                                    message: { ...output, timestamp: Date.now() },
+                                });
+                            }
+                        }
+                        currentBlock = { type: 'thinking', thinkingText: '' };
+                        output.content.push(currentBlock);
+                        stream.push({ type: "thinking_start", contentIndex: blockIndex(), message: { ...output, timestamp: Date.now() } });
+                    }
+                    if (currentBlock.type === 'thinking') {
+                        currentBlock.thinkingText += delta.reasoning_content;
+                        stream.push({
+                            type: "thinking_delta",
+                            contentIndex: blockIndex(),
+                            delta: delta.reasoning_content,
+                            message: { ...output, timestamp: Date.now() },
+                        });
+                    }
+                }
+                // Handle text content
+                if (delta.content) {
+                    accumulatedContent += delta.content;
+                    // Transition from thinking to response if needed
+                    if (!currentBlock || currentBlock.type === 'thinking') {
+                        // End thinking block if exists
+                        if (currentBlock && currentBlock.type === 'thinking') {
+                            stream.push({
+                                type: "thinking_end",
+                                contentIndex: blockIndex(),
+                                content: currentBlock.thinkingText,
+                                message: { ...output, timestamp: Date.now() },
+                            });
+                        }
+                        currentBlock = { type: 'response', content: [{ type: 'text', content: '' }] };
+                        output.content.push(currentBlock);
+                        stream.push({ type: "text_start", contentIndex: blockIndex(), message: { ...output, timestamp: Date.now() } });
+                    }
+                    if (currentBlock.type === 'response') {
+                        const index = currentBlock.content.findIndex((c) => c.type === 'text');
+                        if (index !== -1) {
+                            currentBlock.content[index].content += delta.content;
+                        }
+                        stream.push({
+                            type: "text_delta",
+                            contentIndex: blockIndex(),
+                            delta: delta.content,
+                            message: { ...output, timestamp: Date.now() },
+                        });
+                    }
+                }
+                // Handle tool calls
+                if (delta.tool_calls) {
+                    // End current text/thinking block if we're starting tool calls
+                    if (currentBlock && currentBlock.type !== 'toolCall') {
+                        if (currentBlock.type === 'response') {
+                            stream.push({
+                                type: "text_end",
+                                contentIndex: blockIndex(),
+                                content: currentBlock.content,
+                                message: { ...output, timestamp: Date.now() },
+                            });
+                        }
+                        else if (currentBlock.type === 'thinking') {
+                            stream.push({
+                                type: "thinking_end",
+                                contentIndex: blockIndex(),
+                                content: currentBlock.thinkingText,
+                                message: { ...output, timestamp: Date.now() },
+                            });
+                        }
+                        currentBlock = null;
+                    }
+                    for (const toolCallDelta of delta.tool_calls) {
+                        const toolIndex = toolCallDelta.index;
+                        // Get or create accumulated tool call
+                        if (!accumulatedToolCalls.has(toolIndex)) {
+                            accumulatedToolCalls.set(toolIndex, {
+                                id: toolCallDelta.id || '',
+                                type: 'function',
+                                function: { name: '', arguments: '' }
+                            });
+                        }
+                        const accumulated = accumulatedToolCalls.get(toolIndex);
+                        // Update accumulated data
+                        if (toolCallDelta.id) {
+                            accumulated.id = toolCallDelta.id;
+                        }
+                        if (toolCallDelta.function?.name) {
+                            accumulated.function.name += toolCallDelta.function.name;
+                        }
+                        if (toolCallDelta.function?.arguments) {
+                            accumulated.function.arguments += toolCallDelta.function.arguments;
+                        }
+                        // Get or create tool call block
+                        if (!toolCallBlocks.has(toolIndex)) {
+                            const toolBlock = {
+                                type: 'toolCall',
+                                toolCallId: accumulated.id,
+                                name: accumulated.function.name,
+                                arguments: {},
+                                partialJson: ''
+                            };
+                            toolCallBlocks.set(toolIndex, toolBlock);
+                            output.content.push(toolBlock);
+                            currentBlock = toolBlock;
+                            stream.push({ type: "toolcall_start", contentIndex: blockIndex(), message: { ...output, timestamp: Date.now() } });
+                        }
+                        const toolBlock = toolCallBlocks.get(toolIndex);
+                        // Update tool block with new data
+                        if (toolCallDelta.id) {
+                            toolBlock.toolCallId = accumulated.id;
+                        }
+                        if (toolCallDelta.function?.name) {
+                            toolBlock.name = accumulated.function.name;
+                        }
+                        if (toolCallDelta.function?.arguments) {
+                            toolBlock.partialJson += toolCallDelta.function.arguments;
+                            toolBlock.arguments = parseStreamingJson(toolBlock.partialJson);
+                            stream.push({
+                                type: "toolcall_delta",
+                                contentIndex: blocks.indexOf(toolBlock),
+                                delta: toolCallDelta.function.arguments,
+                                message: { ...output, timestamp: Date.now() },
+                            });
+                        }
+                    }
+                }
+                // Handle finish reason
+                if (choice.finish_reason) {
+                    output.stopReason = mapStopReason(choice.finish_reason);
+                }
+                // Handle usage (typically in the last chunk)
+                if (chunk.usage) {
+                    const usage = chunk.usage;
+                    const cacheHitTokens = usage.prompt_tokens_details?.cached_tokens || 0;
+                    output.usage = {
+                        input: (usage.prompt_tokens || 0) - cacheHitTokens,
+                        output: usage.completion_tokens || 0,
+                        cacheRead: cacheHitTokens,
+                        cacheWrite: 0,
+                        totalTokens: usage.total_tokens || 0,
+                        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+                    };
+                    calculateCost(model, output.usage);
+                }
+            }
+            // End any remaining blocks
+            if (currentBlock) {
+                if (currentBlock.type === 'response') {
+                    stream.push({
+                        type: "text_end",
+                        contentIndex: blockIndex(),
+                        content: currentBlock.content,
+                        message: { ...output, timestamp: Date.now() },
+                    });
+                }
+                else if (currentBlock.type === 'thinking') {
+                    stream.push({
+                        type: "thinking_end",
+                        contentIndex: blockIndex(),
+                        content: currentBlock.thinkingText,
+                        message: { ...output, timestamp: Date.now() },
+                    });
+                }
+            }
+            // End tool call blocks and validate arguments
+            for (const [, toolBlock] of toolCallBlocks) {
+                // Parse final JSON
+                try {
+                    toolBlock.arguments = JSON.parse(toolBlock.partialJson || '{}');
+                }
+                catch {
+                    // Keep the parsed streaming result
+                }
+                // Validate tool arguments if tool definition is available
+                if (context.tools) {
+                    const tool = context.tools.find((t) => t.name === toolBlock.name);
+                    if (tool) {
+                        toolBlock.arguments = validateToolArguments(tool, toolBlock);
+                    }
+                }
+                // Clean up partialJson before sending final event
+                const { partialJson: _, ...cleanToolCall } = toolBlock;
+                stream.push({
+                    type: "toolcall_end",
+                    contentIndex: blocks.indexOf(toolBlock),
+                    toolCall: cleanToolCall,
+                    message: { ...output, timestamp: Date.now() }
+                });
+            }
+            // Clean up partialJson from output content
+            for (const block of output.content) {
+                if ('partialJson' in block) {
+                    delete block.partialJson;
+                }
+            }
+            // Ensure stopReason is toolUse if we have tool calls
+            if (output.content.some((b) => b.type === 'toolCall') && output.stopReason === 'stop') {
+                output.stopReason = 'toolUse';
+            }
+            if (options?.signal?.aborted) {
+                throw new Error("Request was aborted");
+            }
+            if (output.stopReason === "aborted" || output.stopReason === "error") {
+                throw new Error(`Stream ended with status: ${output.stopReason}${output.errorMessage ? ` - ${output.errorMessage}` : ""}`);
+            }
+            // Build final ChatCompletion response
+            finalResponse = {
+                id: `chatcmpl-${id}`,
+                object: "chat.completion",
+                created: Math.floor(startTimestamp / 1000),
+                model: model.id,
+                choices: [{
+                        index: 0,
+                        message: {
+                            role: "assistant",
+                            content: accumulatedContent || null,
+                            refusal: null,
+                            ...(accumulatedToolCalls.size > 0 && {
+                                tool_calls: Array.from(accumulatedToolCalls.values())
+                            })
+                        },
+                        finish_reason: output.stopReason === 'toolUse' ? 'tool_calls' : 'stop',
+                        logprobs: null
+                    }],
+                usage: {
+                    prompt_tokens: output.usage.input + output.usage.cacheRead,
+                    completion_tokens: output.usage.output,
+                    total_tokens: output.usage.totalTokens
+                }
+            };
+            // Add reasoning_content to the message if present
+            if (accumulatedReasoningContent) {
+                finalResponse.choices[0].message.reasoning_content = accumulatedReasoningContent;
+            }
+            stream.push({ type: "done", reason: output.stopReason, message: { ...output, timestamp: Date.now() } });
+            const baseAssistantMessage = {
+                ...output,
+                message: finalResponse,
+                timestamp: Date.now(),
+                duration: Date.now() - startTimestamp
+            };
+            stream.end(baseAssistantMessage);
+        }
+        catch (error) {
+            for (const block of output.content) {
+                delete block.partialJson;
+            }
+            output.stopReason = options?.signal?.aborted ? "aborted" : "error";
+            output.errorMessage = error instanceof Error ? error.message : JSON.stringify(error);
+            stream.push({ type: "error", reason: output.stopReason, message: { ...output, timestamp: Date.now() } });
+            const baseAssistantMessage = {
+                ...output,
+                message: finalResponse,
+                timestamp: Date.now(),
+                duration: Date.now() - startTimestamp
+            };
+            stream.end(baseAssistantMessage);
+        }
+    })();
+    return stream;
+};
+//# sourceMappingURL=stream.js.map

package/dist/providers/zai/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../src/providers/zai/stream.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACzF,OAAO,EAAE,2BAA2B,EAAE,MAAM,6BAA6B,CAAC;AAC1E,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAahD,MAAM,CAAC,MAAM,SAAS,GAA0B,CAC/C,KAAmB,EACnB,OAAgB,EAChB,OAA2B,EAC3B,EAAU,EACT,EAAE;IAEH,MAAM,MAAM,GAAG,IAAI,2BAA2B,EAAS,CAAC;IAExD,CAAC,KAAK,IAAI,EAAE;QAEX,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAClC,MAAM,MAAM,GAAqC;YAChD,IAAI,EAAE,WAAW;YACjB,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK,EAAE,KAAK;YACZ,EAAE;YACF,OAAO,EAAE,EAAE;YACX,KAAK,EAAE;gBACN,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,CAAC;gBACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;aACpE;YACD,UAAU,EAAE,MAAM;YAClB,SAAS,EAAE,cAAc;YACzB,QAAQ,EAAE,CAAC;SACX,CAAC;QAEF,kDAAkD;QAClD,IAAI,aAAa,GAAmB,iBAAiB,EAAE,CAAC;QACxD,IAAI,kBAAkB,GAAG,EAAE,CAAC;QAC5B,IAAI,2BAA2B,GAAG,EAAE,CAAC;QACrC,MAAM,oBAAoB,GAAiG,IAAI,GAAG,EAAE,CAAC;QAErI,IAAI,CAAC;YAEJ,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACpD,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAEpD,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CACrD,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,EAC3B,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAC3B,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;YAE9E,IAAI,YAAY,GAA+G,IAAI,CAAC;YACpI,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC;YAC9B,MAAM,UAAU,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;YAE3C,uCAAuC;YACvC,MAAM,cAAc,GAA6D,IAAI,GAAG,EAAE,CAAC;YAE3F,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;gBACrC,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAChC,IAAI,CAAC,MAAM;oBAAE,SAAS;gBAEtB,MAAM,KAAK,GAAG,MAAM,CAAC,KAA4C,CAAC;gBAElE,oDAAoD;gBACpD,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;oBAC7B,2BAA2B,IAAI,KAAK,CAAC,iBAAiB,CAAC;oBAEvD,iCAAiC;oBACjC,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACvD,+BAA+B;wBAC/B,IAAI,YAAY,EAAE,CAAC;4BAClB,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gCACtC,MAAM,CAAC,IAAI,CAAC;oCACX,IAAI,EAAE,UAAU;oCAChB,YAAY,EAAE,UAAU,EAAE;oCAC1B,OAAO,EAAE,YAAY,CAAC,OAAO;oCAC7B,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;iCAC7C,CAAC,CAAC;4BACJ,CAAC;wBACF,CAAC;wBACD,YAAY,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;wBACtD,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;wBAClC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;oBACpH,CAAC;oBAED,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACtC,YAAY,CAAC,YAAY,IAAI,KAAK,CAAC,iBAAiB,CAAC;wBACrD,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,gBAAgB;4BACtB,YAAY,EAAE,UAAU,EAAE;4BAC1B,KAAK,EAAE,KAAK,CAAC,iBAAiB;4BAC9B,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;yBAC7C,CAAC,CAAC;oBACJ,CAAC;gBACF,CAAC;gBAED,sBAAsB;gBACtB,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBACnB,kBAAkB,IAAI,KAAK,CAAC,OAAO,CAAC;oBAEpC,iDAAiD;oBACjD,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACvD,+BAA+B;wBAC/B,IAAI,YAAY,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;4BACtD,MAAM,CAAC,IAAI,CAAC;gCACX,IAAI,EAAE,cAAc;gCACpB,YAAY,EAAE,UAAU,EAAE;gCAC1B,OAAO,EAAE,YAAY,CAAC,YAAY;gCAClC,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;6BAC7C,CAAC,CAAC;wBACJ,CAAC;wBACD,YAAY,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;wBAC9E,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;wBAClC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;oBAChH,CAAC;oBAED,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACtC,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;wBACvE,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;4BACjB,YAAY,CAAC,OAAO,CAAC,KAAK,CAAiB,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC;wBACvE,CAAC;wBACD,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,YAAY;4BAClB,YAAY,EAAE,UAAU,EAAE;4BAC1B,KAAK,EAAE,KAAK,CAAC,OAAO;4BACpB,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;yBAC7C,CAAC,CAAC;oBACJ,CAAC;gBACF,CAAC;gBAED,oBAAoB;gBACpB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACtB,+DAA+D;oBAC/D,IAAI,YAAY,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACtD,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;4BACtC,MAAM,CAAC,IAAI,CAAC;gCACX,IAAI,EAAE,UAAU;gCAChB,YAAY,EAAE,UAAU,EAAE;gCAC1B,OAAO,EAAE,YAAY,CAAC,OAAO;gCAC7B,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;6BAC7C,CAAC,CAAC;wBACJ,CAAC;6BAAM,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;4BAC7C,MAAM,CAAC,IAAI,CAAC;gCACX,IAAI,EAAE,cAAc;gCACpB,YAAY,EAAE,UAAU,EAAE;gCAC1B,OAAO,EAAE,YAAY,CAAC,YAAY;gCAClC,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;6BAC7C,CAAC,CAAC;wBACJ,CAAC;wBACD,YAAY,GAAG,IAAI,CAAC;oBACrB,CAAC;oBAED,KAAK,MAAM,aAAa,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;wBAC9C,MAAM,SAAS,GAAG,aAAa,CAAC,KAAK,CAAC;wBAEtC,sCAAsC;wBACtC,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;4BAC1C,oBAAoB,CAAC,GAAG,CAAC,SAAS,EAAE;gCACnC,EAAE,EAAE,aAAa,CAAC,EAAE,IAAI,EAAE;gCAC1B,IAAI,EAAE,UAAU;gCAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;6BACrC,CAAC,CAAC;wBACJ,CAAC;wBACD,MAAM,WAAW,GAAG,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;wBAEzD,0BAA0B;wBAC1B,IAAI,aAAa,CAAC,EAAE,EAAE,CAAC;4BACtB,WAAW,CAAC,EAAE,GAAG,aAAa,CAAC,EAAE,CAAC;wBACnC,CAAC;wBACD,IAAI,aAAa,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;4BAClC,WAAW,CAAC,QAAQ,CAAC,IAAI,IAAI,aAAa,CAAC,QAAQ,CAAC,IAAI,CAAC;wBAC1D,CAAC;wBACD,IAAI,aAAa,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;4BACvC,WAAW,CAAC,QAAQ,CAAC,SAAS,IAAI,aAAa,CAAC,QAAQ,CAAC,SAAS,CAAC;wBACpE,CAAC;wBAED,gCAAgC;wBAChC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;4BACpC,MAAM,SAAS,GAAgD;gCAC9D,IAAI,EAAE,UAAU;gCAChB,UAAU,EAAE,WAAW,CAAC,EAAE;gCAC1B,IAAI,EAAE,WAAW,CAAC,QAAQ,CAAC,IAAI;gCAC/B,SAAS,EAAE,EAAE;gCACb,WAAW,EAAE,EAAE;6BACf,CAAC;4BACF,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;4BACzC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;4BAC/B,YAAY,GAAG,SAAS,CAAC;4BACzB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;wBACpH,CAAC;wBAED,MAAM,SAAS,GAAG,cAAc,CAAC,GAAG,CAAC,SAAS,CAAE,CAAC;wBAEjD,kCAAkC;wBAClC,IAAI,aAAa,CAAC,EAAE,EAAE,CAAC;4BACtB,SAAS,CAAC,UAAU,GAAG,WAAW,CAAC,EAAE,CAAC;wBACvC,CAAC;wBACD,IAAI,aAAa,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;4BAClC,SAAS,CAAC,IAAI,GAAG,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC;wBAC5C,CAAC;wBACD,IAAI,aAAa,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;4BACvC,SAAS,CAAC,WAAW,IAAI,aAAa,CAAC,QAAQ,CAAC,SAAS,CAAC;4BAC1D,SAAS,CAAC,SAAS,GAAG,kBAAkB,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;4BAChE,MAAM,CAAC,IAAI,CAAC;gCACX,IAAI,EAAE,gBAAgB;gCACtB,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC;gCACvC,KAAK,EAAE,aAAa,CAAC,QAAQ,CAAC,SAAS;gCACvC,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;6BAC7C,CAAC,CAAC;wBACJ,CAAC;oBACF,CAAC;gBACF,CAAC;gBAED,uBAAuB;gBACvB,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;oBAC1B,MAAM,CAAC,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBACzD,CAAC;gBAED,6CAA6C;gBAC7C,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBACjB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAsC,CAAC;oBAC3D,MAAM,cAAc,GAAG,KAAK,CAAC,qBAAqB,EAAE,aAAa,IAAI,CAAC,CAAC;oBAEvE,MAAM,CAAC,KAAK,GAAG;wBACd,KAAK,EAAE,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC,GAAG,cAAc;wBAClD,MAAM,EAAE,KAAK,CAAC,iBAAiB,IAAI,CAAC;wBACpC,SAAS,EAAE,cAAc;wBACzB,UAAU,EAAE,CAAC;wBACb,WAAW,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;wBACpC,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;qBACpE,CAAC;oBACF,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;gBACpC,CAAC;YACF,CAAC;YAED,2BAA2B;YAC3B,IAAI,YAAY,EAAE,CAAC;gBAClB,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACtC,MAAM,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,UAAU;wBAChB,YAAY,EAAE,UAAU,EAAE;wBAC1B,OAAO,EAAE,YAAY,CAAC,OAAO;wBAC7B,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;qBAC7C,CAAC,CAAC;gBACJ,CAAC;qBAAM,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC7C,MAAM,CAAC,IAAI,CAAC;wBACX,IAAI,EAAE,cAAc;wBACpB,YAAY,EAAE,UAAU,EAAE;wBAC1B,OAAO,EAAE,YAAY,CAAC,YAAY;wBAClC,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;qBAC7C,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAED,8CAA8C;YAC9C,KAAK,MAAM,CAAC,EAAE,SAAS,CAAC,IAAI,cAAc,EAAE,CAAC;gBAC5C,mBAAmB;gBACnB,IAAI,CAAC;oBACJ,SAAS,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC;gBACjE,CAAC;gBAAC,MAAM,CAAC;oBACR,mCAAmC;gBACpC,CAAC;gBAED,0DAA0D;gBAC1D,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;oBACnB,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,CAAC,CAAC;oBAClE,IAAI,IAAI,EAAE,CAAC;wBACV,SAAS,CAAC,SAAS,GAAG,qBAAqB,CAAC,IAAI,EAAE,SAAS,CAAwB,CAAC;oBACrF,CAAC;gBACF,CAAC;gBAED,kDAAkD;gBAClD,MAAM,EAAE,WAAW,EAAE,CAAC,EAAE,GAAG,aAAa,EAAE,GAAG,SAAS,CAAC;gBACvD,MAAM,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,cAAc;oBACpB,YAAY,EAAE,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC;oBACvC,QAAQ,EAAE,aAAa;oBACvB,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;iBAC7C,CAAC,CAAC;YACJ,CAAC;YAED,2CAA2C;YAC3C,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpC,IAAI,aAAa,IAAI,KAAK,EAAE,CAAC;oBAC5B,OAAQ,KAAa,CAAC,WAAW,CAAC;gBACnC,CAAC;YACF,CAAC;YAED,qDAAqD;YACrD,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,MAAM,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;gBACvF,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC;YAC/B,CAAC;YAED,IAAI,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACxC,CAAC;YAED,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,IAAI,MAAM,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;gBACtE,MAAM,IAAI,KAAK,CACd,6BAA6B,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CACzG,CAAC;YACH,CAAC;YAED,sCAAsC;YACtC,aAAa,GAAG;gBACf,EAAE,EAAE,YAAY,EAAE,EAAE;gBACpB,MAAM,EAAE,iBAAiB;gBACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,cAAc,GAAG,IAAI,CAAC;gBAC1C,KAAK,EAAE,KAAK,CAAC,EAAE;gBACf,OAAO,EAAE,CAAC;wBACT,KAAK,EAAE,CAAC;wBACR,OAAO,EAAE;4BACR,IAAI,EAAE,WAAW;4BACjB,OAAO,EAAE,kBAAkB,IAAI,IAAI;4BACnC,OAAO,EAAE,IAAI;4BACb,GAAG,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,IAAI;gCACpC,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC;6BACrD,CAAC;yBACF;wBACD,aAAa,EAAE,MAAM,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM;wBACtE,QAAQ,EAAE,IAAI;qBACd,CAAC;gBACF,KAAK,EAAE;oBACN,aAAa,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,SAAS;oBAC1D,iBAAiB,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;oBACtC,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW;iBACtC;aACD,CAAC;YAEF,kDAAkD;YAClD,IAAI,2BAA2B,EAAE,CAAC;gBAChC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAe,CAAC,iBAAiB,GAAG,2BAA2B,CAAC;YAC3F,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;YAExG,MAAM,oBAAoB,GAAgC;gBACzD,GAAG,MAAM;gBACT,OAAO,EAAE,aAAa;gBACtB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;aACrC,CAAA;YACD,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QAElC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEhB,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpC,OAAQ,KAAa,CAAC,WAAW,CAAC;YACnC,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;YACnE,MAAM,CAAC,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACrF,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;YAEzG,MAAM,oBAAoB,GAAgC;gBACzD,GAAG,MAAM;gBACT,OAAO,EAAE,aAAa;gBACtB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;aACrC,CAAA;YACD,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QAClC,CAAC;IAEF,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,MAAM,CAAC;AACf,CAAC,CAAA"}

package/dist/providers/zai/types.d.ts

@@ -0,0 +1,13 @@
+import type { ChatCompletionCreateParamsBase } from "openai/resources/chat/completions.js";
+export interface ZaiThinkingConfig {
+    type: "enabled" | "disabled";
+    clear_thinking?: boolean;
+}
+type Props = {
+    apiKey?: string;
+    signal?: AbortSignal;
+    thinking?: ZaiThinkingConfig;
+};
+export type ZaiProviderOptions = Omit<ChatCompletionCreateParamsBase, 'model' | 'messages'> & Props;
+export {};
+//# sourceMappingURL=types.d.ts.map

package/dist/providers/zai/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/providers/zai/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,sCAAsC,CAAA;AAE1F,MAAM,WAAW,iBAAiB;IACjC,IAAI,EAAE,SAAS,GAAG,UAAU,CAAC;IAC7B,cAAc,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,KAAK,KAAK,GAAG;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,QAAQ,CAAC,EAAE,iBAAiB,CAAC;CAC7B,CAAA;AAED,MAAM,MAAM,kBAAkB,GAAG,IAAI,CAAC,8BAA8B,EAAE,OAAO,GAAG,UAAU,CAAC,GAAG,KAAK,CAAA"}

package/dist/providers/zai/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/providers/zai/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/providers/zai/types.ts"],"names":[],"mappings":""}

package/dist/providers/zai/utils.d.ts

@@ -0,0 +1,19 @@
+import OpenAI from "openai";
+import { AssistantResponse, Context, Model, StopReason, Tool, Usage } from "../../types.js";
+import type { ChatCompletion, ChatCompletionMessageParam, ChatCompletionTool, ChatCompletionCreateParamsNonStreaming } from "openai/resources/chat/completions.js";
+import { ZaiProviderOptions } from "./types.js";
+export declare function createClient(model: Model<"zai">, apiKey?: string): OpenAI;
+export declare function getResponseAssistantResponse(response: ChatCompletion): AssistantResponse;
+export declare function getAssistantStopReason(response: ChatCompletion): StopReason;
+export declare function getResponseUsage(response: ChatCompletion, model: Model<'zai'>): Usage;
+export declare function buildParams(model: Model<"zai">, context: Context, options: ZaiProviderOptions): ChatCompletionCreateParamsNonStreaming & {
+    thinking?: {
+        type: string;
+        clear_thinking?: boolean;
+    };
+};
+export declare function buildZaiMessages(_model: Model<'zai'>, context: Context): ChatCompletionMessageParam[];
+export declare function convertTools(tools: readonly Tool[]): ChatCompletionTool[];
+export declare function mapStopReason(finishReason: string | null | undefined): StopReason;
+export declare function getMockZaiMessage(): ChatCompletion;
+//# sourceMappingURL=utils.d.ts.map

package/dist/providers/zai/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/providers/zai/utils.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,iBAAiB,EAAwB,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAe,MAAM,gBAAgB,CAAC;AAC/H,OAAO,KAAK,EACX,cAAc,EACd,0BAA0B,EAC1B,kBAAkB,EAClB,sCAAsC,EAGtC,MAAM,sCAAsC,CAAC;AAG9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAahD,wBAAgB,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,EAAE,MAAM,UAehE;AAED,wBAAgB,4BAA4B,CAAC,QAAQ,EAAE,cAAc,GAAG,iBAAiB,CA6CxF;AAED,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,cAAc,GAAG,UAAU,CAG3E;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,cAAc,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,KAAK,CAgBrF;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB;eAIzB;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,cAAc,CAAC,EAAE,OAAO,CAAA;KAAE;EAiC9G;AAED,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,OAAO,GAAG,0BAA0B,EAAE,CAoGrG;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,GAAG,kBAAkB,EAAE,CASzE;AAED,wBAAgB,aAAa,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,UAAU,CAiBjF;AAED,wBAAgB,iBAAiB,IAAI,cAAc,CAsBlD"}

package/dist/providers/zai/utils.js

@@ -0,0 +1,253 @@
+import OpenAI from "openai";
+import { sanitizeSurrogates } from "../../utils/sanitize-unicode.js";
+import { calculateCost } from "../../models.js";
+export function createClient(model, apiKey) {
+    if (!apiKey) {
+        if (!process.env.ZAI_API_KEY) {
+            throw new Error("Z.AI API key is required. Set ZAI_API_KEY environment variable or pass it as an argument.");
+        }
+        apiKey = process.env.ZAI_API_KEY;
+    }
+    return new OpenAI({
+        apiKey,
+        baseURL: model.baseUrl,
+        dangerouslyAllowBrowser: true,
+        defaultHeaders: model.headers,
+    });
+}
+export function getResponseAssistantResponse(response) {
+    const assistantResponse = [];
+    const choice = response.choices[0];
+    if (!choice?.message) {
+        return assistantResponse;
+    }
+    const message = choice.message;
+    // Handle reasoning/thinking content (Z.AI-specific)
+    if (message.reasoning_content) {
+        assistantResponse.push({
+            type: 'thinking',
+            thinkingText: message.reasoning_content
+        });
+    }
+    // Handle text content
+    if (message.content) {
+        assistantResponse.push({
+            type: 'response',
+            content: [{
+                    type: 'text',
+                    content: message.content
+                }]
+        });
+    }
+    // Handle tool calls
+    if (message.tool_calls) {
+        for (const toolCall of message.tool_calls) {
+            // Only handle function tool calls
+            if (toolCall.type === 'function') {
+                assistantResponse.push({
+                    type: 'toolCall',
+                    toolCallId: toolCall.id,
+                    name: toolCall.function.name,
+                    arguments: JSON.parse(toolCall.function.arguments || '{}')
+                });
+            }
+        }
+    }
+    return assistantResponse;
+}
+export function getAssistantStopReason(response) {
+    const finishReason = response.choices[0]?.finish_reason;
+    return mapStopReason(finishReason);
+}
+export function getResponseUsage(response, model) {
+    const responseUsage = response.usage;
+    // Z.AI reports cache hits via prompt_tokens_details.cached_tokens (like OpenAI)
+    const cacheHitTokens = responseUsage?.prompt_tokens_details?.cached_tokens || 0;
+    const usage = {
+        input: (responseUsage?.prompt_tokens || 0) - cacheHitTokens,
+        output: responseUsage?.completion_tokens || 0,
+        cacheRead: cacheHitTokens,
+        cacheWrite: 0,
+        totalTokens: responseUsage?.total_tokens || 0,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
+    };
+    calculateCost(model, usage);
+    return usage;
+}
+export function buildParams(model, context, options) {
+    const messages = buildZaiMessages(model, context);
+    const { apiKey, signal, thinking, ...zaiOptions } = options;
+    const params = {
+        ...zaiOptions,
+        model: model.id,
+        messages,
+        stream: false
+    };
+    // Add thinking configuration if provided, default to enabled for reasoning models
+    if (thinking) {
+        params.thinking = thinking;
+    }
+    else if (model.reasoning) {
+        // Default to enabled for reasoning models
+        params.thinking = { type: "enabled" };
+    }
+    // Add tools if available and supported
+    if (context.tools && context.tools.length > 0 && model.tools.includes('function_calling')) {
+        const tools = [];
+        const convertedTools = convertTools(context.tools);
+        for (const convertedTool of convertedTools) {
+            tools.push(convertedTool);
+        }
+        if (zaiOptions.tools) {
+            for (const optionTool of zaiOptions.tools) {
+                tools.push(optionTool);
+            }
+        }
+        params.tools = tools;
+    }
+    return params;
+}
+export function buildZaiMessages(_model, context) {
+    const messages = [];
+    // Add system prompt
+    if (context.systemPrompt) {
+        messages.push({
+            role: 'system',
+            content: sanitizeSurrogates(context.systemPrompt)
+        });
+    }
+    for (const message of context.messages) {
+        // Handle user messages
+        if (message.role === 'user') {
+            // Z.AI chat completions API supports text content
+            const textContents = message.content
+                .filter(c => c.type === 'text')
+                .map(c => sanitizeSurrogates(c.content))
+                .join('\n');
+            if (textContents) {
+                messages.push({
+                    role: 'user',
+                    content: textContents
+                });
+            }
+        }
+        // Handle tool results
+        if (message.role === 'toolResult') {
+            const textContent = message.content
+                .filter(c => c.type === 'text')
+                .map(c => {
+                const text = c.content;
+                return message.isError ? `[TOOL ERROR] ${text}` : text;
+            })
+                .join('\n');
+            const toolMessage = {
+                role: 'tool',
+                tool_call_id: message.toolCallId,
+                content: sanitizeSurrogates(textContent || (message.isError ? '[TOOL ERROR]' : ''))
+            };
+            messages.push(toolMessage);
+        }
+        // Handle assistant messages
+        if (message.role === 'assistant') {
+            if (message.model.api === 'zai') {
+                // Native Z.AI message - reconstruct from original
+                const baseMessage = message;
+                const originalMessage = baseMessage.message.choices[0]?.message;
+                if (originalMessage) {
+                    const assistantMessage = originalMessage;
+                    messages.push(assistantMessage);
+                }
+            }
+            else {
+                // Convert from other providers using normalized content
+                let textContent = '';
+                const toolCalls = [];
+                let reasoningContent = '';
+                for (const contentBlock of message.content) {
+                    if (contentBlock.type === 'thinking') {
+                        reasoningContent += sanitizeSurrogates(contentBlock.thinkingText);
+                    }
+                    else if (contentBlock.type === 'response') {
+                        const text = contentBlock.content
+                            .filter(c => c.type === 'text')
+                            .map(c => sanitizeSurrogates(c.content))
+                            .join('');
+                        textContent += text;
+                    }
+                    else if (contentBlock.type === 'toolCall') {
+                        toolCalls.push({
+                            id: contentBlock.toolCallId,
+                            type: 'function',
+                            function: {
+                                name: contentBlock.name,
+                                arguments: JSON.stringify(contentBlock.arguments)
+                            }
+                        });
+                    }
+                }
+                const assistantMessage = {
+                    role: 'assistant',
+                    reasoning_content: reasoningContent || undefined,
+                    content: textContent || null
+                };
+                if (toolCalls.length > 0) {
+                    assistantMessage.tool_calls = toolCalls;
+                }
+                messages.push(assistantMessage);
+            }
+        }
+    }
+    return messages;
+}
+export function convertTools(tools) {
+    return tools.map((tool) => ({
+        type: "function",
+        function: {
+            name: tool.name,
+            description: tool.description,
+            parameters: tool.parameters
+        }
+    }));
+}
+export function mapStopReason(finishReason) {
+    if (!finishReason)
+        return "stop";
+    switch (finishReason) {
+        case "stop":
+            return "stop";
+        case "length":
+            return "length";
+        case "tool_calls":
+            return "toolUse";
+        case "sensitive":
+        case "content_filter":
+        case "network_error":
+            return "error";
+        default:
+            return "stop";
+    }
+}
+export function getMockZaiMessage() {
+    return {
+        id: "chatcmpl-123",
+        object: "chat.completion",
+        created: Math.floor(Date.now() / 1000),
+        model: "glm-4.7",
+        choices: [{
+                index: 0,
+                message: {
+                    role: "assistant",
+                    content: "",
+                    refusal: null
+                },
+                finish_reason: "stop",
+                logprobs: null
+            }],
+        usage: {
+            prompt_tokens: 0,
+            completion_tokens: 0,
+            total_tokens: 0
+        }
+    };
+}
+//# sourceMappingURL=utils.js.map

package/dist/providers/zai/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/providers/zai/utils.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAU5B,OAAO,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AACrE,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAchD,MAAM,UAAU,YAAY,CAAC,KAAmB,EAAE,MAAe;IAChE,IAAI,CAAC,MAAM,EAAE,CAAC;QACb,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CACd,2FAA2F,CAC3F,CAAC;QACH,CAAC;QACD,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IAClC,CAAC;IACD,OAAO,IAAI,MAAM,CAAC;QACjB,MAAM;QACN,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,uBAAuB,EAAE,IAAI;QAC7B,cAAc,EAAE,KAAK,CAAC,OAAO;KAC7B,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,QAAwB;IACpE,MAAM,iBAAiB,GAAsB,EAAE,CAAC;IAChD,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAEnC,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;QACtB,OAAO,iBAAiB,CAAC;IAC1B,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,OAA6C,CAAC;IAErE,oDAAoD;IACpD,IAAI,OAAO,CAAC,iBAAiB,EAAE,CAAC;QAC/B,iBAAiB,CAAC,IAAI,CAAC;YACtB,IAAI,EAAE,UAAU;YAChB,YAAY,EAAE,OAAO,CAAC,iBAAiB;SACvC,CAAC,CAAC;IACJ,CAAC;IAED,sBAAsB;IACtB,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACrB,iBAAiB,CAAC,IAAI,CAAC;YACtB,IAAI,EAAE,UAAU;YAChB,OAAO,EAAE,CAAC;oBACT,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,OAAO,CAAC,OAAO;iBACxB,CAAC;SACF,CAAC,CAAC;IACJ,CAAC;IAED,oBAAoB;IACpB,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;QACxB,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YAC3C,kCAAkC;YAClC,IAAI,QAAQ,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAClC,iBAAiB,CAAC,IAAI,CAAC;oBACtB,IAAI,EAAE,UAAU;oBAChB,UAAU,EAAE,QAAQ,CAAC,EAAE;oBACvB,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;oBAC5B,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI,CAAC;iBAC1D,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,iBAAiB,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,QAAwB;IAC9D,MAAM,YAAY,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,CAAC;IACxD,OAAO,aAAa,CAAC,YAAY,CAAC,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAAwB,EAAE,KAAmB;IAC7E,MAAM,aAAa,GAAG,QAAQ,CAAC,KAAuD,CAAC;IAEvF,gFAAgF;IAChF,MAAM,cAAc,GAAG,aAAa,EAAE,qBAAqB,EAAE,aAAa,IAAI,CAAC,CAAC;IAEhF,MAAM,KAAK,GAAU;QACpB,KAAK,EAAE,CAAC,aAAa,EAAE,aAAa,IAAI,CAAC,CAAC,GAAG,cAAc;QAC3D,MAAM,EAAE,aAAa,EAAE,iBAAiB,IAAI,CAAC;QAC7C,SAAS,EAAE,cAAc;QACzB,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,aAAa,EAAE,YAAY,IAAI,CAAC;QAC7C,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;KACpE,CAAC;IACF,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC5B,OAAO,KAAK,CAAC;AACd,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAmB,EAAE,OAAgB,EAAE,OAA2B;IAC7F,MAAM,QAAQ,GAAG,gBAAgB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAElD,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,EAAE,GAAG,OAAO,CAAC;IAC5D,MAAM,MAAM,GAAuG;QAClH,GAAG,UAAU;QACb,KAAK,EAAE,KAAK,CAAC,EAAE;QACf,QAAQ;QACR,MAAM,EAAE,KAAK;KACb,CAAC;IAEF,kFAAkF;IAClF,IAAI,QAAQ,EAAE,CAAC;QACd,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC5B,CAAC;SAAM,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;QAC5B,0CAA0C;QAC1C,MAAM,CAAC,QAAQ,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;IACvC,CAAC;IAED,uCAAuC;IACvC,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC3F,MAAM,KAAK,GAAyB,EAAE,CAAC;QACvC,MAAM,cAAc,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACnD,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE,CAAC;YAC5C,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC3B,CAAC;QAED,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YACtB,KAAK,MAAM,UAAU,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBAC3C,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACxB,CAAC;QACF,CAAC;QAED,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;IACtB,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAoB,EAAE,OAAgB;IACtE,MAAM,QAAQ,GAAiC,EAAE,CAAC;IAElD,oBAAoB;IACpB,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;QAC1B,QAAQ,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,kBAAkB,CAAC,OAAO,CAAC,YAAY,CAAC;SACjD,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACxC,uBAAuB;QACvB,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC7B,kDAAkD;YAClD,MAAM,YAAY,GAAG,OAAO,CAAC,OAAO;iBAClC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBAC9B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,kBAAkB,CAAE,CAAiB,CAAC,OAAO,CAAC,CAAC;iBACxD,IAAI,CAAC,IAAI,CAAC,CAAC;YAEb,IAAI,YAAY,EAAE,CAAC;gBAClB,QAAQ,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,YAAY;iBACrB,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QAED,sBAAsB;QACtB,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACnC,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO;iBACjC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBAC9B,GAAG,CAAC,CAAC,CAAC,EAAE;gBACR,MAAM,IAAI,GAAI,CAAiB,CAAC,OAAO,CAAC;gBACxC,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,gBAAgB,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YACxD,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,CAAC;YAEb,MAAM,WAAW,GAAmC;gBACnD,IAAI,EAAE,MAAM;gBACZ,YAAY,EAAE,OAAO,CAAC,UAAU;gBAChC,OAAO,EAAE,kBAAkB,CAAC,WAAW,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;aACnF,CAAC;YACF,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5B,CAAC;QAED,4BAA4B;QAC5B,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAClC,IAAI,OAAO,CAAC,KAAK,CAAC,GAAG,KAAK,KAAK,EAAE,CAAC;gBACjC,kDAAkD;gBAClD,MAAM,WAAW,GAAG,OAAsC,CAAC;gBAC3D,MAAM,eAAe,GAAG,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;gBAEhE,IAAI,eAAe,EAAE,CAAC;oBACrB,MAAM,gBAAgB,GAAwC,eAAe,CAAC;oBAC9E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBACjC,CAAC;YACF,CAAC;iBAAM,CAAC;gBACP,wDAAwD;gBACxD,IAAI,WAAW,GAAG,EAAE,CAAC;gBACrB,MAAM,SAAS,GAAsD,EAAE,CAAC;gBACxE,IAAI,gBAAgB,GAAG,EAAE,CAAC;gBAE1B,KAAK,MAAM,YAAY,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;oBAC5C,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACtC,gBAAgB,IAAI,kBAAkB,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;oBACnE,CAAC;yBAAM,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAC7C,MAAM,IAAI,GAAG,YAAY,CAAC,OAAO;6BAC/B,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;6BAC9B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,kBAAkB,CAAE,CAAiB,CAAC,OAAO,CAAC,CAAC;6BACxD,IAAI,CAAC,EAAE,CAAC,CAAC;wBACX,WAAW,IAAI,IAAI,CAAC;oBACrB,CAAC;yBAAM,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAC7C,SAAS,CAAC,IAAI,CAAC;4BACd,EAAE,EAAE,YAAY,CAAC,UAAU;4BAC3B,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE;gCACT,IAAI,EAAE,YAAY,CAAC,IAAI;gCACvB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,SAAS,CAAC;6BACjD;yBACD,CAAC,CAAC;oBACJ,CAAC;gBACF,CAAC;gBAED,MAAM,gBAAgB,GAAQ;oBAC7B,IAAI,EAAE,WAAW;oBACjB,iBAAiB,EAAE,gBAAgB,IAAI,SAAS;oBAChD,OAAO,EAAE,WAAW,IAAI,IAAI;iBAC5B,CAAC;gBAEF,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC1B,gBAAgB,CAAC,UAAU,GAAG,SAAS,CAAC;gBACzC,CAAC;gBAED,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACjC,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,QAAQ,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAsB;IAClD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC3B,IAAI,EAAE,UAAmB;QACzB,QAAQ,EAAE;YACT,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,UAAqC;SACtD;KACD,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,YAAuC;IACpE,IAAI,CAAC,YAAY;QAAE,OAAO,MAAM,CAAC;IAEjC,QAAQ,YAAY,EAAE,CAAC;QACtB,KAAK,MAAM;YACV,OAAO,MAAM,CAAC;QACf,KAAK,QAAQ;YACZ,OAAO,QAAQ,CAAC;QACjB,KAAK,YAAY;YAChB,OAAO,SAAS,CAAC;QAClB,KAAK,WAAW,CAAC;QACjB,KAAK,gBAAgB,CAAC;QACtB,KAAK,eAAe;YACnB,OAAO,OAAO,CAAC;QAChB;YACC,OAAO,MAAM,CAAC;IAChB,CAAC;AACF,CAAC;AAED,MAAM,UAAU,iBAAiB;IAChC,OAAO;QACN,EAAE,EAAE,cAAc;QAClB,MAAM,EAAE,iBAAiB;QACzB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QACtC,KAAK,EAAE,SAAS;QAChB,OAAO,EAAE,CAAC;gBACT,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE;oBACR,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,EAAE;oBACX,OAAO,EAAE,IAAI;iBACb;gBACD,aAAa,EAAE,MAAM;gBACrB,QAAQ,EAAE,IAAI;aACd,CAAC;QACF,KAAK,EAAE;YACN,aAAa,EAAE,CAAC;YAChB,iBAAiB,EAAE,CAAC;YACpB,YAAY,EAAE,CAAC;SACf;KACD,CAAC;AACH,CAAC"}