core

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

type Client interface {
	// Chat sends messages and returns a complete response
	//
	// Parameters:
	// - ctx: Context for cancellation and timeout
	// - messages: Conversation messages
	// - opts: Optional parameters (temperature, max tokens, tools, etc.)
	//
	// Returns:
	// - *Response: Complete response with content, usage, and tool calls
	// - error: Error if the request fails
	Chat(ctx context.Context, messages []Message, opts ...Option) (*Response, error)

	// ChatStream sends messages and returns a stream of events
	//
	// Parameters:
	// - ctx: Context for cancellation and timeout
	// - messages: Conversation messages
	// - opts: Optional parameters (temperature, max tokens, tools, etc.)
	//
	// Returns:
	// - *Stream: Stream of events (content, usage, errors)
	// - error: Error if the request fails to start
	ChatStream(ctx context.Context, messages []Message, opts ...Option) (*Stream, error)
}

type ContentBlock struct {
	// Type identifies what kind of content this block contains.
	Type ContentType `json:"type"`

	// Text is the text content (only for ContentTypeText).
	Text string `json:"text,omitempty"`

	// MediaType is the MIME type (only for ContentTypeImage/ContentTypeFile).
	// Examples: "image/png", "image/jpeg", "application/pdf"
	MediaType string `json:"media_type,omitempty"`

	// Data is the base64-encoded content (only for ContentTypeImage/ContentTypeFile).
	Data string `json:"data,omitempty"`

	// FileName is the original filename (only for ContentTypeFile).
	FileName string `json:"file_name,omitempty"`
}

type ContentType string

type Error struct {
	Type    ErrorType
	Message string
	Cause   error
}

type ErrorType string

type FileTokenStore struct {
	Filename string
}

type Message struct {
	Role       string         `json:"role"`
	Content    []ContentBlock `json:"content,omitempty"`
	ToolCalls  []ToolCall     `json:"tool_calls,omitempty"`   // assistant requesting tool use
	ToolCallID string         `json:"tool_call_id,omitempty"` // for role=tool responses
}

type OAuthToken struct {
	Access              string  `json:"access"`
	Refresh             string  `json:"refresh"`
	Expires             int64   `json:"expires"`
	ResourceUrl         *string `json:"resourceUrl,omitempty"`
	NotificationMessage *string `json:"notification_message,omitempty"`
}

type Option func(*Options)

type Options struct {
	Model          string
	Temperature    *float64 // pointer so zero-value is distinguishable from "not set"
	MaxTokens      *int
	TopP           *float64
	Stop           []string
	Tools          []Tool
	SystemPrompt   string // prepended as system message if set
	Thinking       bool   // enables extended thinking/reasoning (provider-dependent)
	ThinkingBudget int    // max tokens for thinking (0 = provider default)
	EnableSearch   bool   // Qwen/compatible-mode search
	UsageCallback  func(Usage)
}

type PKCEHelper struct{}

type Response struct {
	// ID is the unique identifier for this completion.
	// Provider-specific format (e.g., "chatcmpl-abc123" for OpenAI).
	ID string `json:"id,omitempty"`

	// Model is the name of the model that generated this response.
	// May differ from the requested model if the provider substituted it.
	Model string `json:"model,omitempty"`

	// Content is the generated text content.
	// This is a convenience field that concatenates all text blocks from Message.
	// For simple text responses, this is all you need.
	Content string `json:"content"`
	// ReasoningContent is the model's thinking/reasoning process output.
	ReasoningContent string `json:"reasoning_content,omitempty"`

	// Message is the full structured message from the model.
	// Use this when you need access to multimodal content or tool calls.
	Message Message `json:"message"`

	// FinishReason indicates why the model stopped generating.
	// Common values: "stop" (natural end), "length" (max tokens reached),
	// "tool_calls" (model wants to call tools), "content_filter" (filtered).
	FinishReason string `json:"finish_reason"`

	// Usage contains token consumption information.
	// Use this to track costs and monitor usage.
	Usage *Usage `json:"usage,omitempty"`

	// ToolCalls is a convenience field extracted from Message.ToolCalls.
	// Non-empty when the model wants to invoke tools.
	ToolCalls []ToolCall `json:"tool_calls,omitempty"`
}

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

type StreamEvent struct {
	Type    StreamEventType
	Content string
	Usage   *Usage
	Err     error
}

type StreamEventType string

type TokenHelper struct{}

type TokenStore interface {
	Save(token *OAuthToken) error
	Load() (*OAuthToken, error)
}

type Tool struct {
	// Name is the unique identifier for this tool.
	// Must be a valid function name (alphanumeric and underscores).
	Name string `json:"name"`

	// Description explains what the tool does.
	// The model uses this to decide when to call the tool.
	Description string `json:"description"`

	// Parameters is a JSON Schema describing the tool's input parameters.
	// Must be a valid JSON Schema object with "type": "object".
	Parameters json.RawMessage `json:"parameters"`
}

type ToolCall struct {
	// ID is a unique identifier for this tool call.
	// Use this when responding with tool results.
	ID string `json:"id"`

	// Name is the name of the tool being called.
	Name string `json:"name"`

	// Arguments is a JSON string containing the tool's input parameters.
	// Parse this to extract the actual arguments.
	Arguments string `json:"arguments"`
}

type Usage struct {
	// PromptTokens is the number of tokens in the input (your messages).
	PromptTokens int `json:"prompt_tokens"`

	// CompletionTokens is the number of tokens in the output (model's response).
	CompletionTokens int `json:"completion_tokens"`

	// TotalTokens is the sum of PromptTokens and CompletionTokens.
	// This is typically what you're billed for.
	TotalTokens int `json:"total_tokens"`
}