@ai-agent-forge/agent-forge 0.86.4 → 0.86.5

package/dist/core/extensions/capability-runtime.js

@@ -107,11 +107,15 @@ export function createCapabilityRuntime(inputHost, options = {}) {
     const onOutputArtifactCommit = options.onOutputArtifactCommit;
     const toolExecutionContext = options.toolExecutionContext;
     const state = createCapabilityRuntimeState();
-    const { tools, skills, retryPolicies, compactionPolicies, overflowPolicies, branchSummaryPolicies, runLoopStrategies, loopHooks, commands, operations, rpcMethods, contextStrategies, pluginCapabilities, providerHeaderTransforms, contributionPoints, lifecycleDefinitions, lifecycleRegistrations, subscribers, statuses, pluginRegistrations, apiStates, activeOwnerTokens, outputArtifacts, stateByOwner, stateRevisionsByOwner, stateWatchers, stateByInstance, stateRevisionsByInstance, stateWatchersByInstance, logs, spans, lifecycleDeliveryQueues, pluginRejections, } = state;
+    const { tools, skills, retryPolicies, compactionPolicies, overflowPolicies, branchSummaryPolicies, runLoopStrategies, loopHooks, commands, operations, rpcMethods, contextStrategies, pluginCapabilities, providerHeaderTransforms, streamFnWrappers, contributionPoints, lifecycleDefinitions, lifecycleRegistrations, subscribers, statuses, pluginRegistrations, apiStates, activeOwnerTokens, outputArtifacts, stateByOwner, stateRevisionsByOwner, stateWatchers, stateByInstance, stateRevisionsByInstance, stateWatchersByInstance, logs, spans, lifecycleDeliveryQueues, pluginRejections, } = state;
     const runtimeState = state;
     let subscriptionSequence = 0;
     let operationSequence = 0;
     let agentTaskSequence = 0;
+    // 活任务集合（设计 子代理运行时 §4.2 中断分级，2026-10-02）：runtime 级共享，
+    // 每个插件 API 的 createAgentTask 入队、任务 dispose 出队；宿主经
+    // runtime.getActiveAgentTaskCount() 读有界投影。
+    const agentTasks = new Set();
     let artifactSequence = 0;
     let contributionSubscriptionSequence = 0;
     let lifecycleRegistrationSequence = 0;
@@ -167,6 +171,7 @@ export function createCapabilityRuntime(inputHost, options = {}) {
             rpcMethods,
             contextStrategies,
             providerHeaderTransforms,
+            streamFnWrappers,
         ]) {
             for (const [name, entry] of registry) {
                 if (entry.ownerToken === ownerToken) {
@@ -1124,7 +1129,6 @@ export function createCapabilityRuntime(inputHost, options = {}) {
                 owner: point.owner,
             })),
         };
-        const agentTasks = new Set();
         const createAgentTask = async (request) => {
             assertUsable();
             assertAgentTaskRequest(request);
@@ -1648,6 +1652,15 @@ export function createCapabilityRuntime(inputHost, options = {}) {
                 }
                 return own(register("provider-header-transform", name, transform, owner, ownerToken, providerHeaderTransforms));
             },
+            registerStreamFnWrapper(name, wrapper) {
+                assertUsable();
+                if (!name)
+                    throw new Error("Stream function wrapper requires a non-empty name");
+                if (typeof wrapper !== "function") {
+                    throw new Error("Stream function wrapper requires a callable wrapper");
+                }
+                return own(register("stream-fn-wrapper", name, wrapper, owner, ownerToken, streamFnWrappers));
+            },
             registerSkill(definition) {
                 assertUsable();
                 const name = capabilityName(definition, "Skill");
@@ -2097,6 +2110,26 @@ export function createCapabilityRuntime(inputHost, options = {}) {
                 throw new Error(`Tool is not registered: ${name}`);
             return invokeRegisteredCapability(registered, (value) => invokeCapabilityTool(asHandler(value, "Tool"), input, toolExecutionContext, options?.signal));
         },
+        // 中断分级读面（设计 §4.2，2026-10-02）：活任务集合的有界投影。已终态
+        // 任务在显式 dispose 前仍滞留集合（释放钩子语义，见 RuntimeAgentTask），
+        // 因此这里按状态过滤——只计尚未终态的任务。dispose 是多 await 流程，
+        // 任务先置 disposed 再退出集合，中间窗口 status() 会抛——该窗口内的
+        // 任务按非活跃计（宿主按键路径绝不能因投影抛错）。
+        getActiveAgentTaskCount: () => {
+            let active = 0;
+            for (const task of agentTasks) {
+                let status;
+                try {
+                    status = task.status();
+                }
+                catch {
+                    continue;
+                }
+                if (status !== "succeeded" && status !== "failed" && status !== "cancelled")
+                    active += 1;
+            }
+            return active;
+        },
         hasCommand: (name) => commands.has(name),
         async invokeCommand(name, args, options) {
             const registered = commands.get(name);
@@ -2153,6 +2186,9 @@ export function createCapabilityRuntime(inputHost, options = {}) {
             }
             return headers;
         },
+        getStreamFnWrappers: () => Array.from(streamFnWrappers.values())
+            .filter((entry) => activeOwnerTokens.has(entry.ownerToken))
+            .map((entry) => entry.value),
         getOperationStatus: (id) => statuses.get(id),
         drainLifecycleDeliveries: async () => {
             while (true) {