llm

type Adapter interface {
	Chat(ctx context.Context, req ChatRequest) (*ChatCompletion, error)
}

type AgentOrchestrator struct {
	Client        *Client
	Tools         map[string]ToolExecutor
	MaxIterations int
}

type Builder func(ctx context.Context, cfg Config) (Adapter, error)

type ChatCompletion struct {
	ID       string             `json:"id"`
	Model    string             `json:"model"`
	Provider Provider           `json:"provider"`
	Created  time.Time          `json:"created"`
	Choices  []CompletionChoice `json:"choices"`
	Usage    CompletionUsage    `json:"usage"`
	Raw      interface{}        `json:"raw,omitempty"`
}

type ChatRequest struct {
	Messages []Message
	Options  RequestOptions
}

type ChatStreamChunk struct {
	Index        int
	Delta        string
	FinishReason string
	Raw          interface{}
}

type Client struct {
	// contains filtered or unexported fields
}

type CompletionChoice struct {
	Index        int         `json:"index"`
	Message      Message     `json:"message"`
	FinishReason string      `json:"finish_reason"`
	ProviderMeta interface{} `json:"provider_meta,omitempty"`
}

type CompletionUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type Config struct {
	Provider Provider          `json:"provider" yaml:"provider" mapstructure:"provider"`
	Model    string            `json:"model" yaml:"model" mapstructure:"model"`
	BaseURL  string            `json:"base_url,omitempty" yaml:"base_url,omitempty" mapstructure:"base_url"`
	APIKey   string            `json:"api_key,omitempty" yaml:"api_key,omitempty" mapstructure:"api_key"`
	Version  string            `json:"version,omitempty" yaml:"version,omitempty" mapstructure:"version"`
	Timeout  time.Duration     `json:"timeout,omitempty" yaml:"timeout,omitempty" mapstructure:"timeout"`
	Options  RequestOptions    `json:"options,omitempty" yaml:"options,omitempty" mapstructure:"options"`
	Headers  map[string]string `json:"headers,omitempty" yaml:"headers,omitempty" mapstructure:"headers"`
}

type EinoAgent interface {
	Run(ctx context.Context, messages []Message, options RequestOptions) (*ChatCompletion, error)
}

type EinoExecutor interface {
	Chat(ctx context.Context, messages []Message, options RequestOptions) (*ChatCompletion, error)
}

type Message struct {
	Role       MessageRole `json:"role" yaml:"role"`
	Content    string      `json:"content" yaml:"content"`
	Name       string      `json:"name,omitempty" yaml:"name,omitempty"`
	ToolCalls  []ToolCall  `json:"tool_calls,omitempty" yaml:"tool_calls,omitempty"`
	ToolCallID string      `json:"tool_call_id,omitempty" yaml:"tool_call_id,omitempty"`
}

type MessageRole string

type Option func(*RequestOptions)

type Provider string

type RequestOptions struct {
	Temperature *float64          `json:"temperature,omitempty" yaml:"temperature,omitempty"`
	TopP        *float64          `json:"top_p,omitempty" yaml:"top_p,omitempty"`
	MaxTokens   *int              `json:"max_tokens,omitempty" yaml:"max_tokens,omitempty"`
	Stop        []string          `json:"stop,omitempty" yaml:"stop,omitempty"`
	User        string            `json:"user,omitempty" yaml:"user,omitempty"`
	Stream      bool              `json:"stream,omitempty" yaml:"stream,omitempty"`
	Metadata    map[string]any    `json:"metadata,omitempty" yaml:"metadata,omitempty"`
	Timeout     time.Duration     `json:"timeout,omitempty" yaml:"timeout,omitempty"`
	Headers     map[string]string `json:"headers,omitempty" yaml:"headers,omitempty"`
	ProviderRaw map[string]any    `json:"provider_raw,omitempty" yaml:"provider_raw,omitempty"`
	Tools       []ToolDefinition  `json:"tools,omitempty" yaml:"tools,omitempty"`
	ToolChoice  string            `json:"tool_choice,omitempty" yaml:"tool_choice,omitempty"` // auto/none/required/具体工具名（OpenAI 风格）
}

type StreamingAdapter interface {
	Stream(ctx context.Context, req ChatRequest) (<-chan ChatStreamChunk, func(), error)
}

type ToolCall struct {
	ID       string           `json:"id" yaml:"id"`
	Type     string           `json:"type" yaml:"type"` // 目前使用 "function"
	Function ToolFunctionCall `json:"function" yaml:"function"`
}

type ToolDefinition struct {
	Name        string                 `json:"name" yaml:"name"`
	Description string                 `json:"description,omitempty" yaml:"description,omitempty"`
	Parameters  map[string]any         `json:"parameters,omitempty" yaml:"parameters,omitempty"`
	Meta        map[string]interface{} `json:"meta,omitempty" yaml:"meta,omitempty"`
}

type ToolExecutor func(ctx context.Context, call ToolCall) (string, error)

type ToolFunctionCall struct {
	Name      string `json:"name" yaml:"name"`
	Arguments string `json:"arguments" yaml:"arguments"` // 原始 JSON 字符串
}

type Validator interface {
	Validate() error
}