diff --git a/src/main/apiServer/services/models.ts b/src/main/apiServer/services/models.ts index 3e281d68ab..93a1bdb0e1 100644 --- a/src/main/apiServer/services/models.ts +++ b/src/main/apiServer/services/models.ts @@ -1,4 +1,4 @@ -import { ApiModel, ApiModelsRequest, ApiModelsResponse } from '../../../renderer/src/types/apiModels' +import { ApiModel, ApiModelsFilter, ApiModelsResponse } from '../../../renderer/src/types/apiModels' import { loggerService } from '../../services/LoggerService' import { getAvailableProviders, listAllAvailableModels, transformModelToOpenAI } from '../utils' @@ -6,7 +6,7 @@ const logger = loggerService.withContext('ModelsService') // Re-export for backward compatibility -export type ModelsFilter = ApiModelsRequest +export type ModelsFilter = ApiModelsFilter export class ModelsService { async getModels(filter: ModelsFilter): Promise { diff --git a/src/renderer/src/api/agent.ts b/src/renderer/src/api/agent.ts index 1fbbfa4114..0f5ce46379 100644 --- a/src/renderer/src/api/agent.ts +++ b/src/renderer/src/api/agent.ts @@ -3,6 +3,7 @@ import { formatAgentServerError } from '@renderer/utils' import { AddAgentForm, AgentServerErrorSchema, + ApiModelsFilter, ApiModelsResponse, ApiModelsResponseSchema, CreateAgentRequest, @@ -81,7 +82,7 @@ export class AgentApiClient { public getSessionMessagesPath = (agentId: string, sessionId: string) => `/${this.apiVersion}/agents/${agentId}/sessions/${sessionId}/messages` - public modelsPath = (props?: { providerType?: 'anthropic'; limit?: number }) => { + public modelsPath = (props?: ApiModelsFilter) => { const base = `/${this.apiVersion}/models` if (!props) return base if (objectKeys(props).length > 0) { @@ -234,7 +235,7 @@ export class AgentApiClient { } } - public async getModels(props?: { providerType?: 'anthropic'; limit?: number }): Promise { + public async getModels(props?: ApiModelsFilter): Promise { const url = this.modelsPath(props) try { const response = await this.axios.get(url) diff --git a/src/renderer/src/types/apiModels.ts b/src/renderer/src/types/apiModels.ts index 0df2787320..252ebde912 100644 --- a/src/renderer/src/types/apiModels.ts +++ b/src/renderer/src/types/apiModels.ts @@ -32,5 +32,5 @@ export const ApiModelsResponseSchema = z.object({ // Inferred TypeScript types export type ApiModel = z.infer -export type ApiModelsRequest = z.infer +export type ApiModelsFilter = z.infer export type ApiModelsResponse = z.infer