llm

type AnthropicMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type ChatMessage struct {
	Role, Content, ReasoningContent, ToolCallID string
	ToolCalls                                   []ToolCallInfo
}

type ChatRequest struct {
	ProviderID, Model string
	Messages          []ChatMessage
	Temperature       float64
	Stream            bool
	Thinking          string
	Tools             bool
	ContextWindow     int
}

type ChatResponse struct {
	Content, ReasoningContent, Model string
	Usage                            *Usage
}

type GeminiContent struct {
	Role  string `json:"role"`
	Parts []struct {
		Text string `json:"text"`
	} `json:"parts"`
}

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

type ModelInfo struct {
	ID        string
	CreatedAt int64
	OwnedBy   string
}

type OpenAIChatMessage struct {
	Role             string         `json:"role"`
	Content          string         `json:"content"`
	ReasoningContent string         `json:"reasoning_content,omitempty"`
	ToolCallID       string         `json:"tool_call_id,omitempty"`
	ToolCalls        []ToolCallInfo `json:"tool_calls,omitempty"`
}

type OpenAIChatRequest struct {
	Model         string                 `json:"model"`
	Messages      []OpenAIChatMessage    `json:"messages"`
	Temperature   float64                `json:"temperature,omitempty"`
	Stream        bool                   `json:"stream,omitempty"`
	StreamOptions *OpenAIStreamOptions   `json:"stream_options,omitempty"`
	Tools         []OpenAIToolDefinition `json:"tools,omitempty"`
}

type OpenAIChatResponse struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	Created int64  `json:"created"`
	Model   string `json:"model"`
	Choices []struct {
		Index   int `json:"index"`
		Message struct {
			Role    string `json:"role"`
			Content string `json:"content"`
		} `json:"message"`
		FinishReason string `json:"finish_reason"`
	} `json:"choices"`
	Usage *Usage `json:"usage,omitempty"`
}

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

type OpenAIStreamOptions struct {
	IncludeUsage bool `json:"include_usage"`
}

type OpenAIToolCallDelta struct {
	Index    int    `json:"index"`
	ID       string `json:"id,omitempty"`
	Type     string `json:"type,omitempty"`
	Function *struct {
		Name      string `json:"name,omitempty"`
		Arguments string `json:"arguments,omitempty"`
	} `json:"function,omitempty"`
}

type OpenAIToolDefinition struct {
	Type     string            `json:"type"`
	Function OpenAIFunctionDef `json:"function"`
}

type StreamEvent string

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

type ToolCallInfo struct {
	ID, Type string
	Function struct{ Name, Arguments string }
}

type Usage struct{ PromptTokens, CompletionTokens, TotalTokens int }