@mhalder/qdrant-mcp-server 1.1.0

package/build/embeddings/openai.test.js

@@ -0,0 +1,283 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+import { OpenAIEmbeddings } from "./openai.js";
+import OpenAI from "openai";
+vi.mock("openai", () => ({
+    default: vi.fn(),
+}));
+describe("OpenAIEmbeddings", () => {
+    let embeddings;
+    let mockOpenAI;
+    beforeEach(() => {
+        mockOpenAI = {
+            embeddings: {
+                create: vi.fn(),
+            },
+        };
+        vi.mocked(OpenAI).mockImplementation(() => mockOpenAI);
+        embeddings = new OpenAIEmbeddings("test-api-key");
+    });
+    describe("constructor", () => {
+        it("should use default model and dimensions", () => {
+            expect(embeddings.getModel()).toBe("text-embedding-3-small");
+            expect(embeddings.getDimensions()).toBe(1536);
+        });
+        it("should use custom model", () => {
+            const customEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-3-large");
+            expect(customEmbeddings.getModel()).toBe("text-embedding-3-large");
+            expect(customEmbeddings.getDimensions()).toBe(3072);
+        });
+        it("should use custom dimensions", () => {
+            const customEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-3-small", 512);
+            expect(customEmbeddings.getDimensions()).toBe(512);
+        });
+        it("should use default dimensions for text-embedding-ada-002", () => {
+            const adaEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-ada-002");
+            expect(adaEmbeddings.getDimensions()).toBe(1536);
+        });
+    });
+    describe("embed", () => {
+        it("should generate embedding for single text", async () => {
+            const mockEmbedding = Array(1536)
+                .fill(0)
+                .map((_, i) => i * 0.001);
+            mockOpenAI.embeddings.create.mockResolvedValue({
+                data: [{ embedding: mockEmbedding }],
+            });
+            const result = await embeddings.embed("test text");
+            expect(result).toEqual({
+                embedding: mockEmbedding,
+                dimensions: 1536,
+            });
+            expect(mockOpenAI.embeddings.create).toHaveBeenCalledWith({
+                model: "text-embedding-3-small",
+                input: "test text",
+                dimensions: 1536,
+            });
+        });
+        it("should handle long text", async () => {
+            const longText = "word ".repeat(1000);
+            const mockEmbedding = Array(1536).fill(0.5);
+            mockOpenAI.embeddings.create.mockResolvedValue({
+                data: [{ embedding: mockEmbedding }],
+            });
+            const result = await embeddings.embed(longText);
+            expect(result.embedding).toEqual(mockEmbedding);
+            expect(mockOpenAI.embeddings.create).toHaveBeenCalledWith({
+                model: "text-embedding-3-small",
+                input: longText,
+                dimensions: 1536,
+            });
+        });
+        it("should use custom model configuration", async () => {
+            const customEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-3-large", 3072);
+            const mockEmbedding = Array(3072).fill(0.1);
+            mockOpenAI.embeddings.create.mockResolvedValue({
+                data: [{ embedding: mockEmbedding }],
+            });
+            await customEmbeddings.embed("test");
+            expect(mockOpenAI.embeddings.create).toHaveBeenCalledWith({
+                model: "text-embedding-3-large",
+                input: "test",
+                dimensions: 3072,
+            });
+        });
+        it("should propagate errors", async () => {
+            mockOpenAI.embeddings.create.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(1536).fill(0.1),
+                Array(1536).fill(0.2),
+                Array(1536).fill(0.3),
+            ];
+            mockOpenAI.embeddings.create.mockResolvedValue({
+                data: [
+                    { embedding: mockEmbeddings[0] },
+                    { embedding: mockEmbeddings[1] },
+                    { embedding: mockEmbeddings[2] },
+                ],
+            });
+            const texts = ["text1", "text2", "text3"];
+            const results = await embeddings.embedBatch(texts);
+            expect(results).toEqual([
+                { embedding: mockEmbeddings[0], dimensions: 1536 },
+                { embedding: mockEmbeddings[1], dimensions: 1536 },
+                { embedding: mockEmbeddings[2], dimensions: 1536 },
+            ]);
+            expect(mockOpenAI.embeddings.create).toHaveBeenCalledWith({
+                model: "text-embedding-3-small",
+                input: texts,
+                dimensions: 1536,
+            });
+        });
+        it("should handle empty batch", async () => {
+            mockOpenAI.embeddings.create.mockResolvedValue({
+                data: [],
+            });
+            const results = await embeddings.embedBatch([]);
+            expect(results).toEqual([]);
+        });
+        it("should handle single item in batch", async () => {
+            const mockEmbedding = Array(1536).fill(0.5);
+            mockOpenAI.embeddings.create.mockResolvedValue({
+                data: [{ embedding: 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(1536).fill(Math.random()));
+            mockOpenAI.embeddings.create.mockResolvedValue({
+                data: mockEmbeddings.map((embedding) => ({ embedding })),
+            });
+            const texts = Array(batchSize)
+                .fill(null)
+                .map((_, i) => `text ${i}`);
+            const results = await embeddings.embedBatch(texts);
+            expect(results).toHaveLength(batchSize);
+        });
+        it("should propagate errors in batch", async () => {
+            mockOpenAI.embeddings.create.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(1536);
+        });
+        it("should return custom dimensions", () => {
+            const customEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-3-small", 512);
+            expect(customEmbeddings.getDimensions()).toBe(512);
+        });
+    });
+    describe("getModel", () => {
+        it("should return configured model", () => {
+            expect(embeddings.getModel()).toBe("text-embedding-3-small");
+        });
+        it("should return custom model", () => {
+            const customEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-3-large");
+            expect(customEmbeddings.getModel()).toBe("text-embedding-3-large");
+        });
+    });
+    describe("rate limiting", () => {
+        it("should retry on rate limit error (429 status)", async () => {
+            const mockEmbedding = Array(1536).fill(0.5);
+            // Fail first two times with rate limit, succeed on third
+            mockOpenAI.embeddings.create
+                .mockRejectedValueOnce({ status: 429, message: "Rate limit exceeded" })
+                .mockRejectedValueOnce({ status: 429, message: "Rate limit exceeded" })
+                .mockResolvedValue({ data: [{ embedding: mockEmbedding }] });
+            const result = await embeddings.embed("test text");
+            expect(result.embedding).toEqual(mockEmbedding);
+            expect(mockOpenAI.embeddings.create).toHaveBeenCalledTimes(3);
+        });
+        it("should respect Retry-After header when present", async () => {
+            const mockEmbedding = Array(1536).fill(0.5);
+            const rateLimitError = {
+                status: 429,
+                message: "Rate limit exceeded",
+                headers: { "retry-after": "2" },
+            };
+            mockOpenAI.embeddings.create
+                .mockRejectedValueOnce(rateLimitError)
+                .mockResolvedValue({ data: [{ embedding: mockEmbedding }] });
+            const startTime = Date.now();
+            await embeddings.embed("test text");
+            const duration = Date.now() - startTime;
+            // Should wait at least 2 seconds (2000ms)
+            expect(duration).toBeGreaterThanOrEqual(1900); // Allow small margin
+        });
+        it("should fallback to exponential backoff with invalid Retry-After header", async () => {
+            const rateLimitEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-3-small", undefined, {
+                retryAttempts: 2,
+                retryDelayMs: 100, // 100ms for faster tests
+            });
+            const mockEmbedding = Array(1536).fill(0.5);
+            // Test various invalid Retry-After values
+            const invalidRetryAfterError = {
+                status: 429,
+                message: "Rate limit exceeded",
+                headers: { "retry-after": "invalid" }, // Non-numeric value
+            };
+            mockOpenAI.embeddings.create
+                .mockRejectedValueOnce(invalidRetryAfterError)
+                .mockResolvedValue({ data: [{ embedding: mockEmbedding }] });
+            const startTime = Date.now();
+            await rateLimitEmbeddings.embed("test text");
+            const duration = Date.now() - startTime;
+            // Should fallback to exponential backoff (100ms) instead of using invalid header
+            expect(duration).toBeGreaterThanOrEqual(90); // Allow small margin
+            expect(duration).toBeLessThan(500); // Should not wait too long
+        });
+        it("should use exponential backoff when no Retry-After header", async () => {
+            const rateLimitEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-3-small", undefined, {
+                retryAttempts: 3,
+                retryDelayMs: 100, // 100ms for faster tests
+            });
+            const mockEmbedding = Array(1536).fill(0.5);
+            const rateLimitError = {
+                status: 429,
+                message: "Rate limit exceeded",
+            };
+            mockOpenAI.embeddings.create
+                .mockRejectedValueOnce(rateLimitError)
+                .mockRejectedValueOnce(rateLimitError)
+                .mockResolvedValue({ data: [{ embedding: 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); // Allow margin for test execution
+        });
+        it("should throw error after max retries exceeded", async () => {
+            const rateLimitEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-3-small", undefined, {
+                retryAttempts: 2,
+                retryDelayMs: 100,
+            });
+            const rateLimitError = {
+                status: 429,
+                message: "Rate limit exceeded",
+            };
+            mockOpenAI.embeddings.create.mockRejectedValue(rateLimitError);
+            await expect(rateLimitEmbeddings.embed("test text")).rejects.toThrow("OpenAI API rate limit exceeded after 2 retry attempts");
+            // Should try initial + 2 retries = 3 total attempts
+            expect(mockOpenAI.embeddings.create).toHaveBeenCalledTimes(3);
+        });
+        it("should handle rate limit errors in batch operations", async () => {
+            const mockEmbeddings = [Array(1536).fill(0.1), Array(1536).fill(0.2)];
+            mockOpenAI.embeddings.create
+                .mockRejectedValueOnce({ status: 429, message: "Rate limit exceeded" })
+                .mockResolvedValue({
+                data: [
+                    { embedding: mockEmbeddings[0] },
+                    { embedding: mockEmbeddings[1] },
+                ],
+            });
+            const results = await embeddings.embedBatch(["text1", "text2"]);
+            expect(results).toHaveLength(2);
+            expect(mockOpenAI.embeddings.create).toHaveBeenCalledTimes(2);
+        });
+        it("should not retry on non-rate-limit errors", async () => {
+            const apiError = new Error("Invalid API key");
+            mockOpenAI.embeddings.create.mockRejectedValue(apiError);
+            await expect(embeddings.embed("test text")).rejects.toThrow("Invalid API key");
+            expect(mockOpenAI.embeddings.create).toHaveBeenCalledTimes(1);
+        });
+        it("should accept custom rate limit configuration", () => {
+            const customEmbeddings = new OpenAIEmbeddings("test-api-key", "text-embedding-3-small", undefined, {
+                maxRequestsPerMinute: 1000,
+                retryAttempts: 5,
+                retryDelayMs: 2000,
+            });
+            expect(customEmbeddings).toBeDefined();
+        });
+    });
+});
+//# sourceMappingURL=openai.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai.test.js","sourceRoot":"","sources":["../../src/embeddings/openai.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,MAAM,MAAM,QAAQ,CAAC;AAE5B,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;IACvB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;CACjB,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,UAAU,EAAE;gBACV,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;aAChB;SACF,CAAC;QAEF,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,UAAiB,CAAC,CAAC;QAE9D,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,wBAAwB,CAAC,CAAC;YAC7D,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,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACnE,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,wBAAwB,EACxB,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,aAAa,GAAG,IAAI,gBAAgB,CACxC,cAAc,EACd,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,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,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,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,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,WAAW;gBAClB,UAAU,EAAE,IAAI;aACjB,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,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,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,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,QAAQ;gBACf,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,IAAI,CACL,CAAC;YACF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC5C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,CAAC,CAAC;YAEH,MAAM,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAErC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,MAAM;gBACb,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YAEvE,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,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE;oBACJ,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAChC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAChC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;iBACjC;aACF,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,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC;gBACxD,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,KAAK;gBACZ,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACzC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,EAAE;aACT,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,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;aACrC,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,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC;gBAC7C,IAAI,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;aACzD,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,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAC5C,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAC7B,CAAC;YAEF,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,wBAAwB,EACxB,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,wBAAwB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACrE,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,yDAAyD;YACzD,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,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,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,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,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,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;gBAC9B,OAAO,EAAE,EAAE,aAAa,EAAE,GAAG,EAAE;aAChC,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,cAAc,CAAC;iBACrC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,0CAA0C;YAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAC,CAAC,qBAAqB;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG,EAAE,yBAAyB;aAC7C,CACF,CAAC;YAEF,MAAM,aAAa,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5C,0CAA0C;YAC1C,MAAM,sBAAsB,GAAG;gBAC7B,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,qBAAqB;gBAC9B,OAAO,EAAE,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,oBAAoB;aAC5D,CAAC;YAEF,UAAU,CAAC,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,sBAAsB,CAAC;iBAC7C,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,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,iFAAiF;YACjF,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC,CAAC,qBAAqB;YAClE,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,2BAA2B;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,GAAG,EAAE,yBAAyB;aAC7C,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,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,cAAc,CAAC;iBACrC,qBAAqB,CAAC,cAAc,CAAC;iBACrC,iBAAiB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;YAE/D,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,CAAC,kCAAkC;QAClF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,mBAAmB,GAAG,IAAI,gBAAgB,CAC9C,cAAc,EACd,wBAAwB,EACxB,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,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE/D,MAAM,MAAM,CAAC,mBAAmB,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAClE,uDAAuD,CACxD,CAAC;YAEF,oDAAoD;YACpD,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,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,UAAU,CAAC,MAAM;iBACzB,qBAAqB,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,qBAAqB,EAAE,CAAC;iBACtE,iBAAiB,CAAC;gBACjB,IAAI,EAAE;oBACJ,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;oBAChC,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE;iBACjC;aACF,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,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAC9C,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEzD,MAAM,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACzD,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAC3C,cAAc,EACd,wBAAwB,EACxB,SAAS,EACT;gBACE,oBAAoB,EAAE,IAAI;gBAC1B,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/voyage.d.ts

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"voyage.d.ts","sourceRoot":"","sources":["../../src/embeddings/voyage.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAehF,qBAAa,gBAAiB,YAAW,iBAAiB;IACxD,OAAO,CAAC,MAAM,CAAS;IACvB,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;IACxB,OAAO,CAAC,SAAS,CAAC,CAAuB;gBAGvC,MAAM,EAAE,MAAM,EACd,KAAK,GAAE,MAAmB,EAC1B,UAAU,CAAC,EAAE,MAAM,EACnB,eAAe,CAAC,EAAE,eAAe,EACjC,OAAO,GAAE,MAAsC,EAC/C,SAAS,CAAC,EAAE,OAAO,GAAG,UAAU;YA+BpB,gBAAgB;YAiChB,OAAO;IA8Bf,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAiB7C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAiB7D,aAAa,IAAI,MAAM;IAIvB,QAAQ,IAAI,MAAM;CAGnB"}

