@ai-agent-forge/agent-forge 0.88.0 → 0.88.2

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

@@ -31,19 +31,25 @@ export function createCapabilityRuntime(inputHost, options = {}) {
     // snapshot and re-attach to the exposed host afterwards. S4 fourth batch:
     // memoryStorage joins — its store/ledger factories and vector components
     // are host-owned runtime services with the same non-cloneable shape.
+    // D-082 批 3b: modelCatalog joins — its three members are host closures
+    // (catalog/credentials read face for the provider-failover load preflight).
     // Validate BEFORE destructuring: destructuring a null/non-object host would
     // surface a TypeError instead of the contract's validation error.
     assertHost(inputHost);
-    const { communication, turnContext, memoryStorage, ...cloneableHost } = inputHost;
+    const { communication, turnContext, memoryStorage, modelCatalog, ...cloneableHost } = inputHost;
     const host = detachedSnapshot(cloneableHost);
     assertHost(host);
-    const hostWithServices = communication === undefined && turnContext === undefined && memoryStorage === undefined
+    const hostWithServices = communication === undefined &&
+        turnContext === undefined &&
+        memoryStorage === undefined &&
+        modelCatalog === undefined
         ? host
         : Object.freeze({
             ...host,
             ...(communication === undefined ? {} : { communication }),
             ...(turnContext === undefined ? {} : { turnContext }),
             ...(memoryStorage === undefined ? {} : { memoryStorage }),
+            ...(modelCatalog === undefined ? {} : { modelCatalog }),
         });
     const hasSession = host.features.includes("session");
     const hasSessionExport = host.features.includes("session-export");
@@ -110,6 +116,10 @@ export function createCapabilityRuntime(inputHost, options = {}) {
     const state = createCapabilityRuntimeState();
     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;
+    // 契约 A（D-082 批 3b）：store 字段保持基础注册结构类型（store 不感知链序），
+    // 本运行时经此类型视图写入/读取链序元数据；唯一写入方是 registerStreamFnWrapper
+    // 专用注册路径，读取方是 getStreamFnWrappers 与 cleanupRegisteredCapabilities。
+    const streamFnWrapperRegistry = streamFnWrappers;
     let subscriptionSequence = 0;
     let operationSequence = 0;
     let agentTaskSequence = 0;
@@ -124,6 +134,7 @@ export function createCapabilityRuntime(inputHost, options = {}) {
     let spanSequence = 0;
     let traceSequence = 0;
     let loopHookSequence = 0;
+    let streamFnWrapperSequence = 0;
     const createRuntimeId = (kind, sequence, fallback) => {
         if (!options.idFactory)
             return fallback;
@@ -172,7 +183,7 @@ export function createCapabilityRuntime(inputHost, options = {}) {
             rpcMethods,
             contextStrategies,
             providerHeaderTransforms,
-            streamFnWrappers,
+            streamFnWrapperRegistry,
         ]) {
             for (const [name, entry] of registry) {
                 if (entry.ownerToken === ownerToken) {
@@ -1672,14 +1683,35 @@ export function createCapabilityRuntime(inputHost, options = {}) {
                 }
                 return own(register("provider-header-transform", name, transform, owner, ownerToken, providerHeaderTransforms));
             },
-            registerStreamFnWrapper(name, wrapper) {
+            registerStreamFnWrapper(name, wrapper, wrapperOptions) {
                 assertUsable();
                 if (!name)
                     throw new Error(t("Stream function wrapper requires a non-empty name"));
                 if (typeof wrapper !== "function") {
                     throw new Error(t("Stream function wrapper requires a callable wrapper"));
                 }
-                return own(register("stream-fn-wrapper", name, wrapper, owner, ownerToken, streamFnWrappers));
+                // 契约 A（D-082 批 3b）：priority 低值更外层，缺省 0（与既有注册序语义
+                // 等价——同 priority 保持注册序，先注册最外层）。
+                const priority = wrapperOptions?.priority ?? 0;
+                if (typeof priority !== "number" || !Number.isFinite(priority)) {
+                    throw new Error(t("Stream function wrapper priority must be a finite number"));
+                }
+                if (streamFnWrapperRegistry.has(name)) {
+                    throw new Error(t("{kind} is already registered: {name}", { kind: "stream-fn-wrapper", name }));
+                }
+                const entry = {
+                    owner,
+                    ownerToken,
+                    value: wrapper,
+                    priority,
+                    sequence: ++streamFnWrapperSequence,
+                };
+                streamFnWrapperRegistry.set(name, entry);
+                return own(createDisposableRegistration("stream-fn-wrapper", () => {
+                    if (streamFnWrapperRegistry.get(name) !== entry)
+                        return;
+                    streamFnWrapperRegistry.delete(name);
+                }, `registration:${owner}:stream-fn-wrapper:${name}`));
             },
             registerSkill(definition) {
                 assertUsable();
@@ -2206,8 +2238,12 @@ export function createCapabilityRuntime(inputHost, options = {}) {
             }
             return headers;
         },
-        getStreamFnWrappers: () => Array.from(streamFnWrappers.values())
+        getStreamFnWrappers: () => Array.from(streamFnWrapperRegistry.values())
             .filter((entry) => activeOwnerTokens.has(entry.ownerToken))
+            // 契约 A（D-082 批 3b）稳定排序：(priority 升序, 注册序升序)。sequence
+            // 唯一即全序，同 priority 恒保持注册序（缺省 priority 0 时与既有
+            // 注册序语义逐点一致）。
+            .sort((left, right) => left.priority - right.priority || left.sequence - right.sequence)
             .map((entry) => entry.value),
         getOperationStatus: (id) => statuses.get(id),
         drainLifecycleDeliveries: async () => {