Refactor handlers to improve type handling and move Express request type augmentation to a separate definition file.

Author: margostino

src/core/agents/handlers/createEvaluatorHandler.ts

@@ -71,10 +71,10 @@ export const createEvaluatorHandler = ({
       },
     ];
 
-    const evaluatorOutput: EvaluatorOutput = await think({
+    const evaluatorOutput = (await think({
       name: config.name,
       messages: chatCompletionMessages,
-    });
+    })) as EvaluatorOutput;
 
     if (!evaluatorOutput) {
       logger.error(`[${config.name}] Error evaluating:`);

src/core/agents/handlers/createPlannerHandler.ts

@@ -74,10 +74,10 @@ export const createPlannerHandler = ({
       },
     ];
 
-    const plannerOutput: PlannerOutput = await think({
+    const plannerOutput = (await think({
       name: config.name,
       messages: chatCompletionMessages,
-    });
+    })) as PlannerOutput;
 
     if (!plannerOutput) {
       logger.error(`[${config.name}] Error planning:`);

src/core/agents/handlers/createReporterHandler.ts

@@ -55,10 +55,10 @@ export const createReporterHandler = ({
       },
     ];
 
-    const summary = await think({
+    const summary = (await think({
       name: config.name,
       messages: chatCompletionMessages,
-    });
+    })) as string;
 
     if (!summary) {
       logger.error(`[${config.name}] Error summarizing`);

src/core/agents/handlers/createWorkerHandler.ts

@@ -62,7 +62,7 @@ export const createWorkerHandler = ({
       },
     ];
 
-    const llmOutput = await think({ name: config.name, messages: chatCompletionMessages });
+    const llmOutput = (await think({ name: config.name, messages: chatCompletionMessages })) as string;
 
     if (!llmOutput) {
       logger.error(`[${config.name}] Error thinking:`);

src/core/llm/createAiProcessor.ts

@@ -24,7 +24,7 @@ export const createAiProcessor = ({
   }: {
     name: string;
     messages: ChatCompletionMessageParam[];
-  }): Promise<z.infer<typeof schema> | string | null> => {
+  }): Promise<unknown> => {
     logger.info(`[${name}] thinking...`);
 
     const completionMessages = [

src/middlewares/authenticate.ts

@@ -2,16 +2,6 @@ import { NextFunction, Request, Response } from 'express';
 import jwt from 'jsonwebtoken';
 import { env, logger } from '../config';
 
-declare module 'express-serve-static-core' {
-  interface Request {
-    user?: {
-      userId: string;
-      username: string;
-      role: string;
-    };
-  }
-}
-
 export const authenticate = (req: Request, res: Response, next: NextFunction): void => {
   try {
     const authHeader = req.headers.authorization;

src/types/express.d.ts

@@ -0,0 +1,11 @@
+import 'express';
+
+declare module 'express-serve-static-core' {
+  interface Request {
+    user?: {
+      userId: string;
+      username: string;
+      role: string;
+    };
+  }
+}

tests/setup.ts

@@ -5,6 +5,8 @@ jest.setTimeout(10000);
 beforeAll(() => {
   process.env.NODE_ENV = 'test';
   process.env.LOG_LEVEL = 'error';
+  process.env.JWT_SECRET = 'test-secret-key';
+  process.env.JWT_EXPIRES_IN = '7d';
 });
 
 afterEach(() => {