@ank1015/providers 0.0.1

package/dist/providers/openai.js

@@ -0,0 +1,396 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.streamOpenAI = void 0;
+// Return a abstracted event stream
+// Return the final response as it is.
+const openai_1 = __importDefault(require("openai"));
+const event_stream_1 = require("../utils/event-stream");
+const convert_1 = require("./convert");
+const json_parse_1 = require("../utils/json-parse");
+const validation_1 = require("../utils/validation");
+const models_1 = require("../models");
+// takes in model, built in message
+const streamOpenAI = (model, context, options) => {
+    const stream = new event_stream_1.AssistantMessageEventStream();
+    // Start async processing
+    (async () => {
+        const output = {
+            role: "assistant",
+            content: [],
+            api: "openai",
+            model: model.id,
+            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: Date.now(),
+        };
+        let finalResponse = {
+            id: "resp_123",
+            object: "response",
+            created_at: 1740855869,
+            output_text: '',
+            status: "completed",
+            incomplete_details: null,
+            parallel_tool_calls: false,
+            error: null,
+            instructions: null,
+            max_output_tokens: null,
+            model: "gpt-4o-mini-2024-07-18",
+            output: [],
+            previous_response_id: null,
+            temperature: 1,
+            text: {},
+            tool_choice: "auto",
+            tools: [],
+            top_p: 1,
+            truncation: "disabled",
+            usage: {
+                input_tokens: 0,
+                output_tokens: 0,
+                output_tokens_details: {
+                    reasoning_tokens: 0
+                },
+                input_tokens_details: {
+                    cached_tokens: 0
+                },
+                total_tokens: 0
+            },
+            user: undefined,
+            metadata: {}
+        };
+        try {
+            const client = createClient(model, options?.apiKey);
+            const params = buildParams(model, context, options);
+            const openaiStream = await client.responses.create(params, { signal: options?.signal });
+            stream.push({ type: "start", partial: output });
+            let currentItem = null;
+            let currentBlock = null;
+            const blocks = output.content;
+            const blockIndex = () => blocks.length - 1;
+            for await (const event of openaiStream) {
+                // Handle output item start
+                if (event.type === "response.output_item.added") {
+                    const item = event.item;
+                    if (item.type === "reasoning") {
+                        currentItem = item;
+                        currentBlock = { type: "thinking", thinking: "" };
+                        output.content.push(currentBlock);
+                        stream.push({ type: "thinking_start", contentIndex: blockIndex(), partial: output });
+                    }
+                    else if (item.type === "message") {
+                        currentItem = item;
+                        currentBlock = { type: "text", text: "" };
+                        output.content.push(currentBlock);
+                        stream.push({ type: "text_start", contentIndex: blockIndex(), partial: output });
+                    }
+                    else if (item.type === "function_call") {
+                        currentItem = item;
+                        currentBlock = {
+                            type: "toolCall",
+                            id: item.call_id,
+                            name: item.name,
+                            arguments: {},
+                            partialJson: item.arguments || "",
+                        };
+                        output.content.push(currentBlock);
+                        stream.push({ type: "toolcall_start", contentIndex: blockIndex(), partial: output });
+                    }
+                }
+                // Handle reasoning summary deltas
+                else if (event.type === "response.reasoning_summary_part.added") {
+                    if (currentItem && currentItem.type === "reasoning") {
+                        currentItem.summary = currentItem.summary || [];
+                        currentItem.summary.push(event.part);
+                    }
+                }
+                else if (event.type === "response.reasoning_summary_text.delta") {
+                    if (currentItem &&
+                        currentItem.type === "reasoning" &&
+                        currentBlock &&
+                        currentBlock.type === "thinking") {
+                        currentItem.summary = currentItem.summary || [];
+                        const lastPart = currentItem.summary[currentItem.summary.length - 1];
+                        if (lastPart) {
+                            currentBlock.thinking += event.delta;
+                            lastPart.text += event.delta;
+                            stream.push({
+                                type: "thinking_delta",
+                                contentIndex: blockIndex(),
+                                delta: event.delta,
+                                partial: output,
+                            });
+                        }
+                    }
+                }
+                // Add a new line between summary parts (hack...)
+                else if (event.type === "response.reasoning_summary_part.done") {
+                    if (currentItem &&
+                        currentItem.type === "reasoning" &&
+                        currentBlock &&
+                        currentBlock.type === "thinking") {
+                        currentItem.summary = currentItem.summary || [];
+                        const lastPart = currentItem.summary[currentItem.summary.length - 1];
+                        if (lastPart) {
+                            currentBlock.thinking += "\n\n";
+                            lastPart.text += "\n\n";
+                            stream.push({
+                                type: "thinking_delta",
+                                contentIndex: blockIndex(),
+                                delta: "\n\n",
+                                partial: output,
+                            });
+                        }
+                    }
+                }
+                // Handle text output deltas
+                else if (event.type === "response.content_part.added") {
+                    if (currentItem && currentItem.type === "message") {
+                        currentItem.content = currentItem.content || [];
+                        currentItem.content.push(event.part);
+                    }
+                }
+                else if (event.type === "response.output_text.delta") {
+                    if (currentItem && currentItem.type === "message" && currentBlock && currentBlock.type === "text") {
+                        const lastPart = currentItem.content[currentItem.content.length - 1];
+                        if (lastPart && lastPart.type === "output_text") {
+                            currentBlock.text += event.delta;
+                            lastPart.text += event.delta;
+                            stream.push({
+                                type: "text_delta",
+                                contentIndex: blockIndex(),
+                                delta: event.delta,
+                                partial: output,
+                            });
+                        }
+                    }
+                }
+                else if (event.type === "response.refusal.delta") {
+                    if (currentItem && currentItem.type === "message" && currentBlock && currentBlock.type === "text") {
+                        const lastPart = currentItem.content[currentItem.content.length - 1];
+                        if (lastPart && lastPart.type === "refusal") {
+                            currentBlock.text += event.delta;
+                            lastPart.refusal += event.delta;
+                            stream.push({
+                                type: "text_delta",
+                                contentIndex: blockIndex(),
+                                delta: event.delta,
+                                partial: output,
+                            });
+                        }
+                    }
+                }
+                // Handle function call argument deltas
+                else if (event.type === "response.function_call_arguments.delta") {
+                    if (currentItem &&
+                        currentItem.type === "function_call" &&
+                        currentBlock &&
+                        currentBlock.type === "toolCall") {
+                        currentBlock.partialJson += event.delta;
+                        currentBlock.arguments = (0, json_parse_1.parseStreamingJson)(currentBlock.partialJson);
+                        stream.push({
+                            type: "toolcall_delta",
+                            contentIndex: blockIndex(),
+                            delta: event.delta,
+                            partial: output,
+                        });
+                    }
+                }
+                // Handle output item completion
+                else if (event.type === "response.output_item.done") {
+                    const item = event.item;
+                    if (item.type === "reasoning" && currentBlock && currentBlock.type === "thinking") {
+                        currentBlock.thinking = item.summary?.map((s) => s.text).join("\n\n") || "";
+                        stream.push({
+                            type: "thinking_end",
+                            contentIndex: blockIndex(),
+                            content: currentBlock.thinking,
+                            partial: output,
+                        });
+                        currentBlock = null;
+                    }
+                    else if (item.type === "message" && currentBlock && currentBlock.type === "text") {
+                        currentBlock.text = item.content.map((c) => (c.type === "output_text" ? c.text : c.refusal)).join("");
+                        stream.push({
+                            type: "text_end",
+                            contentIndex: blockIndex(),
+                            content: currentBlock.text,
+                            partial: output,
+                        });
+                        currentBlock = null;
+                    }
+                    else if (item.type === "function_call") {
+                        const toolCall = {
+                            type: "toolCall",
+                            id: item.call_id,
+                            name: item.name,
+                            arguments: JSON.parse(item.arguments),
+                        };
+                        // Validate tool arguments if tool definition is available
+                        if (context.tools) {
+                            const tool = context.tools.find((t) => t.name === toolCall.name);
+                            if (tool) {
+                                toolCall.arguments = (0, validation_1.validateToolArguments)(tool, toolCall);
+                            }
+                        }
+                        stream.push({ type: "toolcall_end", contentIndex: blockIndex(), toolCall, partial: output });
+                    }
+                }
+                // Handle completion
+                else if (event.type === "response.completed") {
+                    const response = event.response;
+                    // Update the final Response
+                    finalResponse = response;
+                    if (response?.usage) {
+                        const cachedTokens = response.usage.input_tokens_details?.cached_tokens || 0;
+                        output.usage = {
+                            // OpenAI includes cached tokens in input_tokens, so subtract to get non-cached input
+                            input: (response.usage.input_tokens || 0) - cachedTokens,
+                            output: response.usage.output_tokens || 0,
+                            cacheRead: cachedTokens,
+                            cacheWrite: 0,
+                            totalTokens: response.usage.total_tokens || 0,
+                            cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+                        };
+                    }
+                    (0, models_1.calculateCost)(model, output.usage);
+                    // Map status to stop reason
+                    output.stopReason = mapStopReason(response?.status);
+                    if (output.content.some((b) => b.type === "toolCall") && output.stopReason === "stop") {
+                        output.stopReason = "toolUse";
+                    }
+                }
+                // Handle errors
+                else if (event.type === "error") {
+                    throw new Error(`OpenAI API Error (${event.code}): ${event.message}` || "Unknown OpenAI error");
+                }
+                else if (event.type === "response.failed") {
+                    throw new Error("OpenAI response failed without error details");
+                }
+            }
+            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}` : ""}`);
+            }
+            stream.push({ type: "done", reason: output.stopReason, message: output });
+            stream.end({
+                _provider: 'openai',
+                role: 'assistant',
+                message: finalResponse
+            });
+        }
+        catch (error) {
+            for (const block of output.content)
+                delete block.index;
+            output.stopReason = options?.signal?.aborted ? "aborted" : "error";
+            output.errorMessage = error instanceof Error ? error.message : JSON.stringify(error);
+            stream.push({ type: "error", reason: output.stopReason, error: output });
+            // Update finalResponse to reflect the error state
+            finalResponse.status = options?.signal?.aborted ? "cancelled" : "failed";
+            finalResponse.error = error instanceof Error ? {
+                message: error.message,
+                code: error.code || "unknown_error",
+                type: error.name || "Error"
+            } : { message: String(error) };
+            stream.end({
+                _provider: 'openai',
+                role: 'assistant',
+                message: finalResponse
+            });
+        }
+    })();
+    return stream;
+};
+exports.streamOpenAI = streamOpenAI;
+function createClient(model, apiKey) {
+    if (!apiKey) {
+        if (!process.env.OPENAI_API_KEY) {
+            throw new Error("OpenAI API key is required. Set OPENAI_API_KEY environment variable or pass it as an argument.");
+        }
+        apiKey = process.env.OPENAI_API_KEY;
+    }
+    return new openai_1.default({
+        apiKey,
+        baseURL: model.baseUrl,
+        dangerouslyAllowBrowser: true,
+        defaultHeaders: model.headers,
+    });
+}
+function buildParams(model, context, options) {
+    const messages = (0, convert_1.buildOpenAIMessages)(model, context);
+    const params = {
+        include: ['reasoning.encrypted_content'],
+        input: messages,
+        model: model.id,
+        stream: true
+    };
+    if (options?.maxOutputTokens) {
+        params.max_output_tokens = options.maxOutputTokens;
+    }
+    if (options?.parallelToolCalls) {
+        params.parallel_tool_calls = options.parallelToolCalls;
+    }
+    if (options?.prompt_cache_key) {
+        params.prompt_cache_key = options.prompt_cache_key;
+    }
+    if (options?.promptCacheRetention) {
+        params.prompt_cache_retention = options.promptCacheRetention;
+    }
+    if (options?.reasoning) {
+        params.reasoning = {
+            effort: options.reasoning.effort || "medium",
+            summary: options.reasoning.summary || "auto"
+        };
+    }
+    if (options?.temperature) {
+        params.temperature = options.temperature;
+    }
+    if (context.tools) {
+        params.tools = convertTools(context.tools);
+    }
+    if (options?.truncation) {
+        params.truncation = options.truncation;
+    }
+    return params;
+}
+function convertTools(tools) {
+    return tools.map((tool) => ({
+        type: "function",
+        name: tool.name,
+        description: tool.description,
+        parameters: tool.parameters, // TypeBox already generates JSON Schema
+        strict: null,
+    }));
+}
+function mapStopReason(status) {
+    if (!status)
+        return "stop";
+    switch (status) {
+        case "completed":
+            return "stop";
+        case "incomplete":
+            return "length";
+        case "failed":
+        case "cancelled":
+            return "error";
+        // These two are wonky ...
+        case "in_progress":
+        case "queued":
+            return "stop";
+        default: {
+            const _exhaustive = status;
+            throw new Error(`Unhandled stop reason: ${_exhaustive}`);
+        }
+    }
+}
+//# sourceMappingURL=openai.js.map

