@docyrus/docyrus 0.0.44 → 0.0.45

package/agent-loader.js

@@ -2869,18 +2869,35 @@ async function main() {
   });
   await resourceLoader.reload();
   spinner.update("Creating session...");
-  const { session, modelFallbackMessage, extensionsResult } = await pi.createAgentSession({
-    cwd,
-    agentDir,
-    authStorage,
-    modelRegistry,
-    resourceLoader,
-    settingsManager,
-    sessionManager,
-    tools: buildTools(request.profile, cwd, pi),
-    model: requestedModel,
-    thinkingLevel: request.thinking
-  });
+  const tools = buildTools(request.profile, cwd, pi);
+  const createRuntime = async (options) => {
+    const created = await pi.createAgentSession({
+      cwd: options.cwd,
+      agentDir: options.agentDir,
+      authStorage,
+      modelRegistry,
+      resourceLoader,
+      settingsManager,
+      sessionManager: options.sessionManager,
+      tools,
+      model: requestedModel,
+      thinkingLevel: request.thinking,
+      sessionStartEvent: options.sessionStartEvent
+    });
+    return {
+      ...created,
+      services: {
+        cwd: options.cwd,
+        agentDir: options.agentDir,
+        authStorage,
+        modelRegistry,
+        resourceLoader,
+        settingsManager
+      }
+    };
+  };
+  const initialResult = await createRuntime({ cwd, agentDir, sessionManager });
+  const { session, modelFallbackMessage, extensionsResult } = initialResult;
   if (hasPackagedMcpAdapter) {
     extensionsResult.runtime.flagValues.set("mcp-config", mcpConfigPath);
   }
@@ -2892,15 +2909,7 @@ async function main() {
   if (pipedStdin !== void 0 && !request.print) {
     request.print = true;
   }
-  const runtime = new pi.AgentSessionRuntime(session, {
-    cwd,
-    agentDir,
-    authStorage,
-    modelRegistry,
-    resourceLoader,
-    settingsManager,
-    sessionManager
-  });
+  const runtime = new pi.AgentSessionRuntime(session, initialResult.services, createRuntime);
   if (request.print || pipedStdin !== void 0) {
     if (!session.model) {
       throw new Error(`No models available.