provider

type CachePolicyNormalizer interface {
	NormalizeCachePolicy(cache.CachePolicy) (cache.CachePolicy, error)
}

type CacheRetentionDefault interface {
	DefaultCacheRetention() cache.Retention
}

type EstimateConfidence string

type EventType string

type FinishReason string

type HostedToolDefinition struct {
	Name        string
	Type        string
	Description string
	Parameters  map[string]any
	Options     map[string]any
}

type HostedToolResultData struct {
	Text     string                 `json:"text,omitempty"`
	Results  []HostedToolResultItem `json:"results,omitempty"`
	Error    *HostedToolResultError `json:"error,omitempty"`
	Metadata map[string]any         `json:"metadata,omitempty"`
}

type HostedToolResultError struct {
	Code    string `json:"code,omitempty"`
	Message string `json:"message,omitempty"`
}

type HostedToolResultItem struct {
	Title    string         `json:"title,omitempty"`
	URL      string         `json:"url,omitempty"`
	Snippet  string         `json:"snippet,omitempty"`
	Source   string         `json:"source,omitempty"`
	Metadata map[string]any `json:"metadata,omitempty"`
}

type PayloadHasher interface {
	PayloadHash(Request) (string, error)
}

type Provider interface {
	Stream(context.Context, Request) (<-chan StreamEvent, error)
}

type Request struct {
	RunID            string
	ThreadID         string
	TurnID           string
	PromptScopeID    string
	TraceID          string
	Step             int
	LogicalRequestID string
	Attempt          int
	OverflowRetried  bool
	Provider         string
	Model            string
	Messages         []session.Message
	Tools            []ToolDefinition
	HostedTools      []HostedToolDefinition
	RawPlan          cache.RawPlan
	Cache            cache.CachePolicy
	ContextPolicy    contextpolicy.Policy
	RequestEstimate  contextpolicy.RequestEstimate
	ContextPressure  contextpolicy.ContextPressure
	MaxOutputTokens  int64
	DisableReasoning bool
	PreviousState    *State
	Labels           RequestLabels
}

type RequestLabels struct {
	Correlation map[string]string
	Host        map[string]string
}

type State struct {
	Kind       string            `json:"kind,omitempty"`
	ID         string            `json:"id,omitempty"`
	Attributes map[string]string `json:"attributes,omitempty"`
}

type StreamEvent struct {
	Type          EventType
	Text          string
	ToolCalls     []ToolCall
	ToolCall      ToolCall
	HostedResult  HostedToolResultData
	Reason        string
	Usage         Usage
	ResponseID    string
	ResponseState *State
	Err           error
}

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

type TokenEstimate struct {
	PrefixTokens         int64
	MessageTokens        int64
	ToolDefinitionTokens int64
	EstimatedInputTokens int64
	Source               string
	Method               TokenEstimateMethod
	Confidence           EstimateConfidence
}

type TokenEstimateMethod = contextpolicy.EstimateMethod

type TokenEstimator interface {
	EstimateTokens(context.Context, Request) (TokenEstimate, error)
}

type ToolCall struct {
	ID        string
	Name      string
	Args      string
	Reasoning string
}

type ToolDefinition struct {
	Name         string
	Title        string
	Description  string
	InputSchema  map[string]any
	OutputSchema map[string]any
	Strict       bool
	Annotations  map[string]any
}

type Usage struct {
	InputTokens       int64       `json:"input_tokens,omitempty"`
	OutputTokens      int64       `json:"output_tokens,omitempty"`
	ReasoningTokens   int64       `json:"reasoning_tokens,omitempty"`
	CacheReadTokens   int64       `json:"cache_read_tokens,omitempty"`
	CacheWriteTokens  int64       `json:"cache_write_tokens,omitempty"`
	TotalTokens       int64       `json:"total_tokens,omitempty"`
	CostUSD           float64     `json:"cost_usd,omitempty"`
	Source            UsageSource `json:"source,omitempty"`
	Available         bool        `json:"available,omitempty"`
	WindowInputTokens int64       `json:"window_input_tokens,omitempty"`
}

type UsageSource string