@ank1015/providers 0.0.36 → 0.0.37

package/dist/providers/google-interactions/utils.js

@@ -0,0 +1,252 @@
+import { GoogleGenAI } from "@google/genai";
+import { calculateCost } from "../../models.js";
+import { sanitizeSurrogates } from "../../utils/sanitize-unicode.js";
+export function createClient(model, apiKey) {
+    if (!apiKey) {
+        if (!process.env.GEMINI_API_KEY) {
+            throw new Error("Gemini API key is required. Set GEMINI_API_KEY environment variable or pass it as an argument.");
+        }
+        apiKey = process.env.GEMINI_API_KEY;
+    }
+    return new GoogleGenAI({
+        apiKey,
+        httpOptions: model.headers ? { headers: model.headers } : undefined,
+    });
+}
+export function getResponseAssistantResponse(interaction) {
+    const assistantResponse = [];
+    if (!interaction.outputs)
+        return assistantResponse;
+    for (const output of interaction.outputs) {
+        const outputType = output.type;
+        // Handle text content
+        if (outputType === 'text') {
+            const textOutput = output;
+            if (textOutput.text) {
+                assistantResponse.push({
+                    type: 'response',
+                    content: [{ type: 'text', content: textOutput.text }]
+                });
+            }
+        }
+        // Handle thinking/thought content
+        else if (outputType === 'thought') {
+            const thoughtOutput = output;
+            const thinkingText = thoughtOutput.summary
+                ?.filter(s => s.type === 'text')
+                .map(s => s.text || '')
+                .join('\n\n') || '';
+            if (thinkingText) {
+                assistantResponse.push({
+                    type: 'thinking',
+                    thinkingText
+                });
+            }
+        }
+        // Handle function calls
+        else if (outputType === 'function_call') {
+            const fnOutput = output;
+            assistantResponse.push({
+                type: 'toolCall',
+                toolCallId: fnOutput.id,
+                name: fnOutput.name,
+                arguments: fnOutput.arguments
+            });
+        }
+        // Handle image content
+        else if (outputType === 'image') {
+            const imgOutput = output;
+            if (imgOutput.data) {
+                assistantResponse.push({
+                    type: 'response',
+                    content: [{
+                            type: 'image',
+                            data: imgOutput.data,
+                            mimeType: imgOutput.mime_type || 'image/png'
+                        }]
+                });
+            }
+        }
+        // Other output types (google_search_result, url_context_result, etc.)
+        // are typically intermediate and not directly surfaced
+    }
+    return assistantResponse;
+}
+export function getResponseUsage(interaction, model) {
+    const interactionUsage = interaction.usage;
+    const usage = {
+        input: (interactionUsage?.total_input_tokens || 0) - (interactionUsage?.total_cached_tokens || 0),
+        output: (interactionUsage?.total_output_tokens || 0) + (interactionUsage?.total_reasoning_tokens || 0),
+        cacheRead: interactionUsage?.total_cached_tokens || 0,
+        cacheWrite: 0,
+        totalTokens: interactionUsage?.total_tokens || 0,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
+    };
+    calculateCost(model, usage);
+    return usage;
+}
+export function mapInteractionStatus(status) {
+    switch (status) {
+        case 'completed':
+            return 'stop';
+        case 'in_progress':
+            return 'stop'; // Still processing, but not an error
+        case 'requires_action':
+            return 'toolUse';
+        case 'failed':
+            return 'error';
+        case 'cancelled':
+            return 'aborted';
+        default:
+            return 'stop';
+    }
+}
+export function buildParams(model, context, options) {
+    const input = buildInteractionInput(model, context);
+    const { apiKey, signal, ...interactionOptions } = options;
+    // Build base params
+    const params = {
+        ...interactionOptions,
+        input,
+    };
+    // Set model or agent based on isAgent flag
+    if (model.isAgent) {
+        params.agent = model.id;
+    }
+    else {
+        params.model = model.id;
+    }
+    // Add system instruction if present
+    if (context.systemPrompt) {
+        params.system_instruction = sanitizeSurrogates(context.systemPrompt);
+    }
+    // Add tools if present and supported
+    if (context.tools && context.tools.length > 0 && model.tools.includes('function_calling')) {
+        params.tools = convertTools(context.tools);
+    }
+    return params;
+}
+export function buildInteractionInput(model, context) {
+    // If no messages, return empty
+    if (context.messages.length === 0) {
+        return [];
+    }
+    // If single user message with only text, can return string
+    if (context.messages.length === 1 && context.messages[0].role === 'user') {
+        const msg = context.messages[0];
+        if (msg.role === 'user' && msg.content.length === 1 && msg.content[0].type === 'text') {
+            return sanitizeSurrogates(msg.content[0].content);
+        }
+    }
+    // Build full conversation as turns
+    const turns = [];
+    for (const message of context.messages) {
+        if (message.role === 'user') {
+            const content = [];
+            for (const c of message.content) {
+                if (c.type === 'text') {
+                    content.push({ type: 'text', text: sanitizeSurrogates(c.content) });
+                }
+                else if (c.type === 'image' && model.input.includes('image')) {
+                    content.push({
+                        type: 'image',
+                        data: c.data,
+                        mime_type: c.mimeType
+                    });
+                }
+                else if (c.type === 'file' && model.input.includes('file')) {
+                    content.push({
+                        type: 'document',
+                        data: c.data,
+                        mime_type: c.mimeType
+                    });
+                }
+            }
+            if (content.length > 0) {
+                turns.push({ role: 'user', content });
+            }
+        }
+        else if (message.role === 'toolResult') {
+            // Tool results go as user turn with function_result content
+            const result = {
+                type: 'function_result',
+                call_id: message.toolCallId,
+                name: message.toolName,
+                result: message.content
+                    .filter(c => c.type === 'text')
+                    .map(c => c.content)
+                    .join('\n') || '',
+                is_error: message.isError
+            };
+            turns.push({ role: 'user', content: [result] });
+        }
+        else if (message.role === 'assistant') {
+            // Convert assistant message to model turn
+            const content = [];
+            // Check if this is a google-interactions native message
+            if (message.model.api === 'google-interactions') {
+                const baseMessage = message;
+                // Use outputs directly if available
+                if (baseMessage.message?.outputs) {
+                    turns.push({
+                        role: 'model',
+                        content: baseMessage.message.outputs
+                    });
+                    continue;
+                }
+            }
+            // Convert from normalized content
+            for (const block of message.content) {
+                if (block.type === 'thinking') {
+                    // Wrap thinking in tags for cross-provider context
+                    content.push({
+                        type: 'text',
+                        text: `<thinking>${sanitizeSurrogates(block.thinkingText)}</thinking>`
+                    });
+                }
+                else if (block.type === 'response') {
+                    const textContent = block.content
+                        .filter(c => c.type === 'text')
+                        .map(c => sanitizeSurrogates(c.content))
+                        .join('');
+                    if (textContent) {
+                        content.push({ type: 'text', text: textContent });
+                    }
+                }
+                else if (block.type === 'toolCall') {
+                    content.push({
+                        type: 'function_call',
+                        id: block.toolCallId,
+                        name: block.name,
+                        arguments: block.arguments
+                    });
+                }
+            }
+            if (content.length > 0) {
+                turns.push({ role: 'model', content });
+            }
+        }
+    }
+    return turns;
+}
+export function convertTools(tools) {
+    return tools.map((tool) => ({
+        type: 'function',
+        name: tool.name,
+        description: tool.description,
+        parameters: tool.parameters, // TypeBox already generates JSON Schema
+    }));
+}
+export function getMockInteraction() {
+    return {
+        id: 'interaction_mock',
+        status: 'completed',
+        outputs: [],
+        usage: {
+            total_input_tokens: 0,
+            total_output_tokens: 0,
+            total_tokens: 0
+        }
+    };
+}
+//# sourceMappingURL=utils.js.map

