steroids-cli 0.10.62 → 0.12.0

package/dist/ollama/api-client.js

@@ -0,0 +1,229 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OllamaApiClient = exports.OllamaApiError = void 0;
+const node_util_1 = require("node:util");
+const DEFAULT_TIMEOUT_MS = 30_000;
+class OllamaApiError extends Error {
+    status;
+    responseBody;
+    url;
+    constructor(message, status, responseBody, url) {
+        super(message);
+        this.name = 'OllamaApiError';
+        this.status = status;
+        this.responseBody = responseBody;
+        this.url = url;
+    }
+}
+exports.OllamaApiError = OllamaApiError;
+class OllamaApiClient {
+    endpoint;
+    apiKey;
+    timeoutMs;
+    constructor(config) {
+        this.endpoint = normalizeEndpoint(config.endpoint);
+        this.apiKey = config.apiKey;
+        this.timeoutMs = config.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+    }
+    setConfig(config) {
+        if (config.endpoint) {
+            this.endpoint = normalizeEndpoint(config.endpoint);
+        }
+        if (config.apiKey !== undefined) {
+            this.apiKey = config.apiKey;
+        }
+        if (config.timeoutMs !== undefined) {
+            this.timeoutMs = config.timeoutMs;
+        }
+    }
+    getEndpoint() {
+        return this.endpoint;
+    }
+    async healthCheck() {
+        const response = await this.requestRaw('/');
+        const body = await response.text();
+        return { status: response.status, body };
+    }
+    async getVersion() {
+        return this.requestJson('/api/version');
+    }
+    async listRunningModels() {
+        return this.requestJson('/api/ps');
+    }
+    async listInstalledModels() {
+        return this.requestJson('/api/tags');
+    }
+    async showModel(name) {
+        return this.requestJson('/api/show', {
+            method: 'POST',
+            body: JSON.stringify({ name }),
+            headers: {
+                'Content-Type': 'application/json',
+            },
+        });
+    }
+    async pullModel(name, onProgress, options) {
+        const response = await this.requestRaw('/api/pull', {
+            method: 'POST',
+            body: JSON.stringify({ name, stream: true }),
+            headers: {
+                'Content-Type': 'application/json',
+            },
+        }, {
+            timeoutMs: options?.timeoutMs,
+            operation: 'pull model',
+        });
+        if (!response.body) {
+            return [];
+        }
+        const reader = response.body.getReader();
+        const decoder = new node_util_1.TextDecoder();
+        const updates = [];
+        let buffer = '';
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done) {
+                break;
+            }
+            buffer += decoder.decode(value, { stream: true });
+            buffer = this.consumePullBuffer(buffer, updates, onProgress);
+        }
+        buffer += decoder.decode();
+        buffer = this.consumePullBuffer(buffer, updates, onProgress);
+        const tail = buffer.trim();
+        if (tail) {
+            const progress = this.normalizePullProgress(JSON.parse(tail));
+            updates.push(progress);
+            onProgress?.(progress);
+        }
+        return updates;
+    }
+    async deleteModel(name) {
+        await this.requestRaw('/api/delete', {
+            method: 'DELETE',
+            body: JSON.stringify({ name }),
+            headers: {
+                'Content-Type': 'application/json',
+            },
+        });
+    }
+    async chat(payload) {
+        return this.requestJson('/api/chat', {
+            method: 'POST',
+            body: JSON.stringify(payload),
+            headers: {
+                'Content-Type': 'application/json',
+            },
+        });
+    }
+    async listOpenAIModels() {
+        return this.requestJson('/v1/models');
+    }
+    async getOpenAIModel(model) {
+        return this.requestJson(`/v1/models/${encodeURIComponent(model)}`);
+    }
+    consumePullBuffer(buffer, updates, onProgress) {
+        while (true) {
+            const newlineIndex = buffer.indexOf('\n');
+            if (newlineIndex === -1) {
+                break;
+            }
+            const line = buffer.slice(0, newlineIndex).trim();
+            buffer = buffer.slice(newlineIndex + 1);
+            if (!line) {
+                continue;
+            }
+            const progress = this.normalizePullProgress(JSON.parse(line));
+            updates.push(progress);
+            onProgress?.(progress);
+        }
+        return buffer;
+    }
+    normalizePullProgress(progress) {
+        const total = typeof progress.total === 'number' && Number.isFinite(progress.total) && progress.total > 0
+            ? progress.total
+            : null;
+        const completed = typeof progress.completed === 'number' && Number.isFinite(progress.completed)
+            ? Math.max(0, progress.completed)
+            : null;
+        const percent = total !== null && completed !== null
+            ? Math.max(0, Math.min(100, Math.round((completed / total) * 100)))
+            : null;
+        const status = (progress.status || '').toLowerCase();
+        const done = status.includes('success') || status.includes('complete') || status.includes('finished');
+        const phase = progress.error
+            ? 'error'
+            : done
+                ? 'complete'
+                : status.includes('download')
+                    ? 'downloading'
+                    : status.includes('verify') || status.includes('digest')
+                        ? 'verifying'
+                        : status.includes('pull') || status.includes('resolve')
+                            ? 'starting'
+                            : 'unknown';
+        return {
+            ...progress,
+            percent,
+            phase,
+            done,
+        };
+    }
+    async requestJson(path, init, requestOptions) {
+        const response = await this.requestRaw(path, init, requestOptions);
+        return response.json();
+    }
+    async requestRaw(path, init = {}, requestOptions = {}) {
+        const url = buildUrl(this.endpoint, path);
+        const headers = new Headers(init.headers);
+        const timeoutMs = requestOptions.timeoutMs ?? this.timeoutMs;
+        const operation = requestOptions.operation ?? 'request';
+        if (this.apiKey) {
+            headers.set('Authorization', `Bearer ${this.apiKey}`);
+        }
+        if (!headers.has('Accept')) {
+            headers.set('Accept', 'application/json');
+        }
+        const controller = new AbortController();
+        const timeout = setTimeout(() => controller.abort(), timeoutMs);
+        try {
+            const response = await fetch(url, {
+                ...init,
+                headers,
+                signal: controller.signal,
+            });
+            if (!response.ok) {
+                const body = await response.text();
+                throw new OllamaApiError(`Ollama API request failed: ${response.status}`, response.status, body, url);
+            }
+            return response;
+        }
+        catch (error) {
+            if (error instanceof OllamaApiError) {
+                throw error;
+            }
+            if (error instanceof Error && error.name === 'AbortError') {
+                throw new Error(`Ollama API ${operation} timed out after ${timeoutMs}ms`);
+            }
+            throw error;
+        }
+        finally {
+            clearTimeout(timeout);
+        }
+    }
+}
+exports.OllamaApiClient = OllamaApiClient;
+function normalizeEndpoint(endpoint) {
+    return endpoint.trim().replace(/\/+$/, '');
+}
+function buildUrl(endpoint, path) {
+    const base = normalizeEndpoint(endpoint);
+    if (!path) {
+        return base;
+    }
+    if (path.startsWith('/')) {
+        return `${base}${path}`;
+    }
+    return `${base}/${path}`;
+}
+//# sourceMappingURL=api-client.js.map

