llm

type ChatResult struct {
	Content   string
	ToolCalls []ToolCall
	Usage     *Usage
}

type Client interface {
	// Chat 发送聊天请求，返回完整响应
	Chat(ctx context.Context, messages []Message, tools []ToolDefinition) (*ChatResult, error)
	// StreamChat 发送流式聊天请求，返回流式响应
	StreamChat(ctx context.Context, messages []Message, tools []ToolDefinition) (<-chan StreamResult, error)
	// Generate 简单文本生成（用于关键词提取等）
	Generate(ctx context.Context, prompt string) (string, *Usage, error)
	// Embedding 向量化文本
	Embedding(ctx context.Context, texts []string) ([]float64, error)
}

type FinishReason string

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

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

type Message struct {
	Role       string     `json:"role"`
	Content    string     `json:"content"` // 不可省略
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`
	ToolCallID string     `json:"tool_call_id,omitempty"` // only for role=tool
}

type MessagesLogged []Message

type Option func(*config)

type Response struct {
	Content          string
	ToolCalls        []ToolCall
	Usage            Usage
	CompletionTokens int
}

type StreamOptions struct {
	IncludeUsage bool `json:"include_usage,omitempty"`
}

type StreamResult struct {
	Delta        string
	Think        string
	ToolCalls    []ToolCall
	Done         bool `json:",omitempty"`
	FinishReason FinishReason
	Error        error `json:",omitempty"`
	Model        string
	ResponseID   string

	Usage *Usage
}

type ToolCall struct {
	ID       string       `json:"id"`
	Type     string       `json:"type"`
	Function ToolCallFunc `json:"function"`
}

type ToolCallFunc struct {
	Name      string          `json:"name"`
	Arguments json.RawMessage `json:"arguments"`
	Results   any             `json:"results,omitempty"`
}

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

type ToolLogs []ToolDefinition

type Tools []ToolDefinition

type Usage struct {
	InputTokens  int
	OutputTokens int
	TotalTokens  int
}