tea-rags 0.9.0

package/build/bootstrap/ollama.js

@@ -0,0 +1,56 @@
+export async function checkOllamaAvailability(embeddingProvider, baseUrl, modelName) {
+    if (embeddingProvider !== "ollama")
+        return;
+    const url = baseUrl || "http://localhost:11434";
+    const model = modelName || "jina-embeddings-v2-base-code";
+    const isLocalhost = url.includes("localhost") || url.includes("127.0.0.1");
+    try {
+        const response = await fetch(`${url}/api/version`);
+        if (!response.ok) {
+            throw new Error(`Ollama returned status ${response.status}`);
+        }
+        const tagsResponse = await fetch(`${url}/api/tags`);
+        const { models } = (await tagsResponse.json());
+        const modelExists = models.some((m) => m.name === model || m.name.startsWith(`${model}:`));
+        if (!modelExists) {
+            let msg = `Error: Model '${model}' not found in Ollama.\n`;
+            if (isLocalhost) {
+                msg +=
+                    `Pull it with:\n` +
+                        `  - Using Podman: podman exec ollama ollama pull ${model}\n` +
+                        `  - Using Docker: docker exec ollama ollama pull ${model}\n` +
+                        `  - Or locally: ollama pull ${model}`;
+            }
+            else {
+                msg += `Please ensure the model is available on your Ollama instance:\n  ollama pull ${model}`;
+            }
+            throw new Error(msg);
+        }
+    }
+    catch (error) {
+        // Re-throw "model not found" errors as-is
+        if (error instanceof Error && error.message.startsWith("Error: Model")) {
+            throw error;
+        }
+        const errorMessage = error instanceof Error ? `Error: ${error.message}` : `Error: Ollama is not running at ${url}.\n`;
+        let helpText = "";
+        if (isLocalhost) {
+            helpText =
+                `Please start Ollama:\n` +
+                    `  - Using Podman: podman compose up -d\n` +
+                    `  - Using Docker: docker compose up -d\n` +
+                    `  - Or install locally: curl -fsSL https://ollama.ai/install.sh | sh\n` +
+                    `\nThen pull the embedding model:\n` +
+                    `  ollama pull jina-embeddings-v2-base-code`;
+        }
+        else {
+            helpText =
+                `Please ensure:\n` +
+                    `  - Ollama is running at the specified URL\n` +
+                    `  - The URL is accessible from this machine\n` +
+                    `  - The embedding model is available (e.g., jina-embeddings-v2-base-code)`;
+        }
+        throw new Error(`${errorMessage}\n${helpText}`);
+    }
+}
+//# sourceMappingURL=ollama.js.map

package/build/bootstrap/ollama.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/bootstrap/ollama.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAC3C,iBAAyB,EACzB,OAAgB,EAChB,SAAkB;IAElB,IAAI,iBAAiB,KAAK,QAAQ;QAAE,OAAO;IAE3C,MAAM,GAAG,GAAG,OAAO,IAAI,wBAAwB,CAAC;IAChD,MAAM,KAAK,GAAG,SAAS,IAAI,8BAA8B,CAAC;IAC1D,MAAM,WAAW,GAAG,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IAE3E,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,GAAG,cAAc,CAAC,CAAC;QACnD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/D,CAAC;QAED,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,GAAG,GAAG,WAAW,CAAC,CAAC;QACpD,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,YAAY,CAAC,IAAI,EAAE,CAAmC,CAAC;QACjF,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC;QAE3F,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,IAAI,GAAG,GAAG,iBAAiB,KAAK,0BAA0B,CAAC;YAC3D,IAAI,WAAW,EAAE,CAAC;gBAChB,GAAG;oBACD,iBAAiB;wBACjB,oDAAoD,KAAK,IAAI;wBAC7D,oDAAoD,KAAK,IAAI;wBAC7D,+BAA+B,KAAK,EAAE,CAAC;YAC3C,CAAC;iBAAM,CAAC;gBACN,GAAG,IAAI,gFAAgF,KAAK,EAAE,CAAC;YACjG,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,0CAA0C;QAC1C,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;YACvE,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,UAAU,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,mCAAmC,GAAG,KAAK,CAAC;QAEnG,IAAI,QAAQ,GAAG,EAAE,CAAC;QAClB,IAAI,WAAW,EAAE,CAAC;YAChB,QAAQ;gBACN,wBAAwB;oBACxB,0CAA0C;oBAC1C,0CAA0C;oBAC1C,wEAAwE;oBACxE,oCAAoC;oBACpC,4CAA4C,CAAC;QACjD,CAAC;aAAM,CAAC;YACN,QAAQ;gBACN,kBAAkB;oBAClB,8CAA8C;oBAC9C,+CAA+C;oBAC/C,2EAA2E,CAAC;QAChF,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,GAAG,YAAY,KAAK,QAAQ,EAAE,CAAC,CAAC;IAClD,CAAC;AACH,CAAC"}

package/build/bootstrap/transport/http.d.ts

@@ -0,0 +1,10 @@
+import type { PromptsConfig } from "../../mcp/prompts/index.js";
+import type { AppConfig } from "../config/index.js";
+import { type AppContext } from "../factory.js";
+export interface HttpServerDeps {
+    config: AppConfig;
+    ctx: AppContext;
+    promptsConfig: PromptsConfig | null;
+}
+export declare function startHttpServer(deps: HttpServerDeps): Promise<void>;
+//# sourceMappingURL=http.d.ts.map