package/dist/ollama/api-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api-client.js","sourceRoot":"","sources":["../../src/ollama/api-client.ts"],"names":[],"mappings":";;;AAAA,yCAAwC;AAExC,MAAM,kBAAkB,GAAG,MAAM,CAAC;AA4HlC,MAAa,cAAe,SAAQ,KAAK;IAC9B,MAAM,CAAS;IACf,YAAY,CAAS;IACrB,GAAG,CAAS;IAErB,YAAY,OAAe,EAAE,MAAc,EAAE,YAAoB,EAAE,GAAW;QAC5E,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;CACF;AAZD,wCAYC;AAED,MAAa,eAAe;IAClB,QAAQ,CAAS;IACjB,MAAM,CAAU;IAChB,SAAS,CAAS;IAE1B,YAAY,MAA0B;QACpC,IAAI,CAAC,QAAQ,GAAG,iBAAiB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACnD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,kBAAkB,CAAC;IAC1D,CAAC;IAED,SAAS,CAAC,MAAmC;QAC3C,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,IAAI,CAAC,QAAQ,GAAG,iBAAiB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC9B,CAAC;QACD,IAAI,MAAM,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;YACnC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACpC,CAAC;IACH,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,WAAW,CAAwB,cAAc,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,OAAO,IAAI,CAAC,WAAW,CAAmB,SAAS,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,mBAAmB;QACvB,OAAO,IAAI,CAAC,WAAW,CAAqB,WAAW,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAY;QAC1B,OAAO,IAAI,CAAC,WAAW,CAAqB,WAAW,EAAE;YACvD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC;YAC9B,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;SACF,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,SAAS,CACb,IAAY,EACZ,UAAmD,EACnD,OAAgC;QAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE;YAClD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;YAC5C,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;SACF,EAAE;YACD,SAAS,EAAE,OAAO,EAAE,SAAS;YAC7B,SAAS,EAAE,YAAY;SACxB,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,uBAAW,EAAE,CAAC;QAClC,MAAM,OAAO,GAAyB,EAAE,CAAC;QACzC,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM;YACR,CAAC;YAED,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QAC3B,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;QAC7D,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;QAC3B,IAAI,IAAI,EAAE,CAAC;YACT,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAuB,CAAC,CAAC;YACpF,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvB,UAAU,EAAE,CAAC,QAAQ,CAAC,CAAC;QACzB,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAY;QAC5B,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,EAAE;YACnC,MAAM,EAAE,QAAQ;YAChB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC;YAC9B,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;SACF,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,OAA0B;QACnC,OAAO,IAAI,CAAC,WAAW,CAAqB,WAAW,EAAE;YACvD,MAAM,EAAE,MAAM;YACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAC7B,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;aACnC;SACF,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB;QACpB,OAAO,IAAI,CAAC,WAAW,CAA6B,YAAY,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,OAAO,IAAI,CAAC,WAAW,CAAoB,cAAc,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACxF,CAAC;IAEO,iBAAiB,CACvB,MAAc,EACd,OAA6B,EAC7B,UAAmD;QAEnD,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAC1C,IAAI,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC;gBACxB,MAAM;YACR,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,IAAI,EAAE,CAAC;YAClD,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;YAExC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,SAAS;YACX,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAuB,CAAC,CAAC;YACpF,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvB,UAAU,EAAE,CAAC,QAAQ,CAAC,CAAC;QACzB,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,qBAAqB,CAAC,QAA4B;QACxD,MAAM,KAAK,GAAG,OAAO,QAAQ,CAAC,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,QAAQ,CAAC,KAAK,GAAG,CAAC;YACvG,CAAC,CAAC,QAAQ,CAAC,KAAK;YAChB,CAAC,CAAC,IAAI,CAAC;QACT,MAAM,SAAS,GAAG,OAAO,QAAQ,CAAC,SAAS,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC7F,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,SAAS,CAAC;YACjC,CAAC,CAAC,IAAI,CAAC;QACT,MAAM,OAAO,GAAG,KAAK,KAAK,IAAI,IAAI,SAAS,KAAK,IAAI;YAClD,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;YACnE,CAAC,CAAC,IAAI,CAAC;QACT,MAAM,MAAM,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;QACrD,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACtG,MAAM,KAAK,GAAgC,QAAQ,CAAC,KAAK;YACvD,CAAC,CAAC,OAAO;YACT,CAAC,CAAC,IAAI;gBACJ,CAAC,CAAC,UAAU;gBACZ,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC;oBAC3B,CAAC,CAAC,aAAa;oBACf,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC;wBACtD,CAAC,CAAC,WAAW;wBACb,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC;4BACrD,CAAC,CAAC,UAAU;4BACZ,CAAC,CAAC,SAAS,CAAC;QAEtB,OAAO;YACL,GAAG,QAAQ;YACX,OAAO;YACP,KAAK;YACL,IAAI;SACL,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,WAAW,CAAI,IAAY,EAAE,IAAkB,EAAE,cAA+B;QAC5F,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;QACnE,OAAO,QAAQ,CAAC,IAAI,EAAgB,CAAC;IACvC,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,IAAY,EAAE,OAAoB,EAAE,EAAE,iBAAiC,EAAE;QAChG,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,SAAS,GAAG,cAAc,CAAC,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC;QAC7D,MAAM,SAAS,GAAG,cAAc,CAAC,SAAS,IAAI,SAAS,CAAC;QAExD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;QAC5C,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;QAEhE,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,GAAG,IAAI;gBACP,OAAO;gBACP,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACnC,MAAM,IAAI,cAAc,CACtB,8BAA8B,QAAQ,CAAC,MAAM,EAAE,EAC/C,QAAQ,CAAC,MAAM,EACf,IAAI,EACJ,GAAG,CACJ,CAAC;YACJ,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,cAAc,EAAE,CAAC;gBACpC,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC1D,MAAM,IAAI,KAAK,CAAC,cAAc,SAAS,oBAAoB,SAAS,IAAI,CAAC,CAAC;YAC5E,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,OAAO,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;CACF;AAlPD,0CAkPC;AAED,SAAS,iBAAiB,CAAC,QAAgB;IACzC,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AAC7C,CAAC;AAED,SAAS,QAAQ,CAAC,QAAgB,EAAE,IAAY;IAC9C,MAAM,IAAI,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;IACzC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACzB,OAAO,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC;IAC1B,CAAC;IAED,OAAO,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC;AAC3B,CAAC"}

