refactor: Rename message stream handler and update session creation logic

src/main/apiServer/routes/agents/handlers/messages.ts

@@ -24,13 +24,13 @@ const verifyAgentAndSession = async (agentId: string, sessionId: string) => {
   return session
 }
 
-export const createMessageStream = async (req: Request, res: Response): Promise<void> => {
+export const createMessage = async (req: Request, res: Response): Promise<void> => {
   try {
     const { agentId, sessionId } = req.params
 
     const session = await verifyAgentAndSession(agentId, sessionId)
 
-    const messageData = { ...req.body, session_id: sessionId }
+    const messageData = req.body
 
     logger.info(`Creating streaming message for session: ${sessionId}`)
     logger.debug('Streaming message data:', messageData)
@@ -45,7 +45,7 @@ export const createMessageStream = async (req: Request, res: Response): Promise<
     // Send initial connection event
     res.write('data: {"type":"start"}\n\n')
 
-    const messageStream = sessionMessageService.createSessionMessageStream(session, messageData)
+    const messageStream = sessionMessageService.createSessionMessage(session, messageData)
 
     // Track if the response has ended to prevent further writes
     let responseEnded = false

src/main/apiServer/routes/agents/handlers/sessions.ts

@@ -1,19 +1,18 @@
 import { loggerService } from '@logger'
+import { sessionMessageService, sessionService } from '@main/services/agents'
 import { Request, Response } from 'express'
 
-import { sessionMessageService, sessionService } from '../../../../services/agents'
-
 const logger = loggerService.withContext('ApiServerSessionsHandlers')
 
 export const createSession = async (req: Request, res: Response): Promise<Response> => {
   try {
     const { agentId } = req.params
-    const sessionData = { ...req.body, main_agent_id: agentId }
+    const sessionData = req.body
 
     logger.info(`Creating new session for agent: ${agentId}`)
     logger.debug('Session data:', sessionData)
 
-    const session = await sessionService.createSession(sessionData)
+    const session = await sessionService.createSession(agentId, sessionData)
 
     logger.info(`Session created successfully: ${session.id}`)
     return res.status(201).json(session)
@@ -38,7 +37,7 @@ export const listSessions = async (req: Request, res: Response): Promise<Respons
 
     logger.info(`Listing sessions for agent: ${agentId} with limit=${limit}, offset=${offset}, status=${status}`)
 
-    const result = await sessionService.listSessions(agentId, { limit, offset, status })
+    const result = await sessionService.listSessions(agentId, { limit, offset })
 
     logger.info(`Retrieved ${result.sessions.length} sessions (total: ${result.total}) for agent: ${agentId}`)
     return res.json({
@@ -77,16 +76,16 @@ export const getSession = async (req: Request, res: Response): Promise<Response>
       })
     }
 
-    // Verify session belongs to the agent
-      logger.warn(`Session ${sessionId} does not belong to agent ${agentId}`)
-      return res.status(404).json({
-        error: {
-          message: 'Session not found for this agent',
-          type: 'not_found',
-          code: 'session_not_found'
-        }
-      })
-    }
+    // // Verify session belongs to the agent
+    //   logger.warn(`Session ${sessionId} does not belong to agent ${agentId}`)
+    //   return res.status(404).json({
+    //     error: {
+    //       message: 'Session not found for this agent',
+    //       type: 'not_found',
+    //       code: 'session_not_found'
+    //     }
+    //   })
+    // }
 
     // Fetch session messages
     logger.info(`Fetching messages for session: ${sessionId}`)
@@ -261,7 +260,7 @@ export const listAllSessions = async (req: Request, res: Response): Promise<Resp
 
     logger.info(`Listing all sessions with limit=${limit}, offset=${offset}, status=${status}`)
 
-    const result = await sessionService.listSessions(undefined, { limit, offset, status })
+    const result = await sessionService.listSessions(undefined, { limit, offset })
 
     logger.info(`Retrieved ${result.sessions.length} sessions (total: ${result.total})`)
     return res.json({

src/main/apiServer/routes/agents/index.ts

@@ -188,7 +188,7 @@ const createMessagesRouter = (): express.Router => {
   const messagesRouter = express.Router({ mergeParams: true })
 
   // Message CRUD routes (nested under agent/session)
-  messagesRouter.post('/', validateSessionMessage, handleValidationErrors, messageHandlers.createMessageStream)
+  messagesRouter.post('/', validateSessionMessage, handleValidationErrors, messageHandlers.createMessage)
   return messagesRouter
 }
 

src/main/apiServer/routes/agents/validators/messages.ts

@@ -1,6 +1,5 @@
 import { body } from 'express-validator'
 
 export const validateSessionMessage = [
-  body('role').notEmpty().isIn(['user', 'agent', 'system', 'tool']).withMessage('Valid role is required'),
   body('content').notEmpty().isString().withMessage('Content must be a valid string')
 ]