mslxdff 0.1.215 → 0.1.216

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mslxdff",
-  "version": "0.1.215",
+  "version": "0.1.216",
   "description": "测试项目，请勿使用。",
   "type": "module",
   "bin": {

package/src/providers/codearts/concurrency.js

@@ -10,7 +10,11 @@ export function createGate({ limit = 3 } = {}) {
     if (w) w();
   }
 
-  async function run(fn, onWait) {
+  // 取得一个槽位，返回「释放一次」的幂等函数。超限时 FIFO 排队（并回调 onWait）。
+  // 关键：释放的时机交给调用方——流式请求必须**等流真正读完/取消**才释放，
+  // 否则 runChat 一返回（流还没消费）就放行，上游其实仍占着会话 ⇒ 第 4 路被上游拒
+  // （TM.00001041 并发会话数已达上限）。见 index.js 的 chat 接线。
+  async function acquire(onWait) {
     if (active >= cap) {
       const t0 = Date.now();
       const queued = waiters.length + 1;
@@ -18,15 +22,26 @@ export function createGate({ limit = 3 } = {}) {
       if (onWait) { try { onWait({ waitedMs: Date.now() - t0, queued, limit: cap }); } catch {} }
     }
     active++;
+    let released = false;
+    return function releaseOnce() {
+      if (released) return;
+      released = true;
+      release();
+    };
+  }
+
+  async function run(fn, onWait) {
+    const rel = await acquire(onWait);
     try {
       return await fn();
     } finally {
-      release();
+      rel();
     }
   }
 
   return {
     run,
+    acquire,
     stats: () => ({ active, queued: waiters.length, limit: cap }),
     get active() { return active; },
     get queued() { return waiters.length; },

package/src/providers/codearts/index.js

@@ -82,9 +82,48 @@ export function createCodeartsProvider({
   const onWait = (info) => {
     try { appendEvent({ type: "codearts-queue", ...info }); } catch {}
   };
+  // 流式请求必须在**流真正读完/取消后**才释放并发槽——否则 runChat 一返回（body 还没消费）
+  // 就 release，上游其实仍占着会话 ⇒ 第 4 路被上游拒（TM.00001041 并发会话数已达上限）。
+  // 非流式（JSON）在 runChat 内部已聚合完，可立即释放。返回的 Response 保持原 body 引用，
+  // 只包一层 tee 触发 release；出错也保证释放一次（releaseOnce 幂等）。
+  function holdUntilBodyDone(res, releaseOnce) {
+    const ct = res?.headers?.get?.("content-type") || "";
+    if (!ct.includes("text/event-stream") || !res.body) {
+      releaseOnce();
+      return res;
+    }
+    let done = false;
+    const fin = () => { if (!done) { done = true; releaseOnce(); } };
+    const onErr = () => fin();
+    try {
+      const reader = res.body.getReader();
+      const stream = new ReadableStream({
+        async pull(controller) {
+          try {
+            const { done: d, value } = await reader.read();
+            if (d) { fin(); controller.close(); return; }
+            controller.enqueue(value);
+          } catch (e) { onErr(); controller.error(e); }
+        },
+        cancel(reason) { fin(); try { return reader.cancel(reason); } catch { /* ignore */ } },
+      });
+      return new Response(stream, { status: res.status, statusText: res.statusText, headers: res.headers });
+    } catch {
+      fin();
+      return res;
+    }
+  }
   async function chat(body, opts = {}) {
     if (!ring.size) throw new Error("codearts: missing credential — run: mslxdff -provider codearts login");
-    return gate.run(() => chatSvc.runChat(body, ring, opts), onWait);
+    const rel = await gate.acquire(onWait);
+    let res;
+    try {
+      res = await chatSvc.runChat(body, ring, opts);
+    } catch (err) {
+      rel();
+      throw err;
+    }
+    return holdUntilBodyDone(res, rel);
   }
   async function chatWithKeys(body, keys, opts = {}) {
     const blobs = (keys || []).filter((k) => typeof k === "string" && k.trim().startsWith("{"));
@@ -92,7 +131,15 @@ export function createCodeartsProvider({
     const tmpRing = createKeyRing(blobs, { cooldownMs });
     const tmpPool = createAuthPool({ keys: blobs, fetchImpl, stsHost, clock }); // 借用凭据不写回本节点
     const tmpSvc = createChatService({ id, baseUrl: chatSvcBase, chatPath, fetchImpl, dispatcher: null, authPool: tmpPool, catalog, connectTimeoutMs, clock });
-    return gate.run(() => tmpSvc.runChat(body, tmpRing, opts), onWait);
+    const rel = await gate.acquire(onWait);
+    let res;
+    try {
+      res = await tmpSvc.runChat(body, tmpRing, opts);
+    } catch (err) {
+      rel();
+      throw err;
+    }
+    return holdUntilBodyDone(res, rel);
   }
 
   // 三路发现（主账号）；结果写目录（benefit 判定按账号隔离）