package/dist/ollama/connection.d.ts

@@ -0,0 +1,35 @@
+import { OllamaApiClient } from './api-client.js';
+export declare const DEFAULT_LOCAL_ENDPOINT = "http://localhost:11434";
+export declare const DEFAULT_CLOUD_ENDPOINT = "https://ollama.com";
+export declare const MINIMUM_OLLAMA_VERSION = "0.1.14";
+export type OllamaConnectionMode = 'local' | 'cloud';
+export interface OllamaConnectionConfig {
+    endpoint: string;
+    mode: OllamaConnectionMode;
+    cloudTier: string | null;
+}
+export interface OllamaConnectionStatus {
+    connected: boolean;
+    endpoint: string;
+    mode: OllamaConnectionMode;
+    version?: string;
+    minimumVersionMet?: boolean;
+    loadedModels?: number;
+    error?: string;
+}
+export declare function getOllamaConfigDir(): string;
+export declare function getOllamaConfigPath(): string;
+export declare function getOllamaTokenPath(): string;
+export declare function getDefaultConnectionConfig(): OllamaConnectionConfig;
+export declare function loadConnectionConfig(): OllamaConnectionConfig;
+export declare function saveConnectionConfig(config: OllamaConnectionConfig): void;
+export declare function setLocalConnection(endpoint?: string): OllamaConnectionConfig;
+export declare function setCloudConnection(apiKey: string, endpoint?: string): OllamaConnectionConfig;
+export declare function clearCloudApiKey(): void;
+export declare function getCloudApiKey(): string | undefined;
+export declare function resolveEndpoint(config: OllamaConnectionConfig): string;
+export declare function getResolvedConnectionConfig(): OllamaConnectionConfig;
+export declare function createOllamaApiClient(config?: OllamaConnectionConfig): OllamaApiClient;
+export declare function testConnection(config?: OllamaConnectionConfig): Promise<OllamaConnectionStatus>;
+export declare function isVersionSupported(version: string, minimumVersion?: string): boolean;
+//# sourceMappingURL=connection.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"connection.d.ts","sourceRoot":"","sources":["../../src/ollama/connection.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,eAAe,EAAkB,MAAM,iBAAiB,CAAC;AAElE,eAAO,MAAM,sBAAsB,2BAA2B,CAAC;AAC/D,eAAO,MAAM,sBAAsB,uBAAuB,CAAC;AAC3D,eAAO,MAAM,sBAAsB,WAAW,CAAC;AAE/C,MAAM,MAAM,oBAAoB,GAAG,OAAO,GAAG,OAAO,CAAC;AAErD,MAAM,WAAW,sBAAsB;IACrC,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,oBAAoB,CAAC;IAC3B,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;CAC1B;AAED,MAAM,WAAW,sBAAsB;IACrC,SAAS,EAAE,OAAO,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,oBAAoB,CAAC;IAC3B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAKD,wBAAgB,kBAAkB,IAAI,MAAM,CAE3C;AAED,wBAAgB,mBAAmB,IAAI,MAAM,CAE5C;AAED,wBAAgB,kBAAkB,IAAI,MAAM,CAE3C;AAED,wBAAgB,0BAA0B,IAAI,sBAAsB,CAMnE;AAED,wBAAgB,oBAAoB,IAAI,sBAAsB,CAkB7D;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,sBAAsB,GAAG,IAAI,CAezE;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,GAAE,MAA+B,GAAG,sBAAsB,CAQpG;AAED,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,MAAM,EACd,QAAQ,GAAE,MAA+B,GACxC,sBAAsB,CAkBxB;AAED,wBAAgB,gBAAgB,IAAI,IAAI,CAKvC;AAED,wBAAgB,cAAc,IAAI,MAAM,GAAG,SAAS,CAanD;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,sBAAsB,GAAG,MAAM,CAOtE;AAED,wBAAgB,2BAA2B,IAAI,sBAAsB,CAMpE;AAED,wBAAgB,qBAAqB,CAAC,MAAM,GAAE,sBAAsD,GAAG,eAAe,CAQrH;AAED,wBAAsB,cAAc,CAClC,MAAM,GAAE,sBAAsD,GAC7D,OAAO,CAAC,sBAAsB,CAAC,CAwDjC;AAED,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,cAAc,GAAE,MAA+B,GAAG,OAAO,CAkB5G"}