package/build/bootstrap/transport/http.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.d.ts","sourceRoot":"","sources":["../../../src/bootstrap/transport/http.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAChE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAA+B,KAAK,UAAU,EAAE,MAAM,eAAe,CAAC;AAE7E,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,SAAS,CAAC;IAClB,GAAG,EAAE,UAAU,CAAC;IAChB,aAAa,EAAE,aAAa,GAAG,IAAI,CAAC;CACrC;AAED,wBAAsB,eAAe,CAAC,IAAI,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAuLzE"}

package/build/bootstrap/transport/http.js

@@ -0,0 +1,165 @@
+// src/bootstrap/transport/http.ts
+import { StreamableHTTPServerTransport } from "@modelcontextprotocol/sdk/server/streamableHttp.js";
+import Bottleneck from "bottleneck";
+import express from "express";
+import { createConfiguredServer, pkg } from "../factory.js";
+export async function startHttpServer(deps) {
+    const { config, ctx, promptsConfig } = deps;
+    // Constants for HTTP server configuration
+    const RATE_LIMIT_MAX_REQUESTS = 100;
+    const RATE_LIMIT_WINDOW_MS = 15 * 60 * 1000; // 15 minutes
+    const RATE_LIMIT_MAX_CONCURRENT = 10;
+    const RATE_LIMITER_CLEANUP_INTERVAL_MS = 60 * 60 * 1000; // 1 hour
+    const SHUTDOWN_GRACE_PERIOD_MS = 10 * 1000; // 10 seconds
+    const app = express();
+    app.use(express.json({ limit: "10mb" }));
+    // Configure Express to trust proxy for correct IP detection
+    app.set("trust proxy", true);
+    // Rate limiter group: max 100 requests per 15 minutes per IP, max 10 concurrent per IP
+    const rateLimiterGroup = new Bottleneck.Group({
+        reservoir: RATE_LIMIT_MAX_REQUESTS,
+        reservoirRefreshAmount: RATE_LIMIT_MAX_REQUESTS,
+        reservoirRefreshInterval: RATE_LIMIT_WINDOW_MS,
+        maxConcurrent: RATE_LIMIT_MAX_CONCURRENT,
+    });
+    // Helper function to send JSON-RPC error responses
+    const sendErrorResponse = (res, code, message, httpStatus = 500) => {
+        if (!res.headersSent) {
+            res.status(httpStatus).json({
+                jsonrpc: "2.0",
+                error: { code, message },
+                id: null,
+            });
+        }
+    };
+    // Periodic cleanup of inactive rate limiters to prevent memory leaks
+    const ipLastAccess = new Map();
+    const cleanupIntervalId = setInterval(() => {
+        const now = Date.now();
+        const keysToDelete = [];
+        ipLastAccess.forEach((lastAccess, ip) => {
+            if (now - lastAccess > RATE_LIMITER_CLEANUP_INTERVAL_MS) {
+                keysToDelete.push(ip);
+            }
+        });
+        keysToDelete.forEach((ip) => {
+            void rateLimiterGroup.deleteKey(ip);
+            ipLastAccess.delete(ip);
+        });
+        if (keysToDelete.length > 0) {
+            console.error(`Cleaned up ${keysToDelete.length} inactive rate limiters`);
+        }
+    }, RATE_LIMITER_CLEANUP_INTERVAL_MS);
+    // Rate limiting middleware
+    const rateLimitMiddleware = async (req, res, next) => {
+        const clientIp = req.ip || req.socket.remoteAddress || "unknown";
+        try {
+            // Update last access time for this IP
+            ipLastAccess.set(clientIp, Date.now());
+            // Get or create a limiter for this specific IP
+            const limiter = rateLimiterGroup.key(clientIp);
+            await limiter.schedule(async () => Promise.resolve());
+            next();
+        }
+        catch (error) {
+            // Differentiate between rate limit errors and unexpected errors
+            if (error instanceof Bottleneck.BottleneckError) {
+                console.error(`Rate limit exceeded for IP ${clientIp}:`, error.message);
+            }
+            else {
+                console.error("Unexpected rate limiting error:", error);
+            }
+            sendErrorResponse(res, -32000, "Too many requests", 429);
+        }
+    };
+    // Health check endpoint
+    app.get("/health", (_req, res) => {
+        res.json({
+            status: "ok",
+            mode: config.transportMode,
+            version: pkg.version,
+            embedding_provider: config.embeddingProvider,
+        });
+    });
+    app.post("/mcp", rateLimitMiddleware, async (req, res) => {
+        // Create a new server for each request
+        const requestServer = createConfiguredServer(ctx, promptsConfig);
+        // Create transport with enableJsonResponse
+        const transport = new StreamableHTTPServerTransport({
+            sessionIdGenerator: undefined,
+            enableJsonResponse: true,
+        });
+        // Track cleanup state to prevent double cleanup
+        let cleanedUp = false;
+        const cleanup = async () => {
+            if (cleanedUp)
+                return;
+            cleanedUp = true;
+            await transport.close().catch(() => { });
+            await requestServer.close().catch(() => { });
+        };
+        // Set a timeout for the request to prevent hanging
+        const timeoutId = setTimeout(() => {
+            sendErrorResponse(res, -32000, "Request timeout", 504);
+            cleanup().catch((err) => {
+                console.error("Error during timeout cleanup:", err);
+            });
+        }, config.requestTimeoutMs);
+        try {
+            // Connect server to transport
+            await requestServer.connect(transport);
+            // Handle the request
+            await transport.handleRequest(req, res, req.body);
+            // Clean up AFTER the response finishes
+            const cleanupHandler = () => {
+                clearTimeout(timeoutId);
+                cleanup().catch((err) => {
+                    console.error("Error during response cleanup:", err);
+                });
+            };
+            res.on("finish", cleanupHandler);
+            res.on("close", cleanupHandler);
+            res.on("error", (err) => {
+                console.error("Response stream error:", err);
+                cleanupHandler();
+            });
+        }
+        catch (error) {
+            clearTimeout(timeoutId);
+            console.error("Error handling MCP request:", error);
+            sendErrorResponse(res, -32603, "Internal server error");
+            await cleanup();
+        }
+    });
+    const httpServer = app
+        .listen(config.httpPort, () => {
+        console.error(`Qdrant MCP server running on http://localhost:${config.httpPort}/mcp`);
+    })
+        .on("error", (error) => {
+        console.error("HTTP server error:", error);
+        process.exit(1);
+    });
+    // Graceful shutdown handling
+    let isShuttingDown = false;
+    const shutdown = () => {
+        if (isShuttingDown)
+            return;
+        isShuttingDown = true;
+        console.error("Shutdown signal received, closing HTTP server gracefully...");
+        // Clear the cleanup interval to allow graceful shutdown
+        clearInterval(cleanupIntervalId);
+        // Force shutdown after grace period
+        const forceTimeout = setTimeout(() => {
+            console.error("Forcing shutdown after timeout");
+            process.exit(1);
+        }, SHUTDOWN_GRACE_PERIOD_MS);
+        httpServer.close(() => {
+            clearTimeout(forceTimeout);
+            console.error("HTTP server closed");
+            process.exit(0);
+        });
+    };
+    process.on("SIGTERM", shutdown);
+    process.on("SIGINT", shutdown);
+}
+//# sourceMappingURL=http.js.map

