@springbrand/agent-runtime 0.1.0 → 0.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@springbrand/agent-runtime",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "type": "module",
   "files": [
     "src",

package/src/kernel/profile.ts

@@ -84,7 +84,6 @@ export interface RuntimeProfile {
   denyPolicy?: RuntimeDenyPolicy;
   enabledSubagents: readonly string[];
   mcpServers: readonly RuntimeMcpServer[];
-  enabledExtensions: readonly string[];
   executionLevel: ExecutionLevel;
   memory: RuntimeMemoryProfile;
 }
@@ -109,7 +108,6 @@ export function freezeRuntimeProfile(profile: RuntimeProfile): RuntimeProfile {
       : undefined,
     enabledSubagents: Object.freeze([...profile.enabledSubagents]),
     mcpServers: Object.freeze(profile.mcpServers.map((server) => Object.freeze({ ...server }))),
-    enabledExtensions: Object.freeze([...profile.enabledExtensions]),
     memory: Object.freeze({ ...profile.memory }),
   };
   return Object.freeze(frozen);

package/src/pi/runtime-adapter/assembly.ts

@@ -45,7 +45,7 @@ interface PiAssemblySnapshot {
   >;
   readonly profile: Pick<
     RuntimeProfile,
-    "denyPolicy" | "enabledExtensions" | "mcpServers"
+    "denyPolicy" | "mcpServers"
   >;
   readonly bindings: {
     readonly platform: Pick<RuntimePlatformPort, "loader">;
@@ -453,7 +453,7 @@ async function preparePiAssembly(
   const extensions = await assemblePiExtensions({
     extensions: snapshot.pi.extensions,
     published: extensionNames,
-    enabled: new Set(snapshot.profile.enabledExtensions),
+    enabled: extensionNames,
     authorized: extensionNames,
     load: async (extension) => {
       const workspacePermission =

package/src/pi/runtime-adapter/models.ts

@@ -2,7 +2,9 @@ import type { StreamFn } from "@earendil-works/pi-agent-core";
 import {
   createModels,
   createProvider,
+  lazyStream,
   type Api,
+  type AssistantMessageEvent,
   type Model,
   type MutableModels,
 } from "@earendil-works/pi-ai";
@@ -44,19 +46,57 @@ const CATALOGS = {
 } satisfies Record<RuntimeModelProtocol, readonly Model<Api>[]>;
 
 const PROVIDER_MAX_RETRIES = 2;
+const MODEL_STREAM_STALL_TIMEOUT_MS = 120_000;
+
+async function* stopStalledModelStream(
+  source: AsyncIterable<AssistantMessageEvent>,
+  watchdog: AbortController,
+): AsyncGenerator<AssistantMessageEvent> {
+  const iterator = source[Symbol.asyncIterator]();
+  while (true) {
+    let timer: ReturnType<typeof setTimeout> | undefined;
+    const next = await Promise.race([
+      iterator.next(),
+      new Promise<never>((_, reject) => {
+        timer = setTimeout(() => {
+          const error = new Error(
+            `Model stream stalled for ${MODEL_STREAM_STALL_TIMEOUT_MS} ms`,
+          );
+          watchdog.abort(error);
+          reject(error);
+        }, MODEL_STREAM_STALL_TIMEOUT_MS);
+      }),
+    ]).finally(() => {
+      if (timer !== undefined) clearTimeout(timer);
+    });
+    if (next.done) {
+      throw new Error("Model stream ended without a terminal event");
+    }
+    yield next.value;
+    if (next.value.type === "done" || next.value.type === "error") return;
+  }
+}
 
 /**
- * 给模型请求补上可中断的 provider 瞬时错误重试。
+ * 给模型请求补上可中断的 provider 瞬时错误重试和空闲终止边界。
  *
  * Runtime Turn 和 SubAgent 在把 `Models.streamSimple` 交给 Pi 前调用；显式传入的重试次数优先。
  *
- * Pi 0.83 默认 `maxRetries` 为 0，这里只补 2 次默认值，避免瞬时网络错误直接终止，同时不覆盖调用方策略。
+ * Pi 0.83 默认 `maxRetries` 为 0，且流无事件时会无限等待。这里补 2 次默认重试，
+ * 并在连续 120 秒没有模型事件时中止 provider、返回明确失败终态。
  */
 export function withProviderRetry(streamFn: StreamFn): StreamFn {
   return (model, context, options) =>
-    streamFn(model, context, {
-      ...options,
-      maxRetries: options?.maxRetries ?? PROVIDER_MAX_RETRIES,
+    lazyStream(model, async () => {
+      const watchdog = new AbortController();
+      const source = await streamFn(model, context, {
+        ...options,
+        signal: options?.signal
+          ? AbortSignal.any([options.signal, watchdog.signal])
+          : watchdog.signal,
+        maxRetries: options?.maxRetries ?? PROVIDER_MAX_RETRIES,
+      });
+      return stopStalledModelStream(source, watchdog);
     });
 }
 

package/src/plugins.ts

@@ -837,7 +837,6 @@ class RuntimeBuilder implements RuntimeContributionContext {
       denyPolicy: this.denyPolicy,
       enabledSubagents: [...this.enabledSubagents],
       mcpServers: [...this.connectors.values()],
-      enabledExtensions: [...this.extensions.keys()],
       executionLevel: this.profile.executionLevel,
       memory: this.memoryProfile,
     });