package/dist/ollama/connection.js

@@ -0,0 +1,243 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MINIMUM_OLLAMA_VERSION = exports.DEFAULT_CLOUD_ENDPOINT = exports.DEFAULT_LOCAL_ENDPOINT = void 0;
+exports.getOllamaConfigDir = getOllamaConfigDir;
+exports.getOllamaConfigPath = getOllamaConfigPath;
+exports.getOllamaTokenPath = getOllamaTokenPath;
+exports.getDefaultConnectionConfig = getDefaultConnectionConfig;
+exports.loadConnectionConfig = loadConnectionConfig;
+exports.saveConnectionConfig = saveConnectionConfig;
+exports.setLocalConnection = setLocalConnection;
+exports.setCloudConnection = setCloudConnection;
+exports.clearCloudApiKey = clearCloudApiKey;
+exports.getCloudApiKey = getCloudApiKey;
+exports.resolveEndpoint = resolveEndpoint;
+exports.getResolvedConnectionConfig = getResolvedConnectionConfig;
+exports.createOllamaApiClient = createOllamaApiClient;
+exports.testConnection = testConnection;
+exports.isVersionSupported = isVersionSupported;
+const node_fs_1 = require("node:fs");
+const node_path_1 = require("node:path");
+const global_db_connection_js_1 = require("../runners/global-db-connection.js");
+const api_client_js_1 = require("./api-client.js");
+exports.DEFAULT_LOCAL_ENDPOINT = 'http://localhost:11434';
+exports.DEFAULT_CLOUD_ENDPOINT = 'https://ollama.com';
+exports.MINIMUM_OLLAMA_VERSION = '0.1.14';
+const CONFIG_FILE = 'config.json';
+const TOKEN_FILE = 'token';
+function getOllamaConfigDir() {
+    return (0, node_path_1.join)((0, global_db_connection_js_1.getGlobalSteroidsDir)(), 'ollama');
+}
+function getOllamaConfigPath() {
+    return (0, node_path_1.join)(getOllamaConfigDir(), CONFIG_FILE);
+}
+function getOllamaTokenPath() {
+    return (0, node_path_1.join)(getOllamaConfigDir(), TOKEN_FILE);
+}
+function getDefaultConnectionConfig() {
+    return {
+        endpoint: exports.DEFAULT_LOCAL_ENDPOINT,
+        mode: 'local',
+        cloudTier: null,
+    };
+}
+function loadConnectionConfig() {
+    const configPath = getOllamaConfigPath();
+    if (!(0, node_fs_1.existsSync)(configPath)) {
+        return getDefaultConnectionConfig();
+    }
+    try {
+        const parsed = JSON.parse((0, node_fs_1.readFileSync)(configPath, 'utf8'));
+        const mode = parsed.mode === 'cloud' ? 'cloud' : 'local';
+        const endpoint = normalizeEndpoint(parsed.endpoint || getDefaultConnectionConfig().endpoint);
+        return {
+            endpoint,
+            mode,
+            cloudTier: parsed.cloudTier ?? null,
+        };
+    }
+    catch {
+        return getDefaultConnectionConfig();
+    }
+}
+function saveConnectionConfig(config) {
+    ensureOllamaConfigDir();
+    (0, node_fs_1.writeFileSync)(getOllamaConfigPath(), JSON.stringify({
+        endpoint: normalizeEndpoint(config.endpoint),
+        mode: config.mode,
+        cloudTier: config.cloudTier,
+    }, null, 2), 'utf8');
+}
+function setLocalConnection(endpoint = exports.DEFAULT_LOCAL_ENDPOINT) {
+    const config = {
+        endpoint: normalizeEndpoint(endpoint),
+        mode: 'local',
+        cloudTier: null,
+    };
+    saveConnectionConfig(config);
+    return config;
+}
+function setCloudConnection(apiKey, endpoint = exports.DEFAULT_CLOUD_ENDPOINT) {
+    const token = apiKey.trim();
+    if (!token) {
+        throw new Error('Cloud API key is required');
+    }
+    ensureOllamaConfigDir();
+    const tokenPath = getOllamaTokenPath();
+    (0, node_fs_1.writeFileSync)(tokenPath, token, 'utf8');
+    (0, node_fs_1.chmodSync)(tokenPath, 0o600);
+    const config = {
+        endpoint: normalizeEndpoint(endpoint),
+        mode: 'cloud',
+        cloudTier: null,
+    };
+    saveConnectionConfig(config);
+    return config;
+}
+function clearCloudApiKey() {
+    const tokenPath = getOllamaTokenPath();
+    if ((0, node_fs_1.existsSync)(tokenPath)) {
+        (0, node_fs_1.rmSync)(tokenPath, { force: true });
+    }
+}
+function getCloudApiKey() {
+    const fromEnv = process.env.OLLAMA_API_KEY?.trim();
+    if (fromEnv) {
+        return fromEnv;
+    }
+    const tokenPath = getOllamaTokenPath();
+    if (!(0, node_fs_1.existsSync)(tokenPath)) {
+        return undefined;
+    }
+    const token = (0, node_fs_1.readFileSync)(tokenPath, 'utf8').trim();
+    return token || undefined;
+}
+function resolveEndpoint(config) {
+    const override = getEndpointOverrideFromEnv();
+    if (override) {
+        return override;
+    }
+    return normalizeEndpoint(config.endpoint);
+}
+function getResolvedConnectionConfig() {
+    const config = loadConnectionConfig();
+    return {
+        ...config,
+        endpoint: resolveEndpoint(config),
+    };
+}
+function createOllamaApiClient(config = getResolvedConnectionConfig()) {
+    const endpoint = resolveEndpoint(config);
+    const apiKey = config.mode === 'cloud' ? getCloudApiKey() : undefined;
+    return new api_client_js_1.OllamaApiClient({
+        endpoint,
+        apiKey,
+    });
+}
+async function testConnection(config = getResolvedConnectionConfig()) {
+    const client = createOllamaApiClient(config);
+    const endpoint = client.getEndpoint();
+    try {
+        if (config.mode === 'local') {
+            const health = await client.healthCheck();
+            if (health.status < 200 || health.status >= 300) {
+                return {
+                    connected: false,
+                    endpoint,
+                    mode: config.mode,
+                    error: `Unexpected health status: ${health.status}`,
+                };
+            }
+            if (!isOllamaHealthResponse(health.body)) {
+                return {
+                    connected: false,
+                    endpoint,
+                    mode: config.mode,
+                    error: 'Unexpected health response: endpoint is not an Ollama instance',
+                };
+            }
+        }
+        else {
+            await client.listInstalledModels();
+        }
+        const [versionResult, psResult] = await Promise.allSettled([
+            client.getVersion(),
+            client.listRunningModels(),
+        ]);
+        const version = versionResult.status === 'fulfilled' ? versionResult.value.version : undefined;
+        const loadedModels = psResult.status === 'fulfilled' ? psResult.value.models.length : undefined;
+        return {
+            connected: true,
+            endpoint,
+            mode: config.mode,
+            version,
+            minimumVersionMet: version ? isVersionSupported(version) : undefined,
+            loadedModels,
+        };
+    }
+    catch (error) {
+        if (config.mode === 'cloud' && error instanceof api_client_js_1.OllamaApiError && error.status === 401) {
+            clearCloudApiKey();
+        }
+        return {
+            connected: false,
+            endpoint,
+            mode: config.mode,
+            error: error instanceof Error ? error.message : String(error),
+        };
+    }
+}
+function isVersionSupported(version, minimumVersion = exports.MINIMUM_OLLAMA_VERSION) {
+    const current = parseVersion(version);
+    const required = parseVersion(minimumVersion);
+    for (let i = 0; i < Math.max(current.length, required.length); i += 1) {
+        const currentPart = current[i] ?? 0;
+        const requiredPart = required[i] ?? 0;
+        if (currentPart > requiredPart) {
+            return true;
+        }
+        if (currentPart < requiredPart) {
+            return false;
+        }
+    }
+    return true;
+}
+function parseVersion(version) {
+    const cleaned = version.trim().replace(/^v/i, '');
+    return cleaned
+        .split('.')
+        .map((part) => parseInt(part.replace(/[^0-9].*$/, ''), 10))
+        .map((part) => (Number.isNaN(part) ? 0 : part));
+}
+function ensureOllamaConfigDir() {
+    const dir = getOllamaConfigDir();
+    if (!(0, node_fs_1.existsSync)(dir)) {
+        (0, node_fs_1.mkdirSync)(dir, { recursive: true });
+    }
+}
+function normalizeEndpoint(endpoint) {
+    return endpoint.trim().replace(/\/+$/, '');
+}
+function isOllamaHealthResponse(body) {
+    return body.toLowerCase().includes('ollama is running');
+}
+function getEndpointOverrideFromEnv() {
+    const hostRaw = process.env.STEROIDS_OLLAMA_HOST?.trim();
+    const portRaw = process.env.STEROIDS_OLLAMA_PORT?.trim();
+    if (!hostRaw && !portRaw) {
+        return undefined;
+    }
+    if (!hostRaw) {
+        return normalizeEndpoint(`http://localhost:${portRaw}`);
+    }
+    if (hostRaw.startsWith('http://') || hostRaw.startsWith('https://')) {
+        const endpoint = new URL(hostRaw);
+        if (portRaw) {
+            endpoint.port = portRaw;
+        }
+        return normalizeEndpoint(endpoint.toString());
+    }
+    const port = portRaw || '11434';
+    return normalizeEndpoint(`http://${hostRaw}:${port}`);
+}
+//# sourceMappingURL=connection.js.map