package/build/bootstrap/transport/http.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.js","sourceRoot":"","sources":["../../../src/bootstrap/transport/http.ts"],"names":[],"mappings":"AAAA,kCAAkC;AAClC,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,UAAU,MAAM,YAAY,CAAC;AACpC,OAAO,OAAO,MAAM,SAAS,CAAC;AAI9B,OAAO,EAAE,sBAAsB,EAAE,GAAG,EAAmB,MAAM,eAAe,CAAC;AAQ7E,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,IAAoB;IACxD,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC;IAE5C,0CAA0C;IAC1C,MAAM,uBAAuB,GAAG,GAAG,CAAC;IACpC,MAAM,oBAAoB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;IAC1D,MAAM,yBAAyB,GAAG,EAAE,CAAC;IACrC,MAAM,gCAAgC,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,SAAS;IAClE,MAAM,wBAAwB,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;IAEzD,MAAM,GAAG,GAAG,OAAO,EAAE,CAAC;IACtB,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;IAEzC,4DAA4D;IAC5D,GAAG,CAAC,GAAG,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;IAE7B,uFAAuF;IACvF,MAAM,gBAAgB,GAAG,IAAI,UAAU,CAAC,KAAK,CAAC;QAC5C,SAAS,EAAE,uBAAuB;QAClC,sBAAsB,EAAE,uBAAuB;QAC/C,wBAAwB,EAAE,oBAAoB;QAC9C,aAAa,EAAE,yBAAyB;KACzC,CAAC,CAAC;IAEH,mDAAmD;IACnD,MAAM,iBAAiB,GAAG,CAAC,GAAqB,EAAE,IAAY,EAAE,OAAe,EAAE,UAAU,GAAG,GAAG,EAAE,EAAE;QACnG,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;YACrB,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC;gBAC1B,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;gBACxB,EAAE,EAAE,IAAI;aACT,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;IAEF,qEAAqE;IACrE,MAAM,YAAY,GAAG,IAAI,GAAG,EAAkB,CAAC;IAE/C,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,YAAY,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,EAAE,EAAE;YACtC,IAAI,GAAG,GAAG,UAAU,GAAG,gCAAgC,EAAE,CAAC;gBACxD,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACxB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,YAAY,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;YAC1B,KAAK,gBAAgB,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YACpC,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,KAAK,CAAC,cAAc,YAAY,CAAC,MAAM,yBAAyB,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC,EAAE,gCAAgC,CAAC,CAAC;IAErC,2BAA2B;IAC3B,MAAM,mBAAmB,GAAG,KAAK,EAAE,GAAoB,EAAE,GAAqB,EAAE,IAA0B,EAAE,EAAE;QAC5G,MAAM,QAAQ,GAAG,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI,SAAS,CAAC;QAEjE,IAAI,CAAC;YACH,sCAAsC;YACtC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;YAEvC,+CAA+C;YAC/C,MAAM,OAAO,GAAG,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC/C,MAAM,OAAO,CAAC,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACtD,IAAI,EAAE,CAAC;QACT,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,gEAAgE;YAChE,IAAI,KAAK,YAAY,UAAU,CAAC,eAAe,EAAE,CAAC;gBAChD,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,GAAG,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAC1E,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;YAC1D,CAAC;YACD,iBAAiB,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,mBAAmB,EAAE,GAAG,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC,CAAC;IAEF,wBAAwB;IACxB,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE;QAC/B,GAAG,CAAC,IAAI,CAAC;YACP,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,MAAM,CAAC,aAAa;YAC1B,OAAO,EAAE,GAAG,CAAC,OAAO;YACpB,kBAAkB,EAAE,MAAM,CAAC,iBAAiB;SAC7C,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,mBAAmB,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QACvD,uCAAuC;QACvC,MAAM,aAAa,GAAG,sBAAsB,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;QAEjE,2CAA2C;QAC3C,MAAM,SAAS,GAAG,IAAI,6BAA6B,CAAC;YAClD,kBAAkB,EAAE,SAAS;YAC7B,kBAAkB,EAAE,IAAI;SACzB,CAAC,CAAC;QAEH,gDAAgD;QAChD,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,MAAM,OAAO,GAAG,KAAK,IAAI,EAAE;YACzB,IAAI,SAAS;gBAAE,OAAO;YACtB,SAAS,GAAG,IAAI,CAAC;YACjB,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACxC,MAAM,aAAa,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC;QAEF,mDAAmD;QACnD,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;YAChC,iBAAiB,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,iBAAiB,EAAE,GAAG,CAAC,CAAC;YACvD,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACtB,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,GAAG,CAAC,CAAC;YACtD,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAE5B,IAAI,CAAC;YACH,8BAA8B;YAC9B,MAAM,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAEvC,qBAAqB;YACrB,MAAM,SAAS,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC;YAElD,uCAAuC;YACvC,MAAM,cAAc,GAAG,GAAG,EAAE;gBAC1B,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;oBACtB,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,GAAG,CAAC,CAAC;gBACvD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;YAEF,GAAG,CAAC,EAAE,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;YACjC,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;YAChC,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACtB,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,GAAG,CAAC,CAAC;gBAC7C,cAAc,EAAE,CAAC;YACnB,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,SAAS,CAAC,CAAC;YACxB,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;YACpD,iBAAiB,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,uBAAuB,CAAC,CAAC;YACxD,MAAM,OAAO,EAAE,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,GAAG;SACnB,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,GAAG,EAAE;QAC5B,OAAO,CAAC,KAAK,CAAC,iDAAiD,MAAM,CAAC,QAAQ,MAAM,CAAC,CAAC;IACxF,CAAC,CAAC;SACD,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;QACrB,OAAO,CAAC,KAAK,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;QAC3C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEL,6BAA6B;IAC7B,IAAI,cAAc,GAAG,KAAK,CAAC;IAE3B,MAAM,QAAQ,GAAG,GAAG,EAAE;QACpB,IAAI,cAAc;YAAE,OAAO;QAC3B,cAAc,GAAG,IAAI,CAAC;QAEtB,OAAO,CAAC,KAAK,CAAC,6DAA6D,CAAC,CAAC;QAE7E,wDAAwD;QACxD,aAAa,CAAC,iBAAiB,CAAC,CAAC;QAEjC,oCAAoC;QACpC,MAAM,YAAY,GAAG,UAAU,CAAC,GAAG,EAAE;YACnC,OAAO,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;YAChD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,EAAE,wBAAwB,CAAC,CAAC;QAE7B,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE;YACpB,YAAY,CAAC,YAAY,CAAC,CAAC;YAC3B,OAAO,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACpC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAChC,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;AACjC,CAAC"}

