llmprovider

type CallOption func(*CallOptions)

type CallOptions struct {
	MaxTokens   int
	Temperature float64
	TopP        float64
	TopK        int
	StopWords   []string
	Tools       []Tool
}

type Choice struct {
	// Content is the text content of the response.
	Content string

	// StopReason is why the model stopped generating.
	StopReason string

	// ToolCalls requested by the model. These preserve ThoughtSignature
	// so they can be echoed back in the next request.
	ToolCalls []ToolCallPart

	// Thinking contains reasoning/thinking content if the model produced any.
	Thinking []ThinkingPart

	// GenerationInfo holds arbitrary provider-specific metadata (token
	// counts, safety ratings, etc.).
	GenerationInfo map[string]any
}

type FunctionDef struct {
	Name        string
	Description string
	Parameters  any // JSON Schema
}

type Message struct {
	Role  Role
	Parts []Part
}

type Part interface {
	// contains filtered or unexported methods
}

type Provider interface {
	GenerateContent(ctx context.Context, messages []Message, options ...CallOption) (*Response, error)
}

type Response struct {
	Choices []*Choice
	Usage   Usage
}

type Role string

type TextPart struct {
	Text string
}

type ThinkingPart struct {
	Text      string
	Signature string // Gemini thought_signature or Anthropic thinking signature
	Encrypted string // OpenAI encrypted_content or Anthropic redacted_thinking data
}

type Tool struct {
	Type     string
	Function *FunctionDef
}

type ToolCallPart struct {
	ID        string
	Name      string
	Arguments string // JSON string

	// Thought indicates whether this part was produced during model thinking.
	// Must be echoed back exactly as received from the API.
	Thought bool

	// ThoughtSignature is the opaque token Gemini 3.x attaches to function
	// call parts.  It must be echoed back in subsequent requests or the API
	// returns a 400.  Nil/empty means no signature was provided.
	ThoughtSignature string
}

type ToolResultPart struct {
	ToolCallID string
	Name       string
	Content    string
}

type Usage struct {
	InputTokens  int
	OutputTokens int
	TotalTokens  int
}