package/dist/ollama/connection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection.js","sourceRoot":"","sources":["../../src/ollama/connection.ts"],"names":[],"mappings":";;;AA8BA,gDAEC;AAED,kDAEC;AAED,gDAEC;AAED,gEAMC;AAED,oDAkBC;AAED,oDAeC;AAED,gDAQC;AAED,gDAqBC;AAED,4CAKC;AAED,wCAaC;AAED,0CAOC;AAED,kEAMC;AAED,sDAQC;AAED,wCA0DC;AAED,gDAkBC;AAvPD,qCAAgG;AAChG,yCAAiC;AACjC,gFAA0E;AAC1E,mDAAkE;AAErD,QAAA,sBAAsB,GAAG,wBAAwB,CAAC;AAClD,QAAA,sBAAsB,GAAG,oBAAoB,CAAC;AAC9C,QAAA,sBAAsB,GAAG,QAAQ,CAAC;AAoB/C,MAAM,WAAW,GAAG,aAAa,CAAC;AAClC,MAAM,UAAU,GAAG,OAAO,CAAC;AAE3B,SAAgB,kBAAkB;IAChC,OAAO,IAAA,gBAAI,EAAC,IAAA,8CAAoB,GAAE,EAAE,QAAQ,CAAC,CAAC;AAChD,CAAC;AAED,SAAgB,mBAAmB;IACjC,OAAO,IAAA,gBAAI,EAAC,kBAAkB,EAAE,EAAE,WAAW,CAAC,CAAC;AACjD,CAAC;AAED,SAAgB,kBAAkB;IAChC,OAAO,IAAA,gBAAI,EAAC,kBAAkB,EAAE,EAAE,UAAU,CAAC,CAAC;AAChD,CAAC;AAED,SAAgB,0BAA0B;IACxC,OAAO;QACL,QAAQ,EAAE,8BAAsB;QAChC,IAAI,EAAE,OAAO;QACb,SAAS,EAAE,IAAI;KAChB,CAAC;AACJ,CAAC;AAED,SAAgB,oBAAoB;IAClC,MAAM,UAAU,GAAG,mBAAmB,EAAE,CAAC;IACzC,IAAI,CAAC,IAAA,oBAAU,EAAC,UAAU,CAAC,EAAE,CAAC;QAC5B,OAAO,0BAA0B,EAAE,CAAC;IACtC,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAA,sBAAY,EAAC,UAAU,EAAE,MAAM,CAAC,CAAoC,CAAC;QAC/F,MAAM,IAAI,GAAyB,MAAM,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;QAC/E,MAAM,QAAQ,GAAG,iBAAiB,CAAC,MAAM,CAAC,QAAQ,IAAI,0BAA0B,EAAE,CAAC,QAAQ,CAAC,CAAC;QAC7F,OAAO;YACL,QAAQ;YACR,IAAI;YACJ,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI;SACpC,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,0BAA0B,EAAE,CAAC;IACtC,CAAC;AACH,CAAC;AAED,SAAgB,oBAAoB,CAAC,MAA8B;IACjE,qBAAqB,EAAE,CAAC;IACxB,IAAA,uBAAa,EACX,mBAAmB,EAAE,EACrB,IAAI,CAAC,SAAS,CACZ;QACE,QAAQ,EAAE,iBAAiB,CAAC,MAAM,CAAC,QAAQ,CAAC;QAC5C,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,SAAS,EAAE,MAAM,CAAC,SAAS;KAC5B,EACD,IAAI,EACJ,CAAC,CACF,EACD,MAAM,CACP,CAAC;AACJ,CAAC;AAED,SAAgB,kBAAkB,CAAC,WAAmB,8BAAsB;IAC1E,MAAM,MAAM,GAA2B;QACrC,QAAQ,EAAE,iBAAiB,CAAC,QAAQ,CAAC;QACrC,IAAI,EAAE,OAAO;QACb,SAAS,EAAE,IAAI;KAChB,CAAC;IACF,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAC7B,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAgB,kBAAkB,CAChC,MAAc,EACd,WAAmB,8BAAsB;IAEzC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;IAC5B,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC/C,CAAC;IAED,qBAAqB,EAAE,CAAC;IACxB,MAAM,SAAS,GAAG,kBAAkB,EAAE,CAAC;IACvC,IAAA,uBAAa,EAAC,SAAS,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IACxC,IAAA,mBAAS,EAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAE5B,MAAM,MAAM,GAA2B;QACrC,QAAQ,EAAE,iBAAiB,CAAC,QAAQ,CAAC;QACrC,IAAI,EAAE,OAAO;QACb,SAAS,EAAE,IAAI;KAChB,CAAC;IACF,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAC7B,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAgB,gBAAgB;IAC9B,MAAM,SAAS,GAAG,kBAAkB,EAAE,CAAC;IACvC,IAAI,IAAA,oBAAU,EAAC,SAAS,CAAC,EAAE,CAAC;QAC1B,IAAA,gBAAM,EAAC,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACrC,CAAC;AACH,CAAC;AAED,SAAgB,cAAc;IAC5B,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,EAAE,CAAC;IACnD,IAAI,OAAO,EAAE,CAAC;QACZ,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,SAAS,GAAG,kBAAkB,EAAE,CAAC;IACvC,IAAI,CAAC,IAAA,oBAAU,EAAC,SAAS,CAAC,EAAE,CAAC;QAC3B,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,KAAK,GAAG,IAAA,sBAAY,EAAC,SAAS,EAAE,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;IACrD,OAAO,KAAK,IAAI,SAAS,CAAC;AAC5B,CAAC;AAED,SAAgB,eAAe,CAAC,MAA8B;IAC5D,MAAM,QAAQ,GAAG,0BAA0B,EAAE,CAAC;IAC9C,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,OAAO,iBAAiB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;AAC5C,CAAC;AAED,SAAgB,2BAA2B;IACzC,MAAM,MAAM,GAAG,oBAAoB,EAAE,CAAC;IACtC,OAAO;QACL,GAAG,MAAM;QACT,QAAQ,EAAE,eAAe,CAAC,MAAM,CAAC;KAClC,CAAC;AACJ,CAAC;AAED,SAAgB,qBAAqB,CAAC,SAAiC,2BAA2B,EAAE;IAClG,MAAM,QAAQ,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IACzC,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAEtE,OAAO,IAAI,+BAAe,CAAC;QACzB,QAAQ;QACR,MAAM;KACP,CAAC,CAAC;AACL,CAAC;AAEM,KAAK,UAAU,cAAc,CAClC,SAAiC,2BAA2B,EAAE;IAE9D,MAAM,MAAM,GAAG,qBAAqB,CAAC,MAAM,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;IAEtC,IAAI,CAAC;QACH,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC5B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;YAC1C,IAAI,MAAM,CAAC,MAAM,GAAG,GAAG,IAAI,MAAM,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;gBAChD,OAAO;oBACL,SAAS,EAAE,KAAK;oBAChB,QAAQ;oBACR,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,KAAK,EAAE,6BAA6B,MAAM,CAAC,MAAM,EAAE;iBACpD,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzC,OAAO;oBACL,SAAS,EAAE,KAAK;oBAChB,QAAQ;oBACR,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,KAAK,EAAE,gEAAgE;iBACxE,CAAC;YACJ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,MAAM,CAAC,mBAAmB,EAAE,CAAC;QACrC,CAAC;QAED,MAAM,CAAC,aAAa,EAAE,QAAQ,CAAC,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC;YACzD,MAAM,CAAC,UAAU,EAAE;YACnB,MAAM,CAAC,iBAAiB,EAAE;SAC3B,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;QAC/F,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;QAEhG,OAAO;YACL,SAAS,EAAE,IAAI;YACf,QAAQ;YACR,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,OAAO;YACP,iBAAiB,EAAE,OAAO,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;YACpE,YAAY;SACb,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,YAAY,8BAAc,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACvF,gBAAgB,EAAE,CAAC;QACrB,CAAC;QAED,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,QAAQ;YACR,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC9D,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAgB,kBAAkB,CAAC,OAAe,EAAE,iBAAyB,8BAAsB;IACjG,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,QAAQ,GAAG,YAAY,CAAC,cAAc,CAAC,CAAC;IAE9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACtE,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,YAAY,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEtC,IAAI,WAAW,GAAG,YAAY,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,WAAW,GAAG,YAAY,EAAE,CAAC;YAC/B,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,YAAY,CAAC,OAAe;IACnC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAClD,OAAO,OAAO;SACX,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;SAC1D,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,qBAAqB;IAC5B,MAAM,GAAG,GAAG,kBAAkB,EAAE,CAAC;IACjC,IAAI,CAAC,IAAA,oBAAU,EAAC,GAAG,CAAC,EAAE,CAAC;QACrB,IAAA,mBAAS,EAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACtC,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,QAAgB;IACzC,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AAC7C,CAAC;AAED,SAAS,sBAAsB,CAAC,IAAY;IAC1C,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,0BAA0B;IACjC,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,IAAI,EAAE,CAAC;IACzD,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,IAAI,EAAE,CAAC;IAEzD,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;QACzB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,iBAAiB,CAAC,oBAAoB,OAAO,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QACpE,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;QAClC,IAAI,OAAO,EAAE,CAAC;YACZ,QAAQ,CAAC,IAAI,GAAG,OAAO,CAAC;QAC1B,CAAC;QACD,OAAO,iBAAiB,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,IAAI,OAAO,CAAC;IAChC,OAAO,iBAAiB,CAAC,UAAU,OAAO,IAAI,IAAI,EAAE,CAAC,CAAC;AACxD,CAAC"}

