@@ -75,32 +75,33 @@ describe("memory embedding provider runtime resolution", () => {
|
75 | 75 | |
76 | 76 | it("uses a configured provider api as the memory adapter owner", () => { |
77 | 77 | const ollamaAdapter = createCapabilityAdapter("ollama"); |
| 78 | +const config = { |
| 79 | +models: { |
| 80 | +providers: { |
| 81 | +"ollama-5080": { |
| 82 | +api: "ollama", |
| 83 | +baseUrl: "http://10.0.0.8:11435", |
| 84 | +models: [], |
| 85 | +}, |
| 86 | +}, |
| 87 | +}, |
| 88 | +}; |
78 | 89 | mocks.resolvePluginCapabilityProvider.mockImplementation(({ providerId }) => |
79 | 90 | providerId === "ollama" ? ollamaAdapter : undefined, |
80 | 91 | ); |
81 | 92 | |
82 | | -expect( |
83 | | -runtimeModule.getMemoryEmbeddingProvider("ollama-5080", { |
84 | | -models: { |
85 | | -providers: { |
86 | | -"ollama-5080": { |
87 | | -api: "ollama", |
88 | | -baseUrl: "http://10.0.0.8:11435", |
89 | | -models: [], |
90 | | -}, |
91 | | -}, |
92 | | -}, |
93 | | -} as never), |
94 | | -).toBe(ollamaAdapter); |
| 93 | +expect(runtimeModule.getMemoryEmbeddingProvider("ollama-5080", config as never)).toBe( |
| 94 | +ollamaAdapter, |
| 95 | +); |
95 | 96 | expect(mocks.resolvePluginCapabilityProvider).toHaveBeenCalledWith({ |
96 | 97 | key: "memoryEmbeddingProviders", |
97 | 98 | providerId: "ollama-5080", |
98 | | -cfg: expect.any(Object), |
| 99 | +cfg: config, |
99 | 100 | }); |
100 | 101 | expect(mocks.resolvePluginCapabilityProvider).toHaveBeenCalledWith({ |
101 | 102 | key: "memoryEmbeddingProviders", |
102 | 103 | providerId: "ollama", |
103 | | -cfg: expect.any(Object), |
| 104 | +cfg: config, |
104 | 105 | }); |
105 | 106 | }); |
106 | 107 | |
@@ -134,9 +135,7 @@ describe("memory embedding provider runtime resolution", () => {
|
134 | 135 | }); |
135 | 136 | mocks.resolvePluginCapabilityProviders.mockReturnValue([createCapabilityAdapter("openai")]); |
136 | 137 | |
137 | | -expect(runtimeModule.getMemoryEmbeddingProvider("openai")).toEqual( |
138 | | -expect.objectContaining({ id: "openai" }), |
139 | | -); |
| 138 | +expect(runtimeModule.getMemoryEmbeddingProvider("openai")).toStrictEqual(registered); |
140 | 139 | expect(runtimeModule.listMemoryEmbeddingProviders().map((adapter) => adapter.id)).toEqual([ |
141 | 140 | "openai", |
142 | 141 | ]); |
|
