types

type APIShape = models.APIShape

type CacheInfoMetadata struct {
	CacheEnabled bool   `json:"cache_enabled"`
	Model        string `json:"model,omitempty"`
}

type ChatOption func(*Request)

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"`
	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 EventCallback func(msg Message)

type FollowUpCallback func(ctx context.Context) (*Message, error)

type JsonSchema = jsonschema.JsonSchema

type LogLevel int

type LogType string

type Logger interface {
	Log(ctx context.Context, logType LogType, format string, args ...interface{})
}

type LoggerFunc func(ctx context.Context, logType LogType, format string, args ...interface{})

type Message struct {
	Type MsgType `json:"type"`
	// Annotation for Timestamp
	Time  string `json:"time"`
	Role  Role   `json:"role"`
	Model string `json:"model"`

	// general content
	Content string `json:"content"`
	Error   string `json:"error,omitempty"` // meaningful when: Type == MsgType_StreamEnd, MsgType_Error, MsgType_ToolCall

	// for tool call
	ToolUseID string `json:"tool_use_id,omitempty"`
	ToolName  string `json:"tool_name,omitempty"`

	// StreamID for stream
	StreamID string `json:"stream_id,omitempty"`

	// for message token usage record
	TokenUsage *TokenUsage `json:"token_usage,omitempty"`

	// for message token cost record
	TokenCost *TokenCost `json:"token_cost,omitempty"`

	// Extended structured metadata
	Metadata Metadata `json:"metadata,omitempty"`

	// unix timestamp, accurate
	Timestamp int64 `json:"timestamp,omitempty"`
}

type Messages []Message

type Metadata struct {
	CacheInfo          *CacheInfoMetadata          `json:"cache_info,omitempty"`
	RoundStart         *RoundStartMetadata         `json:"round_start,omitempty"`
	RoundEnd           *RoundEndMetadata           `json:"round_end,omitempty"`
	StreamRequestTool  *StreamRequestToolMetadata  `json:"stream_request_tool,omitempty"`
	StreamResponseTool *StreamResponseToolMetadata `json:"stream_response_tool,omitempty"`
}

type ModelCost = models.ModelCost

type ModelInfo = models.ModelInfo

type MsgType string

type ParamType = jsonschema.ParamType

type Provider = models.Provider

type Request struct {
	Model   string `json:"model"`
	Token   string `json:"token"`
	BaseURL string `json:"base_url"`

	SystemPrompt string    `json:"system_prompt"`
	Message      string    `json:"message"`
	History      []Message `json:"history"`

	MaxRounds       int            `json:"max_rounds"`
	Tools           []string       `json:"tools"`
	ToolFiles       []string       `json:"tool_files"`
	ToolJSONs       []string       `json:"tool_jsons"`
	ToolDefinitions []*UnifiedTool `json:"tool_definitions"`
	DefaultToolCwd  string         `json:"default_tool_cwd"`

	NoCache    bool     `json:"no_cache"`
	MCPServers []string `json:"mcp_servers"`

	Logger Logger `json:"-"`

	// functional options
	EventCallback    EventCallback    `json:"-"` // Cannot be serialized
	ToolCallback     ToolCallback     `json:"-"` // Cannot be serialized
	FollowUpCallback FollowUpCallback `json:"-"` // Cannot be serialized

	// Stream fields for bidirectional tool callback communication
	StreamPair *StreamPair `json:"-"` // Cannot be serialized
}

type Response struct {
	TokenUsage TokenUsage `json:"token_usage"` // Token consumption details
	Cost       *TokenCost `json:"cost"`        // Cost information if available
	StopReason string     `json:"stop_reason"` // Why the conversation stopped
	RoundsUsed int        `json:"rounds_used"` // Number of conversation rounds used

	NumToolCalls int `json:"num_tool_calls"` // Number of tool calls used

	// last response message of the chat
	// assitant respones including msg and tool calls
	// LastAssistantMsg in the exactly the last assistant msg, excluding
	// tool calls
	LastAssistantMsg string `json:"last_assistant_response"`
}

type Role string

type RoundEndMetadata struct {
	Round int `json:"round"`
}

type RoundStartMetadata struct {
	MaxRounds int `json:"max_rounds"`
}

type StdLogger struct{}

type StdinReader interface {
	Subscribe(id string) chan Message
	Unsubscribe(id string)
	Start()
}

type StreamContext interface {
	// ACK will handle it
	ACK(id string) error

	Write(msg Message) error
}

type StreamPair struct {
	Input  io.Reader `json:"-"` // Cannot be serialized - for reading tool callback responses
	Output io.Writer `json:"-"` // Cannot be serialized - for writing tool callback requests
}

type StreamRequestToolMetadata struct {
	DefaultWorkingDir string `json:"default_working_dir"`
}

type StreamResponseToolMetadata struct {
	OK bool `json:"ok"`
}

type StringOrList = interface{}

type TokenCost struct {
	InputUSD       string                  `json:"input_usd"`
	OutputUSD      string                  `json:"output_usd"`
	TotalUSD       string                  `json:"total_usd"`
	InputBreakdown TokenCostInputBreakdown `json:"input_breakdown"`
}

type TokenCostInputBreakdown struct {
	CacheWriteUSD   string `json:"cache_write_usd"`
	CacheReadUSD    string `json:"cache_read_usd"`
	NonCacheReadUSD string `json:"non_cache_read_usd"`
}

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 `json:"usage"`
	Cost  TokenCost  `json:"cost"`
}

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

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

type ToolCall struct {
	ID         string                 `json:"id"`          // Unique identifier for this tool call
	Name       string                 `json:"name"`        // Tool name
	Arguments  map[string]interface{} `json:"arguments"`   // Parsed JSON arguments
	RawArgs    string                 `json:"raw_args"`    // Raw JSON string arguments
	WorkingDir string                 `json:"working_dir"` // Working directory for tool execution
}

type ToolCallback func(ctx context.Context, stream StreamContext, call ToolCall) (ToolResult, bool, error)

type ToolResult struct {
	Content interface{} `json:"content"`         // Result data (must be JSON serializable)
	Error   string      `json:"error,omitempty"` // Tool execution error (if any)
}

type UnifiedTool struct {
	Format      string                 `json:"format,omitempty"`
	Name        string                 `json:"name"`
	Description string                 `json:"description,omitempty"`
	Parameters  *jsonschema.JsonSchema `json:"parameters,omitempty"`

	// command to be executed
	Command []string `json:"command"`

	Handle func(ctx context.Context, stream StreamContext, call ToolCall) (ToolResult, bool, error) `json:"-"`
}