@aibyzero/byz 0.1.3 → 0.1.5

package/dist/runtime/core/agent-session.js

@@ -945,7 +945,7 @@ export class AgentSession {
         if (!command)
             return false;
         // Get command context from extension runner (includes session control methods)
-        const ctx = this._extensionRunner.createCommandContext();
+        const ctx = this._extensionRunner.createCommandContext(command.ownerId, command.sourceInfo.path, command.byzWorkflowOwner);
         try {
             await command.handler(args, ctx);
             return true;
@@ -1201,6 +1201,24 @@ export class AgentSession {
     get resourceLoader() {
         return this._resourceLoader;
     }
+    async replaceByzWorkflowResources(extensionOwner, extensionPath, resources) {
+        if (!this.isIdle || this.isCompacting) {
+            throw new Error("Extension resources cannot be updated while the agent is running.");
+        }
+        if (extensionPath !== "<inline:byz-workflow>") {
+            throw new Error("BYZ workflow resources can only be replaced by the built-in workflow extension.");
+        }
+        if (!this._resourceLoader.replaceByzWorkflowResources) {
+            throw new Error("The active resource loader does not support BYZ workflow resource updates.");
+        }
+        this._resourceLoader.replaceByzWorkflowResources(extensionOwner, {
+            promptPaths: this.buildExtensionResourcePaths((resources.promptPaths ?? []).map((path) => ({ path, extensionOwner, extensionPath }))),
+            skillPaths: this.buildExtensionResourcePaths((resources.skillPaths ?? []).map((path) => ({ path, extensionOwner, extensionPath }))),
+        });
+        this._baseSystemPrompt = this._rebuildSystemPrompt(this.getActiveToolNames());
+        this.agent.state.systemPrompt = this._baseSystemPrompt;
+        this._emit({ type: "resources_changed" });
+    }
     /**
      * Abort current operation and wait for agent to become idle.
      */
@@ -1908,16 +1926,23 @@ export class AgentSession {
         if (!this._extensionRunner.hasHandlers("resources_discover")) {
             return;
         }
-        const { skillPaths, promptPaths, themePaths } = await this._extensionRunner.emitResourcesDiscover(this._cwd, reason);
-        if (skillPaths.length === 0 && promptPaths.length === 0 && themePaths.length === 0) {
-            return;
+        const { skillPaths, promptPaths, themePaths, extensionOwners } = await this._extensionRunner.emitResourcesDiscover(this._cwd, reason);
+        for (const entry of extensionOwners) {
+            if (entry.byzWorkflowOwner) {
+                this._resourceLoader.registerByzWorkflowResourceOwner?.(entry.extensionOwner);
+            }
         }
-        const extensionPaths = {
-            skillPaths: this.buildExtensionResourcePaths(skillPaths),
-            promptPaths: this.buildExtensionResourcePaths(promptPaths),
-            themePaths: this.buildExtensionResourcePaths(themePaths),
-        };
-        this._resourceLoader.extendResources(extensionPaths);
+        const hasResources = skillPaths.length > 0 || promptPaths.length > 0 || themePaths.length > 0;
+        if (hasResources) {
+            const extensionPaths = {
+                skillPaths: this.buildExtensionResourcePaths(skillPaths),
+                promptPaths: this.buildExtensionResourcePaths(promptPaths),
+                themePaths: this.buildExtensionResourcePaths(themePaths),
+            };
+            this._resourceLoader.extendResources(extensionPaths);
+        }
+        if (!hasResources)
+            return;
         this._baseSystemPrompt = this._rebuildSystemPrompt(this.getActiveToolNames());
         this.agent.state.systemPrompt = this._baseSystemPrompt;
     }
@@ -1927,6 +1952,7 @@ export class AgentSession {
             const baseDir = entry.extensionPath.startsWith("<") ? undefined : dirname(entry.extensionPath);
             return {
                 path: entry.path,
+                owner: entry.extensionOwner,
                 metadata: {
                     source,
                     scope: "temporary",
@@ -2036,7 +2062,7 @@ export class AgentSession {
         }, {
             getModel: () => this.model,
             getScopedModels: () => this._scopedModels,
-            isIdle: () => this.isIdle,
+            isIdle: () => this.isIdle && !this.isCompacting,
             isProjectTrusted: () => this.settingsManager.isProjectTrusted(),
             getSignal: () => this.agent.signal,
             abort: () => {
@@ -2065,6 +2091,7 @@ export class AgentSession {
             },
             getSystemPrompt: () => this.systemPrompt,
             getSystemPromptOptions: () => this._baseSystemPromptOptions,
+            replaceByzWorkflowResources: (extensionOwner, extensionPath, resources) => this.replaceByzWorkflowResources(extensionOwner, extensionPath, resources),
         }, {
             registerProvider: (name, config) => {
                 this._modelRuntime.registerProvider(name, config);