kyd-shared-badge 0.3.52 → 0.3.54

package/package.json

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

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

@@ -53,80 +53,24 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
       { role: 'user', content },
     ];
 
-    const upstream = await fetch('https://openrouter.ai/api/v1/chat/completions', {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        'Authorization': `Bearer ${apiKey}`,
-      },
-      body: JSON.stringify({
-        model: 'openai/o4-mini',
-        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 {}
-              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.close();
-                return;
-              }
-              try {
-                const json = JSON.parse(data);
-                const delta = json?.choices?.[0]?.delta?.content ?? '';
-                if (delta) {
-                  assistantText += delta;
-                  controller.enqueue(encoder.encode(delta));
-                }
-              } catch {}
-            }
-            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/plain; charset=utf-8',
-        'Cache-Control': 'no-cache, no-transform',
         'X-Accel-Buffering': 'no',
       },
+      onFinal: async (finalText: string) => {
+        try { if (finalText) await putMessage({ sessionId, role: 'assistant', content: finalText }); } catch {}
+      },
     });
 
   } catch (e) {