From b66c3b4fc3ac1b5d378caa087fd9af32d02ca301 Mon Sep 17 00:00:00 2001 From: Shakker Date: Mon, 11 May 2026 06:49:57 +0100 Subject: [PATCH] test: tighten memory embedding provider assertions --- .../memory-embedding-provider-runtime.test.ts | 35 +++++++++---------- 1 file changed, 17 insertions(+), 18 deletions(-) diff --git a/src/plugins/memory-embedding-provider-runtime.test.ts b/src/plugins/memory-embedding-provider-runtime.test.ts index 96ca02cbecb..70b8efcb746 100644 --- a/src/plugins/memory-embedding-provider-runtime.test.ts +++ b/src/plugins/memory-embedding-provider-runtime.test.ts @@ -75,32 +75,33 @@ describe("memory embedding provider runtime resolution", () => { it("uses a configured provider api as the memory adapter owner", () => { const ollamaAdapter = createCapabilityAdapter("ollama"); + const config = { + models: { + providers: { + "ollama-5080": { + api: "ollama", + baseUrl: "http://10.0.0.8:11435", + models: [], + }, + }, + }, + }; mocks.resolvePluginCapabilityProvider.mockImplementation(({ providerId }) => providerId === "ollama" ? ollamaAdapter : undefined, ); - expect( - runtimeModule.getMemoryEmbeddingProvider("ollama-5080", { - models: { - providers: { - "ollama-5080": { - api: "ollama", - baseUrl: "http://10.0.0.8:11435", - models: [], - }, - }, - }, - } as never), - ).toBe(ollamaAdapter); + expect(runtimeModule.getMemoryEmbeddingProvider("ollama-5080", config as never)).toBe( + ollamaAdapter, + ); expect(mocks.resolvePluginCapabilityProvider).toHaveBeenCalledWith({ key: "memoryEmbeddingProviders", providerId: "ollama-5080", - cfg: expect.any(Object), + cfg: config, }); expect(mocks.resolvePluginCapabilityProvider).toHaveBeenCalledWith({ key: "memoryEmbeddingProviders", providerId: "ollama", - cfg: expect.any(Object), + cfg: config, }); }); @@ -134,9 +135,7 @@ describe("memory embedding provider runtime resolution", () => { }); mocks.resolvePluginCapabilityProviders.mockReturnValue([createCapabilityAdapter("openai")]); - expect(runtimeModule.getMemoryEmbeddingProvider("openai")).toEqual( - expect.objectContaining({ id: "openai" }), - ); + expect(runtimeModule.getMemoryEmbeddingProvider("openai")).toStrictEqual(registered); expect(runtimeModule.listMemoryEmbeddingProviders().map((adapter) => adapter.id)).toEqual([ "openai", ]);