package/build/bootstrap/transport/stdio.d.ts

@@ -0,0 +1,3 @@
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+export declare function startStdioServer(server: McpServer): Promise<void>;
+//# sourceMappingURL=stdio.d.ts.map

package/build/bootstrap/transport/stdio.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stdio.d.ts","sourceRoot":"","sources":["../../../src/bootstrap/transport/stdio.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAGzE,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAIvE"}

package/build/bootstrap/transport/stdio.js

@@ -0,0 +1,7 @@
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+export async function startStdioServer(server) {
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    console.error("Qdrant MCP server running on stdio");
+}
+//# sourceMappingURL=stdio.js.map

package/build/bootstrap/transport/stdio.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stdio.js","sourceRoot":"","sources":["../../../src/bootstrap/transport/stdio.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAEjF,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,MAAiB;IACtD,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAChC,OAAO,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;AACtD,CAAC"}

package/build/core/adapters/embeddings/base.d.ts

@@ -0,0 +1,23 @@
+export interface EmbeddingResult {
+    embedding: number[];
+    dimensions: number;
+}
+export interface RateLimitConfig {
+    maxRequestsPerMinute?: number;
+    retryAttempts?: number;
+    retryDelayMs?: number;
+}
+export interface EmbeddingProvider {
+    embed: (text: string) => Promise<EmbeddingResult>;
+    embedBatch: (texts: string[]) => Promise<EmbeddingResult[]>;
+    getDimensions: () => number;
+    getModel: () => string;
+}
+export interface ProviderConfig {
+    model?: string;
+    dimensions?: number;
+    rateLimitConfig?: RateLimitConfig;
+    apiKey?: string;
+    baseUrl?: string;
+}
+//# sourceMappingURL=base.d.ts.map