package/dist/providers/google-interactions/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/providers/google-interactions/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAgB,MAAM,eAAe,CAAC;AAW1D,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,OAAO,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AAuBrE,MAAM,UAAU,YAAY,CAAC,KAAmC,EAAE,MAAe;IAChF,IAAI,CAAC,MAAM,EAAE,CAAC;QACb,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CACd,gGAAgG,CAChG,CAAC;QACH,CAAC;QACD,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IACrC,CAAC;IACD,OAAO,IAAI,WAAW,CAAC;QACtB,MAAM;QACN,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS;KACnE,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,WAAwB;IACpE,MAAM,iBAAiB,GAAsB,EAAE,CAAC;IAEhD,IAAI,CAAC,WAAW,CAAC,OAAO;QAAE,OAAO,iBAAiB,CAAC;IAEnD,KAAK,MAAM,MAAM,IAAI,WAAW,CAAC,OAAO,EAAE,CAAC;QAC1C,MAAM,UAAU,GAAI,MAAc,CAAC,IAAI,CAAC;QAExC,sBAAsB;QACtB,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;YAC3B,MAAM,UAAU,GAAG,MAAyC,CAAC;YAC7D,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;gBACrB,iBAAiB,CAAC,IAAI,CAAC;oBACtB,IAAI,EAAE,UAAU;oBAChB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC;iBACrD,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,kCAAkC;aAC7B,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;YACnC,MAAM,aAAa,GAAG,MAA+E,CAAC;YACtG,MAAM,YAAY,GAAG,aAAa,CAAC,OAAO;gBACzC,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBAC/B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;iBACtB,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YAErB,IAAI,YAAY,EAAE,CAAC;gBAClB,iBAAiB,CAAC,IAAI,CAAC;oBACtB,IAAI,EAAE,UAAU;oBAChB,YAAY;iBACZ,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,wBAAwB;aACnB,IAAI,UAAU,KAAK,eAAe,EAAE,CAAC;YACzC,MAAM,QAAQ,GAAG,MAAiG,CAAC;YACnH,iBAAiB,CAAC,IAAI,CAAC;gBACtB,IAAI,EAAE,UAAU;gBAChB,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,SAAS,EAAE,QAAQ,CAAC,SAAgC;aACpD,CAAC,CAAC;QACJ,CAAC;QACD,uBAAuB;aAClB,IAAI,UAAU,KAAK,OAAO,EAAE,CAAC;YACjC,MAAM,SAAS,GAAG,MAA8D,CAAC;YACjF,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC;gBACpB,iBAAiB,CAAC,IAAI,CAAC;oBACtB,IAAI,EAAE,UAAU;oBAChB,OAAO,EAAE,CAAC;4BACT,IAAI,EAAE,OAAO;4BACb,IAAI,EAAE,SAAS,CAAC,IAAI;4BACpB,QAAQ,EAAE,SAAS,CAAC,SAAS,IAAI,WAAW;yBAC5C,CAAC;iBACF,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QACD,sEAAsE;QACtE,uDAAuD;IACxD,CAAC;IAED,OAAO,iBAAiB,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,WAAwB,EAAE,KAAmC;IAC7F,MAAM,gBAAgB,GAAG,WAAW,CAAC,KAMxB,CAAC;IAEd,MAAM,KAAK,GAAU;QACpB,KAAK,EAAE,CAAC,gBAAgB,EAAE,kBAAkB,IAAI,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,mBAAmB,IAAI,CAAC,CAAC;QACjG,MAAM,EAAE,CAAC,gBAAgB,EAAE,mBAAmB,IAAI,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,sBAAsB,IAAI,CAAC,CAAC;QACtG,SAAS,EAAE,gBAAgB,EAAE,mBAAmB,IAAI,CAAC;QACrD,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,gBAAgB,EAAE,YAAY,IAAI,CAAC;QAChD,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;IAEF,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC5B,OAAO,KAAK,CAAC;AACd,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAA6B;IACjE,QAAQ,MAAM,EAAE,CAAC;QAChB,KAAK,WAAW;YACf,OAAO,MAAM,CAAC;QACf,KAAK,aAAa;YACjB,OAAO,MAAM,CAAC,CAAC,qCAAqC;QACrD,KAAK,iBAAiB;YACrB,OAAO,SAAS,CAAC;QAClB,KAAK,QAAQ;YACZ,OAAO,OAAO,CAAC;QAChB,KAAK,WAAW;YACf,OAAO,SAAS,CAAC;QAClB;YACC,OAAO,MAAM,CAAC;IAChB,CAAC;AACF,CAAC;AAED,MAAM,UAAU,WAAW,CAC1B,KAAmC,EACnC,OAAgB,EAChB,OAA0C;IAE1C,MAAM,KAAK,GAAG,qBAAqB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAEpD,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,kBAAkB,EAAE,GAAG,OAAO,CAAC;IAE1D,oBAAoB;IACpB,MAAM,MAAM,GAAwB;QACnC,GAAG,kBAAkB;QACrB,KAAK;KACL,CAAC;IAEF,2CAA2C;IAC3C,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QACnB,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC;IACzB,CAAC;SAAM,CAAC;QACP,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC;IACzB,CAAC;IAED,oCAAoC;IACpC,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;QAC1B,MAAM,CAAC,kBAAkB,GAAG,kBAAkB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACtE,CAAC;IAED,qCAAqC;IACrC,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,CAAC,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC5C,CAAC;IAED,OAAO,MAA+C,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,qBAAqB,CACpC,KAAmC,EACnC,OAAgB;IAEhB,+BAA+B;IAC/B,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnC,OAAO,EAAE,CAAC;IACX,CAAC;IAED,2DAA2D;IAC3D,IAAI,OAAO,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC1E,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAChC,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACvF,OAAO,kBAAkB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC;IACF,CAAC;IAED,mCAAmC;IACnC,MAAM,KAAK,GAAW,EAAE,CAAC;IAEzB,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QACxC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC7B,MAAM,OAAO,GAA6B,EAAE,CAAC;YAE7C,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACjC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACvB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBACrE,CAAC;qBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBAChE,OAAO,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,SAAS,EAAE,CAAC,CAAC,QAAQ;qBACrB,CAAC,CAAC;gBACJ,CAAC;qBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC9D,OAAO,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,UAAU;wBAChB,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,SAAS,EAAE,CAAC,CAAC,QAAQ;qBACrB,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;YACvC,CAAC;QACF,CAAC;aACI,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACxC,4DAA4D;YAC5D,MAAM,MAAM,GAA2B;gBACtC,IAAI,EAAE,iBAAiB;gBACvB,OAAO,EAAE,OAAO,CAAC,UAAU;gBAC3B,IAAI,EAAE,OAAO,CAAC,QAAQ;gBACtB,MAAM,EAAE,OAAO,CAAC,OAAO;qBACrB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;qBAC9B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAE,CAAoB,CAAC,OAAO,CAAC;qBACvC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAClB,QAAQ,EAAE,OAAO,CAAC,OAAO;aACzB,CAAC;YAEF,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACjD,CAAC;aACI,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACvC,0CAA0C;YAC1C,MAAM,OAAO,GAA6B,EAAE,CAAC;YAE7C,wDAAwD;YACxD,IAAI,OAAO,CAAC,KAAK,CAAC,GAAG,KAAK,qBAAqB,EAAE,CAAC;gBACjD,MAAM,WAAW,GAAG,OAAsD,CAAC;gBAC3E,oCAAoC;gBACpC,IAAI,WAAW,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;oBAClC,KAAK,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,OAAO;wBACb,OAAO,EAAE,WAAW,CAAC,OAAO,CAAC,OAAmC;qBAChE,CAAC,CAAC;oBACH,SAAS;gBACV,CAAC;YACF,CAAC;YAED,kCAAkC;YAClC,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACrC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC/B,mDAAmD;oBACnD,OAAO,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,aAAa,kBAAkB,CAAC,KAAK,CAAC,YAAY,CAAC,aAAa;qBACtE,CAAC,CAAC;gBACJ,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACtC,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO;yBAC/B,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;yBAC9B,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,kBAAkB,CAAE,CAAoB,CAAC,OAAO,CAAC,CAAC;yBAC3D,IAAI,CAAC,EAAE,CAAC,CAAC;oBAEX,IAAI,WAAW,EAAE,CAAC;wBACjB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;oBACnD,CAAC;gBACF,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACtC,OAAO,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,eAAe;wBACrB,EAAE,EAAE,KAAK,CAAC,UAAU;wBACpB,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,SAAS,EAAE,KAAK,CAAC,SAAS;qBAC1B,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YACxC,CAAC;QACF,CAAC;IACF,CAAC;IAED,OAAO,KAAK,CAAC;AACd,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAsB;IAClD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC3B,IAAI,EAAE,UAAmB;QACzB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,wCAAwC;KACrE,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,kBAAkB;IACjC,OAAO;QACN,EAAE,EAAE,kBAAkB;QACtB,MAAM,EAAE,WAAW;QACnB,OAAO,EAAE,EAAE;QACX,KAAK,EAAE;YACN,kBAAkB,EAAE,CAAC;YACrB,mBAAmB,EAAE,CAAC;YACtB,YAAY,EAAE,CAAC;SACf;KACc,CAAC;AAClB,CAAC"}

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

