@jx-grxf/patchpilot 0.2.0

package/dist/core/json.js

@@ -0,0 +1,95 @@
+import { z } from "zod";
+const toolNameSchema = z.enum([
+    "list_files",
+    "read_file",
+    "search_text",
+    "inspect_document",
+    "write_file",
+    "run_shell"
+]);
+const toolCallSchema = z.object({
+    name: toolNameSchema,
+    arguments: z.record(z.string(), z.unknown()).default({})
+});
+const agentResponseSchema = z.discriminatedUnion("action", [
+    z.object({
+        action: z.literal("tools"),
+        message: z.string().default(""),
+        tool_calls: z.array(toolCallSchema).min(1)
+    }),
+    z.object({
+        action: z.literal("final"),
+        message: z.string()
+    })
+]);
+export function parseAgentResponse(rawContent) {
+    const parsed = normalizeModelJson(JSON.parse(extractJson(rawContent)));
+    return agentResponseSchema.parse(parsed);
+}
+export function formatParseError(error) {
+    if (error instanceof z.ZodError) {
+        const firstIssue = error.issues[0];
+        if (!firstIssue) {
+            return "response did not match the PatchPilot protocol.";
+        }
+        const location = firstIssue.path.length > 0 ? ` at ${firstIssue.path.join(".")}` : "";
+        return `${firstIssue.message}${location}`;
+    }
+    return error instanceof Error ? error.message : String(error);
+}
+function extractJson(rawContent) {
+    const trimmedContent = rawContent.trim();
+    if (trimmedContent.startsWith("{") && trimmedContent.endsWith("}")) {
+        return trimmedContent;
+    }
+    const fencedMatch = trimmedContent.match(/```(?:json)?\s*([\s\S]*?)\s*```/i);
+    if (fencedMatch?.[1]) {
+        return fencedMatch[1].trim();
+    }
+    const firstBrace = trimmedContent.indexOf("{");
+    const lastBrace = trimmedContent.lastIndexOf("}");
+    if (firstBrace >= 0 && lastBrace > firstBrace) {
+        return trimmedContent.slice(firstBrace, lastBrace + 1);
+    }
+    throw new Error("Model response did not contain a JSON object.");
+}
+function normalizeModelJson(parsed) {
+    if (Array.isArray(parsed)) {
+        if (parsed.length === 1) {
+            return normalizeModelJson(parsed[0]);
+        }
+        if (parsed.every(isToolCallLike)) {
+            return {
+                action: "tools",
+                message: "Requesting tools.",
+                tool_calls: parsed
+            };
+        }
+    }
+    if (isRecord(parsed)) {
+        if (!("action" in parsed) && "tool_calls" in parsed) {
+            return {
+                action: "tools",
+                message: readString(parsed.message, "Requesting tools."),
+                tool_calls: parsed.tool_calls
+            };
+        }
+        if (!("action" in parsed) && "message" in parsed) {
+            return {
+                action: "final",
+                message: readString(parsed.message, "")
+            };
+        }
+    }
+    return parsed;
+}
+function isToolCallLike(value) {
+    return isRecord(value) && "name" in value && "arguments" in value;
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null;
+}
+function readString(value, fallback) {
+    return typeof value === "string" ? value : fallback;
+}
+//# sourceMappingURL=json.js.map

package/dist/core/json.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"json.js","sourceRoot":"","sources":["../../src/core/json.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,MAAM,cAAc,GAAG,CAAC,CAAC,IAAI,CAAC;IAC5B,YAAY;IACZ,WAAW;IACX,aAAa;IACb,kBAAkB;IAClB,YAAY;IACZ,WAAW;CACZ,CAAC,CAAC;AAEH,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9B,IAAI,EAAE,cAAc;IACpB,SAAS,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;CACzD,CAAC,CAAC;AAEH,MAAM,mBAAmB,GAAG,CAAC,CAAC,kBAAkB,CAAC,QAAQ,EAAE;IACzD,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QAC1B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/B,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;KAC3C,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QAC1B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;KACpB,CAAC;CACH,CAAC,CAAC;AAEH,MAAM,UAAU,kBAAkB,CAAC,UAAkB;IACnD,MAAM,MAAM,GAAG,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACvE,OAAO,mBAAmB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,IAAI,KAAK,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;QAChC,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,iDAAiD,CAAC;QAC3D,CAAC;QAED,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACtF,OAAO,GAAG,UAAU,CAAC,OAAO,GAAG,QAAQ,EAAE,CAAC;IAC5C,CAAC;IAED,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAChE,CAAC;AAED,SAAS,WAAW,CAAC,UAAkB;IACrC,MAAM,cAAc,GAAG,UAAU,CAAC,IAAI,EAAE,CAAC;IAEzC,IAAI,cAAc,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACnE,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,MAAM,WAAW,GAAG,cAAc,CAAC,KAAK,CAAC,kCAAkC,CAAC,CAAC;IAC7E,IAAI,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrB,OAAO,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC/B,CAAC;IAED,MAAM,UAAU,GAAG,cAAc,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,cAAc,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAClD,IAAI,UAAU,IAAI,CAAC,IAAI,SAAS,GAAG,UAAU,EAAE,CAAC;QAC9C,OAAO,cAAc,CAAC,KAAK,CAAC,UAAU,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;AACnE,CAAC;AAED,SAAS,kBAAkB,CAAC,MAAe;IACzC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC;QAED,IAAI,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,CAAC;YACjC,OAAO;gBACL,MAAM,EAAE,OAAO;gBACf,OAAO,EAAE,mBAAmB;gBAC5B,UAAU,EAAE,MAAM;aACnB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,IAAI,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QACrB,IAAI,CAAC,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,YAAY,IAAI,MAAM,EAAE,CAAC;YACpD,OAAO;gBACL,MAAM,EAAE,OAAO;gBACf,OAAO,EAAE,UAAU,CAAC,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC;gBACxD,UAAU,EAAE,MAAM,CAAC,UAAU;aAC9B,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,CAAC,QAAQ,IAAI,MAAM,CAAC,IAAI,SAAS,IAAI,MAAM,EAAE,CAAC;YACjD,OAAO;gBACL,MAAM,EAAE,OAAO;gBACf,OAAO,EAAE,UAAU,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,CAAC;aACxC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,cAAc,CAAC,KAAc;IACpC,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,MAAM,IAAI,KAAK,IAAI,WAAW,IAAI,KAAK,CAAC;AACpE,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACrD,CAAC;AAED,SAAS,UAAU,CAAC,KAAc,EAAE,QAAgB;IAClD,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC;AACtD,CAAC"}

