@musistudio/claude-code-router 1.0.45 → 1.0.46

package/dist/cli.js

@@ -66532,7 +66532,7 @@ var require_package3 = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "@musistudio/claude-code-router",
-      version: "1.0.45",
+      version: "1.0.46",
       description: "Use Claude Code without an Anthropics account and route it to another LLM provider",
       bin: {
         ccr: "./dist/cli.js"
@@ -81033,10 +81033,11 @@ async function run(options = {}) {
       await router(req, reply, config);
     }
   });
-  server.addHook("onSend", async (req, reply, payload) => {
+  server.addHook("onSend", (req, reply, payload, done) => {
     if (req.sessionId && req.url.startsWith("/v1/messages")) {
       if (payload instanceof ReadableStream) {
         if (req.agents) {
+          const abortController = new AbortController();
           const eventStream = payload.pipeThrough(new SSEParserTransform());
           let currentAgent;
           let currentToolIndex = -1;
@@ -81045,111 +81046,147 @@ async function run(options = {}) {
           let currentToolId = "";
           const toolMessages = [];
           const assistantMessages = [];
-          return rewriteStream(eventStream, async (data, controller) => {
-            if (data.event === "content_block_start" && data?.data?.content_block?.name) {
-              const agent = req.agents.find((name) => agents_default.getAgent(name)?.tools.get(data.data.content_block.name));
-              if (agent) {
-                currentAgent = agents_default.getAgent(agent);
-                currentToolIndex = data.data.index;
-                currentToolName = data.data.content_block.name;
-                currentToolId = data.data.content_block.id;
+          return done(null, rewriteStream(eventStream, async (data, controller) => {
+            try {
+              if (data.event === "content_block_start" && data?.data?.content_block?.name) {
+                const agent = req.agents.find((name) => agents_default.getAgent(name)?.tools.get(data.data.content_block.name));
+                if (agent) {
+                  currentAgent = agents_default.getAgent(agent);
+                  currentToolIndex = data.data.index;
+                  currentToolName = data.data.content_block.name;
+                  currentToolId = data.data.content_block.id;
+                  return void 0;
+                }
+              }
+              if (currentToolIndex > -1 && data.data.index === currentToolIndex && data.data?.delta?.type === "input_json_delta") {
+                currentToolArgs += data.data?.delta?.partial_json;
                 return void 0;
               }
-            }
-            if (currentToolIndex > -1 && data.data.index === currentToolIndex && data.data?.delta?.type === "input_json_delta") {
-              currentToolArgs += data.data?.delta?.partial_json;
-              return void 0;
-            }
-            if (currentToolIndex > -1 && data.data.index === currentToolIndex && data.data.type === "content_block_stop") {
-              try {
-                const args = import_json52.default.parse(currentToolArgs);
-                assistantMessages.push({
-                  type: "tool_use",
-                  id: currentToolId,
-                  name: currentToolName,
-                  input: args
+              if (currentToolIndex > -1 && data.data.index === currentToolIndex && data.data.type === "content_block_stop") {
+                try {
+                  const args = import_json52.default.parse(currentToolArgs);
+                  assistantMessages.push({
+                    type: "tool_use",
+                    id: currentToolId,
+                    name: currentToolName,
+                    input: args
+                  });
+                  const toolResult = await currentAgent?.tools.get(currentToolName)?.handler(args, {
+                    req,
+                    config
+                  });
+                  console.log("result", toolResult);
+                  toolMessages.push({
+                    "tool_use_id": currentToolId,
+                    "type": "tool_result",
+                    "content": toolResult
+                  });
+                  currentAgent = void 0;
+                  currentToolIndex = -1;
+                  currentToolName = "";
+                  currentToolArgs = "";
+                  currentToolId = "";
+                } catch (e) {
+                  console.log(e);
+                }
+                return void 0;
+              }
+              if (data.event === "message_delta" && toolMessages.length) {
+                req.body.messages.push({
+                  role: "assistant",
+                  content: assistantMessages
                 });
-                const toolResult = await currentAgent?.tools.get(currentToolName)?.handler(args, {
-                  req,
-                  config
+                req.body.messages.push({
+                  role: "user",
+                  content: toolMessages
                 });
-                console.log("result", toolResult);
-                toolMessages.push({
-                  "tool_use_id": currentToolId,
-                  "type": "tool_result",
-                  "content": toolResult
+                const response = await fetch(`http://127.0.0.1:${config.PORT}/v1/messages`, {
+                  method: "POST",
+                  headers: {
+                    "x-api-key": config.APIKEY,
+                    "content-type": "application/json"
+                  },
+                  body: JSON.stringify(req.body)
                 });
-                currentAgent = void 0;
-                currentToolIndex = -1;
-                currentToolName = "";
-                currentToolArgs = "";
-                currentToolId = "";
-              } catch (e) {
-                console.log(e);
-              }
-              return void 0;
-            }
-            if (data.event === "message_delta" && toolMessages.length) {
-              req.body.messages.push({
-                role: "assistant",
-                content: assistantMessages
-              });
-              req.body.messages.push({
-                role: "user",
-                content: toolMessages
-              });
-              const response = await fetch(`http://127.0.0.1:${config.PORT}/v1/messages`, {
-                method: "POST",
-                headers: {
-                  "x-api-key": config.APIKEY,
-                  "content-type": "application/json"
-                },
-                body: JSON.stringify(req.body)
-              });
-              if (!response.ok) {
-                return void 0;
-              }
-              const stream = response.body.pipeThrough(new SSEParserTransform());
-              const reader = stream.getReader();
-              while (true) {
-                const { value, done } = await reader.read();
-                if (done) {
-                  break;
+                if (!response.ok) {
+                  return void 0;
                 }
-                if (["message_start", "message_stop"].includes(value.event)) {
-                  continue;
+                const stream = response.body.pipeThrough(new SSEParserTransform());
+                const reader = stream.getReader();
+                while (true) {
+                  try {
+                    const { value, done: done2 } = await reader.read();
+                    if (done2) {
+                      break;
+                    }
+                    if (["message_start", "message_stop"].includes(value.event)) {
+                      continue;
+                    }
+                    if (!controller.desiredSize) {
+                      console.log("Stream backpressure detected");
+                      break;
+                    }
+                    controller.enqueue(value);
+                  } catch (readError) {
+                    if (readError.name === "AbortError" || readError.code === "ERR_STREAM_PREMATURE_CLOSE") {
+                      console.log("Stream reading aborted due to client disconnect");
+                      abortController.abort();
+                      break;
+                    }
+                    throw readError;
+                  }
                 }
-                controller.enqueue(value);
+                return void 0;
               }
-              return void 0;
+              return data;
+            } catch (error) {
+              console.error("Unexpected error in stream processing:", error);
+              if (error.code === "ERR_STREAM_PREMATURE_CLOSE") {
+                console.log("Stream prematurely closed, aborting operations");
+                abortController.abort();
+                return void 0;
+              }
+              throw error;
             }
-            return data;
-          }).pipeThrough(new SSESerializerTransform());
+          }).pipeThrough(new SSESerializerTransform()));
         }
         const [originalStream, clonedStream] = payload.tee();
         const read = async (stream) => {
           const reader = stream.getReader();
-          while (true) {
-            const { done, value } = await reader.read();
-            if (done) break;
-            const dataStr = new TextDecoder().decode(value);
-            if (!dataStr.startsWith("event: message_delta")) {
-              continue;
+          try {
+            while (true) {
+              const { done: done2, value } = await reader.read();
+              if (done2) break;
+              const dataStr = new TextDecoder().decode(value);
+              if (!dataStr.startsWith("event: message_delta")) {
+                continue;
+              }
+              const str = dataStr.slice(27);
+              try {
+                const message = JSON.parse(str);
+                sessionUsageCache.put(req.sessionId, message.usage);
+              } catch {
+              }
             }
-            const str = dataStr.slice(27);
-            try {
-              const message = JSON.parse(str);
-              sessionUsageCache.put(req.sessionId, message.usage);
-            } catch {
+          } catch (readError) {
+            if (readError.name === "AbortError" || readError.code === "ERR_STREAM_PREMATURE_CLOSE") {
+              console.log("Background read stream closed prematurely");
+            } else {
+              console.error("Error in background stream reading:", readError);
             }
+          } finally {
+            reader.releaseLock();
           }
         };
         read(clonedStream);
-        return originalStream;
+        return done(null, originalStream);
       }
       sessionUsageCache.put(req.sessionId, payload.usage);
     }
-    return payload;
+    if (typeof payload === "object" && payload.error) {
+      done(payload.error, null);
+    }
+    done(null, payload);
   });
   server.start();
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@musistudio/claude-code-router",
-  "version": "1.0.45",
+  "version": "1.0.46",
   "description": "Use Claude Code without an Anthropics account and route it to another LLM provider",
   "bin": {
     "ccr": "./dist/cli.js"