Upload folder using huggingface_hub

backend/src/routes/query.js

@@ -40,12 +40,37 @@ router.post('/query/stream', authMiddleware, async (req, res) => {
       question, policy_id, k
     }, { responseType: 'stream' });
 
-    res.setHeader('Content-Type', 'text/plain');
+    res.setHeader('Content-Type', 'text/event-stream');
+    res.setHeader('Cache-Control', 'no-cache');
+    res.setHeader('Connection', 'keep-alive');
     
-    let fullAnswer = "";
+    let answerText = "";
+    let sourcesData = null;
+    let buffer = "";
+
     response.data.on('data', (chunk) => {
-      fullAnswer += chunk.toString();
       res.write(chunk);
+      
+      buffer += chunk.toString();
+      const lines = buffer.split('\n\n');
+      buffer = lines.pop() || '';
+
+      for (const line of lines) {
+        if (line.startsWith('data: ')) {
+          const jsonStr = line.slice(6).trim();
+          if (jsonStr === '[DONE]') continue;
+          try {
+            const parsed = JSON.parse(jsonStr);
+            if (parsed.type === 'token') {
+              answerText += parsed.content;
+            } else if (parsed.type === 'sources') {
+              sourcesData = parsed.sources;
+            }
+          } catch (e) {
+            // ignore partial JSON parse errors
+          }
+        }
+      }
     });
 
     response.data.on('end', async () => {
@@ -55,7 +80,8 @@ router.post('/query/stream', authMiddleware, async (req, res) => {
           user_id: req.user?.id || 'anonymous',
           policy_id,
           question,
-          answer: fullAnswer
+          answer: answerText,
+          sources: sourcesData
         });
       } catch (dbErr) {
         console.error('[query/stream] DB Save Error:', dbErr.message);