From f9fb0f91251cb0773c8696d178a0e4c3453b912e Mon Sep 17 00:00:00 2001
From: icarus <eurfelux@gmail.com>
Date: Thu, 18 Sep 2025 13:19:57 +0800
Subject: [PATCH] feat(agent): add ListAgentsResponse type and update service

Add ListAgentsResponse schema and type to handle agent listing responses
Update AgentService to use the new type for listAgents method
---
 src/main/services/agents/services/AgentService.ts | 3 ++-
 src/renderer/src/types/agent.ts                   | 7 +++++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/main/services/agents/services/AgentService.ts b/src/main/services/agents/services/AgentService.ts
index fb1c839bce..aba60b040a 100644
--- a/src/main/services/agents/services/AgentService.ts
+++ b/src/main/services/agents/services/AgentService.ts
@@ -6,6 +6,7 @@ import type {
   CreateAgentRequest,
   CreateAgentResponse,
   GetAgentResponse,
+  ListAgentsResponse,
   ListOptions,
   UpdateAgentRequest
 } from '@types'
@@ -85,7 +86,7 @@ export class AgentService extends BaseService {
     return agent
   }
 
-  async listAgents(options: ListOptions = {}): Promise<{ agents: GetAgentResponse[]; total: number }> {
+  async listAgents(options: ListOptions = {}): Promise<ListAgentsResponse> {
     this.ensureInitialized() // Build query with pagination
 
     const totalResult = await this.database.select({ count: count() }).from(agentsTable)
diff --git a/src/renderer/src/types/agent.ts b/src/renderer/src/types/agent.ts
index 8dc090f7e6..8e6f1125d0 100644
--- a/src/renderer/src/types/agent.ts
+++ b/src/renderer/src/types/agent.ts
@@ -131,6 +131,13 @@ export const GetAgentResponseSchema = AgentEntitySchema.extend({
 
 export type GetAgentResponse = z.infer<typeof GetAgentResponseSchema>
 
+export const ListAgentsResponseSchema = z.object({
+  agents: z.array(GetAgentResponseSchema),
+  total: z.number()
+})
+
+export type ListAgentsResponse = z.infer<typeof ListAgentsResponseSchema>
+
 export type CreateSessionRequest = AgentBase
 
 export interface UpdateSessionRequest extends Partial<AgentBase> {}