From 230205d210043e48d178462fe6c23c51d81fae18 Mon Sep 17 00:00:00 2001
From: icarus <eurfelux@gmail.com>
Date: Thu, 18 Sep 2025 13:21:37 +0800
Subject: [PATCH] feat(types): add UpdateAgentResponse type and schema

Update agent service to use new response type for consistency
---
 src/main/services/agents/services/AgentService.ts | 5 +++--
 src/renderer/src/types/agent.ts                   | 4 ++++
 2 files changed, 7 insertions(+), 2 deletions(-)

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> {}