From 8632cfeaf2acfa2b94b1eb65f5201f96c2339954 Mon Sep 17 00:00:00 2001
From: shenlan <manbuzhe2009@qq.com>
Date: Thu, 14 Aug 2025 14:13:25 +0800
Subject: [PATCH] fix: handle non-streaming chat responses

---
 ui/homepage/components/AskAIDialog.tsx | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/ui/homepage/components/AskAIDialog.tsx b/ui/homepage/components/AskAIDialog.tsx
index 822d765..7b0e29a 100644
--- a/ui/homepage/components/AskAIDialog.tsx
+++ b/ui/homepage/components/AskAIDialog.tsx
@@ -72,6 +72,16 @@ export function AskAIDialog({
 
       if (!res.ok) throw new Error('Request failed')
 
+      const contentType = res.headers.get('content-type') || ''
+      // Handle non-streaming JSON responses directly.
+      if (!contentType.includes('text/event-stream')) {
+        const data = await res.json().catch(() => ({}))
+        let answer = typeof data === 'string' ? data : data.answer || ''
+        let retrieved = data.chunks || data.sources || []
+        update(answer, retrieved)
+        return { answer, retrieved }
+      }
+
       const reader = res.body?.getReader()
       if (!reader) throw new Error('No reader')