@curie-agent/providers 0.1.2

package/dist/src/google.js

@@ -0,0 +1,295 @@
+import { GoogleGenerativeAI } from '@google/generative-ai';
+let _callId = 0;
+function nextCallId() {
+    return `call_${++_callId}`;
+}
+/** Returns true if the model may return thought_signature fields on function call parts. */
+function isThinkingModelName(model) {
+    const m = model.toLowerCase();
+    return m.startsWith('gemini-2.5') || m.startsWith('gemini-3') || m.includes('flash-latest') || m.includes('pro-latest');
+}
+export class GoogleGeminiProvider {
+    name = 'google';
+    models = [
+        'gemini-2.5-pro',
+        'gemini-2.5-flash',
+        'gemini-2.0-flash-lite',
+        'gemini-2.0-flash',
+    ];
+    defaultModel = 'gemini-2.0-flash';
+    genAI;
+    baseUrl;
+    constructor(apiKey, baseUrl) {
+        const resolvedKey = (apiKey && apiKey.length > 0)
+            ? apiKey
+            : process.env.GOOGLE_API_KEY;
+        if (!resolvedKey) {
+            throw new Error('GoogleGeminiProvider: no API key (pass explicitly or set GOOGLE_API_KEY).');
+        }
+        this.genAI = new GoogleGenerativeAI(resolvedKey);
+        this.baseUrl = baseUrl || '';
+    }
+    isThinkingModel(model) {
+        return isThinkingModelName(model || this.defaultModel);
+    }
+    baseModelParams() {
+        const params = { model: this.defaultModel };
+        if (this.baseUrl) {
+            params['baseUrl'] = this.baseUrl;
+        }
+        return params;
+    }
+    getModel(model, tools) {
+        const m = model || this.defaultModel;
+        const params = { ...this.baseModelParams(), model: m };
+        if (this.isThinkingModel(m)) {
+            params['enable_thoughts'] = false;
+        }
+        if (tools && tools.length > 0) {
+            params['tools'] = [{ functionDeclarations: tools }];
+        }
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        return this.genAI.getGenerativeModel(params);
+    }
+    stream(args) {
+        const model = args.model || this.defaultModel;
+        const tools = args.tools ? this.toFunctionDeclarations(args.tools) : undefined;
+        const genModel = this.getModel(model, tools);
+        const contents = this.mapMessages(args.messages);
+        const requestTools = args.tools ? [{ functionDeclarations: this.toFunctionDeclarations(args.tools) }] : undefined;
+        async function* generator() {
+            const streamResult = await genModel.generateContentStream({
+                contents,
+                systemInstruction: args.system,
+                ...(requestTools ? { tools: requestTools } : {}),
+            });
+            // Collect raw chunk parts as they stream in to preserve thoughtSignature,
+            // which the SDK's aggregated response object drops from its typed interface.
+            const rawParts = [];
+            for await (const chunk of streamResult.stream) {
+                if (args.signal?.aborted) {
+                    break;
+                }
+                const text = chunk.text();
+                if (text) {
+                    yield { type: 'text-delta', text };
+                }
+                // Capture raw parts from each chunk before the SDK strips unknown fields
+                const chunkParts = chunk.candidates?.[0]?.content?.parts ?? [];
+                for (const part of chunkParts) {
+                    rawParts.push(part);
+                }
+            }
+            if (args.signal?.aborted) {
+                yield { type: 'stop', reason: 'aborted' };
+                return;
+            }
+            try {
+                // Prefer raw parts collected during streaming (preserves thoughtSignature).
+                // Fall back to aggregated response parts if streaming collected nothing.
+                const response = await streamResult.response;
+                const aggParts = response.candidates?.[0]?.content?.parts ?? [];
+                const parts = rawParts.length > 0 ? rawParts : aggParts;
+                for (const p of parts) {
+                    if (p.functionCall && typeof p.functionCall === 'object') {
+                        const fc = p.functionCall;
+                        const sig = (typeof p.thoughtSignature === 'string' ? p.thoughtSignature : undefined)
+                            ?? (typeof p.thought_signature === 'string' ? p.thought_signature : undefined);
+                        yield {
+                            type: 'tool-call',
+                            id: nextCallId(),
+                            name: fc.name,
+                            input: (fc.args ?? {}),
+                            ...(sig ? { thoughtSignature: sig } : {}),
+                        };
+                    }
+                }
+            }
+            catch {
+                // Response may throw if prompt was blocked
+            }
+            yield { type: 'stop', reason: 'stop' };
+        }
+        return {
+            iterable: generator(),
+            cancel() {
+                // The Google Gemini SDK doesn't expose a public cancel method.
+                // The signal.aborted check inside the generator loop breaks on the
+                // next iteration, but there's no way to synchronously abort.
+            },
+        };
+    }
+    async check(prompt, args) {
+        const model = args?.model || this.defaultModel;
+        const genModel = this.getModel(model);
+        const response = await genModel.generateContent({
+            contents: [{ role: 'user', parts: [{ text: prompt }] }],
+            systemInstruction: args?.system,
+        });
+        return response.response.text().trim().slice(0, 256);
+    }
+    async complete(args) {
+        const model = args.model || this.defaultModel;
+        const tools = args.tools ? this.toFunctionDeclarations(args.tools) : undefined;
+        const genModel = this.getModel(model, tools);
+        let contents = this.mapMessages(args.messages);
+        const allText = [];
+        let stopReason = 'stop';
+        let toolCalls;
+        for (let turn = 0; turn < 16; turn++) {
+            const requestTools = args.tools ? [{ functionDeclarations: this.toFunctionDeclarations(args.tools) }] : undefined;
+            const response = await genModel.generateContent({
+                contents,
+                systemInstruction: args.system,
+                ...(requestTools ? { tools: requestTools } : {}),
+            });
+            const text = response.response.text().trim();
+            if (text)
+                allText.push(text);
+            // Use raw parts to preserve thought_signature for thinking models.
+            const rawParts = response.response.candidates?.[0]?.content?.parts ?? [];
+            const fcParts = rawParts.filter((p) => p.functionCall);
+            if (fcParts.length === 0) {
+                stopReason = 'stop';
+                break;
+            }
+            toolCalls = fcParts.map((p) => {
+                const fc = p.functionCall;
+                return { id: nextCallId(), name: fc.name, input: (fc.args ?? {}) };
+            });
+            stopReason = 'tool_use';
+            // Build follow-up content echoing thoughtSignature back for thinking models.
+            // thoughtSignature is a top-level Part field (camelCase in JS SDK).
+            contents.push({
+                role: 'model',
+                parts: fcParts.map((p) => {
+                    const raw = p;
+                    const fc = raw.functionCall;
+                    const sig = (typeof raw.thoughtSignature === 'string' ? raw.thoughtSignature : undefined)
+                        ?? (typeof raw.thought_signature === 'string' ? raw.thought_signature : undefined);
+                    const part = { functionCall: { name: fc.name, args: fc.args } };
+                    if (sig)
+                        part.thoughtSignature = sig;
+                    return part;
+                }),
+            });
+            for (const p of fcParts) {
+                const fc = p.functionCall;
+                contents.push({
+                    role: 'user',
+                    parts: [{
+                            functionResponse: {
+                                name: fc.name,
+                                response: { result: fc.args },
+                            },
+                        }],
+                });
+            }
+        }
+        return {
+            text: allText.join('\n'),
+            stopReason,
+            toolCalls,
+        };
+    }
+    toFunctionDeclarations(tools) {
+        return tools.map((t) => ({
+            name: t.name,
+            description: t.description,
+            parameters: this.toFunctionSchema(t.inputSchema),
+        }));
+    }
+    /**
+     * Convert a JSON Schema object to Google's FunctionDeclarationSchema format.
+     * Google's API is strict: it does NOT support the "required" array in
+     * function declaration parameters. Required fields are communicated via
+     * description text instead.
+     */
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    toFunctionSchema(schema) {
+        if (typeof schema !== 'object' || schema === null) {
+            return { type: 'object', properties: {} };
+        }
+        const s = schema;
+        const type = typeof s.type === 'string' ? s.type : 'object';
+        if (type === 'object') {
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const properties = {};
+            if (s.properties && typeof s.properties === 'object') {
+                for (const [key, val] of Object.entries(s.properties)) {
+                    properties[key] = this.toFunctionSchema(val);
+                }
+            }
+            // Google Gemini API does NOT accept "required" in parameters schema.
+            // Omit it; the model will still infer importance from descriptions.
+            return { type: 'object', properties };
+        }
+        // For non-object types, return a simple schema
+        // Google requires "items" for array types
+        if (type === 'array') {
+            const itemsSchema = s.items ? this.toFunctionSchema(s.items) : { type: 'string' };
+            return { type: 'array', items: itemsSchema };
+        }
+        return { type, properties: {} };
+    }
+    mapMessages(messages) {
+        return messages.map((m) => {
+            if (m.role === 'tool') {
+                return {
+                    role: 'user',
+                    parts: [{
+                            functionResponse: {
+                                name: m.content,
+                                response: { output: m.content },
+                            },
+                        }],
+                };
+            }
+            if (typeof m.content === 'string') {
+                return {
+                    role: m.role === 'assistant' ? 'model' : 'user',
+                    parts: [{ text: m.content }],
+                };
+            }
+            if (Array.isArray(m.content)) {
+                const parts = [];
+                for (const c of m.content) {
+                    if (c.type === 'text') {
+                        parts.push({ text: c.text });
+                    }
+                    else if (c.type === 'tool-use') {
+                        // thoughtSignature is a top-level Part field (camelCase in JS SDK)
+                        const fcPart = {
+                            functionCall: { name: c.name, args: c.input },
+                            ...(c.thoughtSignature ? { thoughtSignature: c.thoughtSignature } : {}),
+                        };
+                        parts.push(fcPart);
+                    }
+                    else if (c.type === 'tool-result') {
+                        parts.push({
+                            functionResponse: {
+                                name: c.tool_use_id,
+                                response: { output: c.content },
+                            },
+                        });
+                    }
+                    else if (c.type === 'image') {
+                        parts.push({ text: `[image: ${c.source.media_type}]` });
+                    }
+                    else {
+                        parts.push({ text: JSON.stringify(c) });
+                    }
+                }
+                return {
+                    role: m.role === 'assistant' ? 'model' : 'user',
+                    parts,
+                };
+            }
+            return {
+                role: 'user',
+                parts: [{ text: JSON.stringify(m.content) }],
+            };
+        });
+    }
+}
+//# sourceMappingURL=google.js.map

