@mhalder/qdrant-mcp-server 1.1.0

package/build/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/embeddings/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/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,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC;IAC9C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;IACxD,aAAa,IAAI,MAAM,CAAC;IACxB,QAAQ,IAAI,MAAM,CAAC;CACpB;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/embeddings/base.js

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

package/build/embeddings/base.js.map

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

package/build/embeddings/cohere.d.ts

@@ -0,0 +1,17 @@
+import { EmbeddingProvider, EmbeddingResult, RateLimitConfig } from "./base.js";
+export declare class CohereEmbeddings implements EmbeddingProvider {
+    private client;
+    private model;
+    private dimensions;
+    private limiter;
+    private retryAttempts;
+    private retryDelayMs;
+    private 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/embeddings/cohere.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.d.ts","sourceRoot":"","sources":["../../src/embeddings/cohere.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAQhF,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,OAAO,CAAC,MAAM,CAAe;IAC7B,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,SAAS,CAIA;gBAGf,MAAM,EAAE,MAAM,EACd,KAAK,GAAE,MAA6B,EACpC,UAAU,CAAC,EAAE,MAAM,EACnB,eAAe,CAAC,EAAE,eAAe,EACjC,SAAS,GACL,iBAAiB,GACjB,cAAc,GACd,gBAAgB,GAChB,YAAgC;YA8BxB,gBAAgB;IAkCxB,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/embeddings/cohere.js

@@ -0,0 +1,102 @@
+import { CohereClient } from "cohere-ai";
+import Bottleneck from "bottleneck";
+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;
+        // Default dimensions for different models
+        const defaultDimensions = {
+            "embed-english-v3.0": 1024,
+            "embed-multilingual-v3.0": 1024,
+            "embed-english-light-v3.0": 384,
+            "embed-multilingual-light-v3.0": 384,
+        };
+        this.dimensions = dimensions || defaultDimensions[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(() => 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(() => 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/embeddings/cohere.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.js","sourceRoot":"","sources":["../../src/embeddings/cohere.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,UAAU,MAAM,YAAY,CAAC;AASpC,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAe;IACrB,KAAK,CAAS;IACd,UAAU,CAAS;IACnB,OAAO,CAAa;IACpB,aAAa,CAAS;IACtB,YAAY,CAAS;IACrB,SAAS,CAIA;IAEjB,YACE,MAAc,EACd,QAAgB,oBAAoB,EACpC,UAAmB,EACnB,eAAiC,EACjC,YAImB,iBAAiB;QAEpC,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,0CAA0C;QAC1C,MAAM,iBAAiB,GAA2B;YAChD,oBAAoB,EAAE,IAAI;YAC1B,yBAAyB,EAAE,IAAI;YAC/B,0BAA0B,EAAE,GAAG;YAC/B,+BAA+B,EAAE,GAAG;SACrC,CAAC;QAEF,IAAI,CAAC,UAAU,GAAG,UAAU,IAAI,iBAAiB,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;QAEjE,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,CAC5B,EAAoB,EACpB,UAAkB,CAAC;QAEnB,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,GAAG,EAAE,CAChC,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,GAAG,EAAE,CAChC,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/embeddings/cohere.test.d.ts

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

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

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

package/build/embeddings/cohere.test.js

@@ -0,0 +1,279 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { CohereEmbeddings } from "./cohere.js";
+import { CohereClient } from "cohere-ai";
+vi.mock("cohere-ai", () => ({
+    CohereClient: vi.fn(),
+}));
+describe("CohereEmbeddings", () => {
+    let embeddings;
+    let mockClient;
+    beforeEach(() => {
+        mockClient = {
+            embed: vi.fn(),
+        };
+        vi.mocked(CohereClient).mockImplementation(() => mockClient);
+        embeddings = new CohereEmbeddings("test-api-key");
+    });
+    describe("constructor", () => {
+        it("should use default model and dimensions", () => {
+            expect(embeddings.getModel()).toBe("embed-english-v3.0");
+            expect(embeddings.getDimensions()).toBe(1024);
+        });
+        it("should use custom model", () => {
+            const customEmbeddings = new CohereEmbeddings("test-api-key", "embed-multilingual-v3.0");
+            expect(customEmbeddings.getModel()).toBe("embed-multilingual-v3.0");
+            expect(customEmbeddings.getDimensions()).toBe(1024);
+        });
+        it("should use custom dimensions", () => {
+            const customEmbeddings = new CohereEmbeddings("test-api-key", "embed-english-v3.0", 512);
+            expect(customEmbeddings.getDimensions()).toBe(512);
+        });
+        it("should use default dimensions for light models", () => {
+            const lightEmbeddings = new CohereEmbeddings("test-api-key", "embed-english-light-v3.0");
+            expect(lightEmbeddings.getDimensions()).toBe(384);
+        });
+        it("should default to 1024 for unknown models", () => {
+            const unknownEmbeddings = new CohereEmbeddings("test-api-key", "custom-model");
+            expect(unknownEmbeddings.getDimensions()).toBe(1024);
+        });
+        it("should accept custom input type", () => {
+            const searchQueryEmbeddings = new CohereEmbeddings("test-api-key", "embed-english-v3.0", undefined, undefined, "search_query");
+            expect(searchQueryEmbeddings).toBeInstanceOf(CohereEmbeddings);
+        });
+    });
+    describe("embed", () => {
+        it("should generate embedding for single text", async () => {
+            const mockEmbedding = Array(1024)
+                .fill(0)
+                .map((_, i) => i * 0.001);
+            mockClient.embed.mockResolvedValue({
+                embeddings: [mockEmbedding],
+            });
+            const result = await embeddings.embed("test text");
+            expect(result).toEqual({
+                embedding: mockEmbedding,
+                dimensions: 1024,
+            });
+            expect(mockClient.embed).toHaveBeenCalledWith({
+                texts: ["test text"],
+                model: "embed-english-v3.0",
+                inputType: "search_document",
+                embeddingTypes: ["float"],
+            });
+        });
+        it("should handle long text", async () => {
+            const longText = "word ".repeat(1000);
+            const mockEmbedding = Array(1024).fill(0.5);
+            mockClient.embed.mockResolvedValue({
+                embeddings: [mockEmbedding],
+            });
+            const result = await embeddings.embed(longText);
+            expect(result.embedding).toEqual(mockEmbedding);
+            expect(mockClient.embed).toHaveBeenCalledWith({
+                texts: [longText],
+                model: "embed-english-v3.0",
+                inputType: "search_document",
+                embeddingTypes: ["float"],
+            });
+        });
+        it("should use custom model configuration", async () => {
+            const customEmbeddings = new CohereEmbeddings("test-api-key", "embed-multilingual-v3.0", 1024);
+            const mockEmbedding = Array(1024).fill(0.1);
+            mockClient.embed.mockResolvedValue({
+                embeddings: [mockEmbedding],
+            });
+            await customEmbeddings.embed("test");
+            expect(mockClient.embed).toHaveBeenCalledWith({
+                texts: ["test"],
+                model: "embed-multilingual-v3.0",
+                inputType: "search_document",
+                embeddingTypes: ["float"],
+            });
+        });
+        it("should throw error if no embedding returned", async () => {
+            mockClient.embed.mockResolvedValue({
+                embeddings: [],
+            });
+            await expect(embeddings.embed("test")).rejects.toThrow("No embedding returned from Cohere API");
+        });
+        it("should propagate errors", async () => {
+            mockClient.embed.mockRejectedValue(new Error("API Error"));
+            await expect(embeddings.embed("test")).rejects.toThrow("API Error");
+        });
+    });
+    describe("embedBatch", () => {
+        it("should generate embeddings for multiple texts", async () => {
+            const mockEmbeddings = [
+                Array(1024).fill(0.1),
+                Array(1024).fill(0.2),
+                Array(1024).fill(0.3),
+            ];
+            mockClient.embed.mockResolvedValue({
+                embeddings: mockEmbeddings,
+            });
+            const texts = ["text1", "text2", "text3"];
+            const results = await embeddings.embedBatch(texts);
+            expect(results).toEqual([
+                { embedding: mockEmbeddings[0], dimensions: 1024 },
+                { embedding: mockEmbeddings[1], dimensions: 1024 },
+                { embedding: mockEmbeddings[2], dimensions: 1024 },
+            ]);
+            expect(mockClient.embed).toHaveBeenCalledWith({
+                texts,
+                model: "embed-english-v3.0",
+                inputType: "search_document",
+                embeddingTypes: ["float"],
+            });
+        });
+        it("should handle empty batch", async () => {
+            mockClient.embed.mockResolvedValue({
+                embeddings: [],
+            });
+            const results = await embeddings.embedBatch([]);
+            expect(results).toEqual([]);
+        });
+        it("should handle single item in batch", async () => {
+            const mockEmbedding = Array(1024).fill(0.5);
+            mockClient.embed.mockResolvedValue({
+                embeddings: [mockEmbedding],
+            });
+            const results = await embeddings.embedBatch(["single text"]);
+            expect(results).toHaveLength(1);
+            expect(results[0].embedding).toEqual(mockEmbedding);
+        });
+        it("should handle large batches", async () => {
+            const batchSize = 100;
+            const mockEmbeddings = Array(batchSize)
+                .fill(null)
+                .map(() => Array(1024).fill(Math.random()));
+            mockClient.embed.mockResolvedValue({
+                embeddings: mockEmbeddings,
+            });
+            const texts = Array(batchSize)
+                .fill(null)
+                .map((_, i) => `text ${i}`);
+            const results = await embeddings.embedBatch(texts);
+            expect(results).toHaveLength(batchSize);
+        });
+        it("should throw error if no embeddings returned", async () => {
+            mockClient.embed.mockResolvedValue({});
+            await expect(embeddings.embedBatch(["text1"])).rejects.toThrow("No embeddings returned from Cohere API");
+        });
+        it("should propagate errors in batch", async () => {
+            mockClient.embed.mockRejectedValue(new Error("Batch API Error"));
+            await expect(embeddings.embedBatch(["text1", "text2"])).rejects.toThrow("Batch API Error");
+        });
+    });
+    describe("getDimensions", () => {
+        it("should return configured dimensions", () => {
+            expect(embeddings.getDimensions()).toBe(1024);
+        });
+        it("should return custom dimensions", () => {
+            const customEmbeddings = new CohereEmbeddings("test-api-key", "embed-english-v3.0", 512);
+            expect(customEmbeddings.getDimensions()).toBe(512);
+        });
+    });
+    describe("getModel", () => {
+        it("should return configured model", () => {
+            expect(embeddings.getModel()).toBe("embed-english-v3.0");
+        });
+        it("should return custom model", () => {
+            const customEmbeddings = new CohereEmbeddings("test-api-key", "embed-multilingual-v3.0");
+            expect(customEmbeddings.getModel()).toBe("embed-multilingual-v3.0");
+        });
+    });
+    describe("rate limiting", () => {
+        it("should retry on rate limit error (status 429)", async () => {
+            const mockEmbedding = Array(1024).fill(0.5);
+            mockClient.embed
+                .mockRejectedValueOnce({ status: 429, message: "Rate limit exceeded" })
+                .mockRejectedValueOnce({ status: 429, message: "Rate limit exceeded" })
+                .mockResolvedValue({ embeddings: [mockEmbedding] });
+            const result = await embeddings.embed("test text");
+            expect(result.embedding).toEqual(mockEmbedding);
+            expect(mockClient.embed).toHaveBeenCalledTimes(3);
+        });
+        it("should retry on rate limit error (statusCode 429)", async () => {
+            const mockEmbedding = Array(1024).fill(0.5);
+            mockClient.embed
+                .mockRejectedValueOnce({
+                statusCode: 429,
+                message: "Rate limit exceeded",
+            })
+                .mockResolvedValue({ embeddings: [mockEmbedding] });
+            const result = await embeddings.embed("test text");
+            expect(result.embedding).toEqual(mockEmbedding);
+            expect(mockClient.embed).toHaveBeenCalledTimes(2);
+        });
+        it("should retry on rate limit message", async () => {
+            const mockEmbedding = Array(1024).fill(0.5);
+            mockClient.embed
+                .mockRejectedValueOnce({
+                message: "You have exceeded the rate limit",
+            })
+                .mockResolvedValue({ embeddings: [mockEmbedding] });
+            const result = await embeddings.embed("test text");
+            expect(result.embedding).toEqual(mockEmbedding);
+            expect(mockClient.embed).toHaveBeenCalledTimes(2);
+        });
+        it("should use exponential backoff", async () => {
+            const rateLimitEmbeddings = new CohereEmbeddings("test-api-key", "embed-english-v3.0", undefined, {
+                retryAttempts: 3,
+                retryDelayMs: 100,
+            });
+            const mockEmbedding = Array(1024).fill(0.5);
+            const rateLimitError = {
+                status: 429,
+                message: "Rate limit exceeded",
+            };
+            mockClient.embed
+                .mockRejectedValueOnce(rateLimitError)
+                .mockRejectedValueOnce(rateLimitError)
+                .mockResolvedValue({ embeddings: [mockEmbedding] });
+            const startTime = Date.now();
+            await rateLimitEmbeddings.embed("test text");
+            const duration = Date.now() - startTime;
+            // Should wait: 100ms (first retry) + 200ms (second retry) = 300ms
+            expect(duration).toBeGreaterThanOrEqual(250);
+        });
+        it("should throw error after max retries exceeded", async () => {
+            const rateLimitEmbeddings = new CohereEmbeddings("test-api-key", "embed-english-v3.0", undefined, {
+                retryAttempts: 2,
+                retryDelayMs: 100,
+            });
+            const rateLimitError = {
+                status: 429,
+                message: "Rate limit exceeded",
+            };
+            mockClient.embed.mockRejectedValue(rateLimitError);
+            await expect(rateLimitEmbeddings.embed("test text")).rejects.toThrow("Cohere API rate limit exceeded after 2 retry attempts");
+            expect(mockClient.embed).toHaveBeenCalledTimes(3);
+        });
+        it("should handle rate limit errors in batch operations", async () => {
+            const mockEmbeddings = [Array(1024).fill(0.1), Array(1024).fill(0.2)];
+            mockClient.embed
+                .mockRejectedValueOnce({ status: 429, message: "Rate limit exceeded" })
+                .mockResolvedValue({
+                embeddings: mockEmbeddings,
+            });
+            const results = await embeddings.embedBatch(["text1", "text2"]);
+            expect(results).toHaveLength(2);
+            expect(mockClient.embed).toHaveBeenCalledTimes(2);
+        });
+        it("should not retry on non-rate-limit errors", async () => {
+            const apiError = new Error("Invalid API key");
+            mockClient.embed.mockRejectedValue(apiError);
+            await expect(embeddings.embed("test text")).rejects.toThrow("Invalid API key");
+            expect(mockClient.embed).toHaveBeenCalledTimes(1);
+        });
+        it("should accept custom rate limit configuration", () => {
+            const customEmbeddings = new CohereEmbeddings("test-api-key", "embed-english-v3.0", undefined, {
+                maxRequestsPerMinute: 200,
+                retryAttempts: 5,
+                retryDelayMs: 2000,
+            });
+            expect(customEmbeddings).toBeDefined();
+        });
+    });
+});
+//# sourceMappingURL=cohere.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.test.js","sourceRoot":"","sources":["../../src/embeddings/cohere.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEzC,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1B,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;CACtB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,UAA4B,CAAC;IACjC,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG;YACX,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,UAAiB,CAAC,CAAC;QAEpE,UAAU,GAAG,IAAI,gBAAgB,CAAC,cAAc,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACzD,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,yBAAyB,CAC1B,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YACpE,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,oBAAoB,EACpB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,eAAe,GAAG,IAAI,gBAAgB,CAC1C,cAAc,EACd,0BAA0B,CAC3B,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,iBAAiB,GAAG,IAAI,gBAAgB,CAC5C,cAAc,EACd,cAAc,CACf,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,qBAAqB,GAAG,IAAI,gBAAgB,CAChD,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT,SAAS,EACT,cAAc,CACf,CAAC;YACF,MAAM,CAAC,qBAAqB,CAAC,CAAC,cAAc,CAAC,gBAAgB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC;iBAC9B,IAAI,CAAC,CAAC,CAAC;iBACP,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;YAC5B,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,SAAS,EAAE,aAAa;gBACxB,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,CAAC,WAAW,CAAC;gBACpB,KAAK,EAAE,oBAAoB;gBAC3B,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAEhD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,CAAC,QAAQ,CAAC;gBACjB,KAAK,EAAE,oBAAoB;gBAC3B,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,yBAAyB,EACzB,IAAI,CACL,CAAC;YACF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAErC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK,EAAE,CAAC,MAAM,CAAC;gBACf,KAAK,EAAE,yBAAyB;gBAChC,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACpD,uCAAuC,CACxC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YAE3D,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,cAAc,GAAG;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;aACtB,CAAC;YACF,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAC1C,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;gBACtB,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;gBAClD,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE;aACnD,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC;gBAC5C,KAAK;gBACL,KAAK,EAAE,oBAAoB;gBAC3B,SAAS,EAAE,iBAAiB;gBAC5B,cAAc,EAAE,CAAC,OAAO,CAAC;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAEhD,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,CAAC,aAAa,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;YAE7D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,SAAS,GAAG,GAAG,CAAC;YACtB,MAAM,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC;iBACpC,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAE9C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;gBACjC,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC;iBAC3B,IAAI,CAAC,IAAI,CAAC;iBACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC9B,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEvC,MAAM,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC5D,wCAAwC,CACzC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAEjE,MAAM,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACrE,iBAAiB,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,UAAU,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,oBAAoB,EACpB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,yBAAyB,CAC1B,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC;gBACrB,UAAU,EAAE,GAAG;gBACf,OAAO,EAAE,qBAAqB;aAC/B,CAAC;iBACD,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC;gBACrB,OAAO,EAAE,kCAAkC;aAC5C,CAAC;iBACD,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAChD,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG;aAClB,CACF,CAAC;YAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC,cAAc,CAAC;iBACrC,qBAAqB,CAAC,cAAc,CAAC;iBACrC,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAEtD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,kEAAkE;YAClE,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,GAAG,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG;aAClB,CACF,CAAC;YAEF,MAAM,cAAc,GAAG;gBACrB,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;aAC/B,CAAC;YAEF,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAEnD,MAAM,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAClE,uDAAuD,CACxD,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,cAAc,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAEtE,UAAU,CAAC,KAAK;iBACb,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC;gBACjB,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEL,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAEhE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC9C,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAE7C,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzD,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,oBAAoB,EACpB,SAAS,EACT;gBACE,oBAAoB,EAAE,GAAG;gBACzB,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,IAAI;aACnB,CACF,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/build/embeddings/factory.d.ts

@@ -0,0 +1,10 @@
+import { EmbeddingProvider, ProviderConfig } from "./base.js";
+export type EmbeddingProviderType = "openai" | "cohere" | "voyage" | "ollama";
+export interface FactoryConfig extends ProviderConfig {
+    provider: EmbeddingProviderType;
+}
+export declare class EmbeddingProviderFactory {
+    static create(config: FactoryConfig): EmbeddingProvider;
+    static createFromEnv(): EmbeddingProvider;
+}
+//# sourceMappingURL=factory.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"factory.d.ts","sourceRoot":"","sources":["../../src/embeddings/factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAM9D,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,CAAC;AAE9E,MAAM,WAAW,aAAc,SAAQ,cAAc;IACnD,QAAQ,EAAE,qBAAqB,CAAC;CACjC;AAED,qBAAa,wBAAwB;IACnC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,aAAa,GAAG,iBAAiB;IAsDvD,MAAM,CAAC,aAAa,IAAI,iBAAiB;CA+F1C"}

package/build/embeddings/factory.js

@@ -0,0 +1,98 @@
+import { OpenAIEmbeddings } from "./openai.js";
+import { CohereEmbeddings } from "./cohere.js";
+import { VoyageEmbeddings } from "./voyage.js";
+import { OllamaEmbeddings } from "./ollama.js";
+export class EmbeddingProviderFactory {
+    static create(config) {
+        const { provider, model, dimensions, rateLimitConfig, apiKey, baseUrl } = config;
+        switch (provider) {
+            case "openai":
+                if (!apiKey) {
+                    throw new Error("API key is required for OpenAI provider");
+                }
+                return new OpenAIEmbeddings(apiKey, model || "text-embedding-3-small", dimensions, rateLimitConfig);
+            case "cohere":
+                if (!apiKey) {
+                    throw new Error("API key is required for Cohere provider");
+                }
+                return new CohereEmbeddings(apiKey, model || "embed-english-v3.0", dimensions, rateLimitConfig);
+            case "voyage":
+                if (!apiKey) {
+                    throw new Error("API key is required for Voyage AI provider");
+                }
+                return new VoyageEmbeddings(apiKey, model || "voyage-2", dimensions, rateLimitConfig, baseUrl || "https://api.voyageai.com/v1");
+            case "ollama":
+                return new OllamaEmbeddings(model || "nomic-embed-text", dimensions, rateLimitConfig, baseUrl || "http://localhost:11434");
+            default:
+                throw new Error(`Unknown embedding provider: ${provider}. Supported providers: openai, cohere, voyage, ollama`);
+        }
+    }
+    static createFromEnv() {
+        const provider = (process.env.EMBEDDING_PROVIDER || "ollama").toLowerCase();
+        // Select API key based on provider
+        let apiKey;
+        switch (provider) {
+            case "openai":
+                apiKey = process.env.OPENAI_API_KEY;
+                break;
+            case "cohere":
+                apiKey = process.env.COHERE_API_KEY;
+                break;
+            case "voyage":
+                apiKey = process.env.VOYAGE_API_KEY;
+                break;
+            case "ollama":
+                // No API key needed for local Ollama
+                break;
+        }
+        // Common configuration
+        const model = process.env.EMBEDDING_MODEL;
+        const dimensions = process.env.EMBEDDING_DIMENSIONS
+            ? parseInt(process.env.EMBEDDING_DIMENSIONS, 10)
+            : undefined;
+        // Validate dimensions
+        if (dimensions !== undefined && (isNaN(dimensions) || dimensions <= 0)) {
+            throw new Error(`Invalid EMBEDDING_DIMENSIONS: must be a positive integer, got "${process.env.EMBEDDING_DIMENSIONS}"`);
+        }
+        const baseUrl = process.env.EMBEDDING_BASE_URL;
+        // Rate limiting configuration
+        const maxRequestsPerMinute = process.env.EMBEDDING_MAX_REQUESTS_PER_MINUTE
+            ? parseInt(process.env.EMBEDDING_MAX_REQUESTS_PER_MINUTE, 10)
+            : undefined;
+        // Validate maxRequestsPerMinute
+        if (maxRequestsPerMinute !== undefined &&
+            (isNaN(maxRequestsPerMinute) || maxRequestsPerMinute <= 0)) {
+            throw new Error(`Invalid EMBEDDING_MAX_REQUESTS_PER_MINUTE: must be a positive integer, got "${process.env.EMBEDDING_MAX_REQUESTS_PER_MINUTE}"`);
+        }
+        const retryAttempts = process.env.EMBEDDING_RETRY_ATTEMPTS
+            ? parseInt(process.env.EMBEDDING_RETRY_ATTEMPTS, 10)
+            : undefined;
+        // Validate retryAttempts
+        if (retryAttempts !== undefined &&
+            (isNaN(retryAttempts) || retryAttempts < 0)) {
+            throw new Error(`Invalid EMBEDDING_RETRY_ATTEMPTS: must be a non-negative integer, got "${process.env.EMBEDDING_RETRY_ATTEMPTS}"`);
+        }
+        const retryDelayMs = process.env.EMBEDDING_RETRY_DELAY
+            ? parseInt(process.env.EMBEDDING_RETRY_DELAY, 10)
+            : undefined;
+        // Validate retryDelayMs
+        if (retryDelayMs !== undefined &&
+            (isNaN(retryDelayMs) || retryDelayMs < 0)) {
+            throw new Error(`Invalid EMBEDDING_RETRY_DELAY: must be a non-negative integer, got "${process.env.EMBEDDING_RETRY_DELAY}"`);
+        }
+        const rateLimitConfig = {
+            maxRequestsPerMinute,
+            retryAttempts,
+            retryDelayMs,
+        };
+        return this.create({
+            provider,
+            model,
+            dimensions,
+            rateLimitConfig,
+            apiKey,
+            baseUrl,
+        });
+    }
+}
+//# sourceMappingURL=factory.js.map

package/build/embeddings/factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.js","sourceRoot":"","sources":["../../src/embeddings/factory.ts"],"names":[],"mappings":"AACA,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;AAQ/C,MAAM,OAAO,wBAAwB;IACnC,MAAM,CAAC,MAAM,CAAC,MAAqB;QACjC,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,EAAE,OAAO,EAAE,GACrE,MAAM,CAAC;QAET,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,QAAQ;gBACX,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;gBAC7D,CAAC;gBACD,OAAO,IAAI,gBAAgB,CACzB,MAAM,EACN,KAAK,IAAI,wBAAwB,EACjC,UAAU,EACV,eAAe,CAChB,CAAC;YAEJ,KAAK,QAAQ;gBACX,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;gBAC7D,CAAC;gBACD,OAAO,IAAI,gBAAgB,CACzB,MAAM,EACN,KAAK,IAAI,oBAAoB,EAC7B,UAAU,EACV,eAAe,CAChB,CAAC;YAEJ,KAAK,QAAQ;gBACX,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;gBAChE,CAAC;gBACD,OAAO,IAAI,gBAAgB,CACzB,MAAM,EACN,KAAK,IAAI,UAAU,EACnB,UAAU,EACV,eAAe,EACf,OAAO,IAAI,6BAA6B,CACzC,CAAC;YAEJ,KAAK,QAAQ;gBACX,OAAO,IAAI,gBAAgB,CACzB,KAAK,IAAI,kBAAkB,EAC3B,UAAU,EACV,eAAe,EACf,OAAO,IAAI,wBAAwB,CACpC,CAAC;YAEJ;gBACE,MAAM,IAAI,KAAK,CACb,+BAA+B,QAAQ,uDAAuD,CAC/F,CAAC;QACN,CAAC;IACH,CAAC;IAED,MAAM,CAAC,aAAa;QAClB,MAAM,QAAQ,GAAG,CACf,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,QAAQ,CAC3C,CAAC,WAAW,EAA2B,CAAC;QAEzC,mCAAmC;QACnC,IAAI,MAA0B,CAAC;QAC/B,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,QAAQ;gBACX,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;gBACpC,MAAM;YACR,KAAK,QAAQ;gBACX,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;gBACpC,MAAM;YACR,KAAK,QAAQ;gBACX,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;gBACpC,MAAM;YACR,KAAK,QAAQ;gBACX,qCAAqC;gBACrC,MAAM;QACV,CAAC;QAED,uBAAuB;QACvB,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;QAC1C,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB;YACjD,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,EAAE,EAAE,CAAC;YAChD,CAAC,CAAC,SAAS,CAAC;QAEd,sBAAsB;QACtB,IAAI,UAAU,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,UAAU,IAAI,CAAC,CAAC,EAAE,CAAC;YACvE,MAAM,IAAI,KAAK,CACb,kEAAkE,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,CACtG,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;QAE/C,8BAA8B;QAC9B,MAAM,oBAAoB,GAAG,OAAO,CAAC,GAAG,CAAC,iCAAiC;YACxE,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,iCAAiC,EAAE,EAAE,CAAC;YAC7D,CAAC,CAAC,SAAS,CAAC;QAEd,gCAAgC;QAChC,IACE,oBAAoB,KAAK,SAAS;YAClC,CAAC,KAAK,CAAC,oBAAoB,CAAC,IAAI,oBAAoB,IAAI,CAAC,CAAC,EAC1D,CAAC;YACD,MAAM,IAAI,KAAK,CACb,+EAA+E,OAAO,CAAC,GAAG,CAAC,iCAAiC,GAAG,CAChI,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB;YACxD,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,EAAE,CAAC;YACpD,CAAC,CAAC,SAAS,CAAC;QAEd,yBAAyB;QACzB,IACE,aAAa,KAAK,SAAS;YAC3B,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,aAAa,GAAG,CAAC,CAAC,EAC3C,CAAC;YACD,MAAM,IAAI,KAAK,CACb,0EAA0E,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,CAClH,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,qBAAqB;YACpD,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,EAAE,CAAC;YACjD,CAAC,CAAC,SAAS,CAAC;QAEd,wBAAwB;QACxB,IACE,YAAY,KAAK,SAAS;YAC1B,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC,CAAC,EACzC,CAAC;YACD,MAAM,IAAI,KAAK,CACb,uEAAuE,OAAO,CAAC,GAAG,CAAC,qBAAqB,GAAG,CAC5G,CAAC;QACJ,CAAC;QAED,MAAM,eAAe,GAAG;YACtB,oBAAoB;YACpB,aAAa;YACb,YAAY;SACb,CAAC;QAEF,OAAO,IAAI,CAAC,MAAM,CAAC;YACjB,QAAQ;YACR,KAAK;YACL,UAAU;YACV,eAAe;YACf,MAAM;YACN,OAAO;SACR,CAAC,CAAC;IACL,CAAC;CACF"}

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

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

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

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