@mhalder/qdrant-mcp-server 1.1.0

package/build/embeddings/factory.test.js

@@ -0,0 +1,329 @@
+import { describe, it, expect, beforeEach, afterEach } from "vitest";
+import { EmbeddingProviderFactory } from "./factory.js";
+import { OpenAIEmbeddings } from "./openai.js";
+import { CohereEmbeddings } from "./cohere.js";
+import { VoyageEmbeddings } from "./voyage.js";
+import { OllamaEmbeddings } from "./ollama.js";
+describe("EmbeddingProviderFactory", () => {
+    let originalEnv;
+    beforeEach(() => {
+        // Save original environment
+        originalEnv = { ...process.env };
+    });
+    afterEach(() => {
+        // Restore original environment
+        process.env = originalEnv;
+    });
+    describe("create", () => {
+        describe("Unknown provider", () => {
+            it("should throw error for unknown provider", () => {
+                expect(() => EmbeddingProviderFactory.create({
+                    provider: "unknown",
+                })).toThrow("Unknown embedding provider: unknown");
+            });
+            it("should list supported providers in error message", () => {
+                expect(() => EmbeddingProviderFactory.create({
+                    provider: "invalid",
+                })).toThrow("openai, cohere, voyage, ollama");
+            });
+        });
+        describe("OpenAI provider", () => {
+            it("should throw error if API key is missing", () => {
+                expect(() => EmbeddingProviderFactory.create({
+                    provider: "openai",
+                })).toThrow("API key is required for OpenAI provider");
+            });
+            it("should create OpenAI provider with API key", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "openai",
+                    apiKey: "test-key",
+                });
+                expect(provider).toBeInstanceOf(OpenAIEmbeddings);
+                expect(provider.getModel()).toBe("text-embedding-3-small");
+                expect(provider.getDimensions()).toBe(1536);
+            });
+            it("should use custom model", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "openai",
+                    apiKey: "test-key",
+                    model: "text-embedding-3-large",
+                });
+                expect(provider.getModel()).toBe("text-embedding-3-large");
+                expect(provider.getDimensions()).toBe(3072);
+            });
+            it("should use custom dimensions", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "openai",
+                    apiKey: "test-key",
+                    dimensions: 512,
+                });
+                expect(provider.getDimensions()).toBe(512);
+            });
+            it("should pass rate limit config", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "openai",
+                    apiKey: "test-key",
+                    rateLimitConfig: {
+                        maxRequestsPerMinute: 1000,
+                        retryAttempts: 5,
+                        retryDelayMs: 2000,
+                    },
+                });
+                expect(provider).toBeInstanceOf(OpenAIEmbeddings);
+            });
+        });
+        describe("Cohere provider", () => {
+            it("should throw error if API key is missing", () => {
+                expect(() => EmbeddingProviderFactory.create({
+                    provider: "cohere",
+                })).toThrow("API key is required for Cohere provider");
+            });
+            it("should create Cohere provider with API key", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "cohere",
+                    apiKey: "test-key",
+                });
+                expect(provider).toBeInstanceOf(CohereEmbeddings);
+                expect(provider.getModel()).toBe("embed-english-v3.0");
+                expect(provider.getDimensions()).toBe(1024);
+            });
+            it("should use custom model", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "cohere",
+                    apiKey: "test-key",
+                    model: "embed-multilingual-v3.0",
+                });
+                expect(provider.getModel()).toBe("embed-multilingual-v3.0");
+            });
+            it("should use custom dimensions", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "cohere",
+                    apiKey: "test-key",
+                    dimensions: 384,
+                });
+                expect(provider.getDimensions()).toBe(384);
+            });
+        });
+        describe("Voyage provider", () => {
+            it("should throw error if API key is missing", () => {
+                expect(() => EmbeddingProviderFactory.create({
+                    provider: "voyage",
+                })).toThrow("API key is required for Voyage AI provider");
+            });
+            it("should create Voyage provider with API key", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "voyage",
+                    apiKey: "test-key",
+                });
+                expect(provider).toBeInstanceOf(VoyageEmbeddings);
+                expect(provider.getModel()).toBe("voyage-2");
+                expect(provider.getDimensions()).toBe(1024);
+            });
+            it("should use custom model", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "voyage",
+                    apiKey: "test-key",
+                    model: "voyage-large-2",
+                });
+                expect(provider.getModel()).toBe("voyage-large-2");
+                expect(provider.getDimensions()).toBe(1536);
+            });
+            it("should use default base URL", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "voyage",
+                    apiKey: "test-key",
+                });
+                expect(provider).toBeInstanceOf(VoyageEmbeddings);
+            });
+            it("should use custom base URL", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "voyage",
+                    apiKey: "test-key",
+                    baseUrl: "https://custom.voyageai.com/v1",
+                });
+                expect(provider).toBeInstanceOf(VoyageEmbeddings);
+            });
+        });
+        describe("Ollama provider", () => {
+            it("should not require API key", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "ollama",
+                });
+                expect(provider).toBeInstanceOf(OllamaEmbeddings);
+                expect(provider.getModel()).toBe("nomic-embed-text");
+                expect(provider.getDimensions()).toBe(768);
+            });
+            it("should use custom model", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "ollama",
+                    model: "mxbai-embed-large",
+                });
+                expect(provider.getModel()).toBe("mxbai-embed-large");
+                expect(provider.getDimensions()).toBe(1024);
+            });
+            it("should use default base URL", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "ollama",
+                });
+                expect(provider).toBeInstanceOf(OllamaEmbeddings);
+            });
+            it("should use custom base URL", () => {
+                const provider = EmbeddingProviderFactory.create({
+                    provider: "ollama",
+                    baseUrl: "http://custom:11434",
+                });
+                expect(provider).toBeInstanceOf(OllamaEmbeddings);
+            });
+        });
+    });
+    describe("createFromEnv", () => {
+        it("should default to Ollama provider", () => {
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(OllamaEmbeddings);
+        });
+        it("should create OpenAI provider from environment", () => {
+            process.env.EMBEDDING_PROVIDER = "openai";
+            process.env.OPENAI_API_KEY = "test-openai-key";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(OpenAIEmbeddings);
+        });
+        it("should create Cohere provider from environment", () => {
+            process.env.EMBEDDING_PROVIDER = "cohere";
+            process.env.COHERE_API_KEY = "test-cohere-key";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(CohereEmbeddings);
+        });
+        it("should create Voyage provider from environment", () => {
+            process.env.EMBEDDING_PROVIDER = "voyage";
+            process.env.VOYAGE_API_KEY = "test-voyage-key";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(VoyageEmbeddings);
+        });
+        it("should create Ollama provider from environment", () => {
+            process.env.EMBEDDING_PROVIDER = "ollama";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(OllamaEmbeddings);
+        });
+        it("should be case insensitive for provider name", () => {
+            process.env.EMBEDDING_PROVIDER = "OpenAI";
+            process.env.OPENAI_API_KEY = "test-key";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(OpenAIEmbeddings);
+        });
+        it("should use custom model from environment", () => {
+            process.env.EMBEDDING_PROVIDER = "openai";
+            process.env.OPENAI_API_KEY = "test-key";
+            process.env.EMBEDDING_MODEL = "text-embedding-3-large";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider.getModel()).toBe("text-embedding-3-large");
+        });
+        it("should use custom dimensions from environment", () => {
+            process.env.EMBEDDING_PROVIDER = "openai";
+            process.env.OPENAI_API_KEY = "test-key";
+            process.env.EMBEDDING_DIMENSIONS = "512";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider.getDimensions()).toBe(512);
+        });
+        it("should use custom base URL from environment", () => {
+            process.env.EMBEDDING_PROVIDER = "voyage";
+            process.env.VOYAGE_API_KEY = "test-key";
+            process.env.EMBEDDING_BASE_URL = "https://custom.voyage.com";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(VoyageEmbeddings);
+        });
+        it("should use rate limit config from environment", () => {
+            process.env.EMBEDDING_PROVIDER = "openai";
+            process.env.OPENAI_API_KEY = "test-key";
+            process.env.EMBEDDING_MAX_REQUESTS_PER_MINUTE = "1000";
+            process.env.EMBEDDING_RETRY_ATTEMPTS = "5";
+            process.env.EMBEDDING_RETRY_DELAY = "2000";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(OpenAIEmbeddings);
+        });
+        it("should select correct API key based on provider", () => {
+            process.env.EMBEDDING_PROVIDER = "cohere";
+            process.env.OPENAI_API_KEY = "openai-key";
+            process.env.COHERE_API_KEY = "cohere-key";
+            process.env.VOYAGE_API_KEY = "voyage-key";
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(CohereEmbeddings);
+        });
+        it("should handle Ollama without API key", () => {
+            process.env.EMBEDDING_PROVIDER = "ollama";
+            process.env.OPENAI_API_KEY = "openai-key"; // Should not use this
+            const provider = EmbeddingProviderFactory.createFromEnv();
+            expect(provider).toBeInstanceOf(OllamaEmbeddings);
+        });
+        describe("Environment variable validation", () => {
+            it("should throw error for invalid EMBEDDING_DIMENSIONS (NaN)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_DIMENSIONS = "not-a-number";
+                expect(() => EmbeddingProviderFactory.createFromEnv()).toThrow('Invalid EMBEDDING_DIMENSIONS: must be a positive integer, got "not-a-number"');
+            });
+            it("should throw error for invalid EMBEDDING_DIMENSIONS (negative)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_DIMENSIONS = "-100";
+                expect(() => EmbeddingProviderFactory.createFromEnv()).toThrow('Invalid EMBEDDING_DIMENSIONS: must be a positive integer, got "-100"');
+            });
+            it("should throw error for invalid EMBEDDING_DIMENSIONS (zero)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_DIMENSIONS = "0";
+                expect(() => EmbeddingProviderFactory.createFromEnv()).toThrow('Invalid EMBEDDING_DIMENSIONS: must be a positive integer, got "0"');
+            });
+            it("should throw error for invalid EMBEDDING_MAX_REQUESTS_PER_MINUTE (NaN)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_MAX_REQUESTS_PER_MINUTE = "invalid";
+                expect(() => EmbeddingProviderFactory.createFromEnv()).toThrow('Invalid EMBEDDING_MAX_REQUESTS_PER_MINUTE: must be a positive integer, got "invalid"');
+            });
+            it("should throw error for invalid EMBEDDING_MAX_REQUESTS_PER_MINUTE (negative)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_MAX_REQUESTS_PER_MINUTE = "-50";
+                expect(() => EmbeddingProviderFactory.createFromEnv()).toThrow('Invalid EMBEDDING_MAX_REQUESTS_PER_MINUTE: must be a positive integer, got "-50"');
+            });
+            it("should throw error for invalid EMBEDDING_RETRY_ATTEMPTS (NaN)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_RETRY_ATTEMPTS = "abc";
+                expect(() => EmbeddingProviderFactory.createFromEnv()).toThrow('Invalid EMBEDDING_RETRY_ATTEMPTS: must be a non-negative integer, got "abc"');
+            });
+            it("should throw error for invalid EMBEDDING_RETRY_ATTEMPTS (negative)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_RETRY_ATTEMPTS = "-5";
+                expect(() => EmbeddingProviderFactory.createFromEnv()).toThrow('Invalid EMBEDDING_RETRY_ATTEMPTS: must be a non-negative integer, got "-5"');
+            });
+            it("should throw error for invalid EMBEDDING_RETRY_DELAY (NaN)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_RETRY_DELAY = "xyz";
+                expect(() => EmbeddingProviderFactory.createFromEnv()).toThrow('Invalid EMBEDDING_RETRY_DELAY: must be a non-negative integer, got "xyz"');
+            });
+            it("should throw error for invalid EMBEDDING_RETRY_DELAY (negative)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_RETRY_DELAY = "-1000";
+                expect(() => EmbeddingProviderFactory.createFromEnv()).toThrow('Invalid EMBEDDING_RETRY_DELAY: must be a non-negative integer, got "-1000"');
+            });
+            it("should accept valid EMBEDDING_RETRY_ATTEMPTS (zero)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_RETRY_ATTEMPTS = "0";
+                const provider = EmbeddingProviderFactory.createFromEnv();
+                expect(provider).toBeInstanceOf(OpenAIEmbeddings);
+            });
+            it("should accept valid EMBEDDING_RETRY_DELAY (zero)", () => {
+                process.env.EMBEDDING_PROVIDER = "openai";
+                process.env.OPENAI_API_KEY = "test-key";
+                process.env.EMBEDDING_RETRY_DELAY = "0";
+                const provider = EmbeddingProviderFactory.createFromEnv();
+                expect(provider).toBeInstanceOf(OpenAIEmbeddings);
+            });
+        });
+    });
+});
+//# sourceMappingURL=factory.test.js.map