package/dist/src/google.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.js","sourceRoot":"","sources":["../../src/google.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAc3D,IAAI,OAAO,GAAG,CAAC,CAAC;AAChB,SAAS,UAAU;IACjB,OAAO,QAAQ,EAAE,OAAO,EAAE,CAAC;AAC7B,CAAC;AAED,4FAA4F;AAC5F,SAAS,mBAAmB,CAAC,KAAa;IACxC,MAAM,CAAC,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAC9B,OAAO,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;AAC1H,CAAC;AAED,MAAM,OAAO,oBAAoB;IACtB,IAAI,GAAG,QAAQ,CAAC;IAChB,MAAM,GAAG;QAChB,gBAAgB;QAChB,kBAAkB;QAClB,uBAAuB;QACvB,kBAAkB;KACnB,CAAC;IACO,YAAY,GAAG,kBAAkB,CAAC;IAEnC,KAAK,CAAqB;IAC1B,OAAO,CAAS;IAExB,YAAY,MAAe,EAAE,OAAgB;QAC3C,MAAM,WAAW,GAAG,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;YAC/C,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAC/B,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,2EAA2E,CAC5E,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,IAAI,kBAAkB,CAAC,WAAW,CAAC,CAAC;QACjD,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,EAAE,CAAC;IAC/B,CAAC;IAEO,eAAe,CAAC,KAAc;QACpC,OAAO,mBAAmB,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC;IACzD,CAAC;IAEO,eAAe;QACrB,MAAM,MAAM,GAA4B,EAAE,KAAK,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC;QACrE,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACnC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEQ,QAAQ,CAAC,KAAc,EAAE,KAAe;QAC/C,MAAM,CAAC,GAAG,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QACrC,MAAM,MAAM,GAA4B,EAAE,GAAG,IAAI,CAAC,eAAe,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;QAChF,IAAI,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5B,MAAM,CAAC,iBAAiB,CAAC,GAAG,KAAK,CAAC;QACpC,CAAC;QACD,IAAI,KAAK,IAAK,KAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7C,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAAC,CAAC;QACtD,CAAC;QACD,8DAA8D;QAC9D,OAAO,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,MAAa,CAAC,CAAC;IACtD,CAAC;IAEF,MAAM,CAAC,IAAwB;QAC5B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/E,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAE7C,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAElH,KAAK,SAAS,CAAC,CAAC,SAAS;YACvB,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,qBAAqB,CAAC;gBACxD,QAAQ;gBACR,iBAAiB,EAAE,IAAI,CAAC,MAAM;gBAC9B,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,YAAqB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC1D,CAAC,CAAC;YAEH,0EAA0E;YAC1E,6EAA6E;YAC7E,MAAM,QAAQ,GAAmC,EAAE,CAAC;YAEpD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;gBAC9C,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;oBACzB,MAAM;gBACR,CAAC;gBAED,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;gBAC1B,IAAI,IAAI,EAAE,CAAC;oBACT,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;gBACrC,CAAC;gBAED,yEAAyE;gBACzE,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;gBAC/D,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;oBAC9B,QAAQ,CAAC,IAAI,CAAC,IAA0C,CAAC,CAAC;gBAC5D,CAAC;YACH,CAAC;YAED,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;gBACzB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;gBAC1C,OAAO;YACT,CAAC;YAED,IAAI,CAAC;gBACH,4EAA4E;gBAC5E,yEAAyE;gBACzE,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC;gBAC7C,MAAM,QAAQ,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;gBAChE,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAE,QAAiD,CAAC;gBAElG,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;oBACtB,IAAI,CAAC,CAAC,YAAY,IAAI,OAAO,CAAC,CAAC,YAAY,KAAK,QAAQ,EAAE,CAAC;wBACzD,MAAM,EAAE,GAAG,CAAC,CAAC,YAAgE,CAAC;wBAC9E,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,gBAAgB,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS,CAAC;+BACzE,CAAC,OAAO,CAAC,CAAC,iBAAiB,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;wBACxF,MAAM;4BACJ,IAAI,EAAE,WAAW;4BACjB,EAAE,EAAE,UAAU,EAAE;4BAChB,IAAI,EAAE,EAAE,CAAC,IAAI;4BACb,KAAK,EAAE,CAAC,EAAE,CAAC,IAAI,IAAI,EAAE,CAA4B;4BACjD,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;yBAC1C,CAAC;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,2CAA2C;YAC7C,CAAC;YAED,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;QACzC,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,SAAS,EAAE;YACrB,MAAM;gBACJ,+DAA+D;gBAC/D,mEAAmE;gBACnE,6DAA6D;YAC/D,CAAC;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,IAI3B;QACC,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAEtC,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,eAAe,CAAC;YAC9C,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;YACvD,iBAAiB,EAAE,IAAI,EAAE,MAAM;SAChC,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAA0B;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/E,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAE7C,IAAI,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,IAAI,UAAU,GAAyC,MAAM,CAAC;QAC9D,IAAI,SAA0F,CAAC;QAE/F,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC;YACrC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAElH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,eAAe,CAAC;gBAC9C,QAAQ;gBACR,iBAAiB,EAAE,IAAI,CAAC,MAAM;gBAC9B,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,YAAqB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC1D,CAAC,CAAC;YAEH,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YAC7C,IAAI,IAAI;gBAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE7B,mEAAmE;YACnE,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;YACzE,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAwC,CAAC,YAAY,CAAC,CAAC;YAC/F,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,UAAU,GAAG,MAAM,CAAC;gBACpB,MAAM;YACR,CAAC;YAED,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC5B,MAAM,EAAE,GAAI,CAAwC,CAAC,YAAgE,CAAC;gBACtH,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,IAAI,IAAI,EAAE,CAA4B,EAAE,CAAC;YAChG,CAAC,CAAC,CAAC;YACH,UAAU,GAAG,UAAU,CAAC;YAExB,6EAA6E;YAC7E,oEAAoE;YACpE,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;oBACvB,MAAM,GAAG,GAAG,CAAuC,CAAC;oBACpD,MAAM,EAAE,GAAG,GAAG,CAAC,YAAuC,CAAC;oBACvD,MAAM,GAAG,GAAG,CAAC,OAAO,GAAG,CAAC,gBAAgB,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS,CAAC;2BAC7E,CAAC,OAAO,GAAG,CAAC,iBAAiB,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;oBAC5F,MAAM,IAAI,GAA4B,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;oBACzF,IAAI,GAAG;wBAAE,IAAI,CAAC,gBAAgB,GAAG,GAAG,CAAC;oBACrC,OAAO,IAAI,CAAC;gBACd,CAAC,CAAC;aACM,CAAC,CAAC;YAEZ,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACxB,MAAM,EAAE,GAAI,CAAwC,CAAC,YAAgE,CAAC;gBACtH,QAAQ,CAAC,IAAI,CAAC;oBACZ,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC;4BACN,gBAAgB,EAAE;gCAChB,IAAI,EAAE,EAAE,CAAC,IAAI;gCACb,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,IAAI,EAAE;6BAC9B;yBACF,CAAC;iBACM,CAAC,CAAC;YACd,CAAC;QACH,CAAC;QAED,OAAO;YACL,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;YACxB,UAAU;YACV,SAAS;SACV,CAAC;IACJ,CAAC;IAEO,sBAAsB,CAAC,KAAuB;QACpD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;YAC1B,UAAU,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,WAAW,CAAC;SACjD,CAAC,CAAC,CAAC;IACN,CAAC;IAED;;;;;OAKG;IACH,8DAA8D;IACtD,gBAAgB,CAAC,MAAe;QACtC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;YAClD,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;QAC5C,CAAC;QAED,MAAM,CAAC,GAAG,MAAiC,CAAC;QAC5C,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC;QAE5D,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;YACtB,8DAA8D;YAC9D,MAAM,UAAU,GAAwB,EAAE,CAAC;YAC3C,IAAI,CAAC,CAAC,UAAU,IAAI,OAAO,CAAC,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;gBACrD,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,UAAqC,CAAC,EAAE,CAAC;oBACjF,UAAU,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;gBAC/C,CAAC;YACH,CAAC;YACD,qEAAqE;YACrE,oEAAoE;YACpE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;QACxC,CAAC;QAED,+CAA+C;QAC/C,0CAA0C;QAC1C,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACrB,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;YAClF,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;QAC/C,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAClC,CAAC;IAEO,WAAW,CAAC,QAA2B;QAI7C,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACxB,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACtB,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC;4BACN,gBAAgB,EAAE;gCAChB,IAAI,EAAE,CAAC,CAAC,OAAO;gCACf,QAAQ,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;6BAChC;yBACF,CAAC;iBACH,CAAC;YACJ,CAAC;YAED,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAClC,OAAO;oBACL,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;oBAC/C,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;iBAC7B,CAAC;YACJ,CAAC;YAED,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC7B,MAAM,KAAK,GAA2J,EAAE,CAAC;gBAEzK,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;oBAC1B,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACtB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC/B,CAAC;yBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACjC,mEAAmE;wBACnE,MAAM,MAAM,GAA4B;4BACtC,YAAY,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,KAAe,EAAE;4BACvD,GAAG,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;yBACxE,CAAC;wBACF,KAAK,CAAC,IAAI,CAAC,MAAe,CAAC,CAAC;oBAC9B,CAAC;yBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;wBACpC,KAAK,CAAC,IAAI,CAAC;4BACT,gBAAgB,EAAE;gCAChB,IAAI,EAAE,CAAC,CAAC,WAAW;gCACnB,QAAQ,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;6BAChC;yBACF,CAAC,CAAC;oBACL,CAAC;yBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;wBAC9B,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC,MAAM,CAAC,UAAU,GAAG,EAAE,CAAC,CAAC;oBAC1D,CAAC;yBAAM,CAAC;wBACN,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBAC1C,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;oBAC/C,KAAK;iBACN,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;aAC7C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/src/index.d.ts

