kyd-shared-badge 0.3.38 → 0.3.40

package/package.json

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

package/src/SharedBadgeDisplay.tsx

@@ -583,20 +583,78 @@ const SharedBadgeDisplay = ({ badgeData, chatProps, headless }: { badgeData: Pub
   const AiSection = () => (
     <div className={`${wrapperMaxWidth} mx-auto`}>
       <div className={'rounded-xl shadow-xl p-6 sm:p-8 mt-6 border'} style={{ backgroundColor: 'var(--content-card-background)', borderColor: 'var(--icon-button-secondary)' }}>
+        <Reveal headless={isHeadless} as={'h4'} offsetY={8} className={'text-2xl font-semibold mb-3'} style={{ color: 'var(--text-main)' }}>KYD AI (Beta)</Reveal>
         {(() => {
-          const ai_usage_summary = assessmentResult?.ai_usage_summary;
-          const label = 'AI Transparency';
-          const topMovers = ai_usage_summary?.key_findings || [];
+          const ai = assessmentResult?.ai_usage_summary;
+          if (!ai) return null;
+          const techGauge = {
+            percent: Math.max(0, Math.min(100, Number(ai.originality_score ?? 0))),
+            label: 'Originality',
+          };
+          const riskGauge = {
+            percent: Math.max(0, Math.min(100, Number(ai.transparency_score ?? 0))),
+            label: ai.transparency_descriptor ? `Transparency — ${ai.transparency_descriptor}` : 'Transparency',
+          };
+          const stats: Array<{ label: string; value: string }>= [
+            { label: 'Files analyzed', value: String(ai.files_analyzed ?? 0) },
+            { label: 'Files with AI findings', value: String(ai.files_with_ai_findings ?? 0) },
+            { label: 'Files with disclosure', value: String(ai.files_with_disclosure ?? 0) },
+            { label: 'Repos analyzed', value: String(ai.repos_analyzed ?? 0) },
+            { label: 'Repos with AI findings', value: String(ai.repos_with_ai_findings ?? 0) },
+          ];
+          const findings = Array.isArray(ai.evidence) ? ai.evidence.slice(0, 5) : [];
+          const topMovers = Array.isArray(ai.key_findings) ? ai.key_findings : [];
           return (
-            <GaugeCard
-              key={'ai-card'}
-              title={'KYD AI (Beta)'}
-              description={'Indicates the degree to which AI-assisted code is explicitly disclosed across analyzed files.'}
-              percent={ai_usage_summary?.transparency_score}
-              label={label}
-              topMovers={topMovers.map(t => ({ label: t, uid: 'ai-usage' }))}
-              topMoversTitle={'Key Findings'}
-            />
+            <div className="space-y-8">
+              <div className="grid grid-cols-1 sm:grid-cols-2 gap-4 *:min-h-full">
+                <GaugeCard
+                  key={'ai-originality'}
+                  title={'Originality'}
+                  description={'Estimated share of human-authored code based on linguistic signals in comments.'}
+                  percent={techGauge.percent}
+                  label={techGauge.label}
+                  topMovers={[]}
+                />
+                <GaugeCard
+                  key={'ai-transparency'}
+                  title={'Transparency'}
+                  description={'Proportion of AI-influenced files that include an explicit disclosure.'}
+                  percent={riskGauge.percent}
+                  label={riskGauge.label}
+                  topMovers={topMovers.map(t => ({ label: t, uid: 'ai-usage' }))}
+                  topMoversTitle={'Key Findings'}
+                />
+              </div>
+              <div className="grid grid-cols-2 sm:grid-cols-5 gap-4">
+                {stats.map((s, i) => (
+                  <div key={i} className={'rounded-md p-3 border'} style={{ borderColor: 'var(--icon-button-secondary)', background: 'var(--content-card-background)' }}>
+                    <div className={'text-xs'} style={{ color: 'var(--text-secondary)' }}>{s.label}</div>
+                    <div className={'text-lg font-semibold'} style={{ color: 'var(--text-main)' }}>{s.value}</div>
+                  </div>
+                ))}
+              </div>
+              {findings.length > 0 && (
+                <div>
+                  <div className={'text-sm font-semibold mb-3'} style={{ color: 'var(--text-main)' }}>Sample Findings</div>
+                  <div className="space-y-3">
+                    {findings.map((f, idx) => (
+                      <div key={idx} className={'rounded-md p-3 border'} style={{ borderColor: 'var(--icon-button-secondary)', background: 'var(--content-card-background)' }}>
+                        <div className={'text-xs mb-1'} style={{ color: 'var(--text-secondary)' }}>{f.file_path || 'file'}</div>
+                        {f.snippet ? (
+                          <pre className={'text-xs overflow-auto p-2 rounded'} style={{ background: 'rgba(0,0,0,0.04)', color: 'var(--text-main)' }}>{f.snippet}</pre>
+                        ) : null}
+                        {f.reason ? (
+                          <div className={'text-xs mt-2'} style={{ color: 'var(--text-secondary)' }}>{f.reason}</div>
+                        ) : null}
+                      </div>
+                    ))}
+                  </div>
+                </div>
+              )}
+              {ai.explanation && (
+                <div className={'text-sm'} style={{ color: 'var(--text-secondary)' }}>{ai.explanation}</div>
+              )}
+            </div>
           );
         })()}
       </div>

package/src/chat/ChatWidget.tsx

@@ -54,8 +54,7 @@ export default function ChatWidget({ api = '/api/chat', title = 'KYD Bot', hintT
 
   useEffect(() => {
     // when sidebar is closed, dismiss hint
-    if (open) return;
-    setShowHint(false);
+    if (!open) setShowHint(false);
   }, [open]);
 
   // Sidebar width with bounds and persistence
@@ -278,7 +277,7 @@ export default function ChatWidget({ api = '/api/chat', title = 'KYD Bot', hintT
             )}
           </div>
         ) : (
-          <div style={{ position: 'fixed', right: 0, top: Math.max(headerTop + 16, tabTop), height: 160, width: 44 }}>
+          <div style={{ position: 'fixed', right: 0, top: Math.max(headerTop + 16, tabTop), height: 160, width: 44, overflow: 'hidden' }}>
             <button
               aria-label={'Open chat sidebar'}
               aria-expanded={open}

package/src/lib/routes.ts

@@ -20,6 +20,7 @@ import { createSession } from './chat-store';
 
 export const runtime = 'nodejs';
 
+
 export async function chatStreamRoute(req: NextRequest, userId: string, companyId?: string) {
   try {
 
@@ -41,43 +42,99 @@ 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;
+    if (!apiKey) {
+      return Response.json({ error: 'Server misconfigured: missing OPENROUTER_API_KEY' }, { status: 500 });
+    }
+
+    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}`,
+      },
+      body: JSON.stringify({
+        model: 'openai/o4-mini',
+        messages: chatMessages,
+        stream: true,
+        max_tokens: 1024,
+      }),
+    });
+
+    if (!upstream.ok || !upstream.body) {
+      const errText = await upstream.text().catch(() => '');
+      return Response.json({ error: `Upstream error: ${upstream.status}`, details: errText }, { status: 502 });
+    }
 
-    const openrouter = createOpenAI({
-      apiKey: process.env.OPENROUTER_API_KEY,
-      baseURL: 'https://openrouter.ai/api/v1',
+    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: openrouter('openai/o4-mini'),
-      system,
-      messages: [...history, { role: 'user' as const, content }],
-      maxTokens: 1024,
-      temperature: 0.2,
-      // onFinish: async ({ text }: { text: string }) => {
-      //   // Attempt to capture evidence block if present (UI also parses, but cache server-side)
-      //   const evidence = tryParseEvidenceServer(text);
-      //   await putMessage({ sessionId, role: 'assistant', content: text, evidenceJson: evidence || undefined });
-      // },
+    return new Response(stream, {
+      headers: {
+        'Content-Type': 'text/plain; charset=utf-8',
+        'Cache-Control': 'no-cache, no-transform',
+        'X-Accel-Buffering': 'no',
+      },
     });
 
-    return result.toTextStreamResponse();
   } catch (e) {
     console.error('chat error', e);
     return new Response('An error occurred. Please try again.', { status: 500 });
   }
 }
 
-// function tryParseEvidenceServer(text: string): any | null {
-//   try {
-//     const start = text.indexOf('```json');
-//     const end = text.indexOf('```', start + 7);
-//     const raw = start >= 0 && end > start ? text.slice(start + 7, end).trim() : text.trim();
-//     const obj = JSON.parse(raw);
-//     if (obj && obj.type === 'evidence' && Array.isArray(obj.claims)) return obj;
-//   } catch {}
-//   return null;
-// }
-
 export async function createSessionRoute(req: NextRequest, userId: string, companyId?: string) {
   try {
     const body = await req.json().catch(() => ({}));

package/src/types.ts

@@ -203,10 +203,16 @@ export interface AssessmentResult {
   ai_usage_summary?: {
     explanation: string;
     key_findings: string[];
+    originality_score: number;
+    transparency_score: number;
+    transparency_descriptor?: string;
     files_with_ai_findings: number;
     files_analyzed: number;
-    files_with_disclosure: number;
-    transparency_score: number;
+    files_with_disclosure?: number;
+    repos_analyzed?: number;
+    repos_with_ai_findings?: number;
+    evidence?: Array<{ file_path?: string; snippet?: string; reason?: string }>;
+    findings_by_repo?: Record<string, Array<{ file_path?: string; snippet?: string; reason?: string }>>;
   };
   key_skills?: string[];
   summary_scores?: {
@@ -394,13 +400,19 @@ export interface GraphInsightsPayload {
       top_movers?: Array<{ label?: string; uid?: string }>;
     };
   };
-  ai_usage_summary: {
+  ai_usage_summary?: {
     explanation: string;
     key_findings: string[];
+    originality_score: number;
+    transparency_score: number;
+    transparency_descriptor?: string;
     files_with_ai_findings: number;
     files_analyzed: number;
     files_with_disclosure?: number;
-    transparency_score: number;
+    repos_analyzed?: number;
+    repos_with_ai_findings?: number;
+    evidence?: Array<{ file_path?: string; snippet?: string; reason?: string }>;
+    findings_by_repo?: Record<string, Array<{ file_path?: string; snippet?: string; reason?: string }>>;
   }
 }