feat: Compatible with Claude code JSONL file usage error scenarios, delay closeBlockIfOpen

Author: caozhiyuan

src/routes/messages/responses-stream-translation.ts

@@ -176,8 +176,6 @@ const handleOutputItemDone = (
     state.blockHasDelta.add(blockIndex)
   }
 
-  closeBlockIfOpen(state, blockIndex, events)
-
   return events
 }
 
@@ -232,7 +230,6 @@ const handleFunctionCallArgumentsDone = (
     state.blockHasDelta.add(blockIndex)
   }
 
-  closeBlockIfOpen(state, blockIndex, events)
   state.functionCallStateByOutputIndex.delete(outputIndex)
   return events
 }
@@ -340,8 +337,6 @@ const handleOutputTextDone = (
     })
   }
 
-  closeBlockIfOpen(state, blockIndex, events)
-
   return events
 }
 
@@ -447,6 +442,7 @@ const openTextBlockIfNeeded = (
   }
 
   if (!state.openBlocks.has(blockIndex)) {
+    closeOpenBlocks(state, events)
     events.push({
       type: "content_block_start",
       index: blockIndex,
@@ -478,6 +474,7 @@ const openThinkingBlockIfNeeded = (
   }
 
   if (!state.openBlocks.has(blockIndex)) {
+    closeOpenBlocks(state, events)
     events.push({
       type: "content_block_start",
       index: blockIndex,
@@ -506,13 +503,20 @@ const closeBlockIfOpen = (
   state.blockHasDelta.delete(blockIndex)
 }
 
-const closeAllOpenBlocks = (
+const closeOpenBlocks = (
   state: ResponsesStreamState,
   events: Array<AnthropicStreamEventData>,
 ) => {
   for (const blockIndex of state.openBlocks) {
     closeBlockIfOpen(state, blockIndex, events)
   }
+}
+
+const closeAllOpenBlocks = (
+  state: ResponsesStreamState,
+  events: Array<AnthropicStreamEventData>,
+) => {
+  closeOpenBlocks(state, events)
 
   state.functionCallStateByOutputIndex.clear()
 }
@@ -560,6 +564,7 @@ const openFunctionCallBlock = (
   const { blockIndex } = functionCallState
 
   if (!state.openBlocks.has(blockIndex)) {
+    closeOpenBlocks(state, events)
     events.push({
       type: "content_block_start",
       index: blockIndex,