@@ -0,0 +1,7 @@
+export { createRegistry } from './provider.js';
+export type { Provider, ProviderEvent, ProviderStreamArgs, ProviderMessage, MessageContent, ProviderRegistry, ReasoningEffort, } from './provider.js';
+export { AnthropicProvider } from './anthropic.js';
+export { OpenAIProvider } from './openai.js';
+export { OllamaProvider } from './ollama.js';
+export { GoogleGeminiProvider } from './google.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/src/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,YAAY,EACV,QAAQ,EACR,aAAa,EACb,kBAAkB,EAClB,eAAe,EACf,cAAc,EACd,gBAAgB,EAChB,eAAe,GAChB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC"}

package/dist/src/index.js

@@ -0,0 +1,6 @@
+export { createRegistry } from './provider.js';
+export { AnthropicProvider } from './anthropic.js';
+export { OpenAIProvider } from './openai.js';
+export { OllamaProvider } from './ollama.js';
+export { GoogleGeminiProvider } from './google.js';
+//# sourceMappingURL=index.js.map

package/dist/src/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAU/C,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC"}

package/dist/src/ollama.d.ts

@@ -0,0 +1,26 @@
+import type { Provider, ProviderEvent, ProviderStreamArgs, ProviderCompleteArgs, ProviderCompleteResult } from './provider.js';
+type CancelableIterable<T> = {
+    iterable: AsyncIterable<T>;
+    cancel(): void;
+};
+export declare class OllamaProvider implements Provider {
+    readonly name = "ollama";
+    readonly models: string[];
+    readonly defaultModel = "custom";
+    private client;
+    private _baseUrl;
+    private _allModels;
+    constructor(apiKey?: string, baseUrl?: string);
+    private _fetchModels;
+    stream(args: ProviderStreamArgs): CancelableIterable<ProviderEvent>;
+    check(prompt: string, args?: {
+        model?: string;
+        system?: string;
+        signal?: AbortSignal;
+    }): Promise<string>;
+    complete(args: ProviderCompleteArgs): Promise<ProviderCompleteResult>;
+    private mapMessages;
+    private mapTools;
+}
+export {};
+//# sourceMappingURL=ollama.d.ts.map

