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 { 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 +export const ListAgentsResponseSchema = z.object({ + agents: z.array(GetAgentResponseSchema), + total: z.number() +}) + +export type ListAgentsResponse = z.infer + export type CreateSessionRequest = AgentBase export interface UpdateSessionRequest extends Partial {}