package/build/core/adapters/embeddings/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../../src/core/adapters/embeddings/base.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,eAAe;IAC9B,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,eAAe;IAC9B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,eAAe,CAAC,CAAC;IAClD,UAAU,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;IAC5D,aAAa,EAAE,MAAM,MAAM,CAAC;IAC5B,QAAQ,EAAE,MAAM,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB"}

package/build/core/adapters/embeddings/base.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=base.js.map

package/build/core/adapters/embeddings/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../../src/core/adapters/embeddings/base.ts"],"names":[],"mappings":""}

package/build/core/adapters/embeddings/cohere.d.ts

@@ -0,0 +1,17 @@
+import type { EmbeddingProvider, EmbeddingResult, RateLimitConfig } from "./base.js";
+export declare class CohereEmbeddings implements EmbeddingProvider {
+    private readonly client;
+    private readonly model;
+    private readonly dimensions;
+    private readonly limiter;
+    private readonly retryAttempts;
+    private readonly retryDelayMs;
+    private readonly inputType;
+    constructor(apiKey: string, model?: string, dimensions?: number, rateLimitConfig?: RateLimitConfig, inputType?: "search_document" | "search_query" | "classification" | "clustering");
+    private retryWithBackoff;
+    embed(text: string): Promise<EmbeddingResult>;
+    embedBatch(texts: string[]): Promise<EmbeddingResult[]>;
+    getDimensions(): number;
+    getModel(): string;
+}
+//# sourceMappingURL=cohere.d.ts.map

package/build/core/adapters/embeddings/cohere.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.d.ts","sourceRoot":"","sources":["../../../../src/core/adapters/embeddings/cohere.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AASrF,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAe;IACtC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAa;IACrC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;IACtC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAuE;gBAG/F,MAAM,EAAE,MAAM,EACd,KAAK,SAAuB,EAC5B,UAAU,CAAC,EAAE,MAAM,EACnB,eAAe,CAAC,EAAE,eAAe,EACjC,SAAS,GAAE,iBAAiB,GAAG,cAAc,GAAG,gBAAgB,GAAG,YAAgC;YAsBvF,gBAAgB;IA+BxB,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAwB7C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAwB7D,aAAa,IAAI,MAAM;IAIvB,QAAQ,IAAI,MAAM;CAGnB"}

package/build/core/adapters/embeddings/cohere.js

@@ -0,0 +1,96 @@
+import Bottleneck from "bottleneck";
+import { CohereClient } from "cohere-ai";
+import { getModelDimensions } from "./utils/model-dimensions.js";
+export class CohereEmbeddings {
+    client;
+    model;
+    dimensions;
+    limiter;
+    retryAttempts;
+    retryDelayMs;
+    inputType;
+    constructor(apiKey, model = "embed-english-v3.0", dimensions, rateLimitConfig, inputType = "search_document") {
+        this.client = new CohereClient({ token: apiKey });
+        this.model = model;
+        this.inputType = inputType;
+        this.dimensions = dimensions || getModelDimensions(model) || 1024;
+        // Rate limiting configuration
+        const maxRequestsPerMinute = rateLimitConfig?.maxRequestsPerMinute || 100;
+        this.retryAttempts = rateLimitConfig?.retryAttempts || 3;
+        this.retryDelayMs = rateLimitConfig?.retryDelayMs || 1000;
+        this.limiter = new Bottleneck({
+            reservoir: maxRequestsPerMinute,
+            reservoirRefreshAmount: maxRequestsPerMinute,
+            reservoirRefreshInterval: 60 * 1000,
+            maxConcurrent: 5,
+            minTime: Math.floor((60 * 1000) / maxRequestsPerMinute),
+        });
+    }
+    async retryWithBackoff(fn, attempt = 0) {
+        try {
+            return await fn();
+        }
+        catch (error) {
+            const apiError = error;
+            const isRateLimitError = apiError?.status === 429 ||
+                apiError?.statusCode === 429 ||
+                apiError?.message?.toLowerCase().includes("rate limit");
+            if (isRateLimitError && attempt < this.retryAttempts) {
+                const delayMs = this.retryDelayMs * Math.pow(2, attempt);
+                const waitTimeSeconds = (delayMs / 1000).toFixed(1);
+                console.error(`Rate limit reached. Retrying in ${waitTimeSeconds}s (attempt ${attempt + 1}/${this.retryAttempts})...`);
+                await new Promise((resolve) => setTimeout(resolve, delayMs));
+                return this.retryWithBackoff(fn, attempt + 1);
+            }
+            if (isRateLimitError) {
+                throw new Error(`Cohere API rate limit exceeded after ${this.retryAttempts} retry attempts. Please try again later or reduce request frequency.`);
+            }
+            throw error;
+        }
+    }
+    async embed(text) {
+        return this.limiter.schedule(async () => this.retryWithBackoff(async () => {
+            const response = await this.client.embed({
+                texts: [text],
+                model: this.model,
+                inputType: this.inputType,
+                embeddingTypes: ["float"],
+            });
+            // Cohere v7+ returns embeddings as number[][]
+            const embeddings = response.embeddings;
+            if (!embeddings || embeddings.length === 0) {
+                throw new Error("No embedding returned from Cohere API");
+            }
+            return {
+                embedding: embeddings[0],
+                dimensions: this.dimensions,
+            };
+        }));
+    }
+    async embedBatch(texts) {
+        return this.limiter.schedule(async () => this.retryWithBackoff(async () => {
+            const response = await this.client.embed({
+                texts,
+                model: this.model,
+                inputType: this.inputType,
+                embeddingTypes: ["float"],
+            });
+            // Cohere v7+ returns embeddings as number[][]
+            const embeddings = response.embeddings;
+            if (!embeddings) {
+                throw new Error("No embeddings returned from Cohere API");
+            }
+            return embeddings.map((embedding) => ({
+                embedding,
+                dimensions: this.dimensions,
+            }));
+        }));
+    }
+    getDimensions() {
+        return this.dimensions;
+    }
+    getModel() {
+        return this.model;
+    }
+}
+//# sourceMappingURL=cohere.js.map