package/dist/core/modelClient.d.ts

@@ -0,0 +1,8 @@
+import type { ModelClient, ModelProvider } from "./types.js";
+export declare function createModelClient(options: {
+    provider: ModelProvider;
+    ollamaUrl: string;
+    workspace?: string;
+}): ModelClient;
+export declare function readModelProvider(env?: NodeJS.ProcessEnv): ModelProvider;
+export declare function normalizeModelProvider(value: string): ModelProvider;

package/dist/core/modelClient.js

@@ -0,0 +1,42 @@
+import { CodexCliClient } from "./codex.js";
+import { GeminiClient } from "./gemini.js";
+import { NvidiaClient } from "./nvidia.js";
+import { OllamaClient } from "./ollama.js";
+import { OpenRouterClient } from "./openrouter.js";
+export function createModelClient(options) {
+    if (options.provider === "gemini") {
+        return new GeminiClient();
+    }
+    if (options.provider === "codex") {
+        return new CodexCliClient({
+            workspace: options.workspace ?? process.cwd()
+        });
+    }
+    if (options.provider === "openrouter") {
+        return new OpenRouterClient();
+    }
+    if (options.provider === "nvidia") {
+        return new NvidiaClient();
+    }
+    return new OllamaClient(options.ollamaUrl);
+}
+export function readModelProvider(env = process.env) {
+    return normalizeModelProvider(env.PATCHPILOT_PROVIDER ?? env.PATCHPILOT_MODEL_PROVIDER ?? "ollama");
+}
+export function normalizeModelProvider(value) {
+    const normalizedValue = value.trim().toLowerCase();
+    if (normalizedValue === "gemini" || normalizedValue === "google") {
+        return "gemini";
+    }
+    if (normalizedValue === "codex" || normalizedValue === "openai" || normalizedValue === "openai-codex") {
+        return "codex";
+    }
+    if (normalizedValue === "openrouter" || normalizedValue === "open-router") {
+        return "openrouter";
+    }
+    if (normalizedValue === "nvidia" || normalizedValue === "nim") {
+        return "nvidia";
+    }
+    return "ollama";
+}
+//# sourceMappingURL=modelClient.js.map

package/dist/core/modelClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelClient.js","sourceRoot":"","sources":["../../src/core/modelClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAGnD,MAAM,UAAU,iBAAiB,CAAC,OAIjC;IACC,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAClC,OAAO,IAAI,YAAY,EAAE,CAAC;IAC5B,CAAC;IAED,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QACjC,OAAO,IAAI,cAAc,CAAC;YACxB,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,GAAG,EAAE;SAC9C,CAAC,CAAC;IACL,CAAC;IAED,IAAI,OAAO,CAAC,QAAQ,KAAK,YAAY,EAAE,CAAC;QACtC,OAAO,IAAI,gBAAgB,EAAE,CAAC;IAChC,CAAC;IAED,IAAI,OAAO,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAClC,OAAO,IAAI,YAAY,EAAE,CAAC;IAC5B,CAAC;IAED,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,MAAyB,OAAO,CAAC,GAAG;IACpE,OAAO,sBAAsB,CAAC,GAAG,CAAC,mBAAmB,IAAI,GAAG,CAAC,yBAAyB,IAAI,QAAQ,CAAC,CAAC;AACtG,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,KAAa;IAClD,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IACnD,IAAI,eAAe,KAAK,QAAQ,IAAI,eAAe,KAAK,QAAQ,EAAE,CAAC;QACjE,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,IAAI,eAAe,KAAK,OAAO,IAAI,eAAe,KAAK,QAAQ,IAAI,eAAe,KAAK,cAAc,EAAE,CAAC;QACtG,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,IAAI,eAAe,KAAK,YAAY,IAAI,eAAe,KAAK,aAAa,EAAE,CAAC;QAC1E,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,IAAI,eAAe,KAAK,QAAQ,IAAI,eAAe,KAAK,KAAK,EAAE,CAAC;QAC9D,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/core/nvidia.d.ts