package/build/embeddings/factory.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.test.js","sourceRoot":"","sources":["../../src/embeddings/factory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,wBAAwB,EAAsB,MAAM,cAAc,CAAC;AAC5E,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAE/C,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,IAAI,WAA8B,CAAC;IAEnC,UAAU,CAAC,GAAG,EAAE;QACd,4BAA4B;QAC5B,WAAW,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,+BAA+B;QAC/B,OAAO,CAAC,GAAG,GAAG,WAAW,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;YAChC,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;gBACjD,MAAM,CAAC,GAAG,EAAE,CACV,wBAAwB,CAAC,MAAM,CAAC;oBAC9B,QAAQ,EAAE,SAAgB;iBAC3B,CAAC,CACH,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAC;YACnD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBAC1D,MAAM,CAAC,GAAG,EAAE,CACV,wBAAwB,CAAC,MAAM,CAAC;oBAC9B,QAAQ,EAAE,SAAgB;iBAC3B,CAAC,CACH,CAAC,OAAO,CAAC,gCAAgC,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;YAC/B,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;gBAClD,MAAM,CAAC,GAAG,EAAE,CACV,wBAAwB,CAAC,MAAM,CAAC;oBAC9B,QAAQ,EAAE,QAAQ;iBACnB,CAAC,CACH,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC;YACvD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;gBACpD,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;iBACnB,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;gBAClD,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;gBAC3D,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;gBACjC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;oBAClB,KAAK,EAAE,wBAAwB;iBAChC,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;gBAC3D,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;gBACtC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;oBAClB,UAAU,EAAE,GAAG;iBAChB,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;gBACvC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;oBAClB,eAAe,EAAE;wBACf,oBAAoB,EAAE,IAAI;wBAC1B,aAAa,EAAE,CAAC;wBAChB,YAAY,EAAE,IAAI;qBACnB;iBACF,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;YAC/B,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;gBAClD,MAAM,CAAC,GAAG,EAAE,CACV,wBAAwB,CAAC,MAAM,CAAC;oBAC9B,QAAQ,EAAE,QAAQ;iBACnB,CAAC,CACH,CAAC,OAAO,CAAC,yCAAyC,CAAC,CAAC;YACvD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;gBACpD,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;iBACnB,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;gBAClD,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;gBACvD,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;gBACjC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;oBAClB,KAAK,EAAE,yBAAyB;iBACjC,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;gBACtC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;oBAClB,UAAU,EAAE,GAAG;iBAChB,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;YAC/B,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;gBAClD,MAAM,CAAC,GAAG,EAAE,CACV,wBAAwB,CAAC,MAAM,CAAC;oBAC9B,QAAQ,EAAE,QAAQ;iBACnB,CAAC,CACH,CAAC,OAAO,CAAC,4CAA4C,CAAC,CAAC;YAC1D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;gBACpD,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;iBACnB,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;gBAClD,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBAC7C,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;gBACjC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;oBAClB,KAAK,EAAE,gBAAgB;iBACxB,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBACnD,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;gBACrC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;iBACnB,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;gBACpC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,MAAM,EAAE,UAAU;oBAClB,OAAO,EAAE,gCAAgC;iBAC1C,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;YAC/B,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;gBACpC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;iBACnB,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;gBAClD,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;gBACrD,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;gBACjC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,KAAK,EAAE,mBAAmB;iBAC3B,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;gBACtD,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;gBACrC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;iBACnB,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;gBACpC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,QAAQ;oBAClB,OAAO,EAAE,qBAAqB;iBAC/B,CAAC,CAAC;gBAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,iBAAiB,CAAC;YAE/C,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,iBAAiB,CAAC;YAE/C,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,iBAAiB,CAAC;YAE/C,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAE1C,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;YAExC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;YACxC,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,wBAAwB,CAAC;YAEvD,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;YACxC,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,KAAK,CAAC;YAEzC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;YACxC,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,2BAA2B,CAAC;YAE7D,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;YACxC,OAAO,CAAC,GAAG,CAAC,iCAAiC,GAAG,MAAM,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,GAAG,CAAC;YAC3C,OAAO,CAAC,GAAG,CAAC,qBAAqB,GAAG,MAAM,CAAC;YAE3C,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,YAAY,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,YAAY,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,YAAY,CAAC;YAE1C,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,YAAY,CAAC,CAAC,sBAAsB;YAEjE,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;YAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;YAC/C,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;gBACnE,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,cAAc,CAAC;gBAElD,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAC5D,8EAA8E,CAC/E,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;gBACxE,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,MAAM,CAAC;gBAE1C,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAC5D,sEAAsE,CACvE,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;gBACpE,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,GAAG,CAAC;gBAEvC,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAC5D,mEAAmE,CACpE,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;gBAChF,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,iCAAiC,GAAG,SAAS,CAAC;gBAE1D,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAC5D,sFAAsF,CACvF,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;gBACrF,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,iCAAiC,GAAG,KAAK,CAAC;gBAEtD,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAC5D,kFAAkF,CACnF,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;gBACvE,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,KAAK,CAAC;gBAE7C,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAC5D,6EAA6E,CAC9E,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;gBAC5E,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,IAAI,CAAC;gBAE5C,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAC5D,4EAA4E,CAC7E,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;gBACpE,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,qBAAqB,GAAG,KAAK,CAAC;gBAE1C,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAC5D,0EAA0E,CAC3E,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;gBACzE,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,qBAAqB,GAAG,OAAO,CAAC;gBAE5C,MAAM,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,aAAa,EAAE,CAAC,CAAC,OAAO,CAC5D,4EAA4E,CAC7E,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;gBAC7D,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,GAAG,CAAC;gBAE3C,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;gBAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBAC1D,OAAO,CAAC,GAAG,CAAC,kBAAkB,GAAG,QAAQ,CAAC;gBAC1C,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,UAAU,CAAC;gBACxC,OAAO,CAAC,GAAG,CAAC,qBAAqB,GAAG,GAAG,CAAC;gBAExC,MAAM,QAAQ,GAAG,wBAAwB,CAAC,aAAa,EAAE,CAAC;gBAE1D,MAAM,CAAC,QAAQ,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/build/embeddings/ollama.d.ts