package/dist/ollama/metrics.d.ts

@@ -0,0 +1,32 @@
+interface OllamaMetricsChunk {
+    total_duration?: number;
+    load_duration?: number;
+    prompt_eval_duration?: number;
+    eval_duration?: number;
+    eval_count?: number;
+}
+export interface OllamaTokenMetrics {
+    endpoint: string;
+    totalDurationNs?: number;
+    loadDurationNs?: number;
+    promptEvalDurationNs?: number;
+    evalDurationNs?: number;
+    tokensPerSecond?: number;
+}
+interface RecordOllamaUsageInput {
+    model: string;
+    endpoint: string;
+    role: 'orchestrator' | 'coder' | 'reviewer';
+    promptTokens?: number;
+    completionTokens?: number;
+    totalDurationNs?: number;
+    loadDurationNs?: number;
+    promptEvalDurationNs?: number;
+    evalDurationNs?: number;
+    tokensPerSecond?: number;
+}
+export declare function computeTokensPerSecond(outputTokens?: number, evalDurationNs?: number): number | undefined;
+export declare function buildOllamaTokenMetrics(chunk: OllamaMetricsChunk | undefined, endpoint: string): OllamaTokenMetrics | undefined;
+export declare function recordOllamaUsage(input: RecordOllamaUsageInput): boolean;
+export {};
+//# sourceMappingURL=metrics.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"metrics.d.ts","sourceRoot":"","sources":["../../src/ollama/metrics.ts"],"names":[],"mappings":"AAEA,UAAU,kBAAkB;IAC1B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,UAAU,sBAAsB;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,cAAc,GAAG,OAAO,GAAG,UAAU,CAAC;IAC5C,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,wBAAgB,sBAAsB,CACpC,YAAY,CAAC,EAAE,MAAM,EACrB,cAAc,CAAC,EAAE,MAAM,GACtB,MAAM,GAAG,SAAS,CAKpB;AAED,wBAAgB,uBAAuB,CACrC,KAAK,EAAE,kBAAkB,GAAG,SAAS,EACrC,QAAQ,EAAE,MAAM,GACf,kBAAkB,GAAG,SAAS,CAmBhC;AAED,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,sBAAsB,GAAG,OAAO,CAyCxE"}

