run

type ChatHandler struct {
	APIShape providers.APIShape
	// contains filtered or unexported fields
}

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

type ClientUnion struct {
	OpenAI    *openai.Client
	Anthropic *anthropic.Client
	Gemini    *genai.Client
}

type Config struct {
	Token           string               `json:"token,omitempty"`
	MaxRound        int                  `json:"max_round,omitempty"`
	BaseURL         string               `json:"base_url,omitempty"`
	Model           string               `json:"model,omitempty"`
	SystemPrompt    StringOrList         `json:"system,omitempty"` // can be string or a list of strings
	Tools           []string             `json:"tools,omitempty"`
	ToolCustomFiles []string             `json:"tool_custom_files,omitempty"`
	ToolCustomJSONs []*tools.UnifiedTool `json:"tool_custom_jsons,omitempty"`
	ToolDefaultCwd  string               `json:"tool_default_cwd,omitempty"`
	MCPServers      []string             `json:"mcp_servers,omitempty"`
	Examples        []string             `json:"examples,omitempty"` // a list of example questions this agent can assist with
}

type FullConfig struct {
	Config
	RecordFile         string `json:"record_file,omitempty"`
	NoCache            bool   `json:"no_cache,omitempty"`
	ShowUsage          bool   `json:"show_usage,omitempty"`
	IgnoreDuplicateMsg bool   `json:"ignore_duplicate_msg,omitempty"`
	LogRequest         bool   `json:"log_request,omitempty"`
	LogChat            *bool  `json:"log_chat,omitempty"`
	Verbose            bool   `json:"verbose,omitempty"`
}

type Message struct {
	Type    MsgType `json:"type"`
	Time    string  `json:"time"`
	Role    Role    `json:"role"`
	Model   string  `json:"model"`
	Content string  `json:"content"`

	ToolUseID  string      `json:"tool_use_id,omitempty"`
	ToolName   string      `json:"tool_name,omitempty"`
	TokenUsage *TokenUsage `json:"token_usage,omitempty"`
}

type MessageHistoryUnion struct {
	FullHistory   Messages
	SystemPrompts []string

	OpenAI    []openai.ChatCompletionMessageParamUnion
	Anthropic []anthropic.MessageParam
	Gemini    []*genai.Content
}

type Messages []Message

type MessagesUnion struct {
	OpenAI    []openai.ChatCompletionMessageParamUnion
	Anthropic []anthropic.MessageParam
	Gemini    []*genai.Content
}

type MsgType string

type Number string

type Options struct {
	BaseCmd        string
	DefaultBaseURL string
}

type ResolvedOptions struct {
	AbsDefaultToolCwd string
	Token             string
	BaseUrl           string
}

type ResponseResultAnthropic struct {
	ToolUseNum int

	Messages    []anthropic.ContentBlockParamUnion
	ToolResults []anthropic.ContentBlockParamUnion
	TokenUsage  TokenUsage
}

type ResponseResultGemini struct {
	ToolUseNum  int
	Messages    []*genai.Content
	ToolResults []*genai.Content
	TokenUsage  TokenUsage
}

type ResponseResultOpenAI struct {
	ToolUseNum  int
	Messages    []openai.ChatCompletionMessageParamUnion
	ToolResults []openai.ChatCompletionMessageParamUnion
	TokenUsage  TokenUsage
}

type Role string

type StringOrList = interface{}

type TokenCost struct {
	// the three are available for all providers
	InputUSD  string
	OutputUSD string
	TotalUSD  string

	// Input breakdown
	// anthropic has this detail
	InputBreakdown TokenCostInputBreakdown
}

type TokenCostInputBreakdown struct {
	CacheWriteUSD   string
	CacheReadUSD    string
	NonCacheReadUSD string
}

type TokenUsage struct {
	Input  int64 `json:"input"`
	Output int64 `json:"output"`
	Total  int64 `json:"total"`

	InputBreakdown  TokenUsageInputBreakdown  `json:"input_breakdown"`
	OutputBreakdown TokenUsageOutputBreakdown `json:"output_breakdown"`
}

type TokenUsageCost struct {
	Usage TokenUsage
	Cost  TokenCost
}

type TokenUsageInputBreakdown struct {
	CacheWrite   int64 `json:"cache_write"` // anthropic specific
	CacheRead    int64 `json:"cache_read"`
	NonCacheRead int64 `json:"non_cache_read"`
}

type TokenUsageOutputBreakdown struct {
	CacheOutput int64 `json:"cache_output"`
}

type ToolInfo struct {
	Name    string
	Builtin bool

	ToolDefinition *tools.UnifiedTool

	MCPServer string
	MCPClient *client.Client
}

type ToolInfoMapping map[string]*ToolInfo