provider

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

type CacheRetentionDefault interface {
	DefaultCacheRetention() cache.Retention
}

type EphemeralUserMessage struct {
	Message         session.Message
	HistoryInsertAt int
}

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 PreparedInputTokenLimit interface {
	EnforceInputTokenLimit() bool
}

type PreparedRequest interface {
	Stream(context.Context) (<-chan StreamEvent, error)
	TokenEstimate() TokenEstimate
	PayloadFingerprint() string
	Close() error
}

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

type ReasoningBudget struct {
	MinTokens int64 `json:"min_tokens,omitempty"`
	MaxTokens int64 `json:"max_tokens,omitempty"`
}

type ReasoningCapability struct {
	Kind                      string           `json:"kind,omitempty"`
	SupportedLevels           []ReasoningLevel `json:"supported_levels,omitempty"`
	DefaultLevel              ReasoningLevel   `json:"default_level,omitempty"`
	DisableSupported          bool             `json:"disable_supported,omitempty"`
	DefaultEnabled            *bool            `json:"default_enabled,omitempty"`
	Budget                    ReasoningBudget  `json:"budget,omitempty"`
	DynamicProviderMetadata   bool             `json:"dynamic_provider_metadata,omitempty"`
	WireShape                 string           `json:"wire_shape,omitempty"`
	DisableShape              string           `json:"disable_shape,omitempty"`
	BudgetShape               string           `json:"budget_shape,omitempty"`
	ResponseReasoningFields   []string         `json:"response_reasoning_fields,omitempty"`
	HistoryReplayRequirements []string         `json:"history_replay_requirements,omitempty"`
	SourceURLs                []string         `json:"source_urls,omitempty"`
	SourceCheckedAt           string           `json:"source_checked_at,omitempty"`
	Fixture                   string           `json:"fixture,omitempty"`
}

type ReasoningLevel string

type ReasoningSelection struct {
	Level        ReasoningLevel `json:"level,omitempty"`
	BudgetTokens int64          `json:"budget_tokens,omitempty"`
}

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
	EphemeralUser    *EphemeralUserMessage
	Tools            []ToolDefinition
	HostedTools      []HostedToolDefinition
	RawPlan          cache.RawPlan
	Cache            cache.CachePolicy
	ContextPolicy    contextpolicy.Policy
	RequestEstimate  contextpolicy.RequestEstimate
	ContextPressure  contextpolicy.ContextPressure
	MaxOutputTokens  int64
	Reasoning        ReasoningSelection
	PreviousState    *State
	Labels           RequestLabels
	Prepared         PreparedRequest `json:"-"`
}

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

type RequestPreparer interface {
	PrepareRequest(context.Context, Request) (PreparedRequest, error)
}

type SourceRef struct {
	Title string
	URL   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
	ToolCallStream ToolCallStream
	ToolCalls      []ToolCall
	ToolCall       ToolCall
	HostedResult   HostedToolResultData
	Sources        []SourceRef
	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
	Coverage             TokenEstimateCoverage
}

type TokenEstimateCoverage string

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 ToolCallStream struct {
	ID   string
	Name 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