@@ -57,6 +57,19 @@ function createSettings(params: {
5757} as unknown as ResolvedMemorySearchConfig;
5858}
595960+type MemoryFallbackProviderRequest = NonNullable<
61+ReturnType<typeof resolveMemoryFallbackProviderRequest>
62+>;
63+64+function expectMemoryFallbackRequest(
65+request: ReturnType<typeof resolveMemoryFallbackProviderRequest>,
66+): MemoryFallbackProviderRequest {
67+if (!request) {
68+throw new Error("Expected memory fallback provider request");
69+}
70+return request;
71+}
72+6073describe("memory manager mistral provider wiring", () => {
6174it("stores mistral client when mistral provider is selected", () => {
6275const mistralRuntime: EmbeddingProviderRuntime = {
@@ -72,7 +85,7 @@ describe("memory manager mistral provider wiring", () => {
7285providerUnavailableReason: undefined,
7386});
748775-expect(state.provider?.id).toBe("mistral");
88+expect(state.provider).toEqual(expect.objectContaining({ id: "mistral" }));
7689expect(state.providerRuntime).toBe(mistralRuntime);
7790});
7891@@ -105,7 +118,7 @@ describe("memory manager mistral provider wiring", () => {
105118106119expect(fallbackState.fallbackFrom).toBe("openai");
107120expect(fallbackState.fallbackReason).toBe("forced test");
108-expect(fallbackState.provider?.id).toBe("mistral");
121+expect(fallbackState.provider).toEqual(expect.objectContaining({ id: "mistral" }));
109122expect(fallbackState.providerRuntime).toBe(mistralRuntime);
110123});
111124@@ -116,9 +129,10 @@ describe("memory manager mistral provider wiring", () => {
116129currentProviderId: "openai",
117130});
118131119-expect(request?.provider).toBe("ollama");
120-expect(request?.model).toBe(DEFAULT_OLLAMA_EMBEDDING_MODEL);
121-expect(request?.fallback).toBe("none");
132+const fallbackRequest = expectMemoryFallbackRequest(request);
133+expect(fallbackRequest.provider).toBe("ollama");
134+expect(fallbackRequest.model).toBe(DEFAULT_OLLAMA_EMBEDDING_MODEL);
135+expect(fallbackRequest.fallback).toBe("none");
122136});
123137124138it("includes outputDimensionality in the primary provider request", () => {
@@ -158,8 +172,9 @@ describe("memory manager mistral provider wiring", () => {
158172currentProviderId: "openai",
159173});
160174161-expect(request?.provider).toBe("lmstudio");
162-expect(request?.model).toBe(DEFAULT_LMSTUDIO_EMBEDDING_MODEL);
163-expect(request?.fallback).toBe("none");
175+const fallbackRequest = expectMemoryFallbackRequest(request);
176+expect(fallbackRequest.provider).toBe("lmstudio");
177+expect(fallbackRequest.model).toBe(DEFAULT_LMSTUDIO_EMBEDDING_MODEL);
178+expect(fallbackRequest.fallback).toBe("none");
164179});
165180});
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。