package/build/embeddings/voyage.js

@@ -0,0 +1,113 @@
+import Bottleneck from "bottleneck";
+export class VoyageEmbeddings {
+    apiKey;
+    model;
+    dimensions;
+    limiter;
+    retryAttempts;
+    retryDelayMs;
+    baseUrl;
+    inputType;
+    constructor(apiKey, model = "voyage-2", dimensions, rateLimitConfig, baseUrl = "https://api.voyageai.com/v1", inputType) {
+        this.apiKey = apiKey;
+        this.model = model;
+        this.baseUrl = baseUrl;
+        this.inputType = inputType;
+        // Default dimensions for different models
+        const defaultDimensions = {
+            "voyage-2": 1024,
+            "voyage-large-2": 1536,
+            "voyage-code-2": 1536,
+            "voyage-lite-02-instruct": 1024,
+        };
+        this.dimensions = dimensions || defaultDimensions[model] || 1024;
+        // Rate limiting configuration
+        const maxRequestsPerMinute = rateLimitConfig?.maxRequestsPerMinute || 300;
+        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?.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(`Voyage AI API rate limit exceeded after ${this.retryAttempts} retry attempts. Please try again later or reduce request frequency.`);
+            }
+            throw error;
+        }
+    }
+    async callApi(texts) {
+        const body = {
+            input: texts,
+            model: this.model,
+        };
+        if (this.inputType) {
+            body.input_type = this.inputType;
+        }
+        const response = await fetch(`${this.baseUrl}/embeddings`, {
+            method: "POST",
+            headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${this.apiKey}`,
+            },
+            body: JSON.stringify(body),
+        });
+        if (!response.ok) {
+            const error = {
+                status: response.status,
+                message: await response.text(),
+            };
+            throw error;
+        }
+        return response.json();
+    }
+    async embed(text) {
+        return this.limiter.schedule(() => this.retryWithBackoff(async () => {
+            const response = await this.callApi([text]);
+            if (!response.data || response.data.length === 0) {
+                throw new Error("No embedding returned from Voyage AI API");
+            }
+            return {
+                embedding: response.data[0].embedding,
+                dimensions: this.dimensions,
+            };
+        }));
+    }
+    async embedBatch(texts) {
+        return this.limiter.schedule(() => this.retryWithBackoff(async () => {
+            const response = await this.callApi(texts);
+            if (!response.data) {
+                throw new Error("No embeddings returned from Voyage AI API");
+            }
+            return response.data.map((item) => ({
+                embedding: item.embedding,
+                dimensions: this.dimensions,
+            }));
+        }));
+    }
+    getDimensions() {
+        return this.dimensions;
+    }
+    getModel() {
+        return this.model;
+    }
+}
+//# sourceMappingURL=voyage.js.map

package/build/embeddings/voyage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"voyage.js","sourceRoot":"","sources":["../../src/embeddings/voyage.ts"],"names":[],"mappings":"AAAA,OAAO,UAAU,MAAM,YAAY,CAAC;AAgBpC,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAS;IACf,KAAK,CAAS;IACd,UAAU,CAAS;IACnB,OAAO,CAAa;IACpB,aAAa,CAAS;IACtB,YAAY,CAAS;IACrB,OAAO,CAAS;IAChB,SAAS,CAAwB;IAEzC,YACE,MAAc,EACd,QAAgB,UAAU,EAC1B,UAAmB,EACnB,eAAiC,EACjC,UAAkB,6BAA6B,EAC/C,SAAgC;QAEhC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,0CAA0C;QAC1C,MAAM,iBAAiB,GAA2B;YAChD,UAAU,EAAE,IAAI;YAChB,gBAAgB,EAAE,IAAI;YACtB,eAAe,EAAE,IAAI;YACrB,yBAAyB,EAAE,IAAI;SAChC,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,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,2CAA2C,IAAI,CAAC,aAAa,sEAAsE,CACpI,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,KAAe;QACnC,MAAM,IAAI,GAAQ;YAChB,KAAK,EAAE,KAAK;YACZ,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC;QAEF,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC;QACnC,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,aAAa,EAAE;YACzD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;aACvC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,KAAK,GAAgB;gBACzB,MAAM,EAAE,QAAQ,CAAC,MAAM;gBACvB,OAAO,EAAE,MAAM,QAAQ,CAAC,IAAI,EAAE;aAC/B,CAAC;YACF,MAAM,KAAK,CAAC;QACd,CAAC;QAED,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;IACzB,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,CAAC,IAAI,CAAC,CAAC,CAAC;YAE5C,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACjD,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;YAC9D,CAAC;YAED,OAAO;gBACL,SAAS,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;gBACrC,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,OAAO,CAAC,KAAK,CAAC,CAAC;YAE3C,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;YAC/D,CAAC;YAED,OAAO,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAClC,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,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/voyage.test.d.ts

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

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

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