package/build/core/adapters/embeddings/cohere.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.js","sourceRoot":"","sources":["../../../../src/core/adapters/embeddings/cohere.ts"],"names":[],"mappings":"AAAA,OAAO,UAAU,MAAM,YAAY,CAAC;AACpC,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAGzC,OAAO,EAAE,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAQjE,MAAM,OAAO,gBAAgB;IACV,MAAM,CAAe;IACrB,KAAK,CAAS;IACd,UAAU,CAAS;IACnB,OAAO,CAAa;IACpB,aAAa,CAAS;IACtB,YAAY,CAAS;IACrB,SAAS,CAAuE;IAEjG,YACE,MAAc,EACd,KAAK,GAAG,oBAAoB,EAC5B,UAAmB,EACnB,eAAiC,EACjC,YAAkF,iBAAiB;QAEnG,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAClD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,IAAI,CAAC,UAAU,GAAG,UAAU,IAAI,kBAAkB,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;QAElE,8BAA8B;QAC9B,MAAM,oBAAoB,GAAG,eAAe,EAAE,oBAAoB,IAAI,GAAG,CAAC;QAC1E,IAAI,CAAC,aAAa,GAAG,eAAe,EAAE,aAAa,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,YAAY,GAAG,eAAe,EAAE,YAAY,IAAI,IAAI,CAAC;QAE1D,IAAI,CAAC,OAAO,GAAG,IAAI,UAAU,CAAC;YAC5B,SAAS,EAAE,oBAAoB;YAC/B,sBAAsB,EAAE,oBAAoB;YAC5C,wBAAwB,EAAE,EAAE,GAAG,IAAI;YACnC,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,GAAG,oBAAoB,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAI,EAAoB,EAAE,OAAO,GAAG,CAAC;QACjE,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,QAAQ,GAAG,KAAoB,CAAC;YACtC,MAAM,gBAAgB,GACpB,QAAQ,EAAE,MAAM,KAAK,GAAG;gBACxB,QAAQ,EAAE,UAAU,KAAK,GAAG;gBAC5B,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YAE1D,IAAI,gBAAgB,IAAI,OAAO,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;gBACrD,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;gBACzD,MAAM,eAAe,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACpD,OAAO,CAAC,KAAK,CACX,mCAAmC,eAAe,cAAc,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,aAAa,MAAM,CACxG,CAAC;gBAEF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;gBAC7D,OAAO,IAAI,CAAC,gBAAgB,CAAC,EAAE,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;YAChD,CAAC;YAED,IAAI,gBAAgB,EAAE,CAAC;gBACrB,MAAM,IAAI,KAAK,CACb,wCAAwC,IAAI,CAAC,aAAa,sEAAsE,CACjI,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,IAAI,EAAE,CACtC,IAAI,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACvC,KAAK,EAAE,CAAC,IAAI,CAAC;gBACb,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;YAEH,8CAA8C;YAC9C,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAwB,CAAC;YACrD,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;YAC3D,CAAC;YAED,OAAO;gBACL,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC;gBACxB,UAAU,EAAE,IAAI,CAAC,UAAU;aAC5B,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,IAAI,EAAE,CACtC,IAAI,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACvC,KAAK;gBACL,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;YAEH,8CAA8C;YAC9C,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAwB,CAAC;YACrD,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;YAC5D,CAAC;YAED,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,SAAmB,EAAE,EAAE,CAAC,CAAC;gBAC9C,SAAS;gBACT,UAAU,EAAE,IAAI,CAAC,UAAU;aAC5B,CAAC,CAAC,CAAC;QACN,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;CACF"}

package/build/core/adapters/embeddings/factory.d.ts

@@ -0,0 +1,12 @@
+import type { EmbeddingConfig } from "../../contracts/types/config.js";
+import type { EmbeddingProvider } from "./base.js";
+export type EmbeddingProviderType = "openai" | "cohere" | "voyage" | "ollama" | "onnx";
+export interface EmbeddingPaths {
+    models: string;
+    daemonSocket: string;
+    daemonPid: string;
+}
+export declare class EmbeddingProviderFactory {
+    static create(config: EmbeddingConfig, paths?: EmbeddingPaths): EmbeddingProvider;
+}
+//# sourceMappingURL=factory.d.ts.map