@@ -0,0 +1,18 @@
+import { EmbeddingProvider, EmbeddingResult, RateLimitConfig } from "./base.js";
+export declare class OllamaEmbeddings implements EmbeddingProvider {
+    private model;
+    private dimensions;
+    private limiter;
+    private retryAttempts;
+    private retryDelayMs;
+    private baseUrl;
+    constructor(model?: string, dimensions?: number, rateLimitConfig?: RateLimitConfig, baseUrl?: string);
+    private isOllamaError;
+    private retryWithBackoff;
+    private callApi;
+    embed(text: string): Promise<EmbeddingResult>;
+    embedBatch(texts: string[]): Promise<EmbeddingResult[]>;
+    getDimensions(): number;
+    getModel(): string;
+}
+//# sourceMappingURL=ollama.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../src/embeddings/ollama.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAWhF,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,OAAO,CAAa;IAC5B,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,OAAO,CAAS;gBAGtB,KAAK,GAAE,MAA2B,EAClC,UAAU,CAAC,EAAE,MAAM,EACnB,eAAe,CAAC,EAAE,eAAe,EACjC,OAAO,GAAE,MAAiC;IA4B5C,OAAO,CAAC,aAAa;YAMP,gBAAgB;YAsChB,OAAO;IAyDf,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAiB7C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAkB7D,aAAa,IAAI,MAAM;IAIvB,QAAQ,IAAI,MAAM;CAGnB"}