package/dist/providers/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":";;;;;;AAAA,mCAAmC;AACnC,sCAAsC;AACtC,oDAA4B;AAC5B,wDAAoE;AAEpE,uCAAgD;AAGhD,oDAAyD;AACzD,oDAA4D;AAC5D,sCAA0C;AAiB1C,mCAAmC;AAC5B,MAAM,YAAY,GAA6B,CAClD,KAAsB,EACtB,OAAgB,EAChB,OAA8B,EAChC,EAAE;IAEA,MAAM,MAAM,GAAG,IAAI,0CAA2B,EAAE,CAAC;IAEpD,yBAAyB;IACzB,CAAC,KAAK,IAAI,EAAE;QACX,MAAM,MAAM,GAAqB;YAChC,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,EAAE;YACX,GAAG,EAAE,QAAe;YACpB,KAAK,EAAE,KAAK,CAAC,EAAE;YACf,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,IAAI,CAAC,GAAG,EAAE;SACrB,CAAC;QACF,IAAI,aAAa,GAAa;YAC7B,EAAE,EAAE,UAAU;YACd,MAAM,EAAE,UAAU;YAClB,UAAU,EAAE,UAAU;YACtB,WAAW,EAAE,EAAE;YACf,MAAM,EAAE,WAAW;YACnB,kBAAkB,EAAE,IAAI;YACxB,mBAAmB,EAAE,KAAK;YAC1B,KAAK,EAAE,IAAI;YACX,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;YACvB,KAAK,EAAE,wBAAwB;YAC/B,MAAM,EAAE,EAAE;YACV,oBAAoB,EAAE,IAAI;YAC1B,WAAW,EAAE,CAAC;YACd,IAAI,EAAE,EAAE;YACR,WAAW,EAAE,MAAM;YACnB,KAAK,EAAE,EAAE;YACT,KAAK,EAAE,CAAC;YACR,UAAU,EAAE,UAAU;YACtB,KAAK,EAAE;gBACL,YAAY,EAAE,CAAC;gBACf,aAAa,EAAE,CAAC;gBAChB,qBAAqB,EAAE;oBACxB,gBAAgB,EAAE,CAAC;iBACjB;gBACD,oBAAoB,EAAE;oBACvB,aAAa,EAAE,CAAC;iBACd;gBACD,YAAY,EAAE,CAAC;aAChB;YACD,IAAI,EAAE,SAAS;YACf,QAAQ,EAAE,EAAE;SACV,CAAA;QACH,IAAI,CAAC;YACJ,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;YACpD,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YACxF,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAEhD,IAAI,WAAW,GAAoF,IAAI,CAAC;YACxG,IAAI,YAAY,GAA2G,IAAI,CAAC;YAChI,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC;YAC9B,MAAM,UAAU,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;YAE3C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;gBACxC,2BAA2B;gBAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;oBACjD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;oBACxB,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBAC/B,WAAW,GAAG,IAAI,CAAC;wBACnB,YAAY,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;wBAClD,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,MAAM,EAAE,CAAC,CAAC;oBACtF,CAAC;yBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;wBACpC,WAAW,GAAG,IAAI,CAAC;wBACnB,YAAY,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;wBAC1C,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,MAAM,EAAE,CAAC,CAAC;oBAClF,CAAC;yBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;wBAC1C,WAAW,GAAG,IAAI,CAAC;wBACnB,YAAY,GAAG;4BACd,IAAI,EAAE,UAAU;4BAChB,EAAE,EAAE,IAAI,CAAC,OAAO;4BAChB,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,SAAS,EAAE,EAAE;4BACb,WAAW,EAAE,IAAI,CAAC,SAAS,IAAI,EAAE;yBACjC,CAAC;wBACF,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,MAAM,EAAE,CAAC,CAAC;oBACtF,CAAC;gBACF,CAAC;gBACD,kCAAkC;qBAC7B,IAAI,KAAK,CAAC,IAAI,KAAK,uCAAuC,EAAE,CAAC;oBACjE,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBACrD,WAAW,CAAC,OAAO,GAAG,WAAW,CAAC,OAAO,IAAI,EAAE,CAAC;wBAChD,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACtC,CAAC;gBACF,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,uCAAuC,EAAE,CAAC;oBACnE,IACC,WAAW;wBACX,WAAW,CAAC,IAAI,KAAK,WAAW;wBAChC,YAAY;wBACZ,YAAY,CAAC,IAAI,KAAK,UAAU,EAC/B,CAAC;wBACF,WAAW,CAAC,OAAO,GAAG,WAAW,CAAC,OAAO,IAAI,EAAE,CAAC;wBAChD,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;wBACrE,IAAI,QAAQ,EAAE,CAAC;4BACd,YAAY,CAAC,QAAQ,IAAI,KAAK,CAAC,KAAK,CAAC;4BACrC,QAAQ,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC;4BAC7B,MAAM,CAAC,IAAI,CAAC;gCACX,IAAI,EAAE,gBAAgB;gCACtB,YAAY,EAAE,UAAU,EAAE;gCAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;gCAClB,OAAO,EAAE,MAAM;6BACf,CAAC,CAAC;wBACJ,CAAC;oBACF,CAAC;gBACF,CAAC;gBACD,iDAAiD;qBAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,sCAAsC,EAAE,CAAC;oBAChE,IACC,WAAW;wBACX,WAAW,CAAC,IAAI,KAAK,WAAW;wBAChC,YAAY;wBACZ,YAAY,CAAC,IAAI,KAAK,UAAU,EAC/B,CAAC;wBACF,WAAW,CAAC,OAAO,GAAG,WAAW,CAAC,OAAO,IAAI,EAAE,CAAC;wBAChD,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;wBACrE,IAAI,QAAQ,EAAE,CAAC;4BACd,YAAY,CAAC,QAAQ,IAAI,MAAM,CAAC;4BAChC,QAAQ,CAAC,IAAI,IAAI,MAAM,CAAC;4BACxB,MAAM,CAAC,IAAI,CAAC;gCACX,IAAI,EAAE,gBAAgB;gCACtB,YAAY,EAAE,UAAU,EAAE;gCAC1B,KAAK,EAAE,MAAM;gCACb,OAAO,EAAE,MAAM;6BACf,CAAC,CAAC;wBACJ,CAAC;oBACF,CAAC;gBACF,CAAC;gBACD,4BAA4B;qBACvB,IAAI,KAAK,CAAC,IAAI,KAAK,6BAA6B,EAAE,CAAC;oBACvD,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;wBACnD,WAAW,CAAC,OAAO,GAAG,WAAW,CAAC,OAAO,IAAI,EAAE,CAAC;wBAChD,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,IAAW,CAAC,CAAC;oBAC7C,CAAC;gBACF,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,4BAA4B,EAAE,CAAC;oBACxD,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,IAAI,YAAY,IAAI,YAAY,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACnG,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;wBACrE,IAAI,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;4BACjD,YAAY,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC;4BACjC,QAAQ,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC;4BAC7B,MAAM,CAAC,IAAI,CAAC;gCACX,IAAI,EAAE,YAAY;gCAClB,YAAY,EAAE,UAAU,EAAE;gCAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;gCAClB,OAAO,EAAE,MAAM;6BACf,CAAC,CAAC;wBACJ,CAAC;oBACF,CAAC;gBACF,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,wBAAwB,EAAE,CAAC;oBACpD,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,IAAI,YAAY,IAAI,YAAY,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACnG,MAAM,QAAQ,GAAG,WAAW,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;wBACrE,IAAI,QAAQ,IAAI,QAAQ,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;4BAC7C,YAAY,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC;4BACjC,QAAQ,CAAC,OAAO,IAAI,KAAK,CAAC,KAAK,CAAC;4BAChC,MAAM,CAAC,IAAI,CAAC;gCACX,IAAI,EAAE,YAAY;gCAClB,YAAY,EAAE,UAAU,EAAE;gCAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;gCAClB,OAAO,EAAE,MAAM;6BACf,CAAC,CAAC;wBACJ,CAAC;oBACF,CAAC;gBACF,CAAC;gBACD,uCAAuC;qBAClC,IAAI,KAAK,CAAC,IAAI,KAAK,wCAAwC,EAAE,CAAC;oBAClE,IACC,WAAW;wBACX,WAAW,CAAC,IAAI,KAAK,eAAe;wBACpC,YAAY;wBACZ,YAAY,CAAC,IAAI,KAAK,UAAU,EAC/B,CAAC;wBACF,YAAY,CAAC,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC;wBACxC,YAAY,CAAC,SAAS,GAAG,IAAA,+BAAkB,EAAC,YAAY,CAAC,WAAW,CAAC,CAAC;wBACtE,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,gBAAgB;4BACtB,YAAY,EAAE,UAAU,EAAE;4BAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;4BAClB,OAAO,EAAE,MAAM;yBACf,CAAC,CAAC;oBACJ,CAAC;gBACF,CAAC;gBACD,gCAAgC;qBAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,2BAA2B,EAAE,CAAC;oBACrD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;oBAExB,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,IAAI,YAAY,IAAI,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACnF,YAAY,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;wBAC5E,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,cAAc;4BACpB,YAAY,EAAE,UAAU,EAAE;4BAC1B,OAAO,EAAE,YAAY,CAAC,QAAQ;4BAC9B,OAAO,EAAE,MAAM;yBACf,CAAC,CAAC;wBACH,YAAY,GAAG,IAAI,CAAC;oBACrB,CAAC;yBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,YAAY,IAAI,YAAY,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACpF,YAAY,CAAC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;wBACtG,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,UAAU;4BAChB,YAAY,EAAE,UAAU,EAAE;4BAC1B,OAAO,EAAE,YAAY,CAAC,IAAI;4BAC1B,OAAO,EAAE,MAAM;yBACf,CAAC,CAAC;wBACH,YAAY,GAAG,IAAI,CAAC;oBACrB,CAAC;yBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;wBAC1C,MAAM,QAAQ,GAAsB;4BACnC,IAAI,EAAE,UAAU;4BAChB,EAAE,EAAE,IAAI,CAAC,OAAO;4BAChB,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;yBACrC,CAAC;wBAEF,0DAA0D;wBAC1D,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;4BACnB,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC;4BACjE,IAAI,IAAI,EAAE,CAAC;gCACV,QAAQ,CAAC,SAAS,GAAG,IAAA,kCAAqB,EAAC,IAAI,EAAE,QAAQ,CAAwB,CAAC;4BACnF,CAAC;wBACF,CAAC;wBAED,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;oBAC9F,CAAC;gBACF,CAAC;gBACD,oBAAoB;qBACf,IAAI,KAAK,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;oBAC9C,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;oBAChC,4BAA4B;oBAC5B,aAAa,GAAG,QAAQ,CAAA;oBACxB,IAAI,QAAQ,EAAE,KAAK,EAAE,CAAC;wBACrB,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,oBAAoB,EAAE,aAAa,IAAI,CAAC,CAAC;wBAC7E,MAAM,CAAC,KAAK,GAAG;4BACd,qFAAqF;4BACrF,KAAK,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,YAAY;4BACxD,MAAM,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC;4BACzC,SAAS,EAAE,YAAY;4BACvB,UAAU,EAAE,CAAC;4BACb,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC;4BAC7C,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;yBACpE,CAAC;oBACH,CAAC;oBACD,IAAA,sBAAa,EAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;oBACnC,4BAA4B;oBAC5B,MAAM,CAAC,UAAU,GAAG,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;oBACpD,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;wBACvF,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC;oBAC/B,CAAC;gBACF,CAAC;gBACD,gBAAgB;qBACX,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;oBACjC,MAAM,IAAI,KAAK,CAAC,qBAAqB,KAAK,CAAC,IAAI,MAAM,KAAK,CAAC,OAAO,EAAE,IAAI,sBAAsB,CAAC,CAAC;gBACjG,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;oBAC7C,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;gBACjE,CAAC;YACF,CAAC;YACD,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,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,GAAG,CAAC;gBACV,SAAS,EAAE,QAAQ;gBACnB,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,aAAa;aACtB,CAAC,CAAC;QACJ,CAAC;QAAA,OAAM,KAAK,EAAC,CAAC;YACb,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,OAAO;gBAAE,OAAQ,KAAa,CAAC,KAAK,CAAC;YAChE,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,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAEzE,kDAAkD;YAClD,aAAa,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC;YACzE,aAAa,CAAC,KAAK,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC;gBAC9C,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,IAAI,EAAG,KAAa,CAAC,IAAI,IAAI,eAAe;gBAC5C,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,OAAO;aACpB,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,EAAS,CAAC;YAE7C,MAAM,CAAC,GAAG,CAAC;gBACV,SAAS,EAAE,QAAQ;gBACnB,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,aAAa;aACtB,CAAC,CAAC;QACJ,CAAC;IACC,CAAC,CAAC,EAAE,CAAA;IAEJ,OAAO,MAAM,CAAC;AAClB,CAAC,CAAA;AAvTY,QAAA,YAAY,gBAuTxB;AAGD,SAAS,YAAY,CAAC,KAAsB,EAAE,MAAe;IAC5D,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,gBAAM,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,SAAS,WAAW,CAAC,KAAsB,EAAE,OAAgB,EAAE,OAA+B;IAC7F,MAAM,QAAQ,GAAG,IAAA,6BAAmB,EAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAErD,MAAM,MAAM,GAAkC;QAC7C,OAAO,EAAE,CAAC,6BAA6B,CAAC;QACxC,KAAK,EAAE,QAAQ;QACf,KAAK,EAAE,KAAK,CAAC,EAAE;QACf,MAAM,EAAE,IAAI;KACZ,CAAA;IAED,IAAG,OAAO,EAAE,eAAe,EAAC,CAAC;QAC5B,MAAM,CAAC,iBAAiB,GAAG,OAAO,CAAC,eAAe,CAAC;IACpD,CAAC;IAED,IAAG,OAAO,EAAE,iBAAiB,EAAC,CAAC;QAC9B,MAAM,CAAC,mBAAmB,GAAG,OAAO,CAAC,iBAAiB,CAAC;IACxD,CAAC;IAED,IAAG,OAAO,EAAE,gBAAgB,EAAC,CAAC;QAC7B,MAAM,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IACpD,CAAC;IAED,IAAG,OAAO,EAAE,oBAAoB,EAAC,CAAC;QACjC,MAAM,CAAC,sBAAsB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAC9D,CAAC;IAED,IAAG,OAAO,EAAE,SAAS,EAAC,CAAC;QACtB,MAAM,CAAC,SAAS,GAAG;YAClB,MAAM,EAAE,OAAO,CAAC,SAAS,CAAC,MAAM,IAAK,QAAQ;YAC7C,OAAO,EAAE,OAAO,CAAC,SAAS,CAAC,OAAO,IAAI,MAAM;SAC5C,CAAA;IACF,CAAC;IAED,IAAG,OAAO,EAAE,WAAW,EAAC,CAAC;QACxB,MAAM,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAA;IACzC,CAAC;IAED,IAAG,OAAO,CAAC,KAAK,EAAC,CAAC;QACjB,MAAM,CAAC,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IAC3C,CAAC;IAED,IAAG,OAAO,EAAE,UAAU,EAAC,CAAC;QACvB,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAA;IACvC,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAED,SAAS,YAAY,CAAC,KAAsB;IAC3C,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC3B,IAAI,EAAE,UAAU;QAChB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,UAAU,EAAE,IAAI,CAAC,UAAiB,EAAE,wCAAwC;QAC5E,MAAM,EAAE,IAAI;KACZ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,MAAmD;IACzE,IAAI,CAAC,MAAM;QAAE,OAAO,MAAM,CAAC;IAC3B,QAAQ,MAAM,EAAE,CAAC;QAChB,KAAK,WAAW;YACf,OAAO,MAAM,CAAC;QACf,KAAK,YAAY;YAChB,OAAO,QAAQ,CAAC;QACjB,KAAK,QAAQ,CAAC;QACd,KAAK,WAAW;YACf,OAAO,OAAO,CAAC;QAChB,0BAA0B;QAC1B,KAAK,aAAa,CAAC;QACnB,KAAK,QAAQ;YACZ,OAAO,MAAM,CAAC;QACf,OAAO,CAAC,CAAC,CAAC;YACT,MAAM,WAAW,GAAU,MAAM,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,0BAA0B,WAAW,EAAE,CAAC,CAAC;QAC1D,CAAC;IACF,CAAC;AACF,CAAC"}