package/build/core/adapters/embeddings/factory.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.d.ts","sourceRoot":"","sources":["../../../../src/core/adapters/embeddings/factory.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AACvE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAOnD,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,GAAG,MAAM,CAAC;AAEvF,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;CACnB;AAaD,qBAAa,wBAAwB;IACnC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,eAAe,EAAE,KAAK,CAAC,EAAE,cAAc,GAAG,iBAAiB;CAmElF"}

package/build/core/adapters/embeddings/factory.js

@@ -0,0 +1,53 @@
+import { homedir } from "node:os";
+import { join } from "node:path";
+import { CohereEmbeddings } from "./cohere.js";
+import { OllamaEmbeddings } from "./ollama.js";
+import { DEFAULT_ONNX_MODEL, OnnxEmbeddings } from "./onnx.js";
+import { OpenAIEmbeddings } from "./openai.js";
+import { VoyageEmbeddings } from "./voyage.js";
+/* v8 ignore start -- fallback for backward compat when DI paths not provided */
+function fallbackPaths() {
+    const appData = process.env.TEA_RAGS_DATA_DIR ?? join(homedir(), ".tea-rags");
+    return {
+        models: join(appData, "models"),
+        daemonSocket: join(appData, "onnx.sock"),
+        daemonPid: join(appData, "onnx-daemon.pid"),
+    };
+}
+/* v8 ignore stop */
+export class EmbeddingProviderFactory {
+    static create(config, paths) {
+        const { provider, model, dimensions, baseUrl, tune } = config;
+        const rateLimitConfig = {
+            maxRequestsPerMinute: tune.maxRequestsPerMinute,
+            retryAttempts: tune.retryAttempts,
+            retryDelayMs: tune.retryDelayMs,
+        };
+        switch (provider) {
+            case "openai":
+                if (!config.openaiApiKey) {
+                    throw new Error("API key is required for OpenAI provider");
+                }
+                return new OpenAIEmbeddings(config.openaiApiKey, model || "text-embedding-3-small", dimensions, rateLimitConfig);
+            case "cohere":
+                if (!config.cohereApiKey) {
+                    throw new Error("API key is required for Cohere provider");
+                }
+                return new CohereEmbeddings(config.cohereApiKey, model || "embed-english-v3.0", dimensions, rateLimitConfig);
+            case "voyage":
+                if (!config.voyageApiKey) {
+                    throw new Error("API key is required for Voyage AI provider");
+                }
+                return new VoyageEmbeddings(config.voyageApiKey, model || "voyage-2", dimensions, rateLimitConfig, baseUrl || "https://api.voyageai.com/v1");
+            case "ollama":
+                return new OllamaEmbeddings(model || "unclemusclez/jina-embeddings-v2-base-code:latest", dimensions, rateLimitConfig, baseUrl || "http://localhost:11434", config.ollamaLegacyApi, config.ollamaNumGpu);
+            case "onnx": {
+                const resolved = paths ?? fallbackPaths();
+                return new OnnxEmbeddings(model || DEFAULT_ONNX_MODEL, dimensions, resolved.models, config.device, resolved.daemonSocket, resolved.daemonPid);
+            }
+            default:
+                throw new Error(`Unknown embedding provider: ${String(provider)}. Supported providers: openai, cohere, voyage, ollama, onnx`);
+        }
+    }
+}
+//# sourceMappingURL=factory.js.map

package/build/core/adapters/embeddings/factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.js","sourceRoot":"","sources":["../../../../src/core/adapters/embeddings/factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAIjC,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAC/D,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAU/C,gFAAgF;AAChF,SAAS,aAAa;IACpB,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,WAAW,CAAC,CAAC;IAC9E,OAAO;QACL,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC;QAC/B,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC;QACxC,SAAS,EAAE,IAAI,CAAC,OAAO,EAAE,iBAAiB,CAAC;KAC5C,CAAC;AACJ,CAAC;AACD,oBAAoB;AAEpB,MAAM,OAAO,wBAAwB;IACnC,MAAM,CAAC,MAAM,CAAC,MAAuB,EAAE,KAAsB;QAC3D,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC;QAE9D,MAAM,eAAe,GAAG;YACtB,oBAAoB,EAAE,IAAI,CAAC,oBAAoB;YAC/C,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,YAAY,EAAE,IAAI,CAAC,YAAY;SAChC,CAAC;QAEF,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,QAAQ;gBACX,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;oBACzB,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;gBAC7D,CAAC;gBACD,OAAO,IAAI,gBAAgB,CACzB,MAAM,CAAC,YAAY,EACnB,KAAK,IAAI,wBAAwB,EACjC,UAAU,EACV,eAAe,CAChB,CAAC;YAEJ,KAAK,QAAQ;gBACX,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;oBACzB,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;gBAC7D,CAAC;gBACD,OAAO,IAAI,gBAAgB,CAAC,MAAM,CAAC,YAAY,EAAE,KAAK,IAAI,oBAAoB,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;YAE/G,KAAK,QAAQ;gBACX,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;oBACzB,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;gBAChE,CAAC;gBACD,OAAO,IAAI,gBAAgB,CACzB,MAAM,CAAC,YAAY,EACnB,KAAK,IAAI,UAAU,EACnB,UAAU,EACV,eAAe,EACf,OAAO,IAAI,6BAA6B,CACzC,CAAC;YAEJ,KAAK,QAAQ;gBACX,OAAO,IAAI,gBAAgB,CACzB,KAAK,IAAI,kDAAkD,EAC3D,UAAU,EACV,eAAe,EACf,OAAO,IAAI,wBAAwB,EACnC,MAAM,CAAC,eAAe,EACtB,MAAM,CAAC,YAAY,CACpB,CAAC;YAEJ,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,MAAM,QAAQ,GAAG,KAAK,IAAI,aAAa,EAAE,CAAC;gBAC1C,OAAO,IAAI,cAAc,CACvB,KAAK,IAAI,kBAAkB,EAC3B,UAAU,EACV,QAAQ,CAAC,MAAM,EACf,MAAM,CAAC,MAAM,EACb,QAAQ,CAAC,YAAY,EACrB,QAAQ,CAAC,SAAS,CACnB,CAAC;YACJ,CAAC;YAED;gBACE,MAAM,IAAI,KAAK,CACb,+BAA+B,MAAM,CAAC,QAAQ,CAAC,6DAA6D,CAC7G,CAAC;QACN,CAAC;IACH,CAAC;CACF"}

