diff --git a/packages/engine/src/pi.ts b/packages/engine/src/pi.ts
index 7deca1650..8273c4869 100644
--- a/packages/engine/src/pi.ts
+++ b/packages/engine/src/pi.ts
@@ -249,7 +249,7 @@ async function registerExtensionProviders(cwd: string, modelRegistry: ModelRegis
 export async function createKbAgent(options: AgentOptions): Promise<AgentResult> {
   console.log(`[pi] createKbAgent called (cwd=${options.cwd}, tools=${options.tools}, provider=${options.defaultProvider}, model=${options.defaultModelId})`);
   const authStorage = AuthStorage.create();
-  const modelRegistry = new ModelRegistry(authStorage);
+  const modelRegistry = new ModelRegistry(authStorage, join(getAgentDir(), "models.json"));
   await registerExtensionProviders(options.cwd, modelRegistry);
 
   const tools =