package/dist/stream.d.ts

@@ -0,0 +1,4 @@
+import type { Model, Context, Api, OptionsForApi } from "./types";
+import { AssistantMessageEventStream } from "./utils/event-stream";
+export declare function stream<TApi extends Api>(model: Model<TApi>, context: Context, options?: OptionsForApi<TApi>): AssistantMessageEventStream;
+//# sourceMappingURL=stream.d.ts.map

package/dist/stream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../src/stream.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAClE,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AAqBnE,wBAAgB,MAAM,CAAC,IAAI,SAAS,GAAG,EACtC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,GAC3B,2BAA2B,CAiC7B"}

package/dist/stream.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.stream = stream;
+const openai_1 = require("./providers/openai");
+const google_1 = require("./providers/google");
+const envMap = {
+    openai: "OPENAI_API_KEY",
+    // anthropic: "ANTHROPIC_API_KEY",
+    google: "GEMINI_API_KEY",
+    // groq: "GROQ_API_KEY",
+    // cerebras: "CEREBRAS_API_KEY",
+    // xai: "XAI_API_KEY",
+    // openrouter: "OPENROUTER_API_KEY",
+    // zai: "ZAI_API_KEY",
+};
+function getApiKeyFromEnv(api) {
+    const envVar = envMap[api];
+    return process.env[envVar];
+}
+function stream(model, context, options) {
+    const apiKey = options?.apiKey !== undefined ? options.apiKey : getApiKeyFromEnv(model.api);
+    if (!apiKey) {
+        throw new Error(`No API key for provider: ${model.api}`);
+    }
+    const providerOptions = { ...options, apiKey };
+    // Switch directly on model.api and use type assertions for each provider
+    switch (model.api) {
+        case 'openai':
+            // TypeScript knows this branch only runs when model.api === 'openai'
+            return (0, openai_1.streamOpenAI)(model, context, providerOptions);
+        case 'google':
+            // TypeScript knows this branch only runs when model.api === 'google'
+            return (0, google_1.streamGoogle)(model, context, providerOptions);
+        default: {
+            const _exhaustive = model.api;
+            throw new Error(`Unhandled API: ${_exhaustive}`);
+        }
+    }
+}
+//# sourceMappingURL=stream.js.map