package/build/embeddings/ollama.js

@@ -0,0 +1,135 @@
+import Bottleneck from "bottleneck";
+export class OllamaEmbeddings {
+    model;
+    dimensions;
+    limiter;
+    retryAttempts;
+    retryDelayMs;
+    baseUrl;
+    constructor(model = "nomic-embed-text", dimensions, rateLimitConfig, baseUrl = "http://localhost:11434") {
+        this.model = model;
+        this.baseUrl = baseUrl;
+        // Default dimensions for different models
+        const defaultDimensions = {
+            "nomic-embed-text": 768,
+            "mxbai-embed-large": 1024,
+            "all-minilm": 384,
+        };
+        this.dimensions = dimensions || defaultDimensions[model] || 768;
+        // Rate limiting configuration (more lenient for local models)
+        const maxRequestsPerMinute = rateLimitConfig?.maxRequestsPerMinute || 1000;
+        this.retryAttempts = rateLimitConfig?.retryAttempts || 3;
+        this.retryDelayMs = rateLimitConfig?.retryDelayMs || 500;
+        this.limiter = new Bottleneck({
+            reservoir: maxRequestsPerMinute,
+            reservoirRefreshAmount: maxRequestsPerMinute,
+            reservoirRefreshInterval: 60 * 1000,
+            maxConcurrent: 10,
+            minTime: Math.floor((60 * 1000) / maxRequestsPerMinute),
+        });
+    }
+    isOllamaError(e) {
+        return (typeof e === "object" && e !== null && ("status" in e || "message" in e));
+    }
+    async retryWithBackoff(fn, attempt = 0) {
+        try {
+            return await fn();
+        }
+        catch (error) {
+            // Type guard for OllamaError
+            const apiError = this.isOllamaError(error)
+                ? error
+                : { status: 0, message: String(error) };
+            const isRateLimitError = apiError.status === 429 ||
+                (typeof apiError.message === "string" &&
+                    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(`Ollama API rate limit exceeded after ${this.retryAttempts} retry attempts. Please try again later or reduce request frequency.`);
+            }
+            throw error;
+        }
+    }
+    async callApi(text) {
+        try {
+            const response = await fetch(`${this.baseUrl}/api/embeddings`, {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                },
+                body: JSON.stringify({
+                    model: this.model,
+                    prompt: text,
+                }),
+            });
+            if (!response.ok) {
+                const errorBody = await response.text();
+                const textPreview = text.length > 100 ? text.substring(0, 100) + "..." : text;
+                const error = {
+                    status: response.status,
+                    message: `Ollama API error (${response.status}) for model "${this.model}": ${errorBody}. Text preview: "${textPreview}"`,
+                };
+                throw error;
+            }
+            return response.json();
+        }
+        catch (error) {
+            // Re-throw if it's already an OllamaError from the !response.ok block
+            if (error && typeof error === "object" && "status" in error) {
+                throw error;
+            }
+            // For Error instances (like network errors), enhance the message
+            if (error instanceof Error) {
+                const textPreview = text.length > 100 ? text.substring(0, 100) + "..." : text;
+                throw new Error(`Failed to call Ollama API at ${this.baseUrl} with model ${this.model}: ${error.message}. Text preview: "${textPreview}"`);
+            }
+            // Handle objects with 'message' property - preserve the original error structure
+            // This ensures objects with 'message' property work correctly in tests
+            if (this.isOllamaError(error)) {
+                throw error;
+            }
+            // For other types, create a descriptive error message
+            const textPreview = text.length > 100 ? text.substring(0, 100) + "..." : text;
+            const errorMessage = JSON.stringify(error);
+            throw new Error(`Failed to call Ollama API at ${this.baseUrl} with model ${this.model}: ${errorMessage}. Text preview: "${textPreview}"`);
+        }
+    }
+    async embed(text) {
+        return this.limiter.schedule(() => this.retryWithBackoff(async () => {
+            const response = await this.callApi(text);
+            if (!response.embedding) {
+                throw new Error("No embedding returned from Ollama API");
+            }
+            return {
+                embedding: response.embedding,
+                dimensions: this.dimensions,
+            };
+        }));
+    }
+    async embedBatch(texts) {
+        // Ollama doesn't support batch embeddings natively, so we process in parallel
+        // Process in chunks to avoid overwhelming Ollama and prevent memory issues
+        const CHUNK_SIZE = 50;
+        const results = [];
+        for (let i = 0; i < texts.length; i += CHUNK_SIZE) {
+            const chunk = texts.slice(i, i + CHUNK_SIZE);
+            // The Bottleneck limiter will handle rate limiting and concurrency (maxConcurrent: 10)
+            const chunkResults = await Promise.all(chunk.map((text) => this.embed(text)));
+            results.push(...chunkResults);
+        }
+        return results;
+    }
+    getDimensions() {
+        return this.dimensions;
+    }
+    getModel() {
+        return this.model;
+    }
+}
+//# sourceMappingURL=ollama.js.map

