@rowan-agent/agent 0.9.21 → 0.9.23

package/dist/index.d.ts

@@ -596,12 +596,14 @@ interface SourceInfo {
     baseDir?: string;
     displayName?: string;
 }
+/** Register a Phase by directory path or Phase object; PHASE.md and direct child Skills are loaded atomically when a path is supplied. */
+type PhaseRegistration = string | Phase;
 /**
- * Tool definition for registering LLM-callable tools via `api.registerTool()`.
+ * Tool definition for registering LLM-callable tools via `api.tool.register()`.
  *
  * @example
  * ```typescript
- * api.registerTool({
+ * api.tool.register({
  *   name: "search_docs",
  *   description: "Search documentation",
  *   parameters: { type: "object", properties: { query: { type: "string" } } },
@@ -854,10 +856,13 @@ interface ExtensionAPI {
     on<K extends HookEventType>(eventType: K, handler: HookHandler<K>): void;
     /** Unsubscribe from a hook event. */
     off<K extends HookEventType>(eventType: K, handler: HookHandler<K>): void;
-    /** Register a custom LLM-callable tool. */
-    registerTool(tool: ToolDefinition): void;
-    /** Register a Phase directory bundle. */
-    registerPhase(path: string): Promise<void>;
+    /** Tool capabilities — register and unregister custom tools. */
+    tool: {
+        /** Register a custom LLM-callable tool. */
+        register(tool: ToolDefinition): void;
+        /** Unregister a previously registered tool by name. */
+        unregister(toolName: string): void;
+    };
     /** Register a model provider. */
     registerProvider(config: _rowan_agent_models.ProviderConfig): void;
     /** Unregister a model provider. */