package/dist/src/ollama.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../src/ollama.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,QAAQ,EACR,aAAa,EACb,kBAAkB,EAClB,oBAAoB,EACpB,sBAAsB,EAGvB,MAAM,eAAe,CAAC;AAEvB,KAAK,kBAAkB,CAAC,CAAC,IAAI;IAAE,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC;IAAC,MAAM,IAAI,IAAI,CAAA;CAAE,CAAC;AAe5E,qBAAa,cAAe,YAAW,QAAQ;IAC7C,QAAQ,CAAC,IAAI,YAAY;IACzB,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;IAC1B,QAAQ,CAAC,YAAY,YAAY;IAEjC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,UAAU,CAAyB;gBAE/B,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM;YAY/B,YAAY;IAmB1B,MAAM,CAAC,IAAI,EAAE,kBAAkB,GAAG,kBAAkB,CAAC,aAAa,CAAC;IA0F7D,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE;QACjC,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,WAAW,CAAC;KACtB,GAAG,OAAO,CAAC,MAAM,CAAC;IAiBb,QAAQ,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAwD3E,OAAO,CAAC,WAAW;IA+CnB,OAAO,CAAC,QAAQ;CAUjB"}

package/dist/src/ollama.js

@@ -0,0 +1,245 @@
+import OpenAI from 'openai';
+const FALLBACK_MODELS = [
+    'llama3.3',
+    'mistral',
+    'phi3',
+    'qwen2.5',
+    'deepseek-r1',
+    'llama3.1',
+    'llama3',
+    'codellama',
+    'gemma2',
+    'mixtral',
+];
+export class OllamaProvider {
+    name = 'ollama';
+    models;
+    defaultModel = 'custom';
+    client;
+    _baseUrl;
+    _allModels = null;
+    constructor(apiKey, baseUrl) {
+        const resolvedKey = (apiKey && apiKey.length > 0)
+            ? apiKey
+            : process.env.MODEL_API_KEY;
+        this._baseUrl = baseUrl || 'http://localhost:11434/v1';
+        this.client = new OpenAI({
+            apiKey: resolvedKey || 'sk-not-needed',
+            baseURL: this._baseUrl,
+        });
+        this.models = [...FALLBACK_MODELS];
+    }
+    async _fetchModels() {
+        if (this._allModels)
+            return this._allModels;
+        try {
+            const baseUrl = this._baseUrl.replace(/\/v1(\/)?$/, '');
+            const resp = await fetch(`${baseUrl}/api/tags`, {
+                signal: AbortSignal.timeout(3000),
+            });
+            const data = await resp.json();
+            if (data.models?.length) {
+                this._allModels = data.models.map(m => m.name);
+                return this._allModels;
+            }
+        }
+        catch {
+            /* fallback to static list */
+        }
+        this._allModels = [...FALLBACK_MODELS];
+        return this._allModels;
+    }
+    stream(args) {
+        const self = this;
+        const model = args.model || this.defaultModel;
+        const messages = this.mapMessages(args.messages);
+        const tools = args.tools ? this.mapTools(args.tools) : undefined;
+        const allMessages = args.system
+            ? [{ role: 'system', content: args.system }, ...messages]
+            : messages;
+        // Ollama-specific: omit stream_options, thinking blocks, token caching
+        const streamParams = {
+            model,
+            messages: allMessages,
+            stream: true,
+            ...(tools ? { tools } : {}),
+            ...(args.temperature !== undefined ? { temperature: args.temperature } : {}),
+            ...(args.maxTokens ? { max_tokens: args.maxTokens } : {}),
+        };
+        let sdkStream = null;
+        async function* generator() {
+            sdkStream = await self.client.chat.completions.create(streamParams);
+            if (!sdkStream)
+                return;
+            const pendingTools = new Map();
+            let lastUsage = null;
+            for await (const chunk of sdkStream) {
+                if (args.signal?.aborted) {
+                    break;
+                }
+                if (chunk.usage) {
+                    lastUsage = chunk.usage;
+                }
+                const choice = chunk.choices?.[0];
+                if (!choice)
+                    continue;
+                if (choice.delta?.content) {
+                    yield { type: 'text-delta', text: choice.delta.content };
+                }
+                if (choice.delta?.tool_calls) {
+                    for (const tc of choice.delta.tool_calls) {
+                        const idx = tc.index ?? 0;
+                        if (tc.id) {
+                            pendingTools.set(idx, { id: tc.id, name: tc.function?.name ?? '', inputStr: '' });
+                        }
+                        const pending = pendingTools.get(idx);
+                        if (pending) {
+                            pending.inputStr += tc.function?.arguments ?? '';
+                        }
+                    }
+                }
+            }
+            for (const [, pending] of pendingTools) {
+                let input = {};
+                try {
+                    if (pending.inputStr) {
+                        input = JSON.parse(pending.inputStr);
+                    }
+                }
+                catch { /* empty input on parse failure */ }
+                yield { type: 'tool-call', id: pending.id, name: pending.name, input };
+            }
+            if (lastUsage) {
+                yield {
+                    type: 'usage',
+                    inputTokens: lastUsage.prompt_tokens ?? 0,
+                    outputTokens: lastUsage.completion_tokens ?? 0,
+                };
+            }
+            yield { type: 'stop', reason: args.signal?.aborted ? 'aborted' : 'stop' };
+        }
+        return {
+            iterable: generator(),
+            cancel() {
+                // The signal.aborted check inside the generator loop will break on next iteration
+            },
+        };
+    }
+    async check(prompt, args) {
+        const model = args?.model || this.defaultModel;
+        const messages = [
+            { role: 'user', content: prompt },
+        ];
+        if (args?.system) {
+            messages.unshift({ role: 'system', content: args.system });
+        }
+        const response = await this.client.chat.completions.create({
+            model,
+            messages,
+            max_tokens: 256,
+            temperature: 0,
+        });
+        return response.choices[0]?.message?.content?.trim() ?? '';
+    }
+    async complete(args) {
+        const model = args.model || this.defaultModel;
+        const messages = this.mapMessages(args.messages);
+        const tools = args.tools ? this.mapTools(args.tools) : undefined;
+        const allMessages = args.system
+            ? [{ role: 'system', content: args.system }, ...messages]
+            : messages;
+        const response = await this.client.chat.completions.create({
+            model,
+            messages: allMessages,
+            stream: false,
+            ...(tools ? { tools } : {}),
+            ...(args.temperature !== undefined ? { temperature: args.temperature } : {}),
+            ...(args.maxTokens ? { max_tokens: args.maxTokens } : {}),
+        });
+        const choice = response.choices[0];
+        const text = choice?.message?.content ?? '';
+        const toolCalls = choice?.message?.tool_calls?.map(tc => {
+            let input = {};
+            try {
+                if (tc.function?.arguments) {
+                    input = JSON.parse(tc.function.arguments);
+                }
+            }
+            catch { /* empty input on parse failure */ }
+            return {
+                id: tc.id ?? '',
+                name: tc.function?.name ?? '',
+                input,
+            };
+        }).filter(tc => tc.id) ?? [];
+        const usage = response.usage
+            ? { inputTokens: response.usage.prompt_tokens, outputTokens: response.usage.completion_tokens }
+            : undefined;
+        return {
+            text: text.trim(),
+            stopReason: (choice?.finish_reason === 'stop' || choice?.finish_reason === 'function_call')
+                ? 'stop'
+                : choice?.finish_reason === 'tool_calls'
+                    ? 'tool_use'
+                    : choice?.finish_reason === 'length'
+                        ? 'length'
+                        : choice?.finish_reason === 'content_filter'
+                            ? 'content_filter'
+                            : 'stop',
+            usage,
+            toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
+        };
+    }
+    mapMessages(messages) {
+        return messages.map((m) => {
+            if (m.role === 'tool') {
+                return {
+                    role: 'tool',
+                    content: m.content,
+                    tool_call_id: m.toolUseId,
+                };
+            }
+            if (typeof m.content === 'string') {
+                return { role: m.role, content: m.content };
+            }
+            if (Array.isArray(m.content) && m.role === 'user') {
+                const textParts = m.content
+                    .filter((c) => c.type === 'text')
+                    .map(c => c.text);
+                return { role: 'user', content: textParts.join('\n') };
+            }
+            if (Array.isArray(m.content) && m.role === 'assistant') {
+                const textParts = m.content
+                    .filter((c) => c.type === 'text')
+                    .map(c => c.text);
+                const toolCalls = m.content
+                    .filter((c) => c.type === 'tool-use')
+                    .map(c => ({
+                    id: c.id,
+                    type: 'function',
+                    function: {
+                        name: c.name,
+                        arguments: JSON.stringify(c.input),
+                    },
+                }));
+                return {
+                    role: 'assistant',
+                    content: textParts.join('\n') || null,
+                    ...(toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),
+                };
+            }
+            return { role: m.role, content: JSON.stringify(m.content) };
+        });
+    }
+    mapTools(tools) {
+        return tools.map((t) => ({
+            type: 'function',
+            function: {
+                name: t.name,
+                description: t.description,
+                parameters: t.inputSchema,
+            },
+        }));
+    }
+}
+//# sourceMappingURL=ollama.js.map

