Skip to content

Commit 9adc6e0

Browse files
committed
feat: enhance reasoning handling in translation process
1 parent 65a3568 commit 9adc6e0

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/routes/messages/stream-translation.ts

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,8 +62,11 @@ function handleFinish(
6262
index: state.contentBlockIndex,
6363
})
6464
state.contentBlockOpen = false
65+
state.contentBlockIndex++
6566
}
6667

68+
handleReasoningOpaque(choice.delta, events, state)
69+
6770
events.push(
6871
{
6972
type: "message_delta",
@@ -96,7 +99,7 @@ function handleToolCalls(
9699
events: Array<AnthropicStreamEventData>,
97100
) {
98101
if (delta.tool_calls && delta.tool_calls.length > 0) {
99-
closeThinkingBlockIfOpen(delta, state, events)
102+
closeThinkingBlockIfOpen(state, events)
100103

101104
for (const toolCall of delta.tool_calls) {
102105
if (toolCall.id && toolCall.function?.name) {
@@ -156,7 +159,7 @@ function handleContent(
156159
events: Array<AnthropicStreamEventData>,
157160
) {
158161
if (delta.content && delta.content.length > 0) {
159-
closeThinkingBlockIfOpen(delta, state, events)
162+
closeThinkingBlockIfOpen(state, events)
160163

161164
if (isToolBlockOpen(state)) {
162165
// A tool block was open, so close it before starting a text block.
@@ -312,7 +315,6 @@ function handleThinkingText(
312315
}
313316

314317
function closeThinkingBlockIfOpen(
315-
delta: Delta,
316318
state: AnthropicStreamState,
317319
events: Array<AnthropicStreamEventData>,
318320
): void {
@@ -334,7 +336,6 @@ function closeThinkingBlockIfOpen(
334336
state.contentBlockIndex++
335337
state.thinkingBlockOpen = false
336338
}
337-
handleReasoningOpaque(delta, events, state)
338339
}
339340

340341
export function translateErrorToAnthropicErrorEvent(): AnthropicStreamEventData {

0 commit comments

Comments
 (0)