agent

type AnthropicProvider struct {
	BaseURL    string
	APIKey     string
	HTTPClient *http.Client
	Retry      RetryConfig
}

type BifrostProvider struct {
	BaseURL    string
	HTTPClient *http.Client
	Retry      RetryConfig
	// contains filtered or unexported fields
}

type ChatRequest struct {
	Model       string    `json:"model"`
	Messages    []Message `json:"messages"`
	Tools       []ToolDef `json:"tools,omitempty"`
	Temperature float64   `json:"temperature"`
	MaxTokens   int       `json:"max_tokens,omitempty"`
}

type ChatResponse struct {
	Content          string     `json:"content,omitempty"`
	ReasoningContent string     `json:"reasoning_content,omitempty"` // DeepSeek Reasoner
	ToolCalls        []ToolCall `json:"tool_calls,omitempty"`
	Usage            Usage      `json:"usage"`
}

type ClaudeProvider struct{}

type CostEstimate struct {
	InputTokens  int
	OutputTokens int
	InputCost    float64
	OutputCost   float64
	TotalCost    float64
	Model        string
	Currency     string
}

type Executor interface {
	Execute(ctx context.Context, tc ToolCall) string
}

type LoopConfig struct {
	Provider     Provider
	Executor     Executor
	Model        string
	MaxTurns     int
	MemoryWindow int // -1 = full history (default), 0 = stateless, N = keep last N exchanges
	Temperature  float64
	MaxTokens    int
	SystemPrompt string
	TaskPrompt   string

	// Tools overrides the default tool definitions. Used by MCPExecutor to pass
	// MCP server tools.
	Tools []ToolDef

	// InjectChan receives user messages to inject mid-run (e.g., stage agent_goal).
	// Non-blocking: drained after each tool execution round. Nil = disabled.
	InjectChan <-chan Message
	// MemoryResetChan receives memory window changes mid-run (e.g., break.memory).
	// Values: 0 = reset (system+task only), N = compact to last N exchanges.
	// Non-blocking: checked after each tool execution round. Nil = disabled.
	MemoryResetChan <-chan int
}

type LoopResult struct {
	Turns        int
	Messages     []Message
	TotalUsage   Usage
	FinalOutput  string
	Completed    bool
	Duration     time.Duration
	MemoryWindow int
}

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

type Message struct {
	Role             string     `json:"role"`
	Content          string     `json:"content,omitempty"`
	ReasoningContent string     `json:"reasoning_content,omitempty"` // DeepSeek Reasoner thinking tokens
	ToolCalls        []ToolCall `json:"tool_calls,omitempty"`
	ToolCallID       string     `json:"tool_call_id,omitempty"`
}

type ModelPricing struct {
	InputPerMillion         float64
	OutputPerMillion        float64
	CacheHitInputPerMillion float64
}

type Provider interface {
	Name() string
	Chat(ctx context.Context, req ChatRequest) (*ChatResponse, error)
}

type RateLimitError struct {
	StatusCode int
	Body       string
	RetryAfter time.Duration
}

type RetryConfig struct {
	MaxRetries     int           // max retry attempts (default: 3)
	InitialBackoff time.Duration // starting backoff (default: 2s)
	MaxBackoff     time.Duration // ceiling for backoff (default: 60s)
	Multiplier     float64       // backoff multiplier (default: 2.0)
}

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

type ToolDef struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	Parameters  map[string]any `json:"parameters"`
}

type ToolExecutor struct {
	KubeconfigPath string
	ExtraEnv       []string // Additional env vars for commands (e.g., AWS_ENDPOINT_URL)
}

type Usage struct {
	PromptTokens             int `json:"prompt_tokens"`
	CompletionTokens         int `json:"completion_tokens"`
	CacheCreationInputTokens int `json:"cache_creation_input_tokens,omitempty"`
	CacheReadInputTokens     int `json:"cache_read_input_tokens,omitempty"`
	PromptCacheHitTokens     int `json:"prompt_cache_hit_tokens,omitempty"`
	PromptCacheMissTokens    int `json:"prompt_cache_miss_tokens,omitempty"`
}