kyd-shared-badge 0.3.54 → 0.3.56

package/package.json

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

package/src/components/ReportHeader.tsx

@@ -9,10 +9,10 @@ import { FiInfo, FiAlertTriangle } from 'react-icons/fi';
 // Register English locale once at module import time
 countriesLib.registerLocale(enLocale);
 
-export const getBadgeImageUrl = (score: number) => {
-  if (score >= 75) return '/badgegreen2.png';
-  if (score >= 50) return '/badgeyellow2.png';
-  return '/badgered2.png';
+export const getBadgeImageUrl = (score: number, bland: boolean = false) => {
+  if (score >= 75) return bland ? 'badgegreensimple.png' : '/badgegreen2.png';
+  if (score >= 50) return bland ? 'badgeyellowsimple.png' : '/badgeyellow2.png';
+  return bland ? 'badgeredsimple.png' : '/badgered2.png';
 };
 
 const hexToRgba = (hex: string, alpha: number) => {

package/src/lib/routes.ts

@@ -42,9 +42,9 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
     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;
+    const apiKey = process.env.OPENAI_API_KEY;
     if (!apiKey) {
-      return Response.json({ error: 'Server misconfigured: missing OPENROUTER_API_KEY' }, { status: 500 });
+      return Response.json({ error: 'Server misconfigured: missing OPENAI_API_KEY' }, { status: 500 });
     }
 
     const chatMessages = [
@@ -53,24 +53,21 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
       { role: 'user', content },
     ];
 
-    const openai = createOpenAI({
-      apiKey,
-      baseURL: 'https://openrouter.ai/api/v1',
-    });
+    const openai = createOpenAI({ apiKey });
 
     const result = await streamText({
-      model: openai('openai/o4-mini'),
+      model: openai('gpt-4o-mini'),
       messages: chatMessages as any,
       maxTokens: 1024,
+      onFinish: async ({ text }: { text: string }) => {
+        try { if (text) await putMessage({ sessionId, role: 'assistant', content: text }); } catch {}
+      },
     });
 
-    return result.toAIStreamResponse({
+    return result.toTextStreamResponse({
       headers: {
         'X-Accel-Buffering': 'no',
       },
-      onFinal: async (finalText: string) => {
-        try { if (finalText) await putMessage({ sessionId, role: 'assistant', content: finalText }); } catch {}
-      },
     });
 
   } catch (e) {