package/build/embeddings/ollama.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/embeddings/ollama.ts"],"names":[],"mappings":"AAAA,OAAO,UAAU,MAAM,YAAY,CAAC;AAYpC,MAAM,OAAO,gBAAgB;IACnB,KAAK,CAAS;IACd,UAAU,CAAS;IACnB,OAAO,CAAa;IACpB,aAAa,CAAS;IACtB,YAAY,CAAS;IACrB,OAAO,CAAS;IAExB,YACE,QAAgB,kBAAkB,EAClC,UAAmB,EACnB,eAAiC,EACjC,UAAkB,wBAAwB;QAE1C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAEvB,0CAA0C;QAC1C,MAAM,iBAAiB,GAA2B;YAChD,kBAAkB,EAAE,GAAG;YACvB,mBAAmB,EAAE,IAAI;YACzB,YAAY,EAAE,GAAG;SAClB,CAAC;QAEF,IAAI,CAAC,UAAU,GAAG,UAAU,IAAI,iBAAiB,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC;QAEhE,8DAA8D;QAC9D,MAAM,oBAAoB,GAAG,eAAe,EAAE,oBAAoB,IAAI,IAAI,CAAC;QAC3E,IAAI,CAAC,aAAa,GAAG,eAAe,EAAE,aAAa,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,YAAY,GAAG,eAAe,EAAE,YAAY,IAAI,GAAG,CAAC;QAEzD,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,EAAE;YACjB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,GAAG,oBAAoB,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,CAAU;QAC9B,OAAO,CACL,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,SAAS,IAAI,CAAC,CAAC,CACzE,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,EAAoB,EACpB,UAAkB,CAAC;QAEnB,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,6BAA6B;YAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;gBACxC,CAAC,CAAC,KAAK;gBACP,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YAE1C,MAAM,gBAAgB,GACpB,QAAQ,CAAC,MAAM,KAAK,GAAG;gBACvB,CAAC,OAAO,QAAQ,CAAC,OAAO,KAAK,QAAQ;oBACnC,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;YAE3D,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;IAEO,KAAK,CAAC,OAAO,CAAC,IAAY;QAChC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,iBAAiB,EAAE;gBAC7D,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EAAE,IAAI;iBACb,CAAC;aACH,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,MAAM,WAAW,GACf,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC5D,MAAM,KAAK,GAAgB;oBACzB,MAAM,EAAE,QAAQ,CAAC,MAAM;oBACvB,OAAO,EAAE,qBAAqB,QAAQ,CAAC,MAAM,gBAAgB,IAAI,CAAC,KAAK,MAAM,SAAS,oBAAoB,WAAW,GAAG;iBACzH,CAAC;gBACF,MAAM,KAAK,CAAC;YACd,CAAC;YAED,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;QACzB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,sEAAsE;YACtE,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,QAAQ,IAAI,KAAK,EAAE,CAAC;gBAC5D,MAAM,KAAK,CAAC;YACd,CAAC;YAED,iEAAiE;YACjE,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,WAAW,GACf,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC5D,MAAM,IAAI,KAAK,CACb,gCAAgC,IAAI,CAAC,OAAO,eAAe,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC,OAAO,oBAAoB,WAAW,GAAG,CAC1H,CAAC;YACJ,CAAC;YAED,iFAAiF;YACjF,uEAAuE;YACvE,IAAI,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9B,MAAM,KAAK,CAAC;YACd,CAAC;YAED,sDAAsD;YACtD,MAAM,WAAW,GACf,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;YAC5D,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAE3C,MAAM,IAAI,KAAK,CACb,gCAAgC,IAAI,CAAC,OAAO,eAAe,IAAI,CAAC,KAAK,KAAK,YAAY,oBAAoB,WAAW,GAAG,CACzH,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,EAAE,CAChC,IAAI,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAE1C,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;YAC3D,CAAC;YAED,OAAO;gBACL,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;aAC5B,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,8EAA8E;QAC9E,2EAA2E;QAC3E,MAAM,UAAU,GAAG,EAAE,CAAC;QACtB,MAAM,OAAO,GAAsB,EAAE,CAAC;QAEtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;YAClD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,CAAC;YAC7C,uFAAuF;YACvF,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CACtC,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;CACF"}

package/build/embeddings/ollama.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ollama.test.d.ts.map

package/build/embeddings/ollama.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.test.d.ts","sourceRoot":"","sources":["../../src/embeddings/ollama.test.ts"],"names":[],"mappings":""}