@@ -870,8 +875,12 @@ interface ExtensionAPI {
     context: ExtensionContext;
     /** Shared event bus for inter-extension communication. */
     events: EventBus;
-    /** Phase execution capabilities — rovides Phase In/Out, phase identity, and phase routing. */
+    /** Phase execution capabilities — provides Phase In/Out, phase identity, phase routing, and registration. */
     phase: {
+        /** Register a Phase directory bundle or Phase object. */
+        register(phase: PhaseRegistration): Promise<void>;
+        /** Unregister a previously registered phase by name. */
+        unregister(phaseName: string): void;
         /** Phase In: get payload from previous phase */
         getPayload(): unknown;
         /** Phase Out: set payload for next phase */
@@ -1409,7 +1418,10 @@ declare class ExtensionRunner {
      */
     private createExtensionAPI;
     private registerTool;
+    private unregisterTool;
     private registerPhase;
+    private registerPhaseDefinition;
+    private unregisterPhase;
     private loadRegisteredPhase;
     private registerProvider;
     private unregisterProvider;
@@ -2395,6 +2407,12 @@ declare class InMemoryStore implements DurableStore {
     private writeReceipt;
     private replayOperation;
     private writeOperationReceipt;
+    /**
+     * A Claim receipt exists for idempotent replay of one live Execution Attempt
+     * and carries that Attempt's history. Dropping it when the Attempt ends keeps
+     * one history snapshot per Agent instead of one per Attempt.
+     */
+    private dropClaimReceipt;
     private requireAgent;
     private requireRun;
     private requireToolCall;

package/dist/index.js

@@ -1208,13 +1208,15 @@ function createExtensionAPI(hooks, options, runtime, eventBus) {
       assertActive();
       hooks?.off(eventType, handler);
     },
-    registerTool: (tool) => {
-      assertActive();
-      options?.registerTool?.(tool);
-    },
-    registerPhase: async (registration) => {
-      assertActive();
-      await options?.registerPhase?.(registration);
+    tool: {
+      register: (tool) => {
+        assertActive();
+        options?.registerTool?.(tool);
+      },
+      unregister: (toolName) => {
+        assertActive();
+        options?.unregisterTool?.(toolName);
+      }
     },
     registerProvider: (config) => {
       assertActive();
@@ -1246,6 +1248,14 @@ function createExtensionAPI(hooks, options, runtime, eventBus) {
     }, emit: () => {
     }, has: () => false, count: () => 0 },
     phase: {
+      register: async (registration) => {
+        assertActive();
+        await options?.registerPhase?.(registration);
+      },
+      unregister: (phaseName) => {
+        assertActive();
+        options?.unregisterPhase?.(phaseName);
+      },
       getPayload: () => outputPayload,
       setPayload: (p) => {
         outputPayload = p;
@@ -4925,9 +4935,11 @@ var ExtensionRunner = class {
     };
     return createExtensionAPI(this.hooks, {
       registerPhase: (registration) => this.registerPhase(extension, registration),
+      unregisterPhase: (phaseName) => this.unregisterPhase(extension, phaseName),
       registerProvider: (config) => this.registerProvider(config),
       unregisterProvider: (name) => this.unregisterProvider(name),
       registerTool: (tool) => this.registerTool(extension, tool),
+      unregisterTool: (toolName) => this.unregisterTool(extension, toolName),
       context: extContext,
       manifest,
       trackCleanup: (cleanup) => extension.cleanup.push(cleanup)
@@ -4954,11 +4966,43 @@ var ExtensionRunner = class {
       sourceInfo
     });
   }
-  registerPhase(extension, registration) {
-    if (typeof registration !== "string" || registration.length === 0) {
-      throw new Error(`Phase registration requires a directory path.`);
+  unregisterTool(extension, toolName) {
+    if (extension.tools.has(toolName)) {
+      extension.tools.delete(toolName);
+    }
+  }
+  async registerPhase(extension, registration) {
+    if (typeof registration === "string") {
+      if (registration.length === 0) {
+        throw new Error(`Phase registration requires a directory path.`);
+      }
+      return this.loadRegisteredPhase(extension, registration);
+    }
+    if (typeof registration === "object" && registration !== null && "name" in registration) {
+      return this.registerPhaseDefinition(extension, registration);
+    }
+    throw new Error(`Phase registration requires a directory path or Phase object.`);
+  }
+  registerPhaseDefinition(extension, phase) {
+    const name = phase.name;
+    if (this.phases.has(name)) {
+      throw new Error(`Duplicate phase name: ${name}`);
+    }
+    const registered = {
+      definition: phase,
+      source: { extensionPath: extension.path }
+    };
+    this.phases.set(name, registered);
+    extension.phases.add(name);
+    this._phaseCache = null;
+  }
+  unregisterPhase(extension, phaseName) {
+    const registered = this.phases.get(phaseName);
+    if (registered && registered.source.extensionPath === extension.path) {
+      this.phases.delete(phaseName);
+      extension.phases.delete(phaseName);
+      this._phaseCache = null;
     }
-    return this.loadRegisteredPhase(extension, registration);
   }
   async loadRegisteredPhase(extension, registration) {
     const extensionBase = extension.path.startsWith("<") ? this.cwd : dirname5(extension.path);
@@ -5804,6 +5848,7 @@ function selectedRefs(refs, names) {
 var DEFAULT_CONCURRENCY = 10;
 var DEFAULT_POLL_MS = 25;
 var MAX_CONSUMER_IDLE_POLL_MS = 250;
+var RUN_STATE_CHECK_INTERVAL_MS = 1e3;
 var OWNER_LEASE_MS = 3e4;
 var OWNER_RENEWAL_MS = 1e4;
 var MAX_INLINE_TOOL_RESULT_BYTES = 16 * 1024;
@@ -6608,15 +6653,25 @@ var AgentRuntime = class _AgentRuntime {
   async *observe(runId, options = {}) {
     let cursor = options.after;
     const subscription = this.transientEvents.subscribe(runId);
+    let nextRunStateCheckAtMs = 0;
     try {
       while (true) {
+        const published = subscription.shift();
+        if (published) {
+          yield published;
+          continue;
+        }
         const observationVersion = subscription.checkpoint();
-        const snapshot = await this.owned.snapshotRun(runId);
-        const terminal = ["completed", "failed", "cancelled"].includes(snapshot.state);
-        if (options.signal?.aborted && !terminal) throw abortError();
-        if (terminal) {
-          const pending = await this.owned.listEvents(cursor ? { after: cursor } : {});
-          if (!pending.some((event) => event.runId === runId)) return;
+        const aborted = options.signal?.aborted === true;
+        if (aborted || Date.now() >= nextRunStateCheckAtMs) {
+          nextRunStateCheckAtMs = Date.now() + RUN_STATE_CHECK_INTERVAL_MS;
+          const snapshot = await this.owned.snapshotRun(runId);
+          const terminal = ["completed", "failed", "cancelled"].includes(snapshot.state);
+          if (aborted && !terminal) throw abortError();
+          if (terminal) {
+            const pending = await this.owned.listEvents(cursor ? { after: cursor } : {});
+            if (!pending.some((event) => event.runId === runId)) return;
+          }
         }
         const events = await this.owned.listEvents(cursor ? { after: cursor } : {});
         for (const event of events) {
@@ -6913,6 +6968,7 @@ var InMemoryStore = class _InMemoryStore {
       const failure = indeterminateToolCallIds.length > 0 ? { code: "tool_indeterminate", message, toolCallIds: indeterminateToolCallIds } : { code: "runtime_interrupted", message, ownerEpoch };
       run.state = "failed";
       run.failure = failure;
+      this.dropClaimReceipt(run.execution);
       delete run.execution;
       run.revision += 1;
       run.updatedAt = createTimestamp();
@@ -7142,6 +7198,7 @@ var InMemoryStore = class _InMemoryStore {
           delete run.openInputRequest;
           delete run.checkpoint;
         }
+        this.dropClaimReceipt(run.execution);
         delete run.execution;
         run.state = "cancelled";
         run.cancellationReason = "Run superseded by Message revision.";
@@ -7325,6 +7382,7 @@ var InMemoryStore = class _InMemoryStore {
       delete run.openInteractions;
       delete run.interactionAnswers;
     }
+    this.dropClaimReceipt(run.execution);
     delete run.execution;
     run.revision += 1;
     run.updatedAt = createTimestamp();
@@ -7579,6 +7637,7 @@ var InMemoryStore = class _InMemoryStore {
       };
       run.state = "failed";
       run.failure = failure;
+      this.dropClaimReceipt(run.execution);
       delete run.execution;
       run.revision += 1;
       run.updatedAt = createTimestamp();
@@ -7609,6 +7668,7 @@ var InMemoryStore = class _InMemoryStore {
     run.state = nextState;
     if (input.outcome) run.outcome = clone(input.outcome);
     if (input.failure) run.failure = clone(input.failure);
+    this.dropClaimReceipt(run.execution);
     delete run.execution;
     delete run.openInteractions;
     delete run.interactionAnswers;
@@ -7655,6 +7715,7 @@ var InMemoryStore = class _InMemoryStore {
       run.state = "failed";
       run.failure = failure;
       delete run.cancellationReason;
+      this.dropClaimReceipt(run.execution);
       delete run.execution;
       delete run.openInputRequest;
       delete run.openInteractions;
@@ -7668,6 +7729,7 @@ var InMemoryStore = class _InMemoryStore {
     }
     run.state = "cancelled";
     run.cancellationReason = input.reason;
+    this.dropClaimReceipt(run.execution);
     delete run.execution;
     delete run.openInputRequest;
     delete run.openInteractions;
@@ -7841,6 +7903,14 @@ var InMemoryStore = class _InMemoryStore {
   writeOperationReceipt(key, payload, result) {
     this.operationReceipts.set(key, { payload, result: clone(result) });
   }
+  /**
+   * A Claim receipt exists for idempotent replay of one live Execution Attempt
+   * and carries that Attempt's history. Dropping it when the Attempt ends keeps
+   * one history snapshot per Agent instead of one per Attempt.
+   */
+  dropClaimReceipt(execution) {
+    if (execution) this.operationReceipts.delete(`claim:${execution.executionId}`);
+  }
   requireAgent(agentId) {
     const agent = this.agents.get(agentId);
     if (!agent) throw new RuntimeError("agent_not_found", { agentId });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rowan-agent/agent",
-  "version": "0.9.21",
+  "version": "0.9.23",
   "license": "MIT",
   "repository": {
     "type": "git",