kyd-shared-badge 0.3.51 → 0.3.53

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kyd-shared-badge",
-  "version": "0.3.51",
+  "version": "0.3.53",
   "private": false,
   "main": "./src/index.ts",
   "module": "./src/index.ts",

package/src/SharedBadgeDisplay.tsx

@@ -179,7 +179,7 @@ const SharedBadgeDisplay = ({ badgeData, chatProps, headless }: { badgeData: Pub
                   setActiveTab(t.key);
                   try { if (typeof window !== 'undefined') window.history.replaceState(null, '', `#${t.key}`); } catch {}
                 }}
-                className={`px-3 sm:px-4 py-2 text-sm rounded-t flex items-center gap-2 cursor-pointer ${activeTab === t.key ? 'font-semibold' : ''}`}
+                className={`px-3 text-xs sm:text-base sm:px-4 py-2 text-sm rounded-t flex items-center gap-2 cursor-pointer ${activeTab === t.key ? 'font-semibold' : ''}`}
                 style={{
                   color: 'var(--text-main)',
                   background: activeTab === t.key ? 'var(--content-card-background)' : 'transparent',

package/src/chat/useChatStreaming.ts

@@ -56,7 +56,7 @@ export function useChatStreaming(cfg: UseChatStreamingConfig) {
       }
       const res = await fetch(api, {
         method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
+        headers: { 'Content-Type': 'application/json', 'Accept': 'text/event-stream' },
         body: JSON.stringify({ content, sessionId: sid, badgeId }),
         signal: abortRef.current.signal,
       });
@@ -65,13 +65,23 @@ export function useChatStreaming(cfg: UseChatStreamingConfig) {
 
       const reader = res.body.getReader();
       const decoder = new TextDecoder();
+      let buffer = '';
       let done = false;
       while (!done) {
-        const chunk = await reader.read();
-        done = chunk.done || false;
-        if (chunk.value) {
-          const textPart = decoder.decode(chunk.value, { stream: true });
-          setMessages(m => m.map(msg => msg.id === assistantId ? { ...msg, content: msg.content + textPart } : msg));
+        const { value, done: readerDone } = await reader.read();
+        done = readerDone || false;
+        if (value) {
+          buffer += decoder.decode(value, { stream: true });
+          const lines = buffer.split(/\r?\n/);
+          buffer = lines.pop() ?? '';
+          for (const raw of lines) {
+            const line = raw.trim();
+            if (!line.startsWith('data:')) continue;
+            const data = line.slice(5).trim();
+            if (!data) continue;
+            if (data === '[DONE]') { done = true; break; }
+            setMessages(m => m.map(msg => msg.id === assistantId ? { ...msg, content: msg.content + data } : msg));
+          }
         }
       }
     } catch (e) {

package/src/lib/routes.ts

@@ -2,8 +2,6 @@
 // POST /api/chat { content: string, sessionId?: string }
 
 import { NextRequest } from 'next/server';
-import { streamText } from 'ai';
-import { createOpenAI } from '@ai-sdk/openai';
 
 import { getHistory, putMessage } from './chat-store';
 import { 
@@ -34,19 +32,28 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
       return Response.json({ error: 'Rate limit exceeded' }, { status: 429 });
     }
 
+    // Persist the user message first so it is included in history
     await putMessage({ sessionId, role: 'user', content });
 
-    const badgeUserId = await getBadgeUserId(badgeId);
-    const aggregated = await aggregateUserData(badgeUserId, !!companyId, companyId);
+    // Build required context in parallel to reduce latency
+    const [badgeUserId, history] = await Promise.all([
+      getBadgeUserId(badgeId),
+      getHistory(sessionId, 20),
+    ]);
+    const [aggregated, graphData] = await Promise.all([
+      aggregateUserData(badgeUserId, !!companyId, companyId),
+      getReportGraphData(badgeId),
+    ]);
     const cleaned = cleanDeveloperProfile(aggregated);
-    const graphData = await getReportGraphData(badgeId);
     const system = buildAllContextPrompt(cleaned, graphData, { concise: true });
-    const history = await getHistory(sessionId, 20);
+
     const apiKey = process.env.OPENROUTER_API_KEY;
     if (!apiKey) {
       return Response.json({ error: 'Server misconfigured: missing OPENROUTER_API_KEY' }, { status: 500 });
     }
 
+    const model = process.env.OPENROUTER_MODEL || 'openai/o4-mini';
+
     const chatMessages = [
       { role: 'system', content: system },
       ...history.map((m: any) => ({ role: m.role, content: m.content })),
@@ -58,9 +65,10 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
       headers: {
         'Content-Type': 'application/json',
         'Authorization': `Bearer ${apiKey}`,
+        'Accept': 'text/event-stream',
       },
       body: JSON.stringify({
-        model: 'openai/o4-mini',
+        model,
         messages: chatMessages,
         stream: true,
         max_tokens: 1024,
@@ -84,6 +92,8 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
             const { value, done } = await reader.read();
             if (done) {
               try { if (assistantText) await putMessage({ sessionId, role: 'assistant', content: assistantText }); } catch {}
+              // Send final SSE terminator
+              controller.enqueue(encoder.encode('data: [DONE]\n\n'));
               controller.close();
               return;
             }
@@ -97,6 +107,7 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
               if (!data) continue;
               if (data === '[DONE]') {
                 try { if (assistantText) await putMessage({ sessionId, role: 'assistant', content: assistantText }); } catch {}
+                controller.enqueue(encoder.encode('data: [DONE]\n\n'));
                 controller.close();
                 return;
               }
@@ -105,9 +116,13 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
                 const delta = json?.choices?.[0]?.delta?.content ?? '';
                 if (delta) {
                   assistantText += delta;
-                  controller.enqueue(encoder.encode(delta));
+                  // Emit proper SSE line to the client
+                  controller.enqueue(encoder.encode(`data: ${delta}\n\n`));
                 }
-              } catch {}
+              } catch {
+                // If upstream sends a non-JSON data line, forward as-is
+                controller.enqueue(encoder.encode(`data: ${data}\n\n`));
+              }
             }
             pump();
           } catch (err) {
@@ -123,9 +138,10 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
 
     return new Response(stream, {
       headers: {
-        'Content-Type': 'text/plain; charset=utf-8',
+        'Content-Type': 'text/event-stream; charset=utf-8',
         'Cache-Control': 'no-cache, no-transform',
         'X-Accel-Buffering': 'no',
+        'Connection': 'keep-alive',
       },
     });
 

package/src/types.ts

@@ -466,4 +466,14 @@ export interface TopBusinessRule {
   label: string;
   weight: number;
   uid: string;
-}
+}
+
+export type Role = { 
+	roleId: string; 
+	name: string; 
+	description?: string; 
+	organization?: string; 
+	createdAt?: string; 
+	userId?: string;
+}
+