kyd-shared-badge 0.3.1 → 0.3.2

package/package.json

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

package/src/lib/routes.ts

@@ -5,7 +5,7 @@ import { NextRequest } from 'next/server';
 import { streamText } from 'ai';
 import { bedrock } from '@ai-sdk/amazon-bedrock';
 
-import { verifyCognito } from './auth-verify';
+// import { verifyCognito } from './auth-verify';
 import { getHistory, putMessage } from './chat-store';
 import { aggregateUserData, cleanDeveloperProfile, buildAllContextPrompt, getReportGraphData } from './context';
 import { checkAndConsumeToken } from './rate-limit';
@@ -15,22 +15,23 @@ import { createSession } from './chat-store';
 
 export const runtime = 'nodejs';
 
-export async function chatStreamRoute(req: NextRequest) {
+export async function chatStreamRoute(req: NextRequest, userId: string, companyId?: string) {
   try {
-    const user = await verifyCognito(req);
+    // const user = await verifyCognito(req);
+
     const { sessionId, content, badgeId } = await req.json();
     if (!content || !sessionId) {
       return Response.json({ error: 'Missing sessionId or content' }, { status: 400 });
     }
 
-    const allowed = await checkAndConsumeToken(user.userId, 20);
+    const allowed = await checkAndConsumeToken(userId, 20);
     if (!allowed) {
       return Response.json({ error: 'Rate limit exceeded' }, { status: 429 });
     }
 
     await putMessage({ sessionId, role: 'user', content });
 
-    const aggregated = await aggregateUserData(user.userId, !!user.companyId, user.companyId);
+    const aggregated = await aggregateUserData(userId, !!companyId, companyId);
     const cleaned = cleanDeveloperProfile(aggregated);
     const graphData = await getReportGraphData(badgeId);
     const system = buildAllContextPrompt(cleaned, graphData, { concise: true });
@@ -68,17 +69,13 @@ function tryParseEvidenceServer(text: string): any | null {
   return null;
 }
 
-
-
-
-
-export async function createSessionRoute(req: NextRequest) {
+export async function POST(req: NextRequest, userId: string, companyId?: string) {
   try {
-    const user = await verifyCognito(req);
+    // const user = await verifyCognito(req);
     const body = await req.json().catch(() => ({}));
     const sessionId = await createSession({
-      userId: user.userId,
-      companyId: user.companyId,
+      userId: userId,
+      companyId: companyId,
       model: body?.model,
       promptId: body?.promptId,
       promptVersion: body?.promptVersion,
@@ -89,6 +86,4 @@ export async function createSessionRoute(req: NextRequest) {
     if (e instanceof Response) return e;
     return Response.json({ error: 'Failed to create session' }, { status: 500 });
   }
-}
-
-
+}