package/build/core/adapters/embeddings/ollama.d.ts

@@ -0,0 +1,64 @@
+/**
+ * OllamaEmbeddings - OPTIMIZED with native batch API
+ *
+ * KEY OPTIMIZATION: Uses /api/embed instead of /api/embeddings
+ * - /api/embeddings: Single text per request (OLD)
+ * - /api/embed: Array of texts in one request (NEW, since Ollama 0.2.0)
+ *
+ * Performance improvement: N HTTP requests → 1 request
+ *
+ * Sources:
+ * - https://docs.ollama.com/capabilities/embeddings
+ * - https://ollama.com/blog/embedding-models
+ */
+import type { EmbeddingProvider, EmbeddingResult, RateLimitConfig } from "./base.js";
+export declare class OllamaEmbeddings implements EmbeddingProvider {
+    private readonly model;
+    private readonly dimensions;
+    private readonly limiter;
+    private readonly retryAttempts;
+    private readonly retryDelayMs;
+    private readonly baseUrl;
+    private readonly numGpu;
+    private useNativeBatch;
+    constructor(model?: string, dimensions?: number, rateLimitConfig?: RateLimitConfig, baseUrl?: string, legacyApi?: boolean, numGpu?: number);
+    private isOllamaError;
+    private retryWithBackoff;
+    /**
+     * NEW: Native batch embedding using /api/embed
+     * Sends all texts in ONE request instead of N separate requests
+     */
+    private callBatchApi;
+    /**
+     * Legacy single embedding using /api/embeddings
+     * Fallback for older Ollama versions
+     */
+    private callApi;
+    embed(text: string): Promise<EmbeddingResult>;
+    /**
+     * OPTIMIZED: Native batch embeddings
+     *
+     * OLD: N texts → N HTTP requests (even with Promise.all, still N requests!)
+     * NEW: N texts → 1 HTTP request with input array
+     *
+     * Performance: ~50-100x less network overhead
+     *
+     * Batch size configurable via EMBEDDING_BATCH_SIZE env var:
+     * - 0 = use single requests with INGEST_PIPELINE_CONCURRENCY (fallback mode)
+     * - 32 = conservative (recommended for limited VRAM)
+     * - 64 = balanced (default, good for 8GB+ VRAM)
+     * - 128-512 = aggressive (for high-end GPUs)
+     * - 2048+ = benchmark showed linear scaling with AMD 12GB GPU
+     *
+     * Note: GPU must have num_gpu: 999 enabled (see callBatchApi)
+     */
+    embedBatch(texts: string[]): Promise<EmbeddingResult[]>;
+    /**
+     * Check if Ollama supports native batch API
+     * Can be used to auto-detect and fallback
+     */
+    checkBatchSupport(): Promise<boolean>;
+    getDimensions(): number;
+    getModel(): string;
+}
+//# sourceMappingURL=ollama.d.ts.map

package/build/core/adapters/embeddings/ollama.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../../../src/core/adapters/embeddings/ollama.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAKH,OAAO,KAAK,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAmBrF,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAa;IACrC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;IACtC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,cAAc,CAAU;gBAG9B,KAAK,SAAqD,EAC1D,UAAU,CAAC,EAAE,MAAM,EACnB,eAAe,CAAC,EAAE,eAAe,EACjC,OAAO,SAA2B,EAClC,SAAS,UAAQ,EACjB,MAAM,SAAM;IAwBd,OAAO,CAAC,aAAa;YAIP,gBAAgB;IAgC9B;;;OAGG;YACW,YAAY;IA2B1B;;;OAGG;YACW,OAAO;IAwDf,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IA8BnD;;;;;;;;;;;;;;;;OAgBG;IACG,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IA2C7D;;;OAGG;IACG,iBAAiB,IAAI,OAAO,CAAC,OAAO,CAAC;IAW3C,aAAa,IAAI,MAAM;IAIvB,QAAQ,IAAI,MAAM;CAGnB"}