atom-lite 0.0.0-stage → 0.1.0

package/dist/providers/gemini.js

@@ -0,0 +1,228 @@
+import { parseArgs } from "./tool-calls.js";
+function toGeminiContents(messages) {
+    const systemParts = [];
+    const contents = [];
+    for (const m of messages) {
+        if (m.role === "system") {
+            systemParts.push({ text: m.content });
+            continue;
+        }
+        if (m.role === "user") {
+            contents.push({
+                role: "user",
+                parts: m.content.map((p) => p.type === "text" ? { text: p.text } : p),
+            });
+            continue;
+        }
+        if (m.role === "assistant") {
+            const parts = [];
+            const text = m.content
+                ?.map((p) => (p.type === "text" ? p.text : ""))
+                .join("") ?? "";
+            if (text)
+                parts.push({ text });
+            if (m.toolCalls?.length) {
+                for (const tc of m.toolCalls) {
+                    parts.push({
+                        functionCall: { name: tc.name, args: tc.arguments ?? {} },
+                    });
+                }
+            }
+            // preserve provider state if present
+            const state = m.providerState;
+            contents.push({
+                role: "model",
+                parts,
+                ...(state ? { _state: state } : {}),
+            });
+            continue;
+        }
+        if (m.role === "tool") {
+            contents.push({
+                role: "function",
+                parts: [
+                    {
+                        functionResponse: {
+                            name: m.name,
+                            response: { result: m.content },
+                        },
+                    },
+                ],
+            });
+            continue;
+        }
+    }
+    const systemInstruction = systemParts.length
+        ? { parts: systemParts }
+        : undefined;
+    return { systemInstruction, contents };
+}
+export class GeminiAdapter {
+    id;
+    model;
+    apiKey;
+    baseUrl;
+    contextWindowOverride;
+    modelList;
+    constructor(config) {
+        this.id = config.id;
+        this.model = config.model;
+        this.apiKey = config.apiKey ?? null;
+        this.baseUrl = (config.baseUrl ?? "https://generativelanguage.googleapis.com").replace(/\/$/, "");
+        this.contextWindowOverride = config.contextWindow;
+    }
+    fetchModelList() {
+        if (!this.apiKey)
+            return Promise.resolve([]);
+        if (!this.modelList) {
+            this.modelList = (async () => {
+                const res = await fetch(`${this.baseUrl}/v1beta/models?key=${encodeURIComponent(this.apiKey)}`);
+                if (!res.ok)
+                    return [];
+                const json = (await res.json());
+                return json.models ?? [];
+            })().catch(() => []);
+        }
+        return this.modelList;
+    }
+    async listModels() {
+        const models = await this.fetchModelList();
+        if (!models.length)
+            this.modelList = undefined;
+        const ids = models
+            .map((m) => (m.name ?? "").replace(/^models\//, ""))
+            .filter(Boolean);
+        return ids.length ? ids : [this.model];
+    }
+    async getModelInfo() {
+        const models = await this.fetchModelList();
+        if (!models.length)
+            this.modelList = undefined;
+        const entry = models.find((m) => (m.name ?? "").replace(/^models\//, "") === this.model);
+        const limit = entry?.inputTokenLimit;
+        return {
+            id: this.model,
+            contextWindow: typeof limit === "number" && limit > 0
+                ? Math.floor(limit)
+                : this.contextWindowOverride,
+            supportsTools: true,
+            supportsStreaming: true,
+        };
+    }
+    async generate(request) {
+        if (!this.apiKey)
+            throw new Error("PROVIDER_AUTH_FAILED: missing API key");
+        const { systemInstruction, contents } = toGeminiContents(request.messages);
+        const url = `${this.baseUrl}/v1beta/models/${encodeURIComponent(this.model)}:generateContent?key=${encodeURIComponent(this.apiKey)}`;
+        const body = { contents };
+        if (systemInstruction)
+            body.systemInstruction = systemInstruction;
+        const effort = request.options?.effort;
+        if (effort) {
+            const budget = effort === "low" ? 1024 : effort === "high" ? 16384 : 4096;
+            body.generationConfig = { ...(body.generationConfig ?? {}), thinkingConfig: { thinkingBudget: budget } };
+        }
+        if (request.tools && request.tools.length) {
+            body.tools = [
+                {
+                    functionDeclarations: request.tools.map((t) => ({
+                        name: t.name,
+                        description: t.description,
+                        parameters: t.parameters,
+                    })),
+                },
+            ];
+        }
+        let res;
+        let lastErr;
+        for (let attempt = 0; attempt < 3; attempt++) {
+            try {
+                res = await fetch(url, {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify(body),
+                    signal: request.signal,
+                });
+            }
+            catch (err) {
+                lastErr = new Error(`PROVIDER_CONNECTION_FAILED: ${err instanceof Error ? err.message : String(err)}`);
+                if (attempt < 2) {
+                    await new Promise((r) => setTimeout(r, 200 * (attempt + 1)));
+                    continue;
+                }
+                throw lastErr;
+            }
+            if (res.ok)
+                break;
+            const text = await res.text().catch(() => "");
+            if (res.status === 401 || res.status === 403)
+                throw new Error(`PROVIDER_AUTH_FAILED: ${text}`);
+            if (res.status === 429) {
+                lastErr = new Error(`PROVIDER_RATE_LIMITED: ${text}`);
+                if (attempt < 2) {
+                    await new Promise((r) => setTimeout(r, 500 * (attempt + 1)));
+                    continue;
+                }
+                throw lastErr;
+            }
+            throw new Error(`PROVIDER_INVALID_RESPONSE: ${res.status} ${text}`);
+        }
+        if (!res || !res.ok)
+            throw lastErr ?? new Error("PROVIDER_INVALID_RESPONSE");
+        const json = (await res.json());
+        const candidate = json.candidates?.[0];
+        const parts = candidate?.content?.parts ?? [];
+        const thinkingParts = parts
+            .filter((p) => p.thought && typeof p.text === "string")
+            .map((p) => p.text);
+        const textParts = parts
+            .filter((p) => !p.thought && typeof p.text === "string")
+            .map((p) => p.text);
+        const text = textParts.join("");
+        const thinking = thinkingParts.join("");
+        const toolCalls = parts
+            .filter((p) => p.functionCall)
+            .map((p, i) => ({
+            id: `gemini_${i}`,
+            name: p.functionCall.name,
+            arguments: parseArgs(JSON.stringify(p.functionCall.args ?? {})),
+        }));
+        const usage = json.usageMetadata
+            ? {
+                input: json.usageMetadata.promptTokenCount,
+                // Thinking tokens count toward the next request's context, so fold
+                // them into generated output.
+                output: (json.usageMetadata.candidatesTokenCount ?? 0) +
+                    (json.usageMetadata.thoughtsTokenCount ?? 0),
+                total: json.usageMetadata.totalTokenCount,
+            }
+            : undefined;
+        // safety block handling
+        if (candidate?.finishReason === "SAFETY") {
+            throw new Error("PROVIDER_INVALID_RESPONSE: safety block");
+        }
+        return {
+            message: {
+                role: "assistant",
+                content: text ? [{ type: "text", text }] : undefined,
+                toolCalls: toolCalls.length ? toolCalls : undefined,
+                providerState: candidate?.content,
+            },
+            toolCalls,
+            finishReason: toolCalls.length ? "tool_calls" : "stop",
+            usage,
+            metadata: thinking ? { thinking } : undefined,
+        };
+    }
+    async *stream(request) {
+        const resp = await this.generate(request);
+        const thinking = resp.metadata?.thinking;
+        if (thinking)
+            yield { type: "thinking", text: thinking };
+        if (resp.message.content?.[0]?.type === "text") {
+            yield { type: "text", text: resp.message.content[0].text };
+        }
+        yield { type: "done", response: resp };
+    }
+}
+//# sourceMappingURL=gemini.js.map

package/dist/providers/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/providers/gemini.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAW5C,SAAS,gBAAgB,CAAC,QAAmB;IAI3C,MAAM,WAAW,GAAU,EAAE,CAAC;IAC9B,MAAM,QAAQ,GAAU,EAAE,CAAC;IAC3B,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACxB,WAAW,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YACtC,SAAS;QACX,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtB,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAC9B,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CACzC;aACF,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAU,EAAE,CAAC;YACxB,MAAM,IAAI,GACR,CAAC,CAAC,OAAO;gBACP,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;iBACnD,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACpB,IAAI,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;YAC/B,IAAI,CAAC,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;gBACxB,KAAK,MAAM,EAAE,IAAI,CAAC,CAAC,SAAS,EAAE,CAAC;oBAC7B,KAAK,CAAC,IAAI,CAAC;wBACT,YAAY,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,SAAS,IAAI,EAAE,EAAE;qBAC1D,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YACD,qCAAqC;YACrC,MAAM,KAAK,GAAI,CAAS,CAAC,aAAa,CAAC;YACvC,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,OAAO;gBACb,KAAK;gBACL,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACpC,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QACD,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACtB,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE;oBACL;wBACE,gBAAgB,EAAE;4BAChB,IAAI,EAAE,CAAC,CAAC,IAAI;4BACZ,QAAQ,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;yBAChC;qBACF;iBACF;aACF,CAAC,CAAC;YACH,SAAS;QACX,CAAC;IACH,CAAC;IACD,MAAM,iBAAiB,GAAG,WAAW,CAAC,MAAM;QAC1C,CAAC,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE;QACxB,CAAC,CAAC,SAAS,CAAC;IACd,OAAO,EAAE,iBAAiB,EAAE,QAAQ,EAAE,CAAC;AACzC,CAAC;AAED,MAAM,OAAO,aAAa;IACf,EAAE,CAAS;IACb,KAAK,CAAS;IACb,MAAM,CAAgB;IACtB,OAAO,CAAS;IAChB,qBAAqB,CAAqB;IAC1C,SAAS,CAAkB;IAEnC,YAAY,MAAoB;QAC9B,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC;QACpC,IAAI,CAAC,OAAO,GAAG,CACb,MAAM,CAAC,OAAO,IAAI,2CAA2C,CAC9D,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACrB,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,aAAa,CAAC;IACpD,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7C,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,CAAC,KAAK,IAAI,EAAE;gBAC3B,MAAM,GAAG,GAAG,MAAM,KAAK,CACrB,GAAG,IAAI,CAAC,OAAO,sBAAsB,kBAAkB,CAAC,IAAI,CAAC,MAAO,CAAC,EAAE,CACxE,CAAC;gBACF,IAAI,CAAC,GAAG,CAAC,EAAE;oBAAE,OAAO,EAAE,CAAC;gBACvB,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAQ,CAAC;gBACvC,OAAO,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;YAC3B,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QACvB,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC/C,MAAM,GAAG,GAAG,MAAM;aACf,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;aACxD,MAAM,CAAC,OAAO,CAAC,CAAC;QACnB,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC/C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CACvB,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,CACnE,CAAC;QACF,MAAM,KAAK,GAAG,KAAK,EAAE,eAAe,CAAC;QACrC,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,KAAK;YACd,aAAa,EACX,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,GAAG,CAAC;gBACpC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;gBACnB,CAAC,CAAC,IAAI,CAAC,qBAAqB;YAChC,aAAa,EAAE,IAAI;YACnB,iBAAiB,EAAE,IAAI;SACxB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAqB;QAClC,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3E,MAAM,EAAE,iBAAiB,EAAE,QAAQ,EAAE,GAAG,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC3E,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,kBAAkB,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,wBAAwB,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QACrI,MAAM,IAAI,GAAQ,EAAE,QAAQ,EAAE,CAAC;QAC/B,IAAI,iBAAiB;YAAE,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAClE,MAAM,MAAM,GAAI,OAAO,CAAC,OAAe,EAAE,MAAM,CAAC;QAChD,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,GAAG,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1E,IAAI,CAAC,gBAAgB,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,gBAAgB,IAAI,EAAE,CAAC,EAAE,cAAc,EAAE,EAAE,cAAc,EAAE,MAAM,EAAE,EAAE,CAAC;QAC3G,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YAC1C,IAAI,CAAC,KAAK,GAAG;gBACX;oBACE,oBAAoB,EAAE,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC9C,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;wBAC1B,UAAU,EAAE,CAAC,CAAC,UAAU;qBACzB,CAAC,CAAC;iBACJ;aACF,CAAC;QACJ,CAAC;QACD,IAAI,GAAyB,CAAC;QAC9B,IAAI,OAA0B,CAAC;QAC/B,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;oBACrB,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;oBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;oBAC1B,MAAM,EAAE,OAAO,CAAC,MAAM;iBACvB,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,GAAG,IAAI,KAAK,CACjB,+BAA+B,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAClF,CAAC;gBACF,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAChB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7D,SAAS;gBACX,CAAC;gBACD,MAAM,OAAO,CAAC;YAChB,CAAC;YACD,IAAI,GAAG,CAAC,EAAE;gBAAE,MAAM;YAClB,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9C,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG;gBAC1C,MAAM,IAAI,KAAK,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YACnD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,OAAO,GAAG,IAAI,KAAK,CAAC,0BAA0B,IAAI,EAAE,CAAC,CAAC;gBACtD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBAChB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC7D,SAAS;gBACX,CAAC;gBACD,MAAM,OAAO,CAAC;YAChB,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,8BAA8B,GAAG,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE;YACjB,MAAM,OAAO,IAAI,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC1D,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAQ,CAAC;QACvC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,KAAK,GAAG,SAAS,EAAE,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;QAC9C,MAAM,aAAa,GAAG,KAAK;aACxB,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC3D,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3B,MAAM,SAAS,GAAG,KAAK;aACpB,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC5D,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3B,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChC,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxC,MAAM,SAAS,GAAe,KAAK;aAChC,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC;aAClC,GAAG,CAAC,CAAC,CAAM,EAAE,CAAS,EAAE,EAAE,CAAC,CAAC;YAC3B,EAAE,EAAE,UAAU,CAAC,EAAE;YACjB,IAAI,EAAE,CAAC,CAAC,YAAY,CAAC,IAAI;YACzB,SAAS,EAAE,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;SAChE,CAAC,CAAC,CAAC;QACN,MAAM,KAAK,GAA2B,IAAI,CAAC,aAAa;YACtD,CAAC,CAAC;gBACE,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,gBAAgB;gBAC1C,mEAAmE;gBACnE,8BAA8B;gBAC9B,MAAM,EACJ,CAAC,IAAI,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC,CAAC;oBAC9C,CAAC,IAAI,CAAC,aAAa,CAAC,kBAAkB,IAAI,CAAC,CAAC;gBAC9C,KAAK,EAAE,IAAI,CAAC,aAAa,CAAC,eAAe;aAC1C;YACH,CAAC,CAAC,SAAS,CAAC;QACd,wBAAwB;QACxB,IAAI,SAAS,EAAE,YAAY,KAAK,QAAQ,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO;YACL,OAAO,EAAE;gBACP,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;gBACpD,SAAS,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACnD,aAAa,EAAE,SAAS,EAAE,OAAO;aAClC;YACD,SAAS;YACT,YAAY,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM;YACtD,KAAK;YACL,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS;SAC9C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAqB;QACjC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAI,IAAI,CAAC,QAAgB,EAAE,QAAQ,CAAC;QAClD,IAAI,QAAQ;YAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QACzD,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YAC/C,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAS,CAAC,IAAI,EAAE,CAAC;QACtE,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IACzC,CAAC;CACF"}

package/dist/providers/index.d.ts

@@ -0,0 +1,5 @@
+export { FakeProvider } from "./fake.js";
+export { InMemoryProviderRegistry } from "./registry.js";
+export { OllamaAdapter, OllamaProvider } from "./ollama.js";
+export type { OllamaConfig } from "./ollama.js";
+export type { ProviderAdapter, ProviderRegistry } from "./types.js";

package/dist/providers/index.js

@@ -0,0 +1,4 @@
+export { FakeProvider } from "./fake.js";
+export { InMemoryProviderRegistry } from "./registry.js";
+export { OllamaAdapter, OllamaProvider } from "./ollama.js";
+//# sourceMappingURL=index.js.map

package/dist/providers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,wBAAwB,EAAE,MAAM,eAAe,CAAC;AACzD,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC"}

package/dist/providers/ollama.d.ts

@@ -0,0 +1,64 @@
+import type { ModelRequest, ModelResponse, ModelStreamEvent, ModelInfo } from "../types.js";
+import type { ProviderAdapter } from "./types.js";
+export interface OllamaConfig {
+    id?: string;
+    baseUrl?: string;
+    model: string;
+    apiKey?: string | null;
+    headers?: Record<string, string>;
+    keepAlive?: string | number;
+    think?: boolean | string;
+    options?: Record<string, unknown>;
+    contextWindow?: number;
+    supportsTools?: boolean;
+    supportsStreaming?: boolean;
+}
+export declare class OllamaAdapter implements ProviderAdapter {
+    readonly id: string;
+    private baseUrl;
+    model: string;
+    private headers;
+    private keepAlive;
+    private think;
+    private options;
+    private contextWindow;
+    private supportsToolsFlag;
+    private supportsStreamingFlag;
+    constructor(config: OllamaConfig);
+    get capabilities(): {
+        toolCalling: boolean;
+        streaming: boolean;
+        contextWindow: number | undefined;
+    };
+    get supportsTools(): boolean;
+    get supportsStreaming(): boolean;
+    private buildHeaders;
+    private resolveModel;
+    private toOllamaMessages;
+    private buildBody;
+    private parseToolCalls;
+    listModels(): Promise<string[]>;
+    generate(request: ModelRequest): Promise<ModelResponse>;
+    stream(request: ModelRequest): AsyncIterable<ModelStreamEvent>;
+    getModelInfo(): Promise<ModelInfo | undefined>;
+    doctor(): Promise<{
+        ok: boolean;
+        connectivity: boolean;
+        model: string;
+        supportsTools: boolean;
+        contextWindow: number | undefined;
+        error?: string;
+    }>;
+    healthCheck(): Promise<{
+        ok: boolean;
+        connectivity: boolean;
+        model: string;
+        supportsTools: boolean;
+        contextWindow: number | undefined;
+        error?: string;
+    }>;
+    checkConnectivity(): Promise<boolean>;
+    verifyToolCalling(): Promise<boolean>;
+}
+export declare const OllamaProvider: typeof OllamaAdapter;
+export type OllamaProvider = OllamaAdapter;