package/dist/src/ollama.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/ollama.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAa5B,MAAM,eAAe,GAAG;IACtB,UAAU;IACV,SAAS;IACT,MAAM;IACN,SAAS;IACT,aAAa;IACb,UAAU;IACV,QAAQ;IACR,WAAW;IACX,QAAQ;IACR,SAAS;CACV,CAAC;AAEF,MAAM,OAAO,cAAc;IAChB,IAAI,GAAG,QAAQ,CAAC;IAChB,MAAM,CAAW;IACjB,YAAY,GAAG,QAAQ,CAAC;IAEzB,MAAM,CAAS;IACf,QAAQ,CAAS;IACjB,UAAU,GAAoB,IAAI,CAAC;IAE3C,YAAY,MAAe,EAAE,OAAgB;QAC3C,MAAM,WAAW,GAAG,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;YAC/C,CAAC,CAAC,MAAM;YACR,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,OAAO,IAAI,2BAA2B,CAAC;QACvD,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM,EAAE,WAAW,IAAI,eAAe;YACtC,OAAO,EAAE,IAAI,CAAC,QAAQ;SACvB,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,CAAC,GAAG,eAAe,CAAC,CAAC;IACrC,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO,IAAI,CAAC,UAAU,CAAC;QAC5C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YACxD,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,WAAW,EAAE;gBAC9C,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;aAClC,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAA0C,CAAC;YACvE,IAAI,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;gBACxB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBAC/C,OAAO,IAAI,CAAC,UAAU,CAAC;YACzB,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,6BAA6B;QAC/B,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,CAAC,GAAG,eAAe,CAAC,CAAC;QACvC,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,MAAM,CAAC,IAAwB;QAC7B,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QAE9C,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEjE,MAAM,WAAW,GAAwC,IAAI,CAAC,MAAM;YAClE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,GAAG,QAAQ,CAAC;YACzD,CAAC,CAAC,QAAQ,CAAC;QAEb,uEAAuE;QACvE,MAAM,YAAY,GAAsC;YACtD,KAAK;YACL,QAAQ,EAAE,WAAW;YACrB,MAAM,EAAE,IAAI;YACZ,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3B,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC1D,CAAC;QAEF,IAAI,SAAS,GAAqD,IAAI,CAAC;QAEvE,KAAK,SAAS,CAAC,CAAC,SAAS;YACvB,SAAS,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,CAA8C,CAAC;YAEjH,IAAI,CAAC,SAAS;gBAAE,OAAO;YAEvB,MAAM,YAAY,GAAG,IAAI,GAAG,EAA0D,CAAC;YACvF,IAAI,SAAS,GAAkC,IAAI,CAAC;YAEpD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;gBACpC,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;oBACzB,MAAM;gBACR,CAAC;gBAED,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBAChB,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC;gBAC1B,CAAC;gBAED,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;gBAClC,IAAI,CAAC,MAAM;oBAAE,SAAS;gBAEtB,IAAI,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC;oBAC1B,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;gBAC3D,CAAC;gBAED,IAAI,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC;oBAC7B,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;wBACzC,MAAM,GAAG,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,CAAC;wBAC1B,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;4BACV,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;wBACpF,CAAC;wBACD,MAAM,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;wBACtC,IAAI,OAAO,EAAE,CAAC;4BACZ,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE,CAAC;wBACnD,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,KAAK,MAAM,CAAC,EAAE,OAAO,CAAC,IAAI,YAAY,EAAE,CAAC;gBACvC,IAAI,KAAK,GAA4B,EAAE,CAAC;gBACxC,IAAI,CAAC;oBACH,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;wBACrB,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;oBACvC,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC,CAAC,kCAAkC,CAAC,CAAC;gBAC9C,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC;YACzE,CAAC;YAED,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM;oBACJ,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,SAAS,CAAC,aAAa,IAAI,CAAC;oBACzC,YAAY,EAAE,SAAS,CAAC,iBAAiB,IAAI,CAAC;iBAC/C,CAAC;YACJ,CAAC;YAED,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;QAC5E,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,SAAS,EAAE;YACrB,MAAM;gBACJ,kFAAkF;YACpF,CAAC;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAc,EAAE,IAI3B;QACC,MAAM,KAAK,GAAG,IAAI,EAAE,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QAC/C,MAAM,QAAQ,GAAwC;YACpD,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;SAClC,CAAC;QACF,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC;YACjB,QAAQ,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACtE,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACzD,KAAK;YACL,QAAQ;YACR,UAAU,EAAE,GAAG;YACf,WAAW,EAAE,CAAC;SACf,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAA0B;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QAE9C,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEjE,MAAM,WAAW,GAAwC,IAAI,CAAC,MAAM;YAClE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,GAAG,QAAQ,CAAC;YACzD,CAAC,CAAC,QAAQ,CAAC;QAEb,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACzD,KAAK;YACL,QAAQ,EAAE,WAAW;YACrB,MAAM,EAAE,KAAK;YACb,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3B,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC1D,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,IAAI,GAAG,MAAM,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAE5C,MAAM,SAAS,GAAG,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,EAAE,CAAC,EAAE;YACtD,IAAI,KAAK,GAA4B,EAAE,CAAC;YACxC,IAAI,CAAC;gBACH,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;oBAC3B,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;YAAC,MAAM,CAAC,CAAC,kCAAkC,CAAC,CAAC;YAC9C,OAAO;gBACL,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE;gBACf,IAAI,EAAE,EAAE,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE;gBAC7B,KAAK;aACN,CAAC;QACJ,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QAE7B,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK;YAC1B,CAAC,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa,EAAE,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,iBAAiB,EAAE;YAC/F,CAAC,CAAC,SAAS,CAAC;QAEd,OAAO;YACL,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;YACjB,UAAU,EAAE,CAAC,MAAM,EAAE,aAAa,KAAK,MAAM,IAAI,MAAM,EAAE,aAAa,KAAK,eAAe,CAAC;gBACzF,CAAC,CAAC,MAAM;gBACR,CAAC,CAAC,MAAM,EAAE,aAAa,KAAK,YAAY;oBACxC,CAAC,CAAC,UAAU;oBACZ,CAAC,CAAC,MAAM,EAAE,aAAa,KAAK,QAAQ;wBACpC,CAAC,CAAC,QAAQ;wBACV,CAAC,CAAC,MAAM,EAAE,aAAa,KAAK,gBAAgB;4BAC5C,CAAC,CAAC,gBAAgB;4BAClB,CAAC,CAAC,MAAM;YACV,KAAK;YACL,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;SACxD,CAAC;IACJ,CAAC;IAEO,WAAW,CAAC,QAA2B;QAC7C,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACxB,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACtB,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,CAAC,CAAC,OAAiB;oBAC5B,YAAY,EAAG,CAA2B,CAAC,SAAS;iBACZ,CAAC;YAC7C,CAAC;YAED,IAAI,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAClC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAuC,CAAC;YACnF,CAAC;YAED,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAClD,MAAM,SAAS,GAAG,CAAC,CAAC,OAAO;qBACxB,MAAM,CAAC,CAAC,CAAC,EAAuC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;qBACrE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACpB,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACzD,CAAC;YAED,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACvD,MAAM,SAAS,GAAG,CAAC,CAAC,OAAO;qBACxB,MAAM,CAAC,CAAC,CAAC,EAAuC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;qBACrE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACpB,MAAM,SAAS,GAAG,CAAC,CAAC,OAAO;qBACxB,MAAM,CAAC,CAAC,CAAC,EAAuF,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC;qBACzH,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBACT,EAAE,EAAE,CAAC,CAAC,EAAE;oBACR,IAAI,EAAE,UAAmB;oBACzB,QAAQ,EAAE;wBACR,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC;qBACnC;iBACF,CAAC,CAAC,CAAC;gBAEN,OAAO;oBACL,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI;oBACrC,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACb,CAAC;YAClD,CAAC;YAED,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,QAAQ,CAAC,KAAuB;QACtC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvB,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,UAAU,EAAE,CAAC,CAAC,WAAsC;aACrD;SACF,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}