@@ -0,0 +1,19 @@
+import type { ModelChatOptions, ModelChatResult } from "./types.js";
+export declare const defaultNvidiaModel = "meta/llama-3.1-70b-instruct";
+export declare const defaultNvidiaBaseUrl = "https://integrate.api.nvidia.com/v1";
+type NvidiaRuntimeOptions = {
+    maxTokens: number;
+    temperature: number;
+};
+export declare class NvidiaClient {
+    private readonly apiKey;
+    private readonly baseUrl;
+    private readonly runtimeOptions;
+    constructor(apiKey?: string, baseUrl?: string, runtimeOptions?: NvidiaRuntimeOptions);
+    chat(options: ModelChatOptions): Promise<ModelChatResult>;
+    listModels(): Promise<string[]>;
+    private fetchNvidia;
+    private assertConfigured;
+}
+export declare function readNvidiaApiKey(env?: NodeJS.ProcessEnv): string;
+export {};

package/dist/core/nvidia.js

@@ -0,0 +1,160 @@
+import { attachTokenCost } from "./tokenAccounting.js";
+export const defaultNvidiaModel = "meta/llama-3.1-70b-instruct";
+export const defaultNvidiaBaseUrl = "https://integrate.api.nvidia.com/v1";
+export class NvidiaClient {
+    apiKey;
+    baseUrl;
+    runtimeOptions;
+    constructor(apiKey = readNvidiaApiKey(), baseUrl = process.env.PATCHPILOT_NVIDIA_BASE_URL ?? defaultNvidiaBaseUrl, runtimeOptions = readNvidiaRuntimeOptions()) {
+        this.apiKey = apiKey;
+        this.baseUrl = baseUrl.replace(/\/$/, "");
+        this.runtimeOptions = runtimeOptions;
+    }
+    async chat(options) {
+        this.assertConfigured();
+        const startedAt = Date.now();
+        const response = await this.fetchNvidia("/chat/completions", {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${this.apiKey}`
+            },
+            body: JSON.stringify({
+                model: options.model,
+                messages: options.messages,
+                max_tokens: this.runtimeOptions.maxTokens,
+                temperature: this.runtimeOptions.temperature,
+                response_format: options.formatJson ? agentResponseFormat : undefined
+            }),
+            signal: options.signal
+        });
+        const durationMs = Date.now() - startedAt;
+        const payload = (await readJsonSafely(response));
+        if (!response.ok || payload.error) {
+            const reason = payload.error?.message ? ` ${payload.error.message}` : "";
+            throw new Error(`NVIDIA chat failed for model "${options.model}": HTTP ${response.status}.${reason}`);
+        }
+        const content = payload.choices?.[0]?.message?.content?.trim() ?? "";
+        if (!content) {
+            throw new Error("NVIDIA returned an empty response.");
+        }
+        return {
+            content,
+            telemetry: toTelemetry(payload, durationMs, options.model)
+        };
+    }
+    async listModels() {
+        this.assertConfigured();
+        const response = await this.fetchNvidia("/models");
+        const payload = (await readJsonSafely(response));
+        if (!response.ok || payload.error) {
+            const reason = payload.error?.message ? ` ${payload.error.message}` : "";
+            throw new Error(`NVIDIA models failed with HTTP ${response.status}.${reason}`);
+        }
+        const models = payload.data?.map((model) => model.id?.trim()).filter((model) => Boolean(model)).sort() ?? [];
+        return models.length > 0 ? models : [defaultNvidiaModel];
+    }
+    async fetchNvidia(path, init) {
+        try {
+            return await fetch(`${this.baseUrl}${path}`, init);
+        }
+        catch (error) {
+            const suffix = error instanceof Error ? ` ${error.message}` : "";
+            throw new Error(`Cannot reach NVIDIA API at ${this.baseUrl}.${suffix}`);
+        }
+    }
+    assertConfigured() {
+        if (!this.apiKey) {
+            throw new Error("NVIDIA API key missing. Set NVIDIA_API_KEY in PatchPilot config, .env, or your shell.");
+        }
+    }
+}
+const agentResponseFormat = {
+    type: "json_schema",
+    json_schema: {
+        name: "patchpilot_agent_response",
+        strict: true,
+        schema: {
+            anyOf: [
+                {
+                    type: "object",
+                    additionalProperties: false,
+                    required: ["action", "message", "tool_calls"],
+                    properties: {
+                        action: { const: "tools" },
+                        message: { type: "string" },
+                        tool_calls: {
+                            type: "array",
+                            minItems: 1,
+                            items: {
+                                type: "object",
+                                additionalProperties: false,
+                                required: ["name", "arguments"],
+                                properties: {
+                                    name: {
+                                        enum: ["list_files", "read_file", "search_text", "inspect_document", "write_file", "run_shell"]
+                                    },
+                                    arguments: {
+                                        type: "object",
+                                        additionalProperties: true
+                                    }
+                                }
+                            }
+                        }
+                    }
+                },
+                {
+                    type: "object",
+                    additionalProperties: false,
+                    required: ["action", "message"],
+                    properties: {
+                        action: { const: "final" },
+                        message: { type: "string" }
+                    }
+                }
+            ]
+        }
+    }
+};
+export function readNvidiaApiKey(env = process.env) {
+    return env.NVIDIA_API_KEY?.trim() || env.PATCHPILOT_NVIDIA_API_KEY?.trim() || "";
+}
+function readNvidiaRuntimeOptions(env = process.env) {
+    return {
+        maxTokens: readPositiveInteger(env.PATCHPILOT_NUM_PREDICT, 1024),
+        temperature: readTemperature(env.PATCHPILOT_TEMPERATURE, 0.1)
+    };
+}
+function toTelemetry(payload, durationMs, model) {
+    const promptTokens = payload.usage?.prompt_tokens ?? 0;
+    const responseTokens = payload.usage?.completion_tokens ?? 0;
+    return attachTokenCost({
+        promptTokens,
+        cachedPromptTokens: 0,
+        cacheWriteTokens: 0,
+        responseTokens,
+        totalTokens: payload.usage?.total_tokens ?? promptTokens + responseTokens,
+        evalTokensPerSecond: responseTokens > 0 && durationMs > 0 ? responseTokens / (durationMs / 1000) : null,
+        promptDurationMs: 0,
+        responseDurationMs: durationMs,
+        totalDurationMs: durationMs,
+        tokenSource: "provider"
+    }, "nvidia", model);
+}
+async function readJsonSafely(response) {
+    try {
+        return await response.json();
+    }
+    catch {
+        return {};
+    }
+}
+function readPositiveInteger(value, fallback) {
+    const parsedValue = Number.parseInt(value ?? "", 10);
+    return Number.isFinite(parsedValue) && parsedValue > 0 ? parsedValue : fallback;
+}
+function readTemperature(value, fallback) {
+    const parsedValue = Number.parseFloat(value ?? "");
+    return Number.isFinite(parsedValue) && parsedValue >= 0 ? parsedValue : fallback;
+}
+//# sourceMappingURL=nvidia.js.map

package/dist/core/nvidia.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"nvidia.js","sourceRoot":"","sources":["../../src/core/nvidia.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,MAAM,CAAC,MAAM,kBAAkB,GAAG,6BAA6B,CAAC;AAChE,MAAM,CAAC,MAAM,oBAAoB,GAAG,qCAAqC,CAAC;AAgC1E,MAAM,OAAO,YAAY;IACN,MAAM,CAAS;IACf,OAAO,CAAS;IAChB,cAAc,CAAuB;IAEtD,YACE,MAAM,GAAG,gBAAgB,EAAE,EAC3B,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,0BAA0B,IAAI,oBAAoB,EACxE,cAAc,GAAG,wBAAwB,EAAE;QAE3C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAyB;QAClC,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,mBAAmB,EAAE;YAC3D,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;aACvC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,UAAU,EAAE,IAAI,CAAC,cAAc,CAAC,SAAS;gBACzC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,WAAW;gBAC5C,eAAe,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,SAAS;aACtE,CAAC;YACF,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAC1C,MAAM,OAAO,GAAG,CAAC,MAAM,cAAc,CAAC,QAAQ,CAAC,CAAuB,CAAC;QAEvE,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzE,MAAM,IAAI,KAAK,CAAC,iCAAiC,OAAO,CAAC,KAAK,WAAW,QAAQ,CAAC,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;QACxG,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACrE,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QAED,OAAO;YACL,OAAO;YACP,SAAS,EAAE,WAAW,CAAC,OAAO,EAAE,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC;SAC3D,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,OAAO,GAAG,CAAC,MAAM,cAAc,CAAC,QAAQ,CAAC,CAAyB,CAAC;QACzE,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClC,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzE,MAAM,IAAI,KAAK,CAAC,kCAAkC,QAAQ,CAAC,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;QACjF,CAAC;QAED,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC;QAC9H,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAC3D,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,IAAkB;QACxD,IAAI,CAAC;YACH,OAAO,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,GAAG,IAAI,EAAE,EAAE,IAAI,CAAC,CAAC;QACrD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CAAC,8BAA8B,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IAEO,gBAAgB;QACtB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,uFAAuF,CAAC,CAAC;QAC3G,CAAC;IACH,CAAC;CACF;AAED,MAAM,mBAAmB,GAAG;IAC1B,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE;QACX,IAAI,EAAE,2BAA2B;QACjC,MAAM,EAAE,IAAI;QACZ,MAAM,EAAE;YACN,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,QAAQ;oBACd,oBAAoB,EAAE,KAAK;oBAC3B,QAAQ,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,YAAY,CAAC;oBAC7C,UAAU,EAAE;wBACV,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;wBAC1B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;wBAC3B,UAAU,EAAE;4BACV,IAAI,EAAE,OAAO;4BACb,QAAQ,EAAE,CAAC;4BACX,KAAK,EAAE;gCACL,IAAI,EAAE,QAAQ;gCACd,oBAAoB,EAAE,KAAK;gCAC3B,QAAQ,EAAE,CAAC,MAAM,EAAE,WAAW,CAAC;gCAC/B,UAAU,EAAE;oCACV,IAAI,EAAE;wCACJ,IAAI,EAAE,CAAC,YAAY,EAAE,WAAW,EAAE,aAAa,EAAE,kBAAkB,EAAE,YAAY,EAAE,WAAW,CAAC;qCAChG;oCACD,SAAS,EAAE;wCACT,IAAI,EAAE,QAAQ;wCACd,oBAAoB,EAAE,IAAI;qCAC3B;iCACF;6BACF;yBACF;qBACF;iBACF;gBACD;oBACE,IAAI,EAAE,QAAQ;oBACd,oBAAoB,EAAE,KAAK;oBAC3B,QAAQ,EAAE,CAAC,QAAQ,EAAE,SAAS,CAAC;oBAC/B,UAAU,EAAE;wBACV,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE;wBAC1B,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC5B;iBACF;aACF;SACF;KACF;CACO,CAAC;AAEX,MAAM,UAAU,gBAAgB,CAAC,MAAyB,OAAO,CAAC,GAAG;IACnE,OAAO,GAAG,CAAC,cAAc,EAAE,IAAI,EAAE,IAAI,GAAG,CAAC,yBAAyB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;AACnF,CAAC;AAED,SAAS,wBAAwB,CAAC,MAAyB,OAAO,CAAC,GAAG;IACpE,OAAO;QACL,SAAS,EAAE,mBAAmB,CAAC,GAAG,CAAC,sBAAsB,EAAE,IAAI,CAAC;QAChE,WAAW,EAAE,eAAe,CAAC,GAAG,CAAC,sBAAsB,EAAE,GAAG,CAAC;KAC9D,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,OAA2B,EAAE,UAAkB,EAAE,KAAa;IACjF,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;IACvD,MAAM,cAAc,GAAG,OAAO,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC,CAAC;IAC7D,OAAO,eAAe,CACpB;QACE,YAAY;QACZ,kBAAkB,EAAE,CAAC;QACrB,gBAAgB,EAAE,CAAC;QACnB,cAAc;QACd,WAAW,EAAE,OAAO,CAAC,KAAK,EAAE,YAAY,IAAI,YAAY,GAAG,cAAc;QACzE,mBAAmB,EAAE,cAAc,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI;QACvG,gBAAgB,EAAE,CAAC;QACnB,kBAAkB,EAAE,UAAU;QAC9B,eAAe,EAAE,UAAU;QAC3B,WAAW,EAAE,UAAU;KACxB,EACD,QAAQ,EACR,KAAK,CACN,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,QAAkB;IAC9C,IAAI,CAAC;QACH,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAyB,EAAE,QAAgB;IACtE,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;IACrD,OAAO,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC;AAClF,CAAC;AAED,SAAS,eAAe,CAAC,KAAyB,EAAE,QAAgB;IAClE,MAAM,WAAW,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IACnD,OAAO,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,WAAW,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC;AACnF,CAAC"}

package/dist/core/ollama.d.ts

@@ -0,0 +1,31 @@
+import type { ModelChatOptions, ModelChatResult } from "./types.js";
+export declare const defaultOllamaModel = "qwen2.5-coder:7b";
+export declare const defaultOllamaUrl = "http://127.0.0.1:11434";
+export declare const defaultOllamaPort = 11434;
+export type OllamaRunningModel = {
+    name: string;
+    sizeBytes: number | null;
+    sizeVramBytes: number | null;
+    expiresAt: string | null;
+    contextLength: number | null;
+};
+type OllamaRuntimeOptions = {
+    keepAlive: string;
+    numCtx: number;
+    numPredict: number;
+    temperature: number;
+};
+export declare class OllamaClient {
+    private readonly baseUrl;
+    private readonly runtimeOptions;
+    constructor(baseUrl?: string, runtimeOptions?: OllamaRuntimeOptions);
+    chat(options: ModelChatOptions): Promise<ModelChatResult>;
+    listModels(): Promise<string[]>;
+    listRunningModels(): Promise<OllamaRunningModel[]>;
+    unloadModel(model: string): Promise<void>;
+    private fetchOllama;
+}
+export declare function resolveOllamaBaseUrl(env?: NodeJS.ProcessEnv): string;
+export declare function normalizeOllamaBaseUrl(value: string | undefined): string;
+export declare function readOllamaRuntimeOptions(env?: NodeJS.ProcessEnv): OllamaRuntimeOptions;
+export {};

package/dist/core/ollama.js

@@ -0,0 +1,176 @@
+import { attachTokenCost } from "./tokenAccounting.js";
+export const defaultOllamaModel = "qwen2.5-coder:7b";
+export const defaultOllamaUrl = "http://127.0.0.1:11434";
+export const defaultOllamaPort = 11434;
+export class OllamaClient {
+    baseUrl;
+    runtimeOptions;
+    constructor(baseUrl = defaultOllamaUrl, runtimeOptions = readOllamaRuntimeOptions()) {
+        this.baseUrl = normalizeOllamaBaseUrl(baseUrl);
+        this.runtimeOptions = runtimeOptions;
+    }
+    async chat(options) {
+        const response = await this.fetchOllama("/api/chat", {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json"
+            },
+            body: JSON.stringify({
+                model: options.model,
+                messages: options.messages,
+                stream: false,
+                keep_alive: this.runtimeOptions.keepAlive,
+                options: {
+                    num_ctx: this.runtimeOptions.numCtx,
+                    num_predict: this.runtimeOptions.numPredict,
+                    temperature: this.runtimeOptions.temperature
+                },
+                format: options.formatJson ? "json" : undefined
+            }),
+            signal: options.signal
+        });
+        const payload = (await readJsonSafely(response));
+        if (!response.ok) {
+            const reason = payload.error ? ` ${payload.error}` : "";
+            throw new Error(`Ollama chat failed for model "${options.model}" at ${this.baseUrl}: HTTP ${response.status}.${reason}`);
+        }
+        if (payload.error) {
+            throw new Error(payload.error);
+        }
+        return {
+            content: payload.message?.content?.trim() ?? "",
+            telemetry: toTelemetry(payload, options.model)
+        };
+    }
+    async listModels() {
+        const response = await this.fetchOllama("/api/tags");
+        if (!response.ok) {
+            throw new Error(`Ollama tags failed with HTTP ${response.status}.`);
+        }
+        const payload = (await response.json());
+        return payload.models?.map((model) => model.name).sort() ?? [];
+    }
+    async listRunningModels() {
+        const response = await this.fetchOllama("/api/ps");
+        if (!response.ok) {
+            throw new Error(`Ollama ps failed with HTTP ${response.status}.`);
+        }
+        const payload = (await response.json());
+        return (payload.models
+            ?.map((model) => {
+            const name = model.name?.trim() || model.model?.trim() || "";
+            return name
+                ? {
+                    name,
+                    sizeBytes: readNullableFiniteNumber(model.size),
+                    sizeVramBytes: readNullableFiniteNumber(model.size_vram),
+                    expiresAt: typeof model.expires_at === "string" ? model.expires_at : null,
+                    contextLength: readNullableFiniteNumber(model.context_length ?? model.details?.context_length)
+                }
+                : null;
+        })
+            .filter((model) => model !== null)
+            .sort((left, right) => left.name.localeCompare(right.name)) ?? []);
+    }
+    async unloadModel(model) {
+        const trimmedModel = model.trim();
+        if (!trimmedModel) {
+            return;
+        }
+        const response = await this.fetchOllama("/api/generate", {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json"
+            },
+            body: JSON.stringify({
+                model: trimmedModel,
+                keep_alive: 0
+            })
+        });
+        if (!response.ok) {
+            throw new Error(`Ollama unload failed for model "${trimmedModel}" at ${this.baseUrl}: HTTP ${response.status}.`);
+        }
+    }
+    async fetchOllama(path, init) {
+        try {
+            return await fetch(`${this.baseUrl}${path}`, init);
+        }
+        catch (error) {
+            throw new Error(formatOllamaConnectionError(this.baseUrl, error));
+        }
+    }
+}
+async function readJsonSafely(response) {
+    try {
+        return await response.json();
+    }
+    catch {
+        return {};
+    }
+}
+export function resolveOllamaBaseUrl(env = process.env) {
+    return normalizeOllamaBaseUrl(env.PATCHPILOT_OLLAMA_URL ?? env.OLLAMA_HOST ?? defaultOllamaUrl);
+}
+export function normalizeOllamaBaseUrl(value) {
+    const trimmedValue = value?.trim() ?? "";
+    const normalizedAlias = trimmedValue.toLowerCase();
+    if (!trimmedValue || normalizedAlias === "local" || normalizedAlias === "localhost") {
+        return defaultOllamaUrl;
+    }
+    const rawUrl = /^https?:\/\//i.test(trimmedValue) ? trimmedValue : `http://${trimmedValue}`;
+    const parsedUrl = new URL(rawUrl);
+    if (parsedUrl.hostname === "0.0.0.0" || parsedUrl.hostname === "[::]" || parsedUrl.hostname === "::") {
+        parsedUrl.hostname = "127.0.0.1";
+    }
+    if (!parsedUrl.port && parsedUrl.protocol === "http:") {
+        parsedUrl.port = String(defaultOllamaPort);
+    }
+    if (parsedUrl.pathname === "/api") {
+        parsedUrl.pathname = "";
+    }
+    return parsedUrl.toString().replace(/\/$/, "");
+}
+export function readOllamaRuntimeOptions(env = process.env) {
+    return {
+        keepAlive: env.PATCHPILOT_KEEP_ALIVE?.trim() || "15m",
+        numCtx: readPositiveInteger(env.PATCHPILOT_NUM_CTX, 8192),
+        numPredict: readPositiveInteger(env.PATCHPILOT_NUM_PREDICT, 1024),
+        temperature: readTemperature(env.PATCHPILOT_TEMPERATURE, 0.1)
+    };
+}
+function readPositiveInteger(value, fallback) {
+    const parsedValue = Number.parseInt(value ?? "", 10);
+    return Number.isFinite(parsedValue) && parsedValue > 0 ? parsedValue : fallback;
+}
+function readTemperature(value, fallback) {
+    const parsedValue = Number.parseFloat(value ?? "");
+    return Number.isFinite(parsedValue) && parsedValue >= 0 ? parsedValue : fallback;
+}
+function readNullableFiniteNumber(value) {
+    return typeof value === "number" && Number.isFinite(value) ? value : null;
+}
+function formatOllamaConnectionError(baseUrl, error) {
+    const suffix = error instanceof Error ? ` ${error.message}` : "";
+    return `Cannot reach Ollama at ${baseUrl}. Start Ollama, or run "ollama serve", then try /doctor.${suffix}`;
+}
+function toTelemetry(payload, model) {
+    const promptTokens = payload.prompt_eval_count ?? 0;
+    const responseTokens = payload.eval_count ?? 0;
+    const responseDurationMs = nanosToMillis(payload.eval_duration ?? 0);
+    return attachTokenCost({
+        promptTokens,
+        cachedPromptTokens: 0,
+        cacheWriteTokens: 0,
+        responseTokens,
+        totalTokens: promptTokens + responseTokens,
+        evalTokensPerSecond: responseTokens > 0 && responseDurationMs > 0 ? responseTokens / (responseDurationMs / 1000) : null,
+        promptDurationMs: nanosToMillis(payload.prompt_eval_duration ?? 0),
+        responseDurationMs,
+        totalDurationMs: nanosToMillis(payload.total_duration ?? 0),
+        tokenSource: "provider"
+    }, "ollama", model);
+}
+function nanosToMillis(value) {
+    return Math.round(value / 1_000_000);
+}
+//# sourceMappingURL=ollama.js.map

package/dist/core/ollama.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/core/ollama.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAEvD,MAAM,CAAC,MAAM,kBAAkB,GAAG,kBAAkB,CAAC;AACrD,MAAM,CAAC,MAAM,gBAAgB,GAAG,wBAAwB,CAAC;AACzD,MAAM,CAAC,MAAM,iBAAiB,GAAG,KAAK,CAAC;AAkDvC,MAAM,OAAO,YAAY;IACN,OAAO,CAAS;IAChB,cAAc,CAAuB;IAEtD,YAAY,OAAO,GAAG,gBAAgB,EAAE,cAAc,GAAG,wBAAwB,EAAE;QACjF,IAAI,CAAC,OAAO,GAAG,sBAAsB,CAAC,OAAO,CAAC,CAAC;QAC/C,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAAyB;QAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE;YACnD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,MAAM,EAAE,KAAK;gBACb,UAAU,EAAE,IAAI,CAAC,cAAc,CAAC,SAAS;gBACzC,OAAO,EAAE;oBACP,OAAO,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM;oBACnC,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,UAAU;oBAC3C,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,WAAW;iBAC7C;gBACD,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;aAChD,CAAC;YACF,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,CAAC,MAAM,cAAc,CAAC,QAAQ,CAAC,CAAuB,CAAC;QACvE,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,iCAAiC,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,OAAO,UAAU,QAAQ,CAAC,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC;QAC3H,CAAC;QAED,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACjC,CAAC;QAED,OAAO;YACL,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE;YAC/C,SAAS,EAAE,WAAW,CAAC,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC;SAC/C,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,gCAAgC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAuB,CAAC;QAC9D,OAAO,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QACnD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAqB,CAAC;QAC5D,OAAO,CACL,OAAO,CAAC,MAAM;YACZ,EAAE,GAAG,CAAC,CAAC,KAAK,EAA6B,EAAE;YACzC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;YAC7D,OAAO,IAAI;gBACT,CAAC,CAAC;oBACE,IAAI;oBACJ,SAAS,EAAE,wBAAwB,CAAC,KAAK,CAAC,IAAI,CAAC;oBAC/C,aAAa,EAAE,wBAAwB,CAAC,KAAK,CAAC,SAAS,CAAC;oBACxD,SAAS,EAAE,OAAO,KAAK,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI;oBACzE,aAAa,EAAE,wBAAwB,CAAC,KAAK,CAAC,cAAc,IAAI,KAAK,CAAC,OAAO,EAAE,cAAc,CAAC;iBAC/F;gBACH,CAAC,CAAC,IAAI,CAAC;QACX,CAAC,CAAC;aACD,MAAM,CAAC,CAAC,KAAK,EAA+B,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC;aAC9D,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CACpE,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAa;QAC7B,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,eAAe,EAAE;YACvD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,YAAY;gBACnB,UAAU,EAAE,CAAC;aACd,CAAC;SACH,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,mCAAmC,YAAY,QAAQ,IAAI,CAAC,OAAO,UAAU,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;QACnH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,IAAkB;QACxD,IAAI,CAAC;YACH,OAAO,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,GAAG,IAAI,EAAE,EAAE,IAAI,CAAC,CAAC;QACrD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;CACF;AAED,KAAK,UAAU,cAAc,CAAC,QAAkB;IAC9C,IAAI,CAAC;QACH,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC/B,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAyB,OAAO,CAAC,GAAG;IACvE,OAAO,sBAAsB,CAAC,GAAG,CAAC,qBAAqB,IAAI,GAAG,CAAC,WAAW,IAAI,gBAAgB,CAAC,CAAC;AAClG,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,KAAyB;IAC9D,MAAM,YAAY,GAAG,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACzC,MAAM,eAAe,GAAG,YAAY,CAAC,WAAW,EAAE,CAAC;IACnD,IAAI,CAAC,YAAY,IAAI,eAAe,KAAK,OAAO,IAAI,eAAe,KAAK,WAAW,EAAE,CAAC;QACpF,OAAO,gBAAgB,CAAC;IAC1B,CAAC;IAED,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,UAAU,YAAY,EAAE,CAAC;IAC5F,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC;IAElC,IAAI,SAAS,CAAC,QAAQ,KAAK,SAAS,IAAI,SAAS,CAAC,QAAQ,KAAK,MAAM,IAAI,SAAS,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;QACrG,SAAS,CAAC,QAAQ,GAAG,WAAW,CAAC;IACnC,CAAC;IAED,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,SAAS,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QACtD,SAAS,CAAC,IAAI,GAAG,MAAM,CAAC,iBAAiB,CAAC,CAAC;IAC7C,CAAC;IAED,IAAI,SAAS,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;QAClC,SAAS,CAAC,QAAQ,GAAG,EAAE,CAAC;IAC1B,CAAC;IAED,OAAO,SAAS,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;AACjD,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAyB,OAAO,CAAC,GAAG;IAC3E,OAAO;QACL,SAAS,EAAE,GAAG,CAAC,qBAAqB,EAAE,IAAI,EAAE,IAAI,KAAK;QACrD,MAAM,EAAE,mBAAmB,CAAC,GAAG,CAAC,kBAAkB,EAAE,IAAI,CAAC;QACzD,UAAU,EAAE,mBAAmB,CAAC,GAAG,CAAC,sBAAsB,EAAE,IAAI,CAAC;QACjE,WAAW,EAAE,eAAe,CAAC,GAAG,CAAC,sBAAsB,EAAE,GAAG,CAAC;KAC9D,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAyB,EAAE,QAAgB;IACtE,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;IACrD,OAAO,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC;AAClF,CAAC;AAED,SAAS,eAAe,CAAC,KAAyB,EAAE,QAAgB;IAClE,MAAM,WAAW,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IACnD,OAAO,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,WAAW,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC;AACnF,CAAC;AAED,SAAS,wBAAwB,CAAC,KAAc;IAC9C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;AAC5E,CAAC;AAED,SAAS,2BAA2B,CAAC,OAAe,EAAE,KAAc;IAClE,MAAM,MAAM,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACjE,OAAO,0BAA0B,OAAO,2DAA2D,MAAM,EAAE,CAAC;AAC9G,CAAC;AAED,SAAS,WAAW,CAAC,OAA2B,EAAE,KAAa;IAC7D,MAAM,YAAY,GAAG,OAAO,CAAC,iBAAiB,IAAI,CAAC,CAAC;IACpD,MAAM,cAAc,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,CAAC;IAC/C,MAAM,kBAAkB,GAAG,aAAa,CAAC,OAAO,CAAC,aAAa,IAAI,CAAC,CAAC,CAAC;IAErE,OAAO,eAAe,CACpB;QACE,YAAY;QACZ,kBAAkB,EAAE,CAAC;QACrB,gBAAgB,EAAE,CAAC;QACnB,cAAc;QACd,WAAW,EAAE,YAAY,GAAG,cAAc;QAC1C,mBAAmB,EACjB,cAAc,GAAG,CAAC,IAAI,kBAAkB,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,kBAAkB,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI;QACpG,gBAAgB,EAAE,aAAa,CAAC,OAAO,CAAC,oBAAoB,IAAI,CAAC,CAAC;QAClE,kBAAkB;QAClB,eAAe,EAAE,aAAa,CAAC,OAAO,CAAC,cAAc,IAAI,CAAC,CAAC;QAC3D,WAAW,EAAE,UAAU;KACxB,EACD,QAAQ,EACR,KAAK,CACN,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAAa;IAClC,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC;AACvC,CAAC"}

package/dist/core/openrouter.d.ts

@@ -0,0 +1,27 @@
+import type { ModelChatOptions, ModelChatResult } from "./types.js";
+export declare const defaultOpenRouterModel = "openrouter/auto";
+export declare const defaultOpenRouterBaseUrl = "https://openrouter.ai/api/v1";
+type OpenRouterRuntimeOptions = {
+    maxTokens: number;
+    temperature: number;
+};
+export type OpenRouterTokenRates = {
+    inputPerToken: number;
+    cachedInputPerToken: number;
+    cacheWritePerToken: number;
+    outputPerToken: number;
+};
+export declare class OpenRouterClient {
+    private readonly apiKey;
+    private readonly baseUrl;
+    private readonly runtimeOptions;
+    constructor(apiKey?: string, baseUrl?: string, runtimeOptions?: OpenRouterRuntimeOptions);
+    chat(options: ModelChatOptions): Promise<ModelChatResult>;
+    listModels(): Promise<string[]>;
+    private fetchOpenRouter;
+    private assertConfigured;
+}
+export declare function readOpenRouterApiKey(env?: NodeJS.ProcessEnv): string;
+export declare function isOpenRouterFreeModel(model: string): boolean;
+export declare function getOpenRouterModelRates(model: string): Promise<OpenRouterTokenRates | null>;
+export {};