Tests: fix models list e2e registry mock

2026-03-06 00:43:19 -05:00 · 2026-03-06 00:43:19 -05:00 · 298f9a4f60
commit 298f9a4f60
parent e21f512e5e
1 changed files with 9 additions and 0 deletions
--- a/src/commands/models.list.e2e.test.ts
+++ b/src/commands/models.list.e2e.test.ts
@ -87,6 +87,15 @@ vi.mock("../agents/pi-embedded-runner/model.js", () => ({
  resolveModel: () => {
    throw new Error("resolveModel should not be called from models.list tests");
  },
+  resolveModelWithRegistry: ({
+    provider,
+    modelId,
+    modelRegistry,
+  }: {
+    provider: string;
+    modelId: string;
+    modelRegistry: { find: (provider: string, modelId: string) => unknown };
+  }) => modelRegistry.find(provider, modelId),
 }));

 function makeRuntime() {