import { AbstractAgent } from "@ag-ui/client"

interface AgentConfig {
  agentId?: string // The identifier of the agent
  description?: string // A description of the agent, used by the LLM
  threadId?: string // The conversation thread identifier
  initialMessages?: Message[] // An array of initial messages
  initialState?: State // The initial state of the agent
}

interface MyAgentConfig extends AgentConfig {
  myConfigOption: string
}

class MyAgent extends AbstractAgent {
  private myConfigOption: string

  constructor(config: MyAgentConfig) {
    super(config)
    this.myConfigOption = config.myConfigOption
  }
}

runAgent(parameters?: RunAgentParameters): Observable<RuntimeProtocolEvent>

interface RunAgentParameters {
  runId?: string // Unique ID for this execution run
  tools?: Tool[] // Available tools for the agent
  context?: Context[] // Contextual information
  forwardedProps?: Record<string, any> // Additional properties to forward
}

abortRun(): void

clone(): AbstractAgent

protected abstract run(input: RunAgentInput): RunAgent

protected apply(input: RunAgentInput): ApplyEvents

protected prepareRunAgentInput(parameters?: RunAgentParameters): RunAgentInput

protected onError(error: Error): void
protected onFinalize(): void