@mariozechner/pi-ai 0.5.10

package/dist/providers/google.js

@@ -0,0 +1,337 @@
+import { FunctionCallingConfigMode, GoogleGenAI, } from "@google/genai";
+import { calculateCost } from "../models.js";
+export class GoogleLLM {
+    client;
+    model;
+    constructor(model, apiKey) {
+        if (!apiKey) {
+            if (!process.env.GEMINI_API_KEY) {
+                throw new Error("Gemini API key is required. Set GEMINI_API_KEY environment variable or pass it as an argument.");
+            }
+            apiKey = process.env.GEMINI_API_KEY;
+        }
+        this.client = new GoogleGenAI({ apiKey });
+        this.model = model;
+    }
+    getModel() {
+        return this.model;
+    }
+    async complete(context, options) {
+        try {
+            const contents = this.convertMessages(context.messages);
+            // Build generation config
+            const generationConfig = {};
+            if (options?.temperature !== undefined) {
+                generationConfig.temperature = options.temperature;
+            }
+            if (options?.maxTokens !== undefined) {
+                generationConfig.maxOutputTokens = options.maxTokens;
+            }
+            // Build the config object
+            const config = {
+                ...(Object.keys(generationConfig).length > 0 && generationConfig),
+                ...(context.systemPrompt && { systemInstruction: context.systemPrompt }),
+                ...(context.tools && { tools: this.convertTools(context.tools) }),
+            };
+            // Add tool config if needed
+            if (context.tools && options?.toolChoice) {
+                config.toolConfig = {
+                    functionCallingConfig: {
+                        mode: this.mapToolChoice(options.toolChoice),
+                    },
+                };
+            }
+            // Add thinking config if enabled and model supports it
+            if (options?.thinking?.enabled && this.model.reasoning) {
+                config.thinkingConfig = {
+                    includeThoughts: true,
+                    ...(options.thinking.budgetTokens !== undefined && { thinkingBudget: options.thinking.budgetTokens }),
+                };
+            }
+            // Build the request parameters
+            const params = {
+                model: this.model.id,
+                contents,
+                config,
+            };
+            const stream = await this.client.models.generateContentStream(params);
+            let content = "";
+            let thinking = "";
+            let thoughtSignature;
+            const toolCalls = [];
+            let usage = {
+                input: 0,
+                output: 0,
+                cacheRead: 0,
+                cacheWrite: 0,
+                cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+            };
+            let stopReason = "stop";
+            let inTextBlock = false;
+            let inThinkingBlock = false;
+            // Process the stream
+            for await (const chunk of stream) {
+                // Extract parts from the chunk
+                const candidate = chunk.candidates?.[0];
+                if (candidate?.content?.parts) {
+                    for (const part of candidate.content.parts) {
+                        // Cast to any to access thinking properties not yet in SDK types
+                        const partWithThinking = part;
+                        if (partWithThinking.text !== undefined) {
+                            // Check if it's thinking content using the thought boolean flag
+                            if (partWithThinking.thought === true) {
+                                if (inTextBlock) {
+                                    options?.onText?.("", true);
+                                    inTextBlock = false;
+                                }
+                                thinking += partWithThinking.text;
+                                options?.onThinking?.(partWithThinking.text, false);
+                                inThinkingBlock = true;
+                                // Capture thought signature if present
+                                if (partWithThinking.thoughtSignature) {
+                                    thoughtSignature = partWithThinking.thoughtSignature;
+                                }
+                            }
+                            else {
+                                if (inThinkingBlock) {
+                                    options?.onThinking?.("", true);
+                                    inThinkingBlock = false;
+                                }
+                                content += partWithThinking.text;
+                                options?.onText?.(partWithThinking.text, false);
+                                inTextBlock = true;
+                            }
+                        }
+                        // Handle function calls
+                        if (part.functionCall) {
+                            if (inTextBlock) {
+                                options?.onText?.("", true);
+                                inTextBlock = false;
+                            }
+                            if (inThinkingBlock) {
+                                options?.onThinking?.("", true);
+                                inThinkingBlock = false;
+                            }
+                            // Gemini doesn't provide tool call IDs, so we need to generate them
+                            // Use the function name as part of the ID for better debugging
+                            const toolCallId = `${part.functionCall.name}_${Date.now()}`;
+                            toolCalls.push({
+                                id: toolCallId,
+                                name: part.functionCall.name || "",
+                                arguments: part.functionCall.args,
+                            });
+                        }
+                    }
+                }
+                // Map finish reason
+                if (candidate?.finishReason) {
+                    stopReason = this.mapStopReason(candidate.finishReason);
+                    if (toolCalls.length > 0) {
+                        stopReason = "toolUse";
+                    }
+                }
+                // Capture usage metadata if available
+                if (chunk.usageMetadata) {
+                    usage = {
+                        input: chunk.usageMetadata.promptTokenCount || 0,
+                        output: (chunk.usageMetadata.candidatesTokenCount || 0) + (chunk.usageMetadata.thoughtsTokenCount || 0),
+                        cacheRead: chunk.usageMetadata.cachedContentTokenCount || 0,
+                        cacheWrite: 0,
+                        cost: {
+                            input: 0,
+                            output: 0,
+                            cacheRead: 0,
+                            cacheWrite: 0,
+                            total: 0,
+                        },
+                    };
+                }
+            }
+            // Signal end of blocks
+            if (inTextBlock) {
+                options?.onText?.("", true);
+            }
+            if (inThinkingBlock) {
+                options?.onThinking?.("", true);
+            }
+            // Generate a thinking signature if we have thinking content but no signature from API
+            // This is needed for proper multi-turn conversations with thinking
+            if (thinking && !thoughtSignature) {
+                // Create a base64-encoded signature as Gemini expects
+                // In production, Gemini API should provide this
+                const encoder = new TextEncoder();
+                const data = encoder.encode(thinking);
+                // Create a simple hash-like signature and encode to base64
+                const signature = `gemini_thinking_${data.length}_${Date.now()}`;
+                thoughtSignature = Buffer.from(signature).toString("base64");
+            }
+            // Calculate cost
+            calculateCost(this.model, usage);
+            // Usage metadata is in the last chunk
+            // Already captured during streaming
+            return {
+                role: "assistant",
+                content: content || undefined,
+                thinking: thinking || undefined,
+                thinkingSignature: thoughtSignature,
+                toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
+                provider: this.model.provider,
+                model: this.model.id,
+                usage,
+                stopReason,
+            };
+        }
+        catch (error) {
+            return {
+                role: "assistant",
+                provider: this.model.provider,
+                model: this.model.id,
+                usage: {
+                    input: 0,
+                    output: 0,
+                    cacheRead: 0,
+                    cacheWrite: 0,
+                    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+                },
+                stopReason: "error",
+                error: error instanceof Error ? error.message : String(error),
+            };
+        }
+    }
+    convertMessages(messages) {
+        const contents = [];
+        for (const msg of messages) {
+            if (msg.role === "user") {
+                // Handle both string and array content
+                if (typeof msg.content === "string") {
+                    contents.push({
+                        role: "user",
+                        parts: [{ text: msg.content }],
+                    });
+                }
+                else {
+                    // Convert array content to Google format
+                    const parts = msg.content.map((item) => {
+                        if (item.type === "text") {
+                            return { text: item.text };
+                        }
+                        else {
+                            // Image content - Google uses inlineData
+                            return {
+                                inlineData: {
+                                    mimeType: item.mimeType,
+                                    data: item.data,
+                                },
+                            };
+                        }
+                    });
+                    contents.push({
+                        role: "user",
+                        parts,
+                    });
+                }
+            }
+            else if (msg.role === "assistant") {
+                const parts = [];
+                // Add thinking if present
+                // Note: We include thinkingSignature in our response for multi-turn context,
+                // but don't send it back to Gemini API as it may cause errors
+                if (msg.thinking) {
+                    parts.push({
+                        text: msg.thinking,
+                        thought: true,
+                        // Don't include thoughtSignature when sending back to API
+                        // thoughtSignature: msg.thinkingSignature,
+                    });
+                }
+                if (msg.content) {
+                    parts.push({ text: msg.content });
+                }
+                if (msg.toolCalls) {
+                    for (const toolCall of msg.toolCalls) {
+                        parts.push({
+                            functionCall: {
+                                name: toolCall.name,
+                                args: toolCall.arguments,
+                            },
+                        });
+                    }
+                }
+                if (parts.length > 0) {
+                    contents.push({
+                        role: "model",
+                        parts,
+                    });
+                }
+            }
+            else if (msg.role === "toolResult") {
+                // Tool results are sent as function responses
+                // Extract function name from the tool call ID (format: "functionName_timestamp")
+                const functionName = msg.toolCallId.substring(0, msg.toolCallId.lastIndexOf("_"));
+                contents.push({
+                    role: "user",
+                    parts: [
+                        {
+                            functionResponse: {
+                                name: functionName,
+                                response: {
+                                    result: msg.content,
+                                    isError: msg.isError || false,
+                                },
+                            },
+                        },
+                    ],
+                });
+            }
+        }
+        return contents;
+    }
+    convertTools(tools) {
+        return [
+            {
+                functionDeclarations: tools.map((tool) => ({
+                    name: tool.name,
+                    description: tool.description,
+                    parameters: tool.parameters,
+                })),
+            },
+        ];
+    }
+    mapToolChoice(choice) {
+        switch (choice) {
+            case "auto":
+                return FunctionCallingConfigMode.AUTO;
+            case "none":
+                return FunctionCallingConfigMode.NONE;
+            case "any":
+                return FunctionCallingConfigMode.ANY;
+            default:
+                return FunctionCallingConfigMode.AUTO;
+        }
+    }
+    mapStopReason(reason) {
+        switch (reason) {
+            case "STOP":
+                return "stop";
+            case "MAX_TOKENS":
+                return "length";
+            case "BLOCKLIST":
+            case "PROHIBITED_CONTENT":
+            case "SPII":
+            case "SAFETY":
+            case "IMAGE_SAFETY":
+                return "safety";
+            case "RECITATION":
+                return "safety";
+            case "FINISH_REASON_UNSPECIFIED":
+            case "OTHER":
+            case "LANGUAGE":
+            case "MALFORMED_FUNCTION_CALL":
+            case "UNEXPECTED_TOOL_CALL":
+                return "error";
+            default:
+                return "stop";
+        }
+    }
+}
+//# sourceMappingURL=google.js.map

