From 3a9820c8cba4aeeb35e46e3d3e7f458f363eb918 Mon Sep 17 00:00:00 2001
From: lsdsjy <1356263+lsdsjy@users.noreply.github.com>
Date: Fri, 21 Aug 2026 17:40:32 +0800
Subject: [PATCH] fix(headless): make stream chunk handling exhaustive

---
 packages/bundle/headless/src/index.ts | 47 +++++++++++++++------------
 1 file changed, 27 insertions(+), 20 deletions(-)

diff --git a/packages/bundle/headless/src/index.ts b/packages/bundle/headless/src/index.ts
index 75289736a5..b5b2839b00 100644
--- a/packages/bundle/headless/src/index.ts
+++ b/packages/bundle/headless/src/index.ts
@@ -14,7 +14,7 @@ import z from '@deepseek-ai/schemastery'
 import { installModelSelection } from '@deepseek-ai/dsh-agent'
 import type { Agent, ModelSelectionRef } from '@deepseek-ai/dsh-agent'
 import type {} from '@deepseek-ai/dsh-agent-default-model'
-import { createUserMessage } from '@deepseek-ai/dsh-llm'
+import { assertNever, createUserMessage } from '@deepseek-ai/dsh-llm'
 import { SessionId } from '@deepseek-ai/dsh-session'
 import type { SessionEvent } from '@deepseek-ai/dsh-session'
 // Empty type imports carry the loader Context merge for the settlement await
@@ -113,26 +113,33 @@ function streamReasoning(
     }
     if (!started || event.type !== 'assistant/chunk') return
     const chunk = event.data.chunk
-    if (chunk.type === 'reasoning-delta') {
-      if (chunk.text === '') return
-      if (!open) {
-        stderr.write('dsh: reasoning:\n')
-        open = true
-      }
-      stderr.write(chunk.text)
-      endsWithNewline = chunk.text.endsWith('\n')
-      return
+    switch (chunk.type) {
+      case 'reasoning-delta':
+        if (chunk.text === '') return
+        if (!open) {
+          stderr.write('dsh: reasoning:\n')
+          open = true
+        }
+        stderr.write(chunk.text)
+        endsWithNewline = chunk.text.endsWith('\n')
+        return
+      case 'block-start':
+        if (chunk.blockType !== 'reasoning') close()
+        return
+      case 'block-end':
+        if (chunk.block.type !== 'reasoning') close()
+        return
+      case 'usage':
+        return
+      case 'text-delta':
+      case 'tool-call-delta':
+      case 'finish':
+        close()
+        return
+      /* v8 ignore next -- closed-union exhaustiveness guard */
+      default:
+        return assertNever(chunk, 'headless reasoning stream')
     }
-    if (chunk.type === 'block-start') {
-      if (chunk.blockType !== 'reasoning') close()
-      return
-    }
-    if (chunk.type === 'block-end') {
-      if (chunk.block.type !== 'reasoning') close()
-      return
-    }
-    if (chunk.type === 'usage') return
-    close()
   })
   return () => {
     dispose()