@ank1015/providers 0.0.1 → 0.0.2

package/dist/providers/openai/utils.js

@@ -0,0 +1,285 @@
+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.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({
+        apiKey,
+        baseURL: model.baseUrl,
+        dangerouslyAllowBrowser: true,
+        defaultHeaders: model.headers,
+    });
+}
+export function getResponseAssistantResponse(response) {
+    const assistantResponse = [];
+    if (response.output) {
+        for (const item of response.output) {
+            if (item.type === 'reasoning' && item.summary) {
+                // Convert reasoning to thinking content
+                const thinkingText = item.summary.map(s => s.text).join('\n\n');
+                assistantResponse.push({
+                    type: 'thinking',
+                    thinkingText: thinkingText
+                });
+            }
+            else if (item.type === 'message' && item.content) {
+                // Convert message to text content
+                const textContent = item.content
+                    .map(c => {
+                    if (c.type === 'output_text')
+                        return c.text;
+                    if (c.type === 'refusal')
+                        return c.refusal;
+                    return '';
+                })
+                    .join('');
+                if (textContent) {
+                    assistantResponse.push({
+                        type: 'response',
+                        content: [{
+                                type: 'text',
+                                content: textContent
+                            }]
+                    });
+                }
+            }
+            else if (item.type === 'image_generation_call' && item.result) {
+                assistantResponse.push({
+                    type: 'response',
+                    content: [{
+                            type: 'image',
+                            data: item.result,
+                            mimeType: 'image/png'
+                        }]
+                });
+            }
+            else if (item.type === 'function_call') {
+                // Convert function call to tool call
+                assistantResponse.push({
+                    type: 'toolCall',
+                    toolCallId: item.call_id,
+                    name: item.name,
+                    arguments: JSON.parse(item.arguments || '{}')
+                });
+            }
+        }
+    }
+    return assistantResponse;
+}
+export function getResponseUsage(response, model) {
+    const cachedTokens = response.usage?.input_tokens_details?.cached_tokens || 0;
+    const usage = {
+        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 }
+    };
+    calculateCost(model, usage);
+    return usage;
+}
+export function buildParams(model, context, options) {
+    const messages = buildOpenAIMessages(model, context);
+    const { apiKey, signal, ...openaiOptions } = options;
+    const params = {
+        ...openaiOptions,
+        stream: false
+    };
+    params.model = model.id;
+    params.input = messages;
+    const tools = [];
+    if (context.tools && model.tools.includes('function_calling')) {
+        const convertedTools = convertTools(context.tools);
+        for (const convertedTool of convertedTools) {
+            tools.push(convertedTool);
+        }
+    }
+    if (openaiOptions.tools) {
+        for (const optionTool of openaiOptions.tools) {
+            tools.push(optionTool);
+        }
+    }
+    params.tools = tools;
+    return params;
+}
+export function buildOpenAIMessages(model, context) {
+    const openAIMessages = [];
+    if (context.systemPrompt) {
+        openAIMessages.push({
+            role: 'developer',
+            content: sanitizeSurrogates(context.systemPrompt)
+        });
+    }
+    ;
+    for (const message of context.messages) {
+        if (message.role === 'user') {
+            const contents = [];
+            for (let p = 0; p < message.content.length; p++) {
+                const content = message.content[p];
+                if (content.type === 'text') {
+                    contents.push({
+                        type: 'input_text',
+                        text: sanitizeSurrogates(content.content)
+                    });
+                }
+                if (content.type === 'image' && model.input.includes("image")) {
+                    contents.push({
+                        type: 'input_image',
+                        detail: 'auto',
+                        image_url: `data:${content.mimeType};base64,${content.data}`
+                    });
+                }
+                if (content.type === 'file' && model.input.includes("file")) {
+                    contents.push({
+                        type: 'input_file',
+                        filename: content.filename,
+                        file_data: `data:${content.mimeType};base64,${content.data}`
+                    });
+                }
+            }
+            openAIMessages.push({
+                role: 'user',
+                content: contents
+            });
+        }
+        // normalize for tool results
+        if (message.role === 'toolResult') {
+            const toolOutputs = [];
+            let hasText = false;
+            let hasImg = false;
+            let hasFile = false;
+            for (let p = 0; p < message.content.length; p++) {
+                const content = message.content[p];
+                if (content.type === 'text') {
+                    // Prefix error messages so LLM knows the tool failed
+                    const textContent = message.isError
+                        ? `[TOOL ERROR] ${content.content}`
+                        : content.content;
+                    toolOutputs.push({
+                        type: 'input_text',
+                        text: sanitizeSurrogates(textContent)
+                    });
+                    hasText = true;
+                }
+                if (content.type === 'image' && model.input.includes("image")) {
+                    toolOutputs.push({
+                        type: 'input_image',
+                        detail: 'auto',
+                        image_url: `data:${content.mimeType};base64,${content.data}`
+                    });
+                    hasImg = true;
+                }
+                if (content.type === 'file' && model.input.includes("file")) {
+                    toolOutputs.push({
+                        type: 'input_file',
+                        file_data: `data:${content.mimeType};base64,${content.data}`
+                    });
+                    hasFile = true;
+                }
+            }
+            if (!hasText && (hasImg || hasFile)) {
+                toolOutputs.push({
+                    type: 'input_text',
+                    text: message.isError ? '[TOOL ERROR] (see attached)' : '(see attached)'
+                });
+            }
+            const toolResultInput = {
+                call_id: message.toolCallId,
+                output: toolOutputs,
+                type: 'function_call_output',
+            };
+            openAIMessages.push(toolResultInput);
+        }
+        // normalize for Assistant message
+        if (message.role === 'assistant') {
+            if (message.model.api === 'openai') {
+                const baseMessage = message;
+                for (let p = 0; p < baseMessage.message.output.length; p++) {
+                    const outputPart = baseMessage.message.output[p];
+                    if (outputPart.type === 'function_call' || outputPart.type === 'message' || outputPart.type === 'reasoning') {
+                        openAIMessages.push(outputPart);
+                    }
+                }
+            }
+            // TODO Implement other provider conversions
+            else {
+                throw new Error(`Cannot convert ${message.model.api} assistant message to ${model.api} format. ` +
+                    `Cross-provider conversion for ${message.model.api} → ${model.api} is not yet implemented.`);
+            }
+        }
+    }
+    return openAIMessages;
+}
+export 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,
+    }));
+}
+export 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}`);
+        }
+    }
+}
+export function getMockOpenaiMessage() {
+    return {
+        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: {}
+    };
+}
+//# sourceMappingURL=utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/providers/openai/utils.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAG5B,OAAO,EAAE,kBAAkB,EAAE,MAAM,iCAAiC,CAAC;AACrE,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAGhD,MAAM,UAAU,YAAY,CAAC,KAAsB,EAAE,MAAe;IAChE,IAAI,CAAC,MAAM,EAAE,CAAC;QACV,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CACX,gGAAgG,CACnG,CAAC;QACN,CAAC;QACD,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IACxC,CAAC;IACD,OAAO,IAAI,MAAM,CAAC;QACd,MAAM;QACN,OAAO,EAAE,KAAK,CAAC,OAAO;QACtB,uBAAuB,EAAE,IAAI;QAC7B,cAAc,EAAE,KAAK,CAAC,OAAO;KAChC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,QAAkB;IAC3D,MAAM,iBAAiB,GAAsB,EAAE,CAAC;IAEhD,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QAClB,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACjC,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBAC5C,wCAAwC;gBACxC,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAChE,iBAAiB,CAAC,IAAI,CAAC;oBACnB,IAAI,EAAE,UAAU;oBAChB,YAAY,EAAE,YAAY;iBAC7B,CAAC,CAAC;YACP,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACjD,kCAAkC;gBAClC,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO;qBAC3B,GAAG,CAAC,CAAC,CAAC,EAAE;oBACL,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa;wBAAE,OAAO,CAAC,CAAC,IAAI,CAAC;oBAC5C,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS;wBAAE,OAAO,CAAC,CAAC,OAAO,CAAC;oBAC3C,OAAO,EAAE,CAAC;gBACd,CAAC,CAAC;qBACD,IAAI,CAAC,EAAE,CAAC,CAAC;gBAEd,IAAI,WAAW,EAAE,CAAC;oBACd,iBAAiB,CAAC,IAAI,CAAC;wBACnB,IAAI,EAAE,UAAU;wBAChB,OAAO,EAAE,CAAC;gCACN,IAAI,EAAE,MAAM;gCACZ,OAAO,EAAE,WAAW;6BACvB,CAAC;qBACL,CAAC,CAAC;gBACP,CAAC;YACL,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,uBAAuB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC9D,iBAAiB,CAAC,IAAI,CAAC;oBACnB,IAAI,EAAE,UAAU;oBAChB,OAAO,EAAE,CAAC;4BACN,IAAI,EAAE,OAAO;4BACb,IAAI,EAAE,IAAI,CAAC,MAAM;4BACjB,QAAQ,EAAE,WAAW;yBACxB,CAAC;iBACL,CAAC,CAAA;YACN,CAAC;iBACI,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBACrC,qCAAqC;gBACrC,iBAAiB,CAAC,IAAI,CAAC;oBACnB,IAAI,EAAE,UAAU;oBAChB,UAAU,EAAE,IAAI,CAAC,OAAO;oBACxB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC;iBAChD,CAAC,CAAC;YACP,CAAC;QACL,CAAC;IACL,CAAC;IAED,OAAO,iBAAiB,CAAA;AAC5B,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,QAAkB,EAAE,KAAsB;IACvE,MAAM,YAAY,GAAG,QAAQ,CAAC,KAAK,EAAE,oBAAoB,EAAE,aAAa,IAAI,CAAC,CAAC;IAC9E,MAAM,KAAK,GAAU;QACjB,KAAK,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC,GAAG,YAAY;QACzD,MAAM,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;QAC1C,SAAS,EAAE,YAAY;QACvB,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,QAAQ,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;QAC9C,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;KACvE,CAAC;IACF,aAAa,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC5B,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAsB,EAAE,OAAgB,EAAE,OAA8B;IAChG,MAAM,QAAQ,GAAG,mBAAmB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAErD,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,GAAG,OAAO,CAAA;IACpD,MAAM,MAAM,GAAqC;QAC7C,GAAG,aAAa;QAChB,MAAM,EAAE,KAAK;KAChB,CAAA;IAED,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC;IACxB,MAAM,CAAC,KAAK,GAAG,QAAQ,CAAA;IAEvB,MAAM,KAAK,GAAiB,EAAE,CAAA;IAE9B,IAAI,OAAO,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC5D,MAAM,cAAc,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAClD,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE,CAAC;YACzC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QAC7B,CAAC;IACL,CAAC;IAED,IAAI,aAAa,CAAC,KAAK,EAAE,CAAC;QACtB,KAAK,MAAM,UAAU,IAAI,aAAa,CAAC,KAAK,EAAE,CAAC;YAC3C,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAC1B,CAAC;IACL,CAAC;IAED,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAsB,EAAE,OAAgB;IACxE,MAAM,cAAc,GAAkB,EAAE,CAAC;IACzC,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;QACvB,cAAc,CAAC,IAAI,CAAC;YAChB,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,kBAAkB,CAAC,OAAO,CAAC,YAAY,CAAC;SACpD,CAAC,CAAA;IACN,CAAC;IAAA,CAAC;IAEF,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;QAErC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAoC,EAAE,CAAC;YACrD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACnC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1B,QAAQ,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,YAAY;wBAClB,IAAI,EAAE,kBAAkB,CAAC,OAAO,CAAC,OAAO,CAAC;qBAC5C,CAAC,CAAA;gBACN,CAAC;gBACD,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC5D,QAAQ,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,aAAa;wBACnB,MAAM,EAAE,MAAM;wBACd,SAAS,EAAE,QAAQ,OAAO,CAAC,QAAQ,WAAW,OAAO,CAAC,IAAI,EAAE;qBAC/D,CAAC,CAAA;gBACN,CAAC;gBACD,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC1D,QAAQ,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,YAAY;wBAClB,QAAQ,EAAE,OAAO,CAAC,QAAQ;wBAC1B,SAAS,EAAE,QAAQ,OAAO,CAAC,QAAQ,WAAW,OAAO,CAAC,IAAI,EAAE;qBAC/D,CAAC,CAAA;gBACN,CAAC;YACL,CAAC;YACD,cAAc,CAAC,IAAI,CAAC;gBAChB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,QAAQ;aACpB,CAAC,CAAA;QACN,CAAC;QAED,6BAA6B;QAC7B,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAChC,MAAM,WAAW,GAAuC,EAAE,CAAA;YAC1D,IAAI,OAAO,GAAG,KAAK,CAAC;YACpB,IAAI,MAAM,GAAG,KAAK,CAAC;YACnB,IAAI,OAAO,GAAG,KAAK,CAAC;YACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACnC,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1B,qDAAqD;oBACrD,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO;wBAC/B,CAAC,CAAC,gBAAgB,OAAO,CAAC,OAAO,EAAE;wBACnC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;oBACtB,WAAW,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,YAAY;wBAClB,IAAI,EAAE,kBAAkB,CAAC,WAAW,CAAC;qBACxC,CAAC,CAAA;oBACF,OAAO,GAAG,IAAI,CAAC;gBACnB,CAAC;gBACD,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC5D,WAAW,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,aAAa;wBACnB,MAAM,EAAE,MAAM;wBACd,SAAS,EAAE,QAAQ,OAAO,CAAC,QAAQ,WAAW,OAAO,CAAC,IAAI,EAAE;qBAC/D,CAAC,CAAA;oBACF,MAAM,GAAG,IAAI,CAAA;gBACjB,CAAC;gBACD,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC1D,WAAW,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,YAAY;wBAClB,SAAS,EAAE,QAAQ,OAAO,CAAC,QAAQ,WAAW,OAAO,CAAC,IAAI,EAAE;qBAC/D,CAAC,CAAA;oBACF,OAAO,GAAG,IAAI,CAAA;gBAClB,CAAC;YACL,CAAC;YACD,IAAI,CAAC,OAAO,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,EAAE,CAAC;gBAClC,WAAW,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,YAAY;oBAClB,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,6BAA6B,CAAC,CAAC,CAAC,gBAAgB;iBAC3E,CAAC,CAAA;YACN,CAAC;YACD,MAAM,eAAe,GAAyC;gBAC1D,OAAO,EAAE,OAAO,CAAC,UAAU;gBAC3B,MAAM,EAAE,WAAW;gBACnB,IAAI,EAAE,sBAAsB;aAC/B,CAAA;YACD,cAAc,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QACxC,CAAC;QAED,kCAAkC;QAClC,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/B,IAAI,OAAO,CAAC,KAAK,CAAC,GAAG,KAAK,QAAQ,EAAE,CAAC;gBACjC,MAAM,WAAW,GAAG,OAAyC,CAAA;gBAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACzD,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;oBACjD,IAAI,UAAU,CAAC,IAAI,KAAK,eAAe,IAAI,UAAU,CAAC,IAAI,KAAK,SAAS,IAAI,UAAU,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBAC1G,cAAc,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;oBACpC,CAAC;gBACL,CAAC;YACL,CAAC;YACD,4CAA4C;iBACvC,CAAC;gBACF,MAAM,IAAI,KAAK,CACX,kBAAkB,OAAO,CAAC,KAAK,CAAC,GAAG,yBAAyB,KAAK,CAAC,GAAG,WAAW;oBAChF,iCAAiC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,KAAK,CAAC,GAAG,0BAA0B,CAC9F,CAAC;YACN,CAAC;QACL,CAAC;IAEL,CAAC;IAED,OAAO,cAAc,CAAA;AACzB,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAsB;IAC/C,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACxB,IAAI,EAAE,UAAU;QAChB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,wCAAwC;QACrE,MAAM,EAAE,IAAI;KACf,CAAC,CAAC,CAAC;AACR,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,MAAmD;IAC7E,IAAI,CAAC,MAAM;QAAE,OAAO,MAAM,CAAC;IAC3B,QAAQ,MAAM,EAAE,CAAC;QACb,KAAK,WAAW;YACZ,OAAO,MAAM,CAAC;QAClB,KAAK,YAAY;YACb,OAAO,QAAQ,CAAC;QACpB,KAAK,QAAQ,CAAC;QACd,KAAK,WAAW;YACZ,OAAO,OAAO,CAAC;QACnB,0BAA0B;QAC1B,KAAK,aAAa,CAAC;QACnB,KAAK,QAAQ;YACT,OAAO,MAAM,CAAC;QAClB,OAAO,CAAC,CAAC,CAAC;YACN,MAAM,WAAW,GAAU,MAAM,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,0BAA0B,WAAW,EAAE,CAAC,CAAC;QAC7D,CAAC;IACL,CAAC;AACL,CAAC;AAED,MAAM,UAAU,oBAAoB;IAChC,OAAO;QACH,EAAE,EAAE,UAAU;QACd,MAAM,EAAE,UAAU;QAClB,UAAU,EAAE,UAAU;QACtB,WAAW,EAAE,EAAE;QACf,MAAM,EAAE,WAAW;QACnB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,KAAK;QAC1B,KAAK,EAAE,IAAI;QACX,YAAY,EAAE,IAAI;QAClB,iBAAiB,EAAE,IAAI;QACvB,KAAK,EAAE,wBAAwB;QAC/B,MAAM,EAAE,EAAE;QACV,oBAAoB,EAAE,IAAI;QAC1B,WAAW,EAAE,CAAC;QACd,IAAI,EAAE,EAAE;QACR,WAAW,EAAE,MAAM;QACnB,KAAK,EAAE,EAAE;QACT,KAAK,EAAE,CAAC;QACR,UAAU,EAAE,UAAU;QACtB,KAAK,EAAE;YACH,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,qBAAqB,EAAE;gBACnB,gBAAgB,EAAE,CAAC;aACtB;YACD,oBAAoB,EAAE;gBAClB,aAAa,EAAE,CAAC;aACnB;YACD,YAAY,EAAE,CAAC;SAClB;QACD,IAAI,EAAE,SAAS;QACf,QAAQ,EAAE,EAAE;KACf,CAAA;AACL,CAAC"}

package/dist/types.d.ts

@@ -1,10 +1,11 @@
 import { Response } from "openai/resources/responses/responses.js";
-import type { TSchema } from "@sinclair/typebox";
-import { OpenAIProviderOptions } from "./providers/openai";
-import { AssistantMessageEventStream } from "./utils/event-stream";
-import { GoogleProviderOptions } from "./providers/google";
 import type { GenerateContentResponse } from "@google/genai";
-export type Api = 'openai' | 'google';
+import type { TSchema } from "@sinclair/typebox";
+import { OpenAIProviderOptions } from "./providers/openai/index.js";
+import { GoogleProviderOptions } from "./providers/google/index.js";
+import { AssistantMessageEventStream } from "./utils/event-stream.js";
+export declare const KnownApis: readonly ["openai", "google"];
+export type Api = typeof KnownApis[number];
 export interface Model<TApi extends Api> {
     id: string;
     name: string;
@@ -21,17 +22,40 @@ export interface Model<TApi extends Api> {
     contextWindow: number;
     maxTokens: number;
     headers?: Record<string, string>;
+    tools: string[];
+    excludeSettings?: string[];
+}
+export interface TextContent {
+    type: 'text';
+    content: string;
+    metadata?: Record<string, any>;
 }
+export interface ImageContent {
+    type: "image";
+    data: string;
+    mimeType: string;
+    metadata?: Record<string, any>;
+}
+export interface FileContent {
+    type: "file";
+    data: string;
+    mimeType: string;
+    filename: string;
+    metadata?: Record<string, any>;
+}
+export type Content = (TextContent | ImageContent | FileContent)[];
 export interface UserMessage {
     role: "user";
-    timestamp: number;
-    content: (UserTextContent | UserImageContent | UserFileContent)[];
+    id: string;
+    timestamp?: number;
+    content: Content;
 }
 export interface ToolResultMessage<TDetails = any> {
     role: "toolResult";
+    id: string;
     toolName: string;
-    toolCallId?: string;
-    content: (UserTextContent | UserImageContent | UserFileContent)[];
+    toolCallId: string;
+    content: Content;
     details?: TDetails;
     isError: boolean;
     error?: {
@@ -41,45 +65,22 @@ export interface ToolResultMessage<TDetails = any> {
     };
     timestamp: number;
 }
-export interface UserImageContent {
-    type: "image";
-    data: string;
-    mimeType: string;
-}
-export interface UserFileContent {
-    type: "file";
-    data: string;
-    mimeType: string;
-}
-export interface UserTextContent {
-    type: 'text';
-    content: string;
-}
-export interface NativeOpenAIMessage {
-    role: "assistant";
-    _provider: 'openai';
-    message: Response;
-}
-export interface NativeGoogleMessage {
-    role: "assistant";
-    _provider: 'google';
-    message: GenerateContentResponse;
+export type StopReason = "stop" | "length" | "toolUse" | "error" | "aborted";
+export interface AssistantResponseContent {
+    type: 'response';
+    content: Content;
 }
-export type NativeAssistantMessage = NativeOpenAIMessage | NativeGoogleMessage;
-export type Message = UserMessage | NativeAssistantMessage | ToolResultMessage;
-export interface Tool<TParameters extends TSchema = TSchema, TName extends string = string> {
-    name: TName;
-    description: string;
-    parameters: TParameters;
+export interface AssistantThinkingContent {
+    type: 'thinking';
+    thinkingText: string;
 }
-export type ToolName<TTool extends Tool> = TTool["name"];
-export type ToolNames<TTools extends readonly Tool[]> = TTools[number]["name"];
-export declare function defineTool<TParameters extends TSchema, TName extends string>(tool: Tool<TParameters, TName>): Tool<TParameters, TName>;
-export interface Context<TTools extends readonly Tool[] = readonly Tool[]> {
-    messages: Message[];
-    systemPrompt?: string;
-    tools?: TTools;
+export interface AssistantToolCall {
+    type: "toolCall";
+    name: string;
+    arguments: Record<string, any>;
+    toolCallId: string;
 }
+export type AssistantResponse = (AssistantResponseContent | AssistantThinkingContent | AssistantToolCall)[];
 export interface Usage {
     input: number;
     output: number;
@@ -94,94 +95,101 @@ export interface Usage {
         total: number;
     };
 }
-export type StopReason = "stop" | "length" | "toolUse" | "error" | "aborted";
-export interface AssistantMessage {
+export interface ApiNativeAssistantMessageMap {
+    "openai": Response;
+    "google": GenerateContentResponse;
+}
+export type NativeAssistantMessageForApi<TApi extends Api> = ApiNativeAssistantMessageMap[TApi];
+export interface BaseAssistantMessage<TApi extends Api> {
     role: "assistant";
-    content: (AssistantTextContent | AssistantThinkingContent | AssistantToolCall | AbstractedImageContent)[];
-    api: Api;
-    model: string;
-    usage: Usage;
-    stopReason: StopReason;
+    message: NativeAssistantMessageForApi<TApi>;
+    api: string;
+    id: string;
+    model: Model<TApi>;
     errorMessage?: string;
     timestamp: number;
+    duration: number;
+    stopReason: StopReason;
+    content: AssistantResponse;
+    usage: Usage;
 }
-export interface AssistantTextContent {
-    type: 'text';
-    text: string;
-}
-export interface AssistantThinkingContent {
-    type: 'thinking';
-    thinking: string;
-}
-export interface AssistantToolCall {
-    type: "toolCall";
-    name: string;
-    arguments: Record<string, any>;
-    id?: string;
-}
-export interface AbstractedImageContent {
-    type: "image";
-    data: string;
-    mimeType: string;
+export interface CustomMessage {
+    role: "custom";
+    id: string;
+    content: Record<string, any>;
+    timestamp?: number;
 }
-export type AssistantMessageEvent = {
+export type Message = UserMessage | ToolResultMessage | BaseAssistantMessage<Api> | CustomMessage;
+export type BaseAssistantEventMessage<TApi extends Api> = Omit<BaseAssistantMessage<TApi>, 'message'>;
+export type BaseAssistantEvent<TApi extends Api> = {
     type: "start";
-    partial: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "text_start";
     contentIndex: number;
-    partial: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "text_delta";
     contentIndex: number;
     delta: string;
-    partial: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "text_end";
     contentIndex: number;
-    content: string;
-    partial: AssistantMessage;
+    content: Content;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "thinking_start";
     contentIndex: number;
-    partial: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "thinking_delta";
     contentIndex: number;
     delta: string;
-    partial: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "thinking_end";
     contentIndex: number;
     content: string;
-    partial: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "toolcall_start";
     contentIndex: number;
-    partial: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "toolcall_delta";
     contentIndex: number;
     delta: string;
-    partial: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "toolcall_end";
     contentIndex: number;
     toolCall: AssistantToolCall;
-    partial: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "done";
     reason: Extract<StopReason, "stop" | "length" | "toolUse">;
-    message: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 } | {
     type: "error";
     reason: Extract<StopReason, "aborted" | "error">;
-    error: AssistantMessage;
+    message: BaseAssistantEventMessage<Api>;
 };
+export interface Tool<TParameters extends TSchema = TSchema, TName extends string = string> {
+    name: TName;
+    description: string;
+    parameters: TParameters;
+}
+export interface Context {
+    messages: Message[];
+    systemPrompt?: string;
+    tools?: Tool[];
+}
 export interface ApiOptionsMap {
-    "openai": OpenAIProviderOptions;
-    "google": GoogleProviderOptions;
+    'openai': OpenAIProviderOptions;
+    'google': GoogleProviderOptions;
 }
 export type OptionsForApi<TApi extends Api> = ApiOptionsMap[TApi];
-export type StreamFunction<TApi extends Api> = (model: Model<TApi>, context: Context, options: OptionsForApi<TApi>) => AssistantMessageEventStream;
+export type CompleteFunction<TApi extends Api> = (model: Model<TApi>, context: Context, optionsForApi: OptionsForApi<TApi>, id: string) => Promise<BaseAssistantMessage<TApi>>;
+export type StreamFunction<TApi extends Api> = (model: Model<TApi>, context: Context, optionsForApi: OptionsForApi<TApi>, id: string) => AssistantMessageEventStream<TApi>;
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,yCAAyC,CAAC;AACnE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAE7D,MAAM,MAAM,GAAG,GAAG,QAAQ,GAAG,QAAQ,CAAA;AAErC,MAAM,WAAW,KAAK,CAAC,IAAI,SAAS,GAAG;IACtC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,IAAI,CAAC;IACV,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,OAAO,CAAC;IACnB,KAAK,EAAE,CAAC,MAAM,GAAG,OAAO,GAAG,MAAM,CAAC,EAAE,CAAC;IACrC,IAAI,EAAE;QACL,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACjC;AAKD,MAAM,WAAW,WAAW;IACxB,IAAI,EAAE,MAAM,CAAA;IACZ,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,CAAC,eAAe,GAAG,gBAAgB,GAAG,eAAe,CAAC,EAAE,CAAA;CACpE;AAED,MAAM,WAAW,iBAAiB,CAAC,QAAQ,GAAG,GAAG;IAChD,IAAI,EAAE,YAAY,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,CAAC,eAAe,GAAG,gBAAgB,GAAG,eAAe,CAAC,EAAE,CAAC;IACrE,OAAO,CAAC,EAAE,QAAQ,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE;QACP,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,gBAAgB;IAC7B,IAAI,EAAE,OAAO,CAAA;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,eAAe;IAC5B,IAAI,EAAE,MAAM,CAAA;IACf,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,eAAe;IAC5B,IAAI,EAAE,MAAM,CAAA;IACZ,OAAO,EAAE,MAAM,CAAA;CAClB;AAID,MAAM,WAAW,mBAAmB;IACnC,IAAI,EAAE,WAAW,CAAA;IACd,SAAS,EAAE,QAAQ,CAAA;IACnB,OAAO,EAAE,QAAQ,CAAA;CACpB;AAED,MAAM,WAAW,mBAAmB;IACnC,IAAI,EAAE,WAAW,CAAA;IACd,SAAS,EAAE,QAAQ,CAAA;IACnB,OAAO,EAAE,uBAAuB,CAAA;CACnC;AAED,MAAM,MAAM,sBAAsB,GAAG,mBAAmB,GAAG,mBAAmB,CAAC;AAI/E,MAAM,MAAM,OAAO,GAAG,WAAW,GAAG,sBAAsB,GAAG,iBAAiB,CAAA;AAE9E,MAAM,WAAW,IAAI,CAAC,WAAW,SAAS,OAAO,GAAG,OAAO,EAAE,KAAK,SAAS,MAAM,GAAG,MAAM;IACzF,IAAI,EAAE,KAAK,CAAC;IACZ,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,WAAW,CAAC;CACxB;AAGD,MAAM,MAAM,QAAQ,CAAC,KAAK,SAAS,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;AACzD,MAAM,MAAM,SAAS,CAAC,MAAM,SAAS,SAAS,IAAI,EAAE,IAAI,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;AAK/E,wBAAgB,UAAU,CAAC,WAAW,SAAS,OAAO,EAAE,KAAK,SAAS,MAAM,EAC3E,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,GAC5B,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,CAE1B;AAED,MAAM,WAAW,OAAO,CAAC,MAAM,SAAS,SAAS,IAAI,EAAE,GAAG,SAAS,IAAI,EAAE;IACxE,QAAQ,EAAE,OAAO,EAAE,CAAA;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAA;CACd;AAID,MAAM,WAAW,KAAK;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE;QACL,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE,MAAM,CAAC;KACd,CAAC;CACF;AAED,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG,QAAQ,GAAG,SAAS,GAAG,OAAO,GAAG,SAAS,CAAC;AAG7E,MAAM,WAAW,gBAAgB;IAChC,IAAI,EAAE,WAAW,CAAC;IAClB,OAAO,EAAE,CAAC,oBAAoB,GAAG,wBAAwB,GAAG,iBAAiB,GAAG,sBAAsB,CAAC,EAAE,CAAC;IAC1G,GAAG,EAAE,GAAG,CAAC;IACT,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;IACb,UAAU,EAAE,UAAU,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,oBAAoB;IACjC,IAAI,EAAE,MAAM,CAAA;IACZ,IAAI,EAAE,MAAM,CAAA;CACf;AAED,MAAM,WAAW,wBAAwB;IACrC,IAAI,EAAE,UAAU,CAAA;IAChB,QAAQ,EAAE,MAAM,CAAA;CACnB;AAED,MAAM,WAAW,iBAAiB;IACjC,IAAI,EAAE,UAAU,CAAC;IACd,IAAI,EAAE,MAAM,CAAA;IACf,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,EAAE,CAAC,EAAE,MAAM,CAAA;CACX;AAED,MAAM,WAAW,sBAAsB;IACtC,IAAI,EAAE,OAAO,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,MAAM,qBAAqB,GAC9B;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GAC5C;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GACvE;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GACtF;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GACtF;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GAC3E;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GAC1F;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GAC1F;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GAC3E;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GAC1F;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,iBAAiB,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GACtG;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC,CAAC;IAAC,OAAO,EAAE,gBAAgB,CAAA;CAAE,GACvG;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,SAAS,GAAG,OAAO,CAAC,CAAC;IAAC,KAAK,EAAE,gBAAgB,CAAA;CAAE,CAAC;AAMhG,MAAM,WAAW,aAAa;IAC7B,QAAQ,EAAE,qBAAqB,CAAC;IAChC,QAAQ,EAAE,qBAAqB,CAAC;CAChC;AAED,MAAM,MAAM,aAAa,CAAC,IAAI,SAAS,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC;AAGlE,MAAM,MAAM,cAAc,CAAC,IAAI,SAAS,GAAG,IAAI,CAC9C,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,aAAa,CAAC,IAAI,CAAC,KACxB,2BAA2B,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,yCAAyC,CAAC;AACnE,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,2BAA2B,EAAE,MAAM,yBAAyB,CAAC;AAEtE,eAAO,MAAM,SAAS,+BAAgC,CAAC;AACvD,MAAM,MAAM,GAAG,GAAG,OAAO,SAAS,CAAC,MAAM,CAAC,CAAC;AAC3C,MAAM,WAAW,KAAK,CAAC,IAAI,SAAS,GAAG;IACtC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,IAAI,CAAC;IACV,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,OAAO,CAAC;IACnB,KAAK,EAAE,CAAC,MAAM,GAAG,OAAO,GAAG,MAAM,CAAC,EAAE,CAAC;IACrC,IAAI,EAAE;QACL,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,KAAK,EAAE,MAAM,EAAE,CAAA;IACf,eAAe,CAAC,EAAE,MAAM,EAAE,CAAA;CAC1B;AAMD,MAAM,WAAW,WAAW;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,YAAY;IACzB,IAAI,EAAE,OAAO,CAAA;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,WAAW;IACxB,IAAI,EAAE,MAAM,CAAA;IACf,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC/B;AAED,MAAM,MAAM,OAAO,GAAG,CAAC,WAAW,GAAG,YAAY,GAAG,WAAW,CAAC,EAAE,CAAA;AAElE,MAAM,WAAW,WAAW;IACxB,IAAI,EAAE,MAAM,CAAA;IACZ,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB,CAAC,QAAQ,GAAG,GAAG;IAChD,IAAI,EAAE,YAAY,CAAC;IACnB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACpB,OAAO,CAAC,EAAE,QAAQ,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE;QACP,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;IACF,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG,QAAQ,GAAG,SAAS,GAAG,OAAO,GAAG,SAAS,CAAC;AAG7E,MAAM,WAAW,wBAAwB;IACrC,IAAI,EAAE,UAAU,CAAC;IACjB,OAAO,EAAE,OAAO,CAAA;CACnB;AAED,MAAM,WAAW,wBAAwB;IACrC,IAAI,EAAE,UAAU,CAAC;IACjB,YAAY,EAAE,MAAM,CAAA;CACvB;AAED,MAAM,WAAW,iBAAiB;IACjC,IAAI,EAAE,UAAU,CAAC;IACd,IAAI,EAAE,MAAM,CAAA;IACZ,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAClC,UAAU,EAAE,MAAM,CAAA;CAClB;AAED,MAAM,MAAM,iBAAiB,GAAG,CAAC,wBAAwB,GAAG,wBAAwB,GAAG,iBAAiB,CAAC,EAAE,CAAC;AAE5G,MAAM,WAAW,KAAK;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE;QACL,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE,MAAM,CAAC;KACd,CAAC;CACF;AAED,MAAM,WAAW,4BAA4B;IACzC,QAAQ,EAAE,QAAQ,CAAC;IACnB,QAAQ,EAAE,uBAAuB,CAAC;CACrC;AAED,MAAM,MAAM,4BAA4B,CAAC,IAAI,SAAS,GAAG,IAAI,4BAA4B,CAAC,IAAI,CAAC,CAAA;AAE/F,MAAM,WAAW,oBAAoB,CAAE,IAAI,SAAS,GAAG;IACnD,IAAI,EAAE,WAAW,CAAC;IAClB,OAAO,EAAE,4BAA4B,CAAC,IAAI,CAAC,CAAA;IAC9C,GAAG,EAAE,MAAM,CAAC;IACT,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,UAAU,CAAC;IACvB,OAAO,EAAE,iBAAiB,CAAC;IAC3B,KAAK,EAAE,KAAK,CAAC;CACb;AAED,MAAM,WAAW,aAAa;IAC7B,IAAI,EAAE,QAAQ,CAAC;IACf,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAA;CAClB;AAED,MAAM,MAAM,OAAO,GAAG,WAAW,GAAG,iBAAiB,GAAG,oBAAoB,CAAC,GAAG,CAAC,GAAG,aAAa,CAAA;AAQjG,MAAM,MAAM,yBAAyB,CAAE,IAAI,SAAS,GAAG,IAAK,IAAI,CAAE,oBAAoB,CAAC,IAAI,CAAC,EAAE,SAAS,CAAC,CAAA;AAExG,MAAM,MAAM,kBAAkB,CAAC,IAAI,SAAS,GAAG,IAC7C;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GAC1D;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACrF;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACpG;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACrG;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACzF;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACxG;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACxG;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACzF;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACxG;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,YAAY,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,iBAAiB,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACpH;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,MAAM,GAAG,QAAQ,GAAG,SAAS,CAAC,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,GACrH;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,MAAM,EAAE,OAAO,CAAC,UAAU,EAAE,SAAS,GAAG,OAAO,CAAC,CAAC;IAAC,OAAO,EAAE,yBAAyB,CAAC,GAAG,CAAC,CAAA;CAAE,CAAC;AAO/G,MAAM,WAAW,IAAI,CAAC,WAAW,SAAS,OAAO,GAAG,OAAO,EAAE,KAAK,SAAS,MAAM,GAAG,MAAM;IACzF,IAAI,EAAE,KAAK,CAAC;IACZ,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,WAAW,CAAC;CACxB;AAED,MAAM,WAAW,OAAO;IACvB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;CACf;AAED,MAAM,WAAW,aAAa;IAC7B,QAAQ,EAAE,qBAAqB,CAAA;IAC/B,QAAQ,EAAE,qBAAqB,CAAA;CAC/B;AAED,MAAM,MAAM,aAAa,CAAC,IAAI,SAAS,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,CAAA;AAEjE,MAAM,MAAM,gBAAgB,CAAC,IAAI,SAAS,GAAG,IAAI,CAChD,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,aAAa,EAAE,aAAa,CAAC,IAAI,CAAC,EAClC,EAAE,EAAE,MAAM,KACN,OAAO,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC,CAAA;AAExC,MAAM,MAAM,cAAc,CAAC,IAAI,SAAS,GAAG,IAAI,CAC9C,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,EAClB,OAAO,EAAE,OAAO,EAChB,aAAa,EAAE,aAAa,CAAC,IAAI,CAAC,EAClC,EAAE,EAAE,MAAM,KACN,2BAA2B,CAAC,IAAI,CAAC,CAAA"}

package/dist/types.js

@@ -1,10 +1,2 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.defineTool = defineTool;
-// Helper function to create a tool with better type inference
-// Use 'as const' on the tool array for best autocomplete:
-// const tools = [defineTool({ name: "calculator", ... }), ...] as const
-function defineTool(tool) {
-    return tool;
-}
+export const KnownApis = ['openai', 'google'];
 //# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":";;AAqGA,gCAIC;AAPD,8DAA8D;AAC9D,0DAA0D;AAC1D,wEAAwE;AACxE,SAAgB,UAAU,CACzB,IAA8B;IAE9B,OAAO,IAAI,CAAC;AACb,CAAC"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAOA,MAAM,CAAC,MAAM,SAAS,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAU,CAAC"}

package/dist/utils/event-stream.d.ts

@@ -1,4 +1,4 @@
-import { NativeAssistantMessage, AssistantMessageEvent } from "../types";
+import { BaseAssistantMessage, BaseAssistantEvent, Api } from "../types.js";
 export declare class EventStream<T, R = T> implements AsyncIterable<T> {
     private queue;
     private waiting;
@@ -11,6 +11,6 @@ export declare class EventStream<T, R = T> implements AsyncIterable<T> {
     [Symbol.asyncIterator](): AsyncIterator<T>;
     result(): Promise<R>;
 }
-export declare class AssistantMessageEventStream extends EventStream<AssistantMessageEvent, NativeAssistantMessage> {
+export declare class AssistantMessageEventStream<TApi extends Api> extends EventStream<BaseAssistantEvent<TApi>, BaseAssistantMessage<TApi>> {
 }
 //# sourceMappingURL=event-stream.d.ts.map

package/dist/utils/event-stream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"event-stream.d.ts","sourceRoot":"","sources":["../../src/utils/event-stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,MAAM,UAAU,CAAC;AAGzE,qBAAa,WAAW,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAE,YAAW,aAAa,CAAC,CAAC,CAAC;IAC7D,OAAO,CAAC,KAAK,CAAW;IACxB,OAAO,CAAC,OAAO,CAA8C;IAC7D,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,kBAAkB,CAAa;IACvC,OAAO,CAAC,kBAAkB,CAAuB;;IAQjD,IAAI,CAAC,KAAK,EAAE,CAAC,GAAG,IAAI;IAYpB,GAAG,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI;IAYb,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC;IAcjD,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC;CAGpB;AAED,qBAAa,2BAA4B,SAAQ,WAAW,CAAC,qBAAqB,EAAE,sBAAsB,CAAC;CAAG"}
+{"version":3,"file":"event-stream.d.ts","sourceRoot":"","sources":["../../src/utils/event-stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,GAAG,EAAE,MAAM,aAAa,CAAC;AAG5E,qBAAa,WAAW,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAE,YAAW,aAAa,CAAC,CAAC,CAAC;IAC7D,OAAO,CAAC,KAAK,CAAW;IACxB,OAAO,CAAC,OAAO,CAA8C;IAC7D,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,kBAAkB,CAAa;IACvC,OAAO,CAAC,kBAAkB,CAAuB;;IAQjD,IAAI,CAAC,KAAK,EAAE,CAAC,GAAG,IAAI;IAYpB,GAAG,CAAC,MAAM,EAAE,CAAC,GAAG,IAAI;IAYb,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC;IAcjD,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC;CAGpB;AAED,qBAAa,2BAA2B,CAAC,IAAI,SAAS,GAAG,CAAE,SAAQ,WAAW,CAAC,kBAAkB,CAAC,IAAI,CAAC,EAAE,oBAAoB,CAAC,IAAI,CAAC,CAAC;CAAG"}