package/dist/providers/google.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.js","sourceRoot":"","sources":["../../src/providers/google.ts"],"names":[],"mappings":"AAAA,OAAO,EAGN,yBAAyB,EAGzB,WAAW,GAEX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAsB7C,MAAM,OAAO,SAAS;IACb,MAAM,CAAc;IACpB,KAAK,CAAQ;IAErB,YAAY,KAAY,EAAE,MAAe;QACxC,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CACd,gGAAgG,CAChG,CAAC;YACH,CAAC;YACD,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QACrC,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACpB,CAAC;IAED,QAAQ;QACP,OAAO,IAAI,CAAC,KAAK,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAgB,EAAE,OAA0B;QAC1D,IAAI,CAAC;YACJ,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAExD,0BAA0B;YAC1B,MAAM,gBAAgB,GAA0B,EAAE,CAAC;YACnD,IAAI,OAAO,EAAE,WAAW,KAAK,SAAS,EAAE,CAAC;gBACxC,gBAAgB,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;YACpD,CAAC;YACD,IAAI,OAAO,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;gBACtC,gBAAgB,CAAC,eAAe,GAAG,OAAO,CAAC,SAAS,CAAC;YACtD,CAAC;YAED,0BAA0B;YAC1B,MAAM,MAAM,GAA0B;gBACrC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,gBAAgB,CAAC;gBACjE,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,iBAAiB,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC;gBACxE,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;aACjE,CAAC;YAEF,4BAA4B;YAC5B,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,EAAE,UAAU,EAAE,CAAC;gBAC1C,MAAM,CAAC,UAAU,GAAG;oBACnB,qBAAqB,EAAE;wBACtB,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,UAAU,CAAC;qBAC5C;iBACD,CAAC;YACH,CAAC;YAED,uDAAuD;YACvD,IAAI,OAAO,EAAE,QAAQ,EAAE,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;gBACxD,MAAM,CAAC,cAAc,GAAG;oBACvB,eAAe,EAAE,IAAI;oBACrB,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,KAAK,SAAS,IAAI,EAAE,cAAc,EAAE,OAAO,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC;iBACrG,CAAC;YACH,CAAC;YAED,+BAA+B;YAC/B,MAAM,MAAM,GAA8B;gBACzC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE;gBACpB,QAAQ;gBACR,MAAM;aACN,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;YAEtE,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,IAAI,QAAQ,GAAG,EAAE,CAAC;YAClB,IAAI,gBAAoC,CAAC;YACzC,MAAM,SAAS,GAAe,EAAE,CAAC;YACjC,IAAI,KAAK,GAAU;gBAClB,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,UAAU,EAAE,CAAC;gBACb,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;aACpE,CAAC;YACF,IAAI,UAAU,GAAe,MAAM,CAAC;YACpC,IAAI,WAAW,GAAG,KAAK,CAAC;YACxB,IAAI,eAAe,GAAG,KAAK,CAAC;YAE5B,qBAAqB;YACrB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAClC,+BAA+B;gBAC/B,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;gBACxC,IAAI,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;oBAC/B,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;wBAC5C,iEAAiE;wBACjE,MAAM,gBAAgB,GAAG,IAAI,CAAC;wBAC9B,IAAI,gBAAgB,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;4BACzC,gEAAgE;4BAChE,IAAI,gBAAgB,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;gCACvC,IAAI,WAAW,EAAE,CAAC;oCACjB,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;oCAC5B,WAAW,GAAG,KAAK,CAAC;gCACrB,CAAC;gCACD,QAAQ,IAAI,gBAAgB,CAAC,IAAI,CAAC;gCAClC,OAAO,EAAE,UAAU,EAAE,CAAC,gBAAgB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gCACpD,eAAe,GAAG,IAAI,CAAC;gCACvB,uCAAuC;gCACvC,IAAI,gBAAgB,CAAC,gBAAgB,EAAE,CAAC;oCACvC,gBAAgB,GAAG,gBAAgB,CAAC,gBAAgB,CAAC;gCACtD,CAAC;4BACF,CAAC;iCAAM,CAAC;gCACP,IAAI,eAAe,EAAE,CAAC;oCACrB,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;oCAChC,eAAe,GAAG,KAAK,CAAC;gCACzB,CAAC;gCACD,OAAO,IAAI,gBAAgB,CAAC,IAAI,CAAC;gCACjC,OAAO,EAAE,MAAM,EAAE,CAAC,gBAAgB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;gCAChD,WAAW,GAAG,IAAI,CAAC;4BACpB,CAAC;wBACF,CAAC;wBAED,wBAAwB;wBACxB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;4BACvB,IAAI,WAAW,EAAE,CAAC;gCACjB,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gCAC5B,WAAW,GAAG,KAAK,CAAC;4BACrB,CAAC;4BACD,IAAI,eAAe,EAAE,CAAC;gCACrB,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;gCAChC,eAAe,GAAG,KAAK,CAAC;4BACzB,CAAC;4BAED,oEAAoE;4BACpE,+DAA+D;4BAC/D,MAAM,UAAU,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;4BAC7D,SAAS,CAAC,IAAI,CAAC;gCACd,EAAE,EAAE,UAAU;gCACd,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE;gCAClC,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,IAA2B;6BACxD,CAAC,CAAC;wBACJ,CAAC;oBACF,CAAC;gBACF,CAAC;gBAED,oBAAoB;gBACpB,IAAI,SAAS,EAAE,YAAY,EAAE,CAAC;oBAC7B,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;oBACxD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC1B,UAAU,GAAG,SAAS,CAAC;oBACxB,CAAC;gBACF,CAAC;gBAED,sCAAsC;gBACtC,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;oBACzB,KAAK,GAAG;wBACP,KAAK,EAAE,KAAK,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC;wBAChD,MAAM,EACL,CAAC,KAAK,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,kBAAkB,IAAI,CAAC,CAAC;wBAChG,SAAS,EAAE,KAAK,CAAC,aAAa,CAAC,uBAAuB,IAAI,CAAC;wBAC3D,UAAU,EAAE,CAAC;wBACb,IAAI,EAAE;4BACL,KAAK,EAAE,CAAC;4BACR,MAAM,EAAE,CAAC;4BACT,SAAS,EAAE,CAAC;4BACZ,UAAU,EAAE,CAAC;4BACb,KAAK,EAAE,CAAC;yBACR;qBACD,CAAC;gBACH,CAAC;YACF,CAAC;YAED,uBAAuB;YACvB,IAAI,WAAW,EAAE,CAAC;gBACjB,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YAC7B,CAAC;YACD,IAAI,eAAe,EAAE,CAAC;gBACrB,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACjC,CAAC;YAED,sFAAsF;YACtF,mEAAmE;YACnE,IAAI,QAAQ,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACnC,sDAAsD;gBACtD,gDAAgD;gBAChD,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;gBAClC,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACtC,2DAA2D;gBAC3D,MAAM,SAAS,GAAG,mBAAmB,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;gBACjE,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAC9D,CAAC;YAED,iBAAiB;YACjB,aAAa,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YAEjC,sCAAsC;YACtC,oCAAoC;YAEpC,OAAO;gBACN,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,OAAO,IAAI,SAAS;gBAC7B,QAAQ,EAAE,QAAQ,IAAI,SAAS;gBAC/B,iBAAiB,EAAE,gBAAgB;gBACnC,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACvD,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ;gBAC7B,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE;gBACpB,KAAK;gBACL,UAAU;aACV,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO;gBACN,IAAI,EAAE,WAAW;gBACjB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ;gBAC7B,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE;gBACpB,KAAK,EAAE;oBACN,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,CAAC;oBACT,SAAS,EAAE,CAAC;oBACZ,UAAU,EAAE,CAAC;oBACb,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;iBACpE;gBACD,UAAU,EAAE,OAAO;gBACnB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC7D,CAAC;QACH,CAAC;IACF,CAAC;IAEO,eAAe,CAAC,QAAmB;QAC1C,MAAM,QAAQ,GAAc,EAAE,CAAC;QAE/B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC5B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACzB,uCAAuC;gBACvC,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACrC,QAAQ,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;qBAC9B,CAAC,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACP,yCAAyC;oBACzC,MAAM,KAAK,GAAW,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;wBAC9C,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BAC1B,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC;wBAC5B,CAAC;6BAAM,CAAC;4BACP,yCAAyC;4BACzC,OAAO;gCACN,UAAU,EAAE;oCACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;oCACvB,IAAI,EAAE,IAAI,CAAC,IAAI;iCACf;6BACD,CAAC;wBACH,CAAC;oBACF,CAAC,CAAC,CAAC;oBACH,QAAQ,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,MAAM;wBACZ,KAAK;qBACL,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACrC,MAAM,KAAK,GAAW,EAAE,CAAC;gBAEzB,0BAA0B;gBAC1B,6EAA6E;gBAC7E,8DAA8D;gBAC9D,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;oBAClB,KAAK,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,GAAG,CAAC,QAAQ;wBAClB,OAAO,EAAE,IAAI;wBACb,0DAA0D;wBAC1D,2CAA2C;qBAC3C,CAAC,CAAC;gBACJ,CAAC;gBAED,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;oBACjB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;gBACnC,CAAC;gBAED,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;oBACnB,KAAK,MAAM,QAAQ,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;wBACtC,KAAK,CAAC,IAAI,CAAC;4BACV,YAAY,EAAE;gCACb,IAAI,EAAE,QAAQ,CAAC,IAAI;gCACnB,IAAI,EAAE,QAAQ,CAAC,SAAS;6BACxB;yBACD,CAAC,CAAC;oBACJ,CAAC;gBACF,CAAC;gBAED,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtB,QAAQ,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,OAAO;wBACb,KAAK;qBACL,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBACtC,8CAA8C;gBAC9C,iFAAiF;gBACjF,MAAM,YAAY,GAAG,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;gBAClF,QAAQ,CAAC,IAAI,CAAC;oBACb,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE;wBACN;4BACC,gBAAgB,EAAE;gCACjB,IAAI,EAAE,YAAY;gCAClB,QAAQ,EAAE;oCACT,MAAM,EAAE,GAAG,CAAC,OAAO;oCACnB,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,KAAK;iCAC7B;6BACD;yBACD;qBACD;iBACD,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;QAED,OAAO,QAAQ,CAAC;IACjB,CAAC;IAEO,YAAY,CAAC,KAAa;QACjC,OAAO;YACN;gBACC,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAC1C,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;iBAC3B,CAAC,CAAC;aACH;SACD,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,MAAc;QACnC,QAAQ,MAAM,EAAE,CAAC;YAChB,KAAK,MAAM;gBACV,OAAO,yBAAyB,CAAC,IAAI,CAAC;YACvC,KAAK,MAAM;gBACV,OAAO,yBAAyB,CAAC,IAAI,CAAC;YACvC,KAAK,KAAK;gBACT,OAAO,yBAAyB,CAAC,GAAG,CAAC;YACtC;gBACC,OAAO,yBAAyB,CAAC,IAAI,CAAC;QACxC,CAAC;IACF,CAAC;IAEO,aAAa,CAAC,MAAoB;QACzC,QAAQ,MAAM,EAAE,CAAC;YAChB,KAAK,MAAM;gBACV,OAAO,MAAM,CAAC;YACf,KAAK,YAAY;gBAChB,OAAO,QAAQ,CAAC;YACjB,KAAK,WAAW,CAAC;YACjB,KAAK,oBAAoB,CAAC;YAC1B,KAAK,MAAM,CAAC;YACZ,KAAK,QAAQ,CAAC;YACd,KAAK,cAAc;gBAClB,OAAO,QAAQ,CAAC;YACjB,KAAK,YAAY;gBAChB,OAAO,QAAQ,CAAC;YACjB,KAAK,2BAA2B,CAAC;YACjC,KAAK,OAAO,CAAC;YACb,KAAK,UAAU,CAAC;YAChB,KAAK,yBAAyB,CAAC;YAC/B,KAAK,sBAAsB;gBAC1B,OAAO,OAAO,CAAC;YAChB;gBACC,OAAO,MAAM,CAAC;QAChB,CAAC;IACF,CAAC;CACD"}

