diff --git a/src/main/services/agents/services/AgentService.ts b/src/main/services/agents/services/AgentService.ts
index aba60b040a..a247243d26 100644
--- a/src/main/services/agents/services/AgentService.ts
+++ b/src/main/services/agents/services/AgentService.ts
@@ -8,7 +8,8 @@ import type {
   GetAgentResponse,
   ListAgentsResponse,
   ListOptions,
-  UpdateAgentRequest
+  UpdateAgentRequest,
+  UpdateAgentResponse
 } from '@types'
 import { count, eq } from 'drizzle-orm'
 
@@ -111,7 +112,7 @@ export class AgentService extends BaseService {
     return { agents, total: totalResult[0].count }
   }
 
-  async updateAgent(id: string, updates: UpdateAgentRequest): Promise<GetAgentResponse | null> {
+  async updateAgent(id: string, updates: UpdateAgentRequest): Promise<UpdateAgentResponse | null> {
     this.ensureInitialized()
 
     // Check if agent exists
diff --git a/src/renderer/src/types/agent.ts b/src/renderer/src/types/agent.ts
index 8e6f1125d0..1ee069f5cf 100644
--- a/src/renderer/src/types/agent.ts
+++ b/src/renderer/src/types/agent.ts
@@ -138,6 +138,10 @@ export const ListAgentsResponseSchema = z.object({
 
 export type ListAgentsResponse = z.infer<typeof ListAgentsResponseSchema>
 
+export const UpdateAgentResponseSchema = GetAgentResponseSchema
+
+export type UpdateAgentResponse = GetAgentResponse
+
 export type CreateSessionRequest = AgentBase
 
 export interface UpdateSessionRequest extends Partial<AgentBase> {}