package/dist/ollama/metrics.js

@@ -0,0 +1,70 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.computeTokensPerSecond = computeTokensPerSecond;
+exports.buildOllamaTokenMetrics = buildOllamaTokenMetrics;
+exports.recordOllamaUsage = recordOllamaUsage;
+const global_db_connection_js_1 = require("../runners/global-db-connection.js");
+function computeTokensPerSecond(outputTokens, evalDurationNs) {
+    if (!isFinitePositiveInt(outputTokens) || !isFinitePositiveInt(evalDurationNs)) {
+        return undefined;
+    }
+    return (outputTokens / evalDurationNs) * 1_000_000_000;
+}
+function buildOllamaTokenMetrics(chunk, endpoint) {
+    if (!endpoint) {
+        return undefined;
+    }
+    const totalDurationNs = sanitizeNullableInt(chunk?.total_duration);
+    const loadDurationNs = sanitizeNullableInt(chunk?.load_duration);
+    const promptEvalDurationNs = sanitizeNullableInt(chunk?.prompt_eval_duration);
+    const evalDurationNs = sanitizeNullableInt(chunk?.eval_duration);
+    const evalCount = sanitizeNullableInt(chunk?.eval_count);
+    return {
+        endpoint,
+        totalDurationNs: totalDurationNs ?? undefined,
+        loadDurationNs: loadDurationNs ?? undefined,
+        promptEvalDurationNs: promptEvalDurationNs ?? undefined,
+        evalDurationNs: evalDurationNs ?? undefined,
+        tokensPerSecond: computeTokensPerSecond(evalCount ?? undefined, evalDurationNs ?? undefined),
+    };
+}
+function recordOllamaUsage(input) {
+    if (!input.endpoint) {
+        return false;
+    }
+    const promptTokens = sanitizeNullableInt(input.promptTokens);
+    const completionTokens = sanitizeNullableInt(input.completionTokens);
+    const totalDurationNs = sanitizeNullableInt(input.totalDurationNs);
+    const loadDurationNs = sanitizeNullableInt(input.loadDurationNs);
+    const promptEvalDurationNs = sanitizeNullableInt(input.promptEvalDurationNs);
+    const evalDurationNs = sanitizeNullableInt(input.evalDurationNs);
+    const tokensPerSecond = sanitizeNullableFloat(input.tokensPerSecond ?? computeTokensPerSecond(completionTokens ?? undefined, evalDurationNs ?? undefined));
+    const createdAt = Date.now();
+    const { db, close } = (0, global_db_connection_js_1.openGlobalDatabase)();
+    try {
+        db.prepare(`INSERT INTO ollama_usage (
+         model, endpoint, role, prompt_tokens, completion_tokens, total_duration_ns,
+         load_duration_ns, prompt_eval_duration_ns, eval_duration_ns, tokens_per_second, created_at
+       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`).run(input.model, input.endpoint, input.role, promptTokens, completionTokens, totalDurationNs, loadDurationNs, promptEvalDurationNs, evalDurationNs, tokensPerSecond, createdAt);
+        return true;
+    }
+    finally {
+        close();
+    }
+}
+function isFinitePositiveInt(value) {
+    return typeof value === 'number' && Number.isFinite(value) && value > 0;
+}
+function sanitizeNullableInt(value) {
+    if (!isFinitePositiveInt(value) && value !== 0) {
+        return null;
+    }
+    return Number(value);
+}
+function sanitizeNullableFloat(value) {
+    if (typeof value !== 'number' || !Number.isFinite(value) || value <= 0) {
+        return null;
+    }
+    return value;
+}
+//# sourceMappingURL=metrics.js.map