@@ -0,0 +1,3 @@
+import { CompleteFunction } from "../../types.js";
+export declare const completeZai: CompleteFunction<'zai'>;
+//# sourceMappingURL=complete.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"complete.d.ts","sourceRoot":"","sources":["../../../src/providers/zai/complete.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAA4B,MAAM,gBAAgB,CAAC;AAM5E,eAAO,MAAM,WAAW,EAAE,gBAAgB,CAAC,KAAK,CAoE/C,CAAC"}

package/dist/providers/zai/complete.js

@@ -0,0 +1,58 @@
+import { createClient, buildParams, getResponseAssistantResponse, getResponseUsage, getAssistantStopReason } from "./utils.js";
+export const completeZai = async (model, context, options, id) => {
+    const startTimestamp = Date.now();
+    const client = createClient(model, options?.apiKey);
+    const params = buildParams(model, context, options);
+    try {
+        const response = await client.chat.completions.create(params, { signal: options?.signal });
+        // Cache processed content for performance and consistency
+        const content = getResponseAssistantResponse(response);
+        const usage = getResponseUsage(response, model);
+        let stopReason = getAssistantStopReason(response);
+        // Ensure stopReason is toolUse if we have tool calls
+        const hasToolCall = content.some(c => c.type === 'toolCall');
+        if (hasToolCall && stopReason === 'stop') {
+            stopReason = 'toolUse';
+        }
+        return {
+            role: "assistant",
+            message: response,
+            id,
+            api: model.api,
+            model,
+            timestamp: Date.now(),
+            duration: Date.now() - startTimestamp,
+            stopReason,
+            content,
+            usage
+        };
+    }
+    catch (error) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        const isAborted = options.signal?.aborted;
+        const stopReason = isAborted ? "aborted" : "error";
+        // Return error response with empty content and zero usage
+        const emptyUsage = {
+            input: 0,
+            output: 0,
+            cacheRead: 0,
+            cacheWrite: 0,
+            totalTokens: 0,
+            cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
+        };
+        return {
+            role: "assistant",
+            message: {}, // Empty response object for error case
+            id,
+            api: model.api,
+            model,
+            errorMessage,
+            timestamp: Date.now(),
+            duration: Date.now() - startTimestamp,
+            stopReason,
+            content: [],
+            usage: emptyUsage
+        };
+    }
+};
+//# sourceMappingURL=complete.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"complete.js","sourceRoot":"","sources":["../../../src/providers/zai/complete.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,4BAA4B,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAG/H,MAAM,CAAC,MAAM,WAAW,GAA4B,KAAK,EACxD,KAAmB,EACnB,OAAO,EACP,OAA2B,EAC3B,EAAU,EACT,EAAE;IACH,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAClC,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IACpD,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAEpD,IAAI,CAAC;QACJ,MAAM,QAAQ,GAAmB,MAAM,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CACpE,MAAM,EACN,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAC3B,CAAC;QAEF,0DAA0D;QAC1D,MAAM,OAAO,GAAG,4BAA4B,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAChD,IAAI,UAAU,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QAElD,qDAAqD;QACrD,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QAC7D,IAAI,WAAW,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;YAC1C,UAAU,GAAG,SAAS,CAAC;QACxB,CAAC;QAED,OAAO;YACN,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,QAAQ;YACjB,EAAE;YACF,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK;YACL,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;YACrC,UAAU;YACV,OAAO;YACP,KAAK;SACL,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC;QAC1C,MAAM,UAAU,GAAe,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;QAE/D,0DAA0D;QAC1D,MAAM,UAAU,GAAU;YACzB,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;YACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;SACpE,CAAC;QAEF,OAAO;YACN,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAoB,EAAE,uCAAuC;YACtE,EAAE;YACF,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK;YACL,YAAY;YACZ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;YACrC,UAAU;YACV,OAAO,EAAE,EAAE;YACX,KAAK,EAAE,UAAU;SACjB,CAAC;IACH,CAAC;AACF,CAAC,CAAC"}

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

@@ -0,0 +1,5 @@
+export { completeZai } from "./complete.js";
+export { streamZai } from "./stream.js";
+export { ZaiProviderOptions, ZaiThinkingConfig } from "./types.js";
+export { getMockZaiMessage } from "./utils.js";
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/providers/zai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACnE,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC"}

package/dist/providers/zai/index.js

@@ -0,0 +1,4 @@
+export { completeZai } from "./complete.js";
+export { streamZai } from "./stream.js";
+export { getMockZaiMessage } from "./utils.js";
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/providers/zai/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAExC,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC"}

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

@@ -0,0 +1,3 @@
+import { StreamFunction } from "../../types.js";
+export declare const streamZai: StreamFunction<'zai'>;
+//# sourceMappingURL=stream.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../../src/providers/zai/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAA0I,cAAc,EAAe,MAAM,gBAAgB,CAAA;AAoBpM,eAAO,MAAM,SAAS,EAAE,cAAc,CAAC,KAAK,CA4W3C,CAAA"}