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 = StreamingQuerier

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 MockCall struct {
	Method string
	Prompt string
}

type MockClient interface {
	Client

	// SetResponse sets a fixed response for all queries.
	SetResponse(response string)

	// SetResponseQueue sets a queue of responses consumed in order.
	SetResponseQueue(responses ...string)

	// SetResponseFunc sets a function that generates responses based on prompt.
	SetResponseFunc(fn func(prompt string) string)

	// SetError configures the client to return an error on queries.
	SetError(err error)

	// Calls returns all recorded calls for test assertions.
	Calls() []MockCall

	// CallCount returns the total number of calls made.
	CallCount() int

	// LastCall returns the most recent call.
	LastCall() MockCall

	// Reset clears all recorded calls and resets to default response.
	Reset()
}

type Provider string

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

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

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

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

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 StreamingQuerier interface {
	Querier

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

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

type TokenCounter interface {
	Count(text string) int
}

type WordBasedCounter struct{}

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