package/dist/ollama/metrics.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"metrics.js","sourceRoot":"","sources":["../../src/ollama/metrics.ts"],"names":[],"mappings":";;AAgCA,wDAQC;AAED,0DAsBC;AAED,8CAyCC;AA3GD,gFAAwE;AAgCxE,SAAgB,sBAAsB,CACpC,YAAqB,EACrB,cAAuB;IAEvB,IAAI,CAAC,mBAAmB,CAAC,YAAY,CAAC,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,EAAE,CAAC;QAC/E,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,CAAC,YAAY,GAAG,cAAc,CAAC,GAAG,aAAa,CAAC;AACzD,CAAC;AAED,SAAgB,uBAAuB,CACrC,KAAqC,EACrC,QAAgB;IAEhB,IAAI,CAAC,QAAQ,EAAE,CAAC;QACd,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,eAAe,GAAG,mBAAmB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;IACnE,MAAM,cAAc,GAAG,mBAAmB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IACjE,MAAM,oBAAoB,GAAG,mBAAmB,CAAC,KAAK,EAAE,oBAAoB,CAAC,CAAC;IAC9E,MAAM,cAAc,GAAG,mBAAmB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IACjE,MAAM,SAAS,GAAG,mBAAmB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAEzD,OAAO;QACL,QAAQ;QACR,eAAe,EAAE,eAAe,IAAI,SAAS;QAC7C,cAAc,EAAE,cAAc,IAAI,SAAS;QAC3C,oBAAoB,EAAE,oBAAoB,IAAI,SAAS;QACvD,cAAc,EAAE,cAAc,IAAI,SAAS;QAC3C,eAAe,EAAE,sBAAsB,CAAC,SAAS,IAAI,SAAS,EAAE,cAAc,IAAI,SAAS,CAAC;KAC7F,CAAC;AACJ,CAAC;AAED,SAAgB,iBAAiB,CAAC,KAA6B;IAC7D,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QACpB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,YAAY,GAAG,mBAAmB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC7D,MAAM,gBAAgB,GAAG,mBAAmB,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;IACrE,MAAM,eAAe,GAAG,mBAAmB,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;IACnE,MAAM,cAAc,GAAG,mBAAmB,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IACjE,MAAM,oBAAoB,GAAG,mBAAmB,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;IAC7E,MAAM,cAAc,GAAG,mBAAmB,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IACjE,MAAM,eAAe,GAAG,qBAAqB,CAC3C,KAAK,CAAC,eAAe,IAAI,sBAAsB,CAAC,gBAAgB,IAAI,SAAS,EAAE,cAAc,IAAI,SAAS,CAAC,CAC5G,CAAC;IAEF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,IAAA,4CAAkB,GAAE,CAAC;IAC3C,IAAI,CAAC;QACH,EAAE,CAAC,OAAO,CACR;;;kDAG4C,CAC7C,CAAC,GAAG,CACH,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,QAAQ,EACd,KAAK,CAAC,IAAI,EACV,YAAY,EACZ,gBAAgB,EAChB,eAAe,EACf,cAAc,EACd,oBAAoB,EACpB,cAAc,EACd,eAAe,EACf,SAAS,CACV,CAAC;QAEF,OAAO,IAAI,CAAC;IACd,CAAC;YAAS,CAAC;QACT,KAAK,EAAE,CAAC;IACV,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAc;IACzC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC;AAC1E,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAc;IACzC,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;QAC/C,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAc;IAC3C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QACvE,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}