archbyte 0.2.9 → 0.3.1

package/dist/agents/providers/ollama.js

@@ -1,162 +0,0 @@
-export class OllamaProvider {
-    name = "ollama";
-    baseUrl;
-    constructor(baseUrl = "http://localhost:11434") {
-        this.baseUrl = baseUrl.replace(/\/$/, "");
-    }
-    async chat(params) {
-        const messages = [
-            { role: "system", content: params.system },
-            ...params.messages.map((m) => this.toOllamaMessage(m)),
-        ];
-        const body = {
-            model: params.model,
-            messages,
-            stream: false,
-        };
-        if (params.tools?.length) {
-            body.tools = params.tools.map((t) => ({
-                type: "function",
-                function: {
-                    name: t.name,
-                    description: t.description,
-                    parameters: t.input_schema,
-                },
-            }));
-        }
-        const response = await fetch(`${this.baseUrl}/api/chat`, {
-            method: "POST",
-            headers: { "Content-Type": "application/json" },
-            body: JSON.stringify(body),
-        });
-        if (!response.ok) {
-            const text = await response.text();
-            throw new Error(`Ollama error ${response.status}: ${text}`);
-        }
-        const data = (await response.json());
-        const content = [];
-        if (data.message.content) {
-            content.push({ type: "text", text: data.message.content });
-        }
-        let hasToolCalls = false;
-        if (data.message.tool_calls) {
-            hasToolCalls = true;
-            for (const tc of data.message.tool_calls) {
-                content.push({
-                    type: "tool_use",
-                    id: tc.id ?? `call_${Date.now()}`,
-                    name: tc.function.name,
-                    input: JSON.parse(tc.function.arguments),
-                });
-            }
-        }
-        return {
-            content,
-            stopReason: hasToolCalls ? "tool_use" : "end_turn",
-            usage: {
-                inputTokens: data.prompt_eval_count ?? 0,
-                outputTokens: data.eval_count ?? 0,
-            },
-        };
-    }
-    async *stream(params) {
-        const messages = [
-            { role: "system", content: params.system },
-            ...params.messages.map((m) => this.toOllamaMessage(m)),
-        ];
-        const body = {
-            model: params.model,
-            messages,
-            stream: true,
-        };
-        if (params.tools?.length) {
-            body.tools = params.tools.map((t) => ({
-                type: "function",
-                function: {
-                    name: t.name,
-                    description: t.description,
-                    parameters: t.input_schema,
-                },
-            }));
-        }
-        const response = await fetch(`${this.baseUrl}/api/chat`, {
-            method: "POST",
-            headers: { "Content-Type": "application/json" },
-            body: JSON.stringify(body),
-        });
-        if (!response.ok) {
-            const text = await response.text();
-            throw new Error(`Ollama error ${response.status}: ${text}`);
-        }
-        const reader = response.body?.getReader();
-        if (!reader)
-            throw new Error("No response body");
-        const decoder = new TextDecoder();
-        let buffer = "";
-        while (true) {
-            const { done, value } = await reader.read();
-            if (done)
-                break;
-            buffer += decoder.decode(value, { stream: true });
-            const lines = buffer.split("\n");
-            buffer = lines.pop() ?? "";
-            for (const line of lines) {
-                if (!line.trim())
-                    continue;
-                const data = JSON.parse(line);
-                if (data.message?.content) {
-                    yield { type: "text", text: data.message.content };
-                }
-                if (data.message?.tool_calls) {
-                    for (const tc of data.message.tool_calls) {
-                        yield {
-                            type: "tool_use_start",
-                            toolName: tc.function.name,
-                            toolId: tc.id ?? `call_${Date.now()}`,
-                        };
-                        yield { type: "tool_use_input", input: tc.function.arguments };
-                    }
-                }
-                if (data.done) {
-                    yield { type: "done" };
-                }
-            }
-        }
-    }
-    toOllamaMessage(m) {
-        if (typeof m.content === "string") {
-            return { role: m.role, content: m.content };
-        }
-        const toolResults = m.content.filter((b) => b.type === "tool_result");
-        if (toolResults.length > 0) {
-            return {
-                role: "tool",
-                content: toolResults[0].content ?? "",
-                tool_call_id: toolResults[0].tool_use_id,
-            };
-        }
-        const toolUses = m.content.filter((b) => b.type === "tool_use");
-        if (toolUses.length > 0) {
-            const text = m.content
-                .filter((b) => b.type === "text")
-                .map((b) => b.text ?? "")
-                .join("");
-            return {
-                role: "assistant",
-                content: text,
-                tool_calls: toolUses.map((b) => ({
-                    id: b.id ?? "",
-                    type: "function",
-                    function: {
-                        name: b.name ?? "",
-                        arguments: JSON.stringify(b.input ?? {}),
-                    },
-                })),
-            };
-        }
-        return {
-            role: m.role,
-            content: m.content.map((b) => b.text ?? "").join(""),
-        };
-    }
-}