package/dist/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../src/stream.ts"],"names":[],"mappings":";;AAsBA,wBAqCC;AAzDD,+CAAyE;AACzE,+CAAyE;AAEzE,MAAM,MAAM,GAAwB;IACnC,MAAM,EAAE,gBAAgB;IACxB,kCAAkC;IAClC,MAAM,EAAE,gBAAgB;IACxB,wBAAwB;IACxB,gCAAgC;IAChC,sBAAsB;IACtB,oCAAoC;IACpC,sBAAsB;CACtB,CAAC;AAGF,SAAS,gBAAgB,CAAC,GAAQ;IAC9B,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAA;IAC1B,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;AAC9B,CAAC;AAED,SAAgB,MAAM,CACrB,KAAkB,EAClB,OAAgB,EAChB,OAA6B;IAG1B,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAC3F,IAAG,CAAC,MAAM,EAAC,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;IACvD,CAAC;IAEJ,MAAM,eAAe,GAAG,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,CAAC;IAE/C,yEAAyE;IACzE,QAAQ,KAAK,CAAC,GAAG,EAAE,CAAC;QACb,KAAK,QAAQ;YACT,qEAAqE;YACrE,OAAO,IAAA,qBAAY,EAC3B,KAAwB,EACxB,OAAO,EACP,eAAwC,CACxC,CAAC;QAEH,KAAK,QAAQ;YACH,qEAAqE;YAC9E,OAAO,IAAA,qBAAY,EAClB,KAAwB,EACxB,OAAO,EACP,eAAwC,CACxC,CAAC;QAEG,OAAO,CAAC,CAAC,CAAC;YACf,MAAM,WAAW,GAAU,KAAK,CAAC,GAAG,CAAC;YACrC,MAAM,IAAI,KAAK,CAAC,kBAAkB,WAAW,EAAE,CAAC,CAAC;QAC5C,CAAC;IACL,CAAC;AAEL,CAAC"}

