kyd-shared-badge 0.3.53 → 0.3.54

package/package.json

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

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', 'Accept': 'text/event-stream' },
+        headers: { 'Content-Type': 'application/json' },
         body: JSON.stringify({ content, sessionId: sid, badgeId }),
         signal: abortRef.current.signal,
       });
@@ -65,23 +65,13 @@ export function useChatStreaming(cfg: UseChatStreamingConfig) {
 
       const reader = res.body.getReader();
       const decoder = new TextDecoder();
-      let buffer = '';
       let done = false;
       while (!done) {
-        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));
-          }
+        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));
         }
       }
     } catch (e) {

package/src/components/ReportHeader.tsx

@@ -9,7 +9,7 @@ import { FiInfo, FiAlertTriangle } from 'react-icons/fi';
 // Register English locale once at module import time
 countriesLib.registerLocale(enLocale);
 
-const getBadgeImageUrl = (score: number) => {
+export const getBadgeImageUrl = (score: number) => {
   if (score >= 75) return '/badgegreen2.png';
   if (score >= 50) return '/badgeyellow2.png';
   return '/badgered2.png';

package/src/index.ts

@@ -4,5 +4,4 @@ export { default as PrintableBadgeDisplay } from './PrintableBadgeDisplay';
 export { default as ChatWindowStreaming } from './chat/ChatWindowStreaming';
 export { default as ChatWidget } from './chat/ChatWidget';
 export * from './utils/date';
-
-
+export { getBadgeImageUrl } from './components/ReportHeader';

package/src/lib/routes.ts

@@ -2,6 +2,8 @@
 // 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 { 
@@ -32,116 +34,42 @@ 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 });
 
-    // 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 badgeUserId = await getBadgeUserId(badgeId);
+    const aggregated = await aggregateUserData(badgeUserId, !!companyId, companyId);
     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 })),
       { role: 'user', content },
     ];
 
-    const upstream = await fetch('https://openrouter.ai/api/v1/chat/completions', {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        'Authorization': `Bearer ${apiKey}`,
-        'Accept': 'text/event-stream',
-      },
-      body: JSON.stringify({
-        model,
-        messages: chatMessages,
-        stream: true,
-        max_tokens: 1024,
-      }),
+    const openai = createOpenAI({
+      apiKey,
+      baseURL: 'https://openrouter.ai/api/v1',
     });
 
-    if (!upstream.ok || !upstream.body) {
-      const errText = await upstream.text().catch(() => '');
-      return Response.json({ error: `Upstream error: ${upstream.status}`, details: errText }, { status: 502 });
-    }
-
-    const encoder = new TextEncoder();
-    let sseBuffer = '';
-    let assistantText = '';
-    const stream = new ReadableStream<Uint8Array>({
-      start(controller) {
-        const reader = upstream.body!.getReader();
-        const decoder = new TextDecoder();
-        const pump = async (): Promise<void> => {
-          try {
-            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;
-            }
-            sseBuffer += decoder.decode(value, { stream: true });
-            const lines = sseBuffer.split(/\r?\n/);
-            sseBuffer = lines.pop() ?? '';
-            for (const l of lines) {
-              const line = l.trim();
-              if (!line.startsWith('data:')) continue;
-              const data = line.slice(5).trim();
-              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;
-              }
-              try {
-                const json = JSON.parse(data);
-                const delta = json?.choices?.[0]?.delta?.content ?? '';
-                if (delta) {
-                  assistantText += delta;
-                  // Emit proper SSE line to the client
-                  controller.enqueue(encoder.encode(`data: ${delta}\n\n`));
-                }
-              } catch {
-                // If upstream sends a non-JSON data line, forward as-is
-                controller.enqueue(encoder.encode(`data: ${data}\n\n`));
-              }
-            }
-            pump();
-          } catch (err) {
-            controller.error(err);
-          }
-        };
-        pump();
-      },
-      cancel() {
-        try { (upstream as any).body?.cancel?.(); } catch {}
-      },
+    const result = await streamText({
+      model: openai('openai/o4-mini'),
+      messages: chatMessages as any,
+      maxTokens: 1024,
     });
 
-    return new Response(stream, {
+    return result.toAIStreamResponse({
       headers: {
-        'Content-Type': 'text/event-stream; charset=utf-8',
-        'Cache-Control': 'no-cache, no-transform',
         'X-Accel-Buffering': 'no',
-        'Connection': 'keep-alive',
+      },
+      onFinal: async (finalText: string) => {
+        try { if (finalText) await putMessage({ sessionId, role: 'assistant', content: finalText }); } catch {}
       },
     });