@armadra/agent 0.3.0 → 0.4.0

package/dist/ai/sse.js

@@ -6,6 +6,7 @@
  * 第一个空格去掉；多条 data 以 LF 连接。为兼容不规范服务端，流结束时未以空行收尾的最后一个
  * 事件也会分派。
  */
+import { IdleTimeoutError } from "./http.js";
 export class SseParser {
     buffer = "";
     event;
@@ -102,8 +103,12 @@ export class SseParser {
         return hadContent ? event : undefined;
     }
 }
-/** 从字节流读 SSE 事件；`signal` 中止时停止读取（调用方据此产出 aborted）。 */
-export async function* readSseEvents(body, signal) {
+/**
+ * 从字节流读 SSE 事件；`signal` 中止时停止读取（调用方据此产出 aborted）。
+ * `idleTimeoutMs`：等待下一块字节的上限，每收到一块即重新计时（消费者处理事件的时间不计）；
+ * 超时取消底层流并抛 `IdleTimeoutError`（phase `stream`）。
+ */
+export async function* readSseEvents(body, signal, idleTimeoutMs) {
     const reader = body.getReader();
     const decoder = new TextDecoder("utf-8");
     const parser = new SseParser();
@@ -112,11 +117,25 @@ export async function* readSseEvents(body, signal) {
     };
     signal?.addEventListener("abort", onAbort, { once: true });
     let drained = false;
+    let idle;
+    let stalled = false;
+    const arm = () => {
+        if (idleTimeoutMs === undefined || idleTimeoutMs <= 0)
+            return;
+        idle = setTimeout(() => {
+            stalled = true;
+            reader.cancel().catch(() => undefined);
+        }, idleTimeoutMs);
+    };
     try {
         while (true) {
             if (signal?.aborted)
                 return;
+            arm();
             const { value, done } = await reader.read();
+            clearTimeout(idle);
+            if (stalled && !signal?.aborted)
+                throw new IdleTimeoutError(idleTimeoutMs ?? 0, "stream");
             if (done)
                 break;
             if (signal?.aborted)
@@ -131,6 +150,7 @@ export async function* readSseEvents(body, signal) {
             yield event;
     }
     finally {
+        clearTimeout(idle);
         signal?.removeEventListener("abort", onAbort);
         // 消费者提前退出（读到终止事件即 break）或中止：取消底层流，释放连接
         if (!drained)

package/dist/ai/types.d.ts

@@ -27,7 +27,8 @@ export type CacheRetention = "none" | "short" | "long";
  * 真实回合，`summary` 是压缩 / 分支摘要，`warm` 是缓存保温重放，`probe` 是 `models check` /
  * `cache-probe` 一类的探测。缺省视为 `turn`。协议层只透传，不改变请求体。
  */
-export type RequestPurpose = "turn" | "summary" | "warm" | "probe";
+/** `classify`：auto 权限模式的分类请求（独立请求，不经会话层缓存观测、不触发保温）。 */
+export type RequestPurpose = "turn" | "summary" | "warm" | "probe" | "classify";
 /** `provider/model-id` 拆开后的引用。 */
 export interface ModelRef {
     provider: string;
@@ -342,6 +343,11 @@ export interface StreamOptions {
     /** 值为 null 表示删除该头。 */
     headers?: Record<string, string | null>;
     timeoutMs?: number;
+    /**
+     * 空闲超时：等响应头、以及流中两块数据之间的最长间隔（每收到字节即重新计时）。缺省 300 000，
+     * 0 关闭。超时报 `idle timeout` 错误，会话层按可重试处理。
+     */
+    idleTimeoutMs?: number;
     maxTokens?: number;
     temperature?: number;
     thinkingLevel?: ModelThinkingLevel;
@@ -434,7 +440,8 @@ export type ModelLookup = {
     provider: ProviderData;
 } | {
     ok: false;
-    reason: "not_found" | "ambiguous" | "provider_not_found";
+    /** `channel_not_found`：模型在，`@渠道` 不在（候选是该模型可用的 `provider/model@渠道`）。 */
+    reason: "not_found" | "ambiguous" | "provider_not_found" | "channel_not_found";
     candidates: string[];
 };
 export interface ProviderRegistryApi {