@ank1015/providers 0.0.37 → 0.0.39

package/dist/providers/cerebras/stream.js

@@ -0,0 +1,321 @@
+import { createClient, buildParams, mapStopReason, getMockCerebrasMessage } 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 streamCerebras = (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 = getMockCerebrasMessage();
+        let accumulatedContent = '';
+        let accumulatedReasoning = '';
+        const accumulatedToolCalls = new Map();
+        try {
+            const client = createClient(model, options?.apiKey);
+            const params = buildParams(model, context, options);
+            const cerebrasStream = 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 cerebrasStream) {
+                const choice = chunk.choices[0];
+                if (!choice)
+                    continue;
+                const delta = choice.delta;
+                // Handle reasoning content (Cerebras uses 'reasoning' field when reasoning_format=parsed)
+                if (delta.reasoning) {
+                    accumulatedReasoning += delta.reasoning;
+                    // 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;
+                        stream.push({
+                            type: "thinking_delta",
+                            contentIndex: blockIndex(),
+                            delta: delta.reasoning,
+                            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 to the message if present
+            if (accumulatedReasoning) {
+                finalResponse.choices[0].message.reasoning = accumulatedReasoning;
+            }
+            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/cerebras/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../src/providers/cerebras/stream.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,aAAa,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAC9F,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,cAAc,GAA+B,CACzD,KAAwB,EACxB,OAAgB,EAChB,OAAgC,EAChC,EAAU,EACT,EAAE;IAEH,MAAM,MAAM,GAAG,IAAI,2BAA2B,EAAc,CAAC;IAE7D,CAAC,KAAK,IAAI,EAAE;QAEX,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAClC,MAAM,MAAM,GAA0C;YACrD,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,sBAAsB,EAAE,CAAC;QAC7D,IAAI,kBAAkB,GAAG,EAAE,CAAC;QAC5B,IAAI,oBAAoB,GAAG,EAAE,CAAC;QAC9B,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,cAAc,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAC1D,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,cAAc,EAAE,CAAC;gBAC1C,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAChC,IAAI,CAAC,MAAM;oBAAE,SAAS;gBAEtB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAiD,CAAC;gBAEvE,0FAA0F;gBAC1F,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;oBACrB,oBAAoB,IAAI,KAAK,CAAC,SAAS,CAAC;oBAExC,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,SAAS,CAAC;wBAC7C,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,gBAAgB;4BACtB,YAAY,EAAE,UAAU,EAAE;4BAC1B,KAAK,EAAE,KAAK,CAAC,SAAS;4BACtB,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,KAA2C,CAAC;oBAChE,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,0CAA0C;YAC1C,IAAI,oBAAoB,EAAE,CAAC;gBACzB,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAe,CAAC,SAAS,GAAG,oBAAoB,CAAC;YAC5E,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,GAAqC;gBAC9D,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,GAAqC;gBAC9D,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/cerebras/types.d.ts

@@ -0,0 +1,14 @@
+import type { ChatCompletionCreateParamsBase } from "openai/resources/chat/completions.js";
+export type CerebrasReasoningFormat = "parsed" | "raw" | "hidden" | "none";
+export type CerebrasReasoningEffort = "low" | "medium" | "high";
+type Props = {
+    apiKey?: string;
+    signal?: AbortSignal;
+    reasoning_format?: CerebrasReasoningFormat;
+    reasoning_effort?: CerebrasReasoningEffort;
+    disable_reasoning?: boolean;
+    clear_thinking?: boolean;
+};
+export type CerebrasProviderOptions = Omit<ChatCompletionCreateParamsBase, 'model' | 'messages'> & Props;
+export {};
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/providers/cerebras/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,sCAAsC,CAAA;AAE1F,MAAM,MAAM,uBAAuB,GAAG,QAAQ,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;AAC3E,MAAM,MAAM,uBAAuB,GAAG,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;AAEhE,KAAK,KAAK,GAAG;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,WAAW,CAAC;IAErB,gBAAgB,CAAC,EAAE,uBAAuB,CAAC;IAE3C,gBAAgB,CAAC,EAAE,uBAAuB,CAAC;IAE3C,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAE5B,cAAc,CAAC,EAAE,OAAO,CAAC;CACzB,CAAA;AAED,MAAM,MAAM,uBAAuB,GAAG,IAAI,CAAC,8BAA8B,EAAE,OAAO,GAAG,UAAU,CAAC,GAAG,KAAK,CAAA"}

package/dist/providers/cerebras/types.js

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

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

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

package/dist/providers/cerebras/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 { CerebrasProviderOptions } from "./types.js";
+export declare function createClient(model: Model<"cerebras">, 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<'cerebras'>): Usage;
+export declare function buildParams(model: Model<"cerebras">, context: Context, options: CerebrasProviderOptions): ChatCompletionCreateParamsNonStreaming & {
+    reasoning_format?: string;
+    reasoning_effort?: string;
+    disable_reasoning?: boolean;
+    clear_thinking?: boolean;
+};
+export declare function buildCerebrasMessages(_model: Model<'cerebras'>, context: Context): ChatCompletionMessageParam[];
+export declare function convertTools(tools: readonly Tool[]): ChatCompletionTool[];
+export declare function mapStopReason(finishReason: string | null | undefined): StopReason;
+export declare function getMockCerebrasMessage(): ChatCompletion;
+//# sourceMappingURL=utils.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/providers/cerebras/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,uBAAuB,EAAE,MAAM,YAAY,CAAC;AAarD,wBAAgB,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,EAAE,MAAM,UAkBrE;AAED,wBAAgB,4BAA4B,CAAC,QAAQ,EAAE,cAAc,GAAG,iBAAiB,CA4CxF;AAED,wBAAgB,sBAAsB,CAAC,QAAQ,EAAE,cAAc,GAAG,UAAU,CAG3E;AAED,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,cAAc,EAAE,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,GAAG,KAAK,CAgB1F;AAED,wBAAgB,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,uBAAuB;uBAcnF,MAAM;uBACN,MAAM;wBACL,OAAO;qBACV,OAAO;EAkDzB;AAED,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,OAAO,GAAG,0BAA0B,EAAE,CAmG/G;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,GAAG,kBAAkB,EAAE,CAmBzE;AAED,wBAAgB,aAAa,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,GAAG,UAAU,CAejF;AAED,wBAAgB,sBAAsB,IAAI,cAAc,CAsBvD"}

package/dist/providers/cerebras/utils.js

@@ -0,0 +1,274 @@
+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.CEREBRAS_API_KEY) {
+            throw new Error("Cerebras API key is required. Set CEREBRAS_API_KEY environment variable or pass it as an argument.");
+        }
+        apiKey = process.env.CEREBRAS_API_KEY;
+    }
+    return new OpenAI({
+        apiKey,
+        baseURL: model.baseUrl,
+        dangerouslyAllowBrowser: true,
+        defaultHeaders: {
+            "User-Agent": "cerebras-providers-sdk",
+            ...model.headers,
+        },
+    });
+}
+export function getResponseAssistantResponse(response) {
+    const assistantResponse = [];
+    const choice = response.choices[0];
+    if (!choice?.message) {
+        return assistantResponse;
+    }
+    const message = choice.message;
+    // Handle reasoning content (Cerebras uses 'reasoning' field when reasoning_format=parsed)
+    if (message.reasoning) {
+        assistantResponse.push({
+            type: 'thinking',
+            thinkingText: message.reasoning
+        });
+    }
+    // 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) {
+            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;
+    // Cerebras reports cache hits via prompt_tokens_details.cached_tokens
+    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 = buildCerebrasMessages(model, context);
+    const { apiKey, signal, reasoning_format, reasoning_effort, disable_reasoning, clear_thinking, ...cerebrasOptions } = options;
+    const params = {
+        ...cerebrasOptions,
+        model: model.id,
+        messages,
+        stream: false
+    };
+    // Add reasoning format - default to 'parsed' to get structured reasoning
+    if (reasoning_format) {
+        params.reasoning_format = reasoning_format;
+    }
+    else if (model.reasoning) {
+        // Default to parsed for reasoning models
+        params.reasoning_format = 'parsed';
+    }
+    // Model-specific reasoning parameters
+    if (reasoning_effort !== undefined) {
+        // GPT-OSS specific
+        params.reasoning_effort = reasoning_effort;
+    }
+    if (disable_reasoning !== undefined) {
+        // GLM specific
+        params.disable_reasoning = disable_reasoning;
+    }
+    if (clear_thinking !== undefined) {
+        // GLM specific
+        params.clear_thinking = clear_thinking;
+    }
+    // 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 (cerebrasOptions.tools) {
+            for (const optionTool of cerebrasOptions.tools) {
+                tools.push(optionTool);
+            }
+        }
+        params.tools = tools;
+    }
+    return params;
+}
+export function buildCerebrasMessages(_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') {
+            // Cerebras 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 === 'cerebras') {
+                // Native Cerebras 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 = [];
+                for (const contentBlock of message.content) {
+                    if (contentBlock.type === 'thinking') {
+                        // For Cerebras, we include thinking in <think> tags for GLM/Qwen compatibility
+                        textContent = `<think>${sanitizeSurrogates(contentBlock.thinkingText)}</think>` + textContent;
+                    }
+                    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',
+                    content: textContent || null
+                };
+                if (toolCalls.length > 0) {
+                    assistantMessage.tool_calls = toolCalls;
+                }
+                messages.push(assistantMessage);
+            }
+        }
+    }
+    return messages;
+}
+export function convertTools(tools) {
+    return tools.map((tool) => {
+        const params = tool.parameters;
+        // Add additionalProperties: false for Cerebras strict mode compatibility
+        const parameters = {
+            ...params,
+            additionalProperties: false
+        };
+        return {
+            type: "function",
+            function: {
+                name: tool.name,
+                description: tool.description,
+                parameters,
+                strict: true
+            }
+        };
+    });
+}
+export function mapStopReason(finishReason) {
+    if (!finishReason)
+        return "stop";
+    switch (finishReason) {
+        case "stop":
+            return "stop";
+        case "length":
+            return "length";
+        case "tool_calls":
+            return "toolUse";
+        case "content_filter":
+            return "error";
+        default:
+            return "stop";
+    }
+}
+export function getMockCerebrasMessage() {
+    return {
+        id: "chatcmpl-123",
+        object: "chat.completion",
+        created: Math.floor(Date.now() / 1000),
+        model: "zai-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