diff --git a/src/auto-reply/reply.directive.directive-behavior.prefers-alias-matches-fuzzy-selection-is-ambiguous.test.ts b/src/auto-reply/reply.directive.directive-behavior.prefers-alias-matches-fuzzy-selection-is-ambiguous.test.ts index 271027882..b636b85d6 100644 --- a/src/auto-reply/reply.directive.directive-behavior.prefers-alias-matches-fuzzy-selection-is-ambiguous.test.ts +++ b/src/auto-reply/reply.directive.directive-behavior.prefers-alias-matches-fuzzy-selection-is-ambiguous.test.ts @@ -104,10 +104,11 @@ describe("directive behavior", () => { ); const text = Array.isArray(res) ? res[0]?.text : res?.text; - expect(text).toContain("Unrecognized model: ki"); - expect(text).toContain("Did you mean: moonshot/kimi-k2-0905-preview"); - expect(text).toContain("Try: /model moonshot/kimi-k2-0905-preview"); - assertModelSelection(storePath); + expect(text).toContain("Model set to Kimi (moonshot/kimi-k2-0905-preview)."); + assertModelSelection(storePath, { + provider: "moonshot", + model: "kimi-k2-0905-preview", + }); expect(runEmbeddedPiAgent).not.toHaveBeenCalled(); }); }); diff --git a/src/auto-reply/reply.directive.directive-behavior.supports-fuzzy-model-matches-model-directive.test.ts b/src/auto-reply/reply.directive.directive-behavior.supports-fuzzy-model-matches-model-directive.test.ts index cc7f54b22..2f0a6cc21 100644 --- a/src/auto-reply/reply.directive.directive-behavior.supports-fuzzy-model-matches-model-directive.test.ts +++ b/src/auto-reply/reply.directive.directive-behavior.supports-fuzzy-model-matches-model-directive.test.ts @@ -95,10 +95,11 @@ describe("directive behavior", () => { ); const text = Array.isArray(res) ? res[0]?.text : res?.text; - expect(text).toContain("Unrecognized model: kimi"); - expect(text).toContain("Did you mean: moonshot/kimi-k2-0905-preview"); - expect(text).toContain("Try: /model moonshot/kimi-k2-0905-preview"); - assertModelSelection(storePath); + expect(text).toContain("Model set to moonshot/kimi-k2-0905-preview."); + assertModelSelection(storePath, { + provider: "moonshot", + model: "kimi-k2-0905-preview", + }); expect(runEmbeddedPiAgent).not.toHaveBeenCalled(); }); }); @@ -142,10 +143,11 @@ describe("directive behavior", () => { ); const text = Array.isArray(res) ? res[0]?.text : res?.text; - expect(text).toContain("Unrecognized model: kimi-k2-0905-preview"); - expect(text).toContain("Did you mean: moonshot/kimi-k2-0905-preview"); - expect(text).toContain("Try: /model moonshot/kimi-k2-0905-preview"); - assertModelSelection(storePath); + expect(text).toContain("Model set to moonshot/kimi-k2-0905-preview."); + assertModelSelection(storePath, { + provider: "moonshot", + model: "kimi-k2-0905-preview", + }); expect(runEmbeddedPiAgent).not.toHaveBeenCalled(); }); }); @@ -184,10 +186,11 @@ describe("directive behavior", () => { ); const text = Array.isArray(res) ? res[0]?.text : res?.text; - expect(text).toContain("Unrecognized model: moonshot/kimi"); - expect(text).toContain("Did you mean: moonshot/kimi-k2-0905-preview"); - expect(text).toContain("Try: /model moonshot/kimi-k2-0905-preview"); - assertModelSelection(storePath); + expect(text).toContain("Model set to moonshot/kimi-k2-0905-preview."); + assertModelSelection(storePath, { + provider: "moonshot", + model: "kimi-k2-0905-preview", + }); expect(runEmbeddedPiAgent).not.toHaveBeenCalled(); }); });