llm

type ChatResult struct {
	Content   string
	ToolCalls []ToolCall
}

type Client struct {
	Provider    string
	BaseURL     string
	APIKey      string
	Model       string
	MaxTokens   int
	Temperature *float64
	Headers     map[string]string
	HTTP        HTTPDoer
}

type FunctionDefinition struct {
	Name        string     `json:"name"`
	Description string     `json:"description,omitempty"`
	Parameters  JSONSchema `json:"parameters"`
}

type HTTPDoer interface {
	Do(req *http.Request) (*http.Response, error)
}

type JSONSchema struct {
	Type        string                `json:"type,omitempty"`
	Description string                `json:"description,omitempty"`
	Properties  map[string]JSONSchema `json:"properties,omitempty"`
	Required    []string              `json:"required,omitempty"`
	Enum        []string              `json:"enum,omitempty"`
	Items       *JSONSchema           `json:"items,omitempty"`

	// Raw can be used when the schema cannot be expressed with this subset.
	// If set, other fields should be left empty.
	Raw json.RawMessage `json:"-"`
}

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

type ToolCall struct {
	ID        string
	Name      string
	Arguments json.RawMessage
}

type ToolCallPayload struct {
	ID       string              `json:"id"`
	Type     string              `json:"type"`
	Function ToolCallPayloadFunc `json:"function"`
}

type ToolCallPayloadFunc struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type ToolDefinition struct {
	Type     string             `json:"type"`
	Function FunctionDefinition `json:"function"`
}