import z from "zod";
import { ServiceClient } from "../../service-client";
import { CreateAgentSchema, GetAgentSchema, ListModelsSchema } from "./schema";
import { Agent, AIModel } from "./types";


export class AgentService extends ServiceClient {
    protected getServiceName(): string {
        return 'com.appconda.service.agent';
    }

    public async CreateAgent(payload: z.infer<typeof CreateAgentSchema>): Promise<Agent> {
        return await this.actionCall('CreateAgent', payload);
    }

    public async GetAgent(payload: z.infer<typeof GetAgentSchema>): Promise<Agent> {
        return await this.actionCall('GetAgent', payload);
    }

    public async ListAgents(): Promise<Agent[]> {
        return await this.actionCall('ListAgents', {});
    }

    public async ListModelProviders(): Promise<Agent[]> {
        return await this.actionCall('ListModelProviders', {});
    }

    public async ListModels(payload: z.infer<typeof ListModelsSchema>): Promise<AIModel[]> {
        return await this.actionCall('ListModels', payload);
    }


}