kyd-shared-badge 0.3.14 → 0.3.15

package/package.json

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

package/src/lib/context.ts

@@ -150,4 +150,9 @@ async function getJsonFromS3(bucket: string, key: string): Promise<any> {
 export async function getReportGraphData(badgeId: string): Promise<GraphInsightsPayload | undefined> {
   const badge = await doc.send(new GetCommand({ TableName: badgeTableName, Key: { badgeId } }));
   return badge.Item?.assessmentResult?.graph_insights as GraphInsightsPayload | undefined;
+}
+
+export async function getBadgeUserId(badgeId: string): Promise<string> {
+  const badge = await doc.send(new GetCommand({ TableName: badgeTableName, Key: { badgeId } }));
+  return badge.Item?.userId as string;
 }

package/src/lib/routes.ts

@@ -6,7 +6,13 @@ import { streamText } from 'ai';
 import { bedrock } from '@ai-sdk/amazon-bedrock';
 
 import { getHistory, putMessage } from './chat-store';
-import { aggregateUserData, cleanDeveloperProfile, buildAllContextPrompt, getReportGraphData } from './context';
+import { 
+  aggregateUserData, 
+  cleanDeveloperProfile, 
+  buildAllContextPrompt, 
+  getReportGraphData,
+  getBadgeUserId
+} from './context';
 import { checkAndConsumeToken } from './rate-limit';
 
 import { createSession } from './chat-store';
@@ -29,7 +35,8 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
 
     await putMessage({ sessionId, role: 'user', content });
 
-    const aggregated = await aggregateUserData(userId, !!companyId, companyId);
+    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 });