client

type CharBasedCounter struct{}

type CircuitBreakerConfig struct {
	// Name identifies the circuit breaker.
	Name string

	// MaxRequests is the maximum number of requests allowed in half-open state.
	MaxRequests uint32

	// Interval is the cyclic period for clearing counts in closed state.
	Interval time.Duration

	// Timeout is the duration in open state before transitioning to half-open.
	Timeout time.Duration

	// ConsecutiveFailures is the number of consecutive failures to trip the breaker.
	ConsecutiveFailures uint32

	// OnStateChange is called when the circuit breaker state changes.
	OnStateChange func(name string, from, to gobreaker.State)
}

type Client interface {
	// Query sends a prompt and returns the complete response.
	Query(ctx context.Context, prompt string) (string, error)

	// QueryStream sends a prompt and streams the response.
	QueryStream(ctx context.Context, prompt string) (<-chan Message, <-chan error)

	// WithSystemPrompt returns a new client with the given system prompt.
	WithSystemPrompt(prompt string) Client

	// WithTools returns a new client with the given tools.
	WithTools(tools ...*Tool) Client

	// Close releases resources associated with the client.
	Close() error
}

type ClientOption func(*clientOptions)

type CompactionMessage struct {
	Role    string
	Content string
	Tokens  int // Estimated token count
}

type Compactor interface {
	// ShouldCompact returns true if compaction is needed.
	ShouldCompact(messages []CompactionMessage, maxTokens int) bool

	// Compact summarizes messages to reduce token count.
	Compact(ctx context.Context, messages []CompactionMessage) ([]CompactionMessage, error)

	// EstimateTokens estimates token count for a message.
	EstimateTokens(content string) int
}

type CompactorConfig struct {
	// Threshold is the percentage of max tokens that triggers compaction (0.0-1.0).
	Threshold float64

	// KeepRecent is the number of recent messages to preserve uncompacted.
	KeepRecent int

	// SummaryPrompt is the prompt used for summarization.
	SummaryPrompt string

	// MaxSummaryTokens limits the summary size.
	MaxSummaryTokens int
}

type Message = claude.Message

type Provider string

type ProviderConfig struct {
	Provider Provider
	Model    string
	APIKey   string
	BaseURL  string
}

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

type RateLimitError struct {
	RetryAfter time.Duration
	Message    string
}

type RateLimiterConfig struct {
	// InitialBackoff is the initial backoff after a 429.
	InitialBackoff time.Duration

	// MaxBackoff is the maximum backoff duration.
	MaxBackoff time.Duration

	// BackoffMultiplier increases backoff on consecutive 429s.
	BackoffMultiplier float64
}

type ResilienceConfig struct {
	// Circuit breaker settings
	CircuitBreaker *CircuitBreakerConfig

	// Retry settings
	Retry *RetryConfig

	// Rate limiter settings
	RateLimiter *RateLimiterConfig
}

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

type RetryConfig struct {
	// MaxRetries is the maximum number of retry attempts.
	MaxRetries uint64

	// InitialInterval is the initial backoff interval.
	InitialInterval time.Duration

	// MaxInterval is the maximum backoff interval.
	MaxInterval time.Duration

	// Multiplier is the factor by which the interval increases.
	Multiplier float64

	// RandomizationFactor adds jitter to prevent thundering herd.
	RandomizationFactor float64
}

type ServerError struct {
	StatusCode int
	Message    string
}

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

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

type TokenCounter interface {
	Count(text string) int
}

type Tool struct {
	Name        string
	Description string
	InputSchema map[string]any
	Handler     func(ctx context.Context, input map[string]any) (any, error)
}

type WordBasedCounter struct{}