codesearch-mcp 0.1.0-alpha → 0.2.0-alpha

package/dist/src/services/OllamaEmbedder.js

@@ -1,111 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.OllamaEmbedder = void 0;
-const ollama_1 = require("ollama");
-const LoggerService_1 = require("./LoggerService");
-const reranker_1 = require("./reranker");
-const constants_1 = require("../constants");
-class OllamaEmbedder {
-    config;
-    logger;
-    dimensions;
-    client;
-    rerank;
-    constructor(config, logger = LoggerService_1.LoggerService.silent(), deps = {}) {
-        this.config = config;
-        this.logger = logger;
-        this.client = new ollama_1.Ollama({ host: config.embedding.baseURL });
-        this.dimensions = config.embedding.dimensions ?? (0, constants_1.registryDimension)(config.embedding.model) ?? 0;
-        this.rerank = (0, reranker_1.makeReranker)(config, logger, deps);
-    }
-    // Ollama has no query/document asymmetry — inputType is accepted and ignored.
-    async embed(text, _inputType) {
-        const { embedding } = await this.client.embeddings({
-            model: this.config.embedding.model,
-            prompt: text,
-        });
-        return embedding;
-    }
-    async embedBatch(texts, _inputType) {
-        if (texts.length === 0)
-            return [];
-        try {
-            // /api/embed accepts an array input and returns one vector per input.
-            const res = await this.client.embed({ model: this.config.embedding.model, input: texts });
-            if (res.embeddings?.length === texts.length)
-                return res.embeddings;
-            this.logger.warn('OllamaEmbedder: batch embed count mismatch, falling back to sequential', {
-                expected: texts.length, got: res.embeddings?.length ?? 0,
-            });
-        }
-        catch (err) {
-            this.logger.warn('OllamaEmbedder: batch embed failed, falling back to sequential', {
-                reason: err instanceof Error ? err.message : String(err),
-            });
-        }
-        // Sequential fallback — the contract (one vector per input, in order) holds.
-        const out = [];
-        for (const t of texts)
-            out.push(await this.embed(t));
-        return out;
-    }
-    async listModels() {
-        const res = await this.client.list();
-        return res.models.map(m => m.name).filter(Boolean).sort();
-    }
-    /**
-     * Pull the configured model if it is not already present locally, so a plain
-     * `rag-index` works after switching to (or back to) a local model that was
-     * never pulled — `rag-setup` is no longer the only place that fetches models.
-     * A tagless model name (`bge-m3`) matches its `:latest`/tagged install.
-     * Streams pull progress to the console: a live single-line bar on a TTY, or
-     * de-duplicated status lines otherwise.
-     */
-    async ensureModelAvailable() {
-        const model = this.config.embedding.model;
-        try {
-            const res = await this.client.list();
-            const present = res.models.some(m => m.name === model || m.name.startsWith(`${model}:`));
-            if (present)
-                return;
-        }
-        catch (err) {
-            // Can't list (Ollama unreachable?) — try the pull; its error is clearer.
-            this.logger.warn('OllamaEmbedder: could not list models before pull', {
-                reason: err instanceof Error ? err.message : String(err),
-            });
-        }
-        console.log(`Pulling Ollama model "${model}" (not found locally)...`);
-        this.logger.info('OllamaEmbedder: pulling model', { model });
-        const stream = await this.client.pull({ model, stream: true });
-        const isTty = Boolean(process.stdout.isTTY);
-        let lastLen = 0;
-        let lastLog = '';
-        for await (const part of stream) {
-            const out = '  ' + OllamaEmbedder.formatPullProgress(part);
-            if (isTty) {
-                process.stdout.write('\r' + out.padEnd(lastLen));
-                lastLen = out.length;
-            }
-            else if (out !== lastLog) {
-                console.log(out);
-                lastLog = out;
-            }
-        }
-        if (isTty && lastLen > 0)
-            process.stdout.write('\r' + ' '.repeat(lastLen) + '\r');
-        console.log(`  ✓ ${model} ready.`);
-    }
-    static formatPullProgress(p) {
-        const status = p.status ?? 'pulling';
-        if (p.total && p.total > 0) {
-            const pct = Math.min(100, Math.floor(((p.completed ?? 0) / p.total) * 100));
-            const width = 24;
-            const filled = Math.round((pct / 100) * width);
-            const bar = '[' + '█'.repeat(filled) + '░'.repeat(width - filled) + ']';
-            return `${status} ${bar} ${pct}% (${((p.completed ?? 0) / 1e6).toFixed(1)}/${(p.total / 1e6).toFixed(1)} MB)`;
-        }
-        return status;
-    }
-}
-exports.OllamaEmbedder = OllamaEmbedder;