diff --git a/src/memory/embeddings-voyage.test.ts b/src/memory/embeddings-voyage.test.ts index e47710689e..08e59474ae 100644 --- a/src/memory/embeddings-voyage.test.ts +++ b/src/memory/embeddings-voyage.test.ts @@ -19,6 +19,28 @@ const createFetchMock = () => { return withFetchPreconnect(fetchMock); }; +function mockVoyageApiKey() { + vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({ + apiKey: "voyage-key-123", + mode: "api-key", + source: "test", + }); +} + +async function createDefaultVoyageProvider( + model: string, + fetchMock: ReturnType, +) { + vi.stubGlobal("fetch", fetchMock); + mockVoyageApiKey(); + return createVoyageEmbeddingProvider({ + config: {} as never, + provider: "voyage", + model, + fallback: "none", + }); +} + describe("voyage embedding provider", () => { afterEach(() => { vi.resetAllMocks(); @@ -27,20 +49,7 @@ describe("voyage embedding provider", () => { it("configures client with correct defaults and headers", async () => { const fetchMock = createFetchMock(); - vi.stubGlobal("fetch", fetchMock); - - vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({ - apiKey: "voyage-key-123", - mode: "api-key", - source: "test", - }); - - const result = await createVoyageEmbeddingProvider({ - config: {} as never, - provider: "voyage", - model: "voyage-4-large", - fallback: "none", - }); + const result = await createDefaultVoyageProvider("voyage-4-large", fetchMock); await result.provider.embedQuery("test query"); @@ -105,20 +114,7 @@ describe("voyage embedding provider", () => { ), ), ); - vi.stubGlobal("fetch", fetchMock); - - vi.mocked(authModule.resolveApiKeyForProvider).mockResolvedValue({ - apiKey: "voyage-key-123", - mode: "api-key", - source: "test", - }); - - const result = await createVoyageEmbeddingProvider({ - config: {} as never, - provider: "voyage", - model: "voyage-4-large", - fallback: "none", - }); + const result = await createDefaultVoyageProvider("voyage-4-large", fetchMock); await result.provider.embedBatch(["doc1", "doc2"]);