package/dist/src/openai.d.ts

@@ -0,0 +1,23 @@
+import type { Provider, ProviderEvent, ProviderStreamArgs, ProviderCompleteArgs, ProviderCompleteResult } from './provider.js';
+type CancelableIterable<T> = {
+    iterable: AsyncIterable<T>;
+    cancel(): void;
+};
+export declare class OpenAIProvider implements Provider {
+    readonly name = "openai";
+    readonly models: string[];
+    readonly defaultModel = "gpt-4o";
+    private client;
+    constructor(apiKey?: string, baseUrl?: string);
+    stream(args: ProviderStreamArgs): CancelableIterable<ProviderEvent>;
+    check(prompt: string, args?: {
+        model?: string;
+        system?: string;
+        signal?: AbortSignal;
+    }): Promise<string>;
+    complete(args: ProviderCompleteArgs): Promise<ProviderCompleteResult>;
+    private mapMessages;
+    private mapTools;
+}
+export {};
+//# sourceMappingURL=openai.d.ts.map

package/dist/src/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/openai.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,QAAQ,EACR,aAAa,EACb,kBAAkB,EAClB,oBAAoB,EACpB,sBAAsB,EAKvB,MAAM,eAAe,CAAC;AAEvB,KAAK,kBAAkB,CAAC,CAAC,IAAI;IAAE,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC;IAAC,MAAM,IAAI,IAAI,CAAA;CAAE,CAAC;AAmB5E,qBAAa,cAAe,YAAW,QAAQ;IAC7C,QAAQ,CAAC,IAAI,YAAY;IACzB,QAAQ,CAAC,MAAM,WAWb;IACF,QAAQ,CAAC,YAAY,YAAY;IAEjC,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM;IAe9C,MAAM,CAAC,IAAI,EAAE,kBAAkB,GAAG,kBAAkB,CAAC,aAAa,CAAC;IA2G5D,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE;QACjC,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,WAAW,CAAC;KACtB,GAAG,OAAO,CAAC,MAAM,CAAC;IAiBb,QAAQ,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAgE3E,OAAO,CAAC,WAAW;IA+CnB,OAAO,CAAC,QAAQ;CAUjB"}