package/dist/providers/openai-completions.d.ts

@@ -0,0 +1,21 @@
+import type { AssistantMessage, Context, LLM, LLMOptions, Model } from "../types.js";
+export interface OpenAICompletionsLLMOptions extends LLMOptions {
+    toolChoice?: "auto" | "none" | "required" | {
+        type: "function";
+        function: {
+            name: string;
+        };
+    };
+    reasoningEffort?: "low" | "medium" | "high";
+}
+export declare class OpenAICompletionsLLM implements LLM<OpenAICompletionsLLMOptions> {
+    private client;
+    private modelInfo;
+    constructor(model: Model, apiKey?: string);
+    getModel(): Model;
+    complete(request: Context, options?: OpenAICompletionsLLMOptions): Promise<AssistantMessage>;
+    private convertMessages;
+    private convertTools;
+    private mapStopReason;
+}
+//# sourceMappingURL=openai-completions.d.ts.map

package/dist/providers/openai-completions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai-completions.d.ts","sourceRoot":"","sources":["../../src/providers/openai-completions.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EACX,gBAAgB,EAChB,OAAO,EACP,GAAG,EACH,UAAU,EAEV,KAAK,EAKL,MAAM,aAAa,CAAC;AAErB,MAAM,WAAW,2BAA4B,SAAQ,UAAU;IAC9D,UAAU,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,UAAU,GAAG;QAAE,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,EAAE;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,CAAC;IAC7F,eAAe,CAAC,EAAE,KAAK,GAAG,QAAQ,GAAG,MAAM,CAAC;CAC5C;AAED,qBAAa,oBAAqB,YAAW,GAAG,CAAC,2BAA2B,CAAC;IAC5E,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAQ;gBAEb,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,EAAE,MAAM;IAazC,QAAQ,IAAI,KAAK;IAIX,QAAQ,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,2BAA2B,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA+MlG,OAAO,CAAC,eAAe;IAiFvB,OAAO,CAAC,YAAY;IAWpB,OAAO,CAAC,aAAa;CAerB"}