From 7dffd8160a255152a2ac1c84da30b04da69f1441 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Tue, 31 Mar 2026 23:49:05 +0900
Subject: [PATCH] test(extensions): use ModelRegistry factory

---
 extensions/openai/index.test.ts     | 2 +-
 extensions/openrouter/index.test.ts | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/extensions/openai/index.test.ts b/extensions/openai/index.test.ts
index 963b9f16ab1..da5df9c6b38 100644
--- a/extensions/openai/index.test.ts
+++ b/extensions/openai/index.test.ts
@@ -58,7 +58,7 @@ function resolveTemplateModelId(modelId: string) {
 }
 
 function createTemplateModelRegistry(modelId: string): ModelRegistry {
-  const registry = new ModelRegistry(AuthStorage.inMemory());
+  const registry = ModelRegistry.inMemory(AuthStorage.inMemory());
   const template = getModel("openai", resolveTemplateModelId(modelId));
   registry.registerProvider("openai", {
     apiKey: "test",
diff --git a/extensions/openrouter/index.test.ts b/extensions/openrouter/index.test.ts
index 3de534486bb..6a3c24a1ea3 100644
--- a/extensions/openrouter/index.test.ts
+++ b/extensions/openrouter/index.test.ts
@@ -41,7 +41,7 @@ describeLive("openrouter plugin live", () => {
     const resolved = provider.resolveDynamicModel?.({
       provider: "openrouter",
       modelId: LIVE_MODEL_ID,
-      modelRegistry: new ModelRegistry(AuthStorage.inMemory()),
+      modelRegistry: ModelRegistry.inMemory(AuthStorage.inMemory()),
     });
     if (!resolved) {
       throw new Error(`openrouter provider did not resolve ${LIVE_MODEL_ID}`);