package/dist/test-google-agent-loop.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=test-google-agent-loop.d.ts.map

package/dist/test-google-agent-loop.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-google-agent-loop.d.ts","sourceRoot":"","sources":["../src/test-google-agent-loop.ts"],"names":[],"mappings":""}

package/dist/test-google-agent-loop.js

@@ -0,0 +1,186 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const agent_loop_1 = require("./agent/agent-loop");
+const models_generated_1 = require("./models.generated");
+const typebox_1 = require("@sinclair/typebox");
+const fs_1 = require("fs");
+const path_1 = require("path");
+const genai_1 = require("@google/genai");
+// Create logs directory
+const logsDir = (0, path_1.join)(process.cwd(), "logs");
+try {
+    (0, fs_1.mkdirSync)(logsDir, { recursive: true });
+}
+catch (e) {
+    // Directory might already exist
+}
+const eventsFile = (0, path_1.join)(logsDir, "agent-events.txt");
+const resultFile = (0, path_1.join)(logsDir, "agent-result.json");
+// Clear previous logs
+(0, fs_1.writeFileSync)(eventsFile, "=== Agent Loop Events Log ===\n", "utf-8");
+(0, fs_1.writeFileSync)(resultFile, "", "utf-8");
+console.log("🧪 Testing Agent Loop...\n");
+// Define calculator parameter schema
+const CalculatorParams = typebox_1.Type.Object({
+    operation: typebox_1.Type.Union([
+        typebox_1.Type.Literal("addition"),
+        typebox_1.Type.Literal("add"),
+        typebox_1.Type.Literal("divide"),
+        typebox_1.Type.Literal("multiply"),
+        typebox_1.Type.Literal("subtraction"),
+        typebox_1.Type.Literal("multiplication"),
+        typebox_1.Type.Literal("division"),
+    ]),
+    a: typebox_1.Type.Number({ description: "First number" }),
+    b: typebox_1.Type.Number({ description: "Second number" }),
+});
+// Define a calculator tool with execution logic
+const calculatorTool = {
+    name: "calculator",
+    label: "Calculator",
+    description: "Performs basic arithmetic operations",
+    parameters: CalculatorParams,
+    async execute(toolCallId, params, signal) {
+        // Simulate some work
+        await new Promise(resolve => setTimeout(resolve, 100));
+        let result;
+        switch (params.operation) {
+            case "addition":
+                result = params.a + params.b;
+                break;
+            case "add":
+                result = params.a + params.b;
+                break;
+            case "subtraction":
+                result = params.a - params.b;
+                break;
+            case "multiplication":
+                result = params.a * params.b;
+                break;
+            case "multiply":
+                result = params.a * params.b;
+                break;
+            case "division":
+                if (params.b === 0) {
+                    throw new Error("Division by zero is not allowed");
+                }
+                result = params.a / params.b;
+                break;
+            case "divide":
+                if (params.b === 0) {
+                    throw new Error("Division by zero is not allowed");
+                }
+                result = params.a / params.b;
+                break;
+        }
+        return {
+            content: [
+                {
+                    type: "text",
+                    content: `The result of ${params.operation} of ${params.a} and ${params.b} is ${result}`,
+                },
+            ],
+            details: {
+                operation: params.operation,
+                operands: [params.a, params.b],
+                result,
+            },
+        };
+    },
+};
+// Create a user message
+const userMessage = {
+    role: "user",
+    timestamp: Date.now(),
+    content: [
+        {
+            type: "text",
+            content: "Can you calculate 1005 + 27032 for me? Then multiply the result by 2.",
+        },
+    ],
+};
+// Create agent context
+const agentContext = {
+    systemPrompt: "You are a helpful assistant with access to a calculator tool. Always use the calculator tool for math operations.",
+    messages: [],
+    tools: [calculatorTool],
+};
+// Get a model
+const model = models_generated_1.MODELS.google["gemini-3-pro-preview"];
+// Create config
+const config = {
+    model,
+    providerOptions: {
+        thinkingConfig: {
+            thinkingLevel: genai_1.ThinkingLevel.LOW
+        }
+    },
+};
+console.log(`📦 Model: ${model.name}`);
+console.log(`🔧 Tools: ${calculatorTool.name}`);
+const firstContent = userMessage.content[0];
+const messageText = firstContent.type === "text" ? firstContent.content : `[${firstContent.type}]`;
+console.log(`💬 User: ${messageText}\n`);
+console.log("📝 Streaming events to:", eventsFile);
+console.log("💾 Final result to:", resultFile);
+console.log("\n--- Starting Agent Loop ---\n");
+// Helper to log events
+function logEvent(event) {
+    const timestamp = new Date().toISOString();
+    const eventLog = `[${timestamp}] ${event.type}\n${JSON.stringify(event, null, 2)}\n${"=".repeat(80)}\n`;
+    (0, fs_1.appendFileSync)(eventsFile, eventLog, "utf-8");
+    console.log(`📨 Event: ${event.type}`);
+}
+// Run the agent loop
+(async () => {
+    try {
+        const eventStream = (0, agent_loop_1.agentLoop)(userMessage, agentContext, config);
+        // Iterate through events
+        for await (const event of eventStream) {
+            logEvent(event);
+            // Show some key events in console
+            if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
+                process.stdout.write(event.assistantMessageEvent.delta);
+            }
+            else if (event.type === "tool_execution_start") {
+                console.log(`\n🔧 Executing tool: ${event.toolName}(${JSON.stringify(event.args)})`);
+            }
+            else if (event.type === "tool_execution_end") {
+                if (event.isError) {
+                    console.log(`❌ Tool failed: ${typeof event.result === "string" ? event.result : "See details"}`);
+                }
+                else {
+                    console.log(`✅ Tool succeeded`);
+                }
+            }
+            else if (event.type === "turn_end") {
+                console.log(`\n--- Turn Complete (${event.toolResults.length} tool results) ---\n`);
+            }
+            else if (event.type === "agent_end") {
+                console.log(`\n🏁 Agent completed with status: ${event.status}\n`);
+            }
+        }
+        // Get the final result
+        console.log("\n--- Agent Loop Complete ---\n");
+        console.log("⏳ Getting final messages...");
+        const finalMessages = await eventStream.result();
+        console.log(`✅ Received ${finalMessages.length} messages!`);
+        // Save final result
+        const resultData = {
+            timestamp: new Date().toISOString(),
+            messageCount: finalMessages.length,
+            messages: finalMessages,
+        };
+        (0, fs_1.writeFileSync)(resultFile, JSON.stringify(resultData, null, 2), "utf-8");
+        console.log("\n✨ Test complete!");
+        console.log(`📝 Events logged to: ${eventsFile}`);
+        console.log(`💾 Final result saved to: ${resultFile}`);
+        console.log(`📊 Total messages: ${finalMessages.length}`);
+    }
+    catch (error) {
+        console.error("\n❌ Error during agent loop:", error);
+        (0, fs_1.appendFileSync)(eventsFile, `\n\nERROR: ${error}\n`, "utf-8");
+        process.exit(1);
+    }
+})();
+//# sourceMappingURL=test-google-agent-loop.js.map