Tests: update pi SDK mocks

src/agents/model-catalog.test.ts

@@ -38,8 +38,12 @@ describe("loadModelCatalog", () => {
         throw new Error("boom");
       }
       return {
-        discoverAuthStorage: () => ({}),
-        discoverModels: () => [{ id: "gpt-4.1", name: "GPT-4.1", provider: "openai" }],
+        AuthStorage: class {},
+        ModelRegistry: class {
+          getAll() {
+            return [{ id: "gpt-4.1", name: "GPT-4.1", provider: "openai" }];
+          }
+        },
       } as unknown as PiSdkModule;
     });
 
@@ -59,19 +63,21 @@ describe("loadModelCatalog", () => {
     __setModelCatalogImportForTest(
       async () =>
         ({
-          discoverAuthStorage: () => ({}),
-          discoverModels: () => ({
-            getAll: () => [
-              { id: "gpt-4.1", name: "GPT-4.1", provider: "openai" },
-              {
-                get id() {
-                  throw new Error("boom");
+          AuthStorage: class {},
+          ModelRegistry: class {
+            getAll() {
+              return [
+                { id: "gpt-4.1", name: "GPT-4.1", provider: "openai" },
+                {
+                  get id() {
+                    throw new Error("boom");
+                  },
+                  provider: "openai",
+                  name: "bad",
                 },
-                provider: "openai",
-                name: "bad",
-              },
-            ],
-          }),
+              ];
+            }
+          },
         }) as unknown as PiSdkModule,
     );
 

src/agents/pi-embedded-runner/model.test.ts

@@ -1,8 +1,14 @@
 import { describe, expect, it, vi } from "vitest";
 
 vi.mock("@mariozechner/pi-coding-agent", () => ({
-  discoverAuthStorage: vi.fn(() => ({ mocked: true })),
-  discoverModels: vi.fn(() => ({ find: vi.fn(() => null) })),
+  AuthStorage: class {
+    mocked = true;
+  },
+  ModelRegistry: class {
+    find() {
+      return null;
+    }
+  },
 }));
 
 import type { OpenClawConfig } from "../../config/config.js";