feat(agents, sessions): implement replace functionality for agent and session updates

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

@@ -1,8 +1,9 @@
 import { loggerService } from '@logger'
-import { ListAgentsResponse } from '@types'
+import { ListAgentsResponse,type ReplaceAgentRequest, type UpdateAgentRequest } from '@types'
 import { Request, Response } from 'express'
 
 import { agentService } from '../../../../services/agents'
+import type { ValidationRequest } from '../validators/zodValidator'
 
 const logger = loggerService.withContext('ApiServerAgentsHandlers')
 
@@ -263,7 +264,10 @@ export const updateAgent = async (req: Request, res: Response): Promise<Response
     logger.info(`Updating agent: ${agentId}`)
     logger.debug('Update data:', req.body)
 
-    const agent = await agentService.updateAgent(agentId, req.body)
+    const { validatedBody } = req as ValidationRequest
+    const replacePayload = (validatedBody ?? {}) as ReplaceAgentRequest
+
+    const agent = await agentService.updateAgent(agentId, replacePayload, { replace: true })
 
     if (!agent) {
       logger.warn(`Agent not found for update: ${agentId}`)
@@ -395,7 +399,10 @@ export const patchAgent = async (req: Request, res: Response): Promise<Response>
     logger.info(`Partially updating agent: ${agentId}`)
     logger.debug('Partial update data:', req.body)
 
-    const agent = await agentService.updateAgent(agentId, req.body)
+    const { validatedBody } = req as ValidationRequest
+    const updatePayload = (validatedBody ?? {}) as UpdateAgentRequest
+
+    const agent = await agentService.updateAgent(agentId, updatePayload)
 
     if (!agent) {
       logger.warn(`Agent not found for partial update: ${agentId}`)

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

@@ -1,8 +1,10 @@
 import { loggerService } from '@logger'
 import { sessionMessageService, sessionService } from '@main/services/agents'
-import { CreateSessionResponse, ListAgentSessionsResponse } from '@types'
+import { CreateSessionResponse, ListAgentSessionsResponse,type ReplaceSessionRequest } from '@types'
 import { Request, Response } from 'express'
 
+import type { ValidationRequest } from '../validators/zodValidator'
+
 const logger = loggerService.withContext('ApiServerSessionsHandlers')
 
 export const createSession = async (req: Request, res: Response): Promise<Response> => {
@@ -131,9 +133,10 @@ export const updateSession = async (req: Request, res: Response): Promise<Respon
       })
     }
 
-    // For PUT, we replace the entire resource
-    const sessionData = { ...req.body, main_agent_id: agentId }
-    const session = await sessionService.updateSession(agentId, sessionId, sessionData)
+    const { validatedBody } = req as ValidationRequest
+    const replacePayload = (validatedBody ?? {}) as ReplaceSessionRequest
+
+    const session = await sessionService.updateSession(agentId, sessionId, replacePayload)
 
     if (!session) {
       logger.warn(`Session not found for update: ${sessionId}`)

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

@@ -5,11 +5,13 @@ import { checkAgentExists, handleValidationErrors } from './middleware'
 import {
   validateAgent,
   validateAgentId,
+  validateAgentReplace,
   validateAgentUpdate,
   validatePagination,
   validateSession,
   validateSessionId,
   validateSessionMessage,
+  validateSessionReplace,
   validateSessionUpdate
 } from './validators'
 
@@ -152,7 +154,13 @@ const agentsRouter = express.Router()
 agentsRouter.post('/', validateAgent, handleValidationErrors, agentHandlers.createAgent)
 agentsRouter.get('/', validatePagination, handleValidationErrors, agentHandlers.listAgents)
 agentsRouter.get('/:agentId', validateAgentId, handleValidationErrors, agentHandlers.getAgent)
-agentsRouter.put('/:agentId', validateAgentId, validateAgentUpdate, handleValidationErrors, agentHandlers.updateAgent)
+agentsRouter.put(
+  '/:agentId',
+  validateAgentId,
+  validateAgentReplace,
+  handleValidationErrors,
+  agentHandlers.updateAgent
+)
 agentsRouter.patch('/:agentId', validateAgentId, validateAgentUpdate, handleValidationErrors, agentHandlers.patchAgent)
 agentsRouter.delete('/:agentId', validateAgentId, handleValidationErrors, agentHandlers.deleteAgent)
 
@@ -167,7 +175,7 @@ const createSessionsRouter = (): express.Router => {
   sessionsRouter.put(
     '/:sessionId',
     validateSessionId,
-    validateSessionUpdate,
+    validateSessionReplace,
     handleValidationErrors,
     sessionHandlers.updateSession
   )

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

@@ -1,4 +1,9 @@
-import { AgentIdParamSchema, CreateAgentRequestSchema, UpdateAgentRequestSchema } from '@types'
+import {
+  AgentIdParamSchema,
+  CreateAgentRequestSchema,
+  ReplaceAgentRequestSchema,
+  UpdateAgentRequestSchema
+} from '@types'
 
 import { createZodValidator } from './zodValidator'
 
@@ -6,6 +11,10 @@ export const validateAgent = createZodValidator({
   body: CreateAgentRequestSchema
 })
 
+export const validateAgentReplace = createZodValidator({
+  body: ReplaceAgentRequestSchema
+})
+
 export const validateAgentUpdate = createZodValidator({
   body: UpdateAgentRequestSchema
 })

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

@@ -1,4 +1,9 @@
-import { CreateSessionRequestSchema, SessionIdParamSchema, UpdateSessionRequestSchema } from '@types'
+import {
+  CreateSessionRequestSchema,
+  ReplaceSessionRequestSchema,
+  SessionIdParamSchema,
+  UpdateSessionRequestSchema
+} from '@types'
 
 import { createZodValidator } from './zodValidator'
 
@@ -6,6 +11,10 @@ export const validateSession = createZodValidator({
   body: CreateSessionRequestSchema
 })
 
+export const validateSessionReplace = createZodValidator({
+  body: ReplaceSessionRequestSchema
+})
+
 export const validateSessionUpdate = createZodValidator({
   body: UpdateSessionRequestSchema
 })