kyd-shared-badge 0.3.13 → 0.3.15

package/package.json

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

package/src/lib/context.ts

@@ -97,6 +97,8 @@ export function buildAllContextPrompt(cleanedData: any, reportGraphData: GraphIn
     'Return natural language for the main reply.',
     'When referring to the input JSON, use the term "Report Information" or "Report"',
     'You are speaking to a non-technical user, ensure you do not use variable-esk language like "based on linkedin_provider", instead use "based the LinkedIn profile"',
+    'Do NOT recommend reviewing external sources or contacting anyone. Do NOT suggest actions such as "reach out", "discuss directly", "interview", or "collect more information". If the report lacks information to answer, state that plainly and stop.',
+    'Avoid advisory phrasing like "I recommend", "you should", or "consider". Provide only what the Report contains or explicitly lacks.',
     // 'Optionally append a fenced JSON evidence block at the end if relevant.',
     // 'If you output evidence, follow the exact evidence schema described.',
   ];
@@ -148,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 });