llm

type AnthropicContentBlock struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

type AnthropicError struct {
	Type    string `json:"type"`
	Message string `json:"message"`
}

type AnthropicMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type AnthropicProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

type AnthropicRequest struct {
	Model       string             `json:"model"`
	MaxTokens   int                `json:"max_tokens"`
	Messages    []AnthropicMessage `json:"messages"`
	System      string             `json:"system,omitempty"`
	Temperature float64            `json:"temperature,omitempty"`
}

type AnthropicResponse struct {
	ID           string                  `json:"id"`
	Type         string                  `json:"type"`
	Role         string                  `json:"role"`
	Content      []AnthropicContentBlock `json:"content"`
	Model        string                  `json:"model"`
	StopReason   string                  `json:"stop_reason"`
	StopSequence string                  `json:"stop_sequence"`
	Usage        AnthropicUsage          `json:"usage"`
}

type AnthropicUsage struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
}

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

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

type ChatResponse struct {
	Content      string `json:"content"`
	FinishReason string `json:"finish_reason,omitempty"`
	Usage        *Usage `json:"usage,omitempty"`
}

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

type LMStudioChoice struct {
	Index        int             `json:"index"`
	Message      LMStudioMessage `json:"message"`
	FinishReason string          `json:"finish_reason"`
}

type LMStudioMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type LMStudioProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

type LMStudioRequest struct {
	Model       string            `json:"model"`
	Messages    []LMStudioMessage `json:"messages"`
	MaxTokens   int               `json:"max_tokens,omitempty"`
	Temperature float64           `json:"temperature,omitempty"`
	Stream      bool              `json:"stream"`
}

type LMStudioResponse struct {
	ID      string           `json:"id"`
	Object  string           `json:"object"`
	Created int64            `json:"created"`
	Model   string           `json:"model"`
	Choices []LMStudioChoice `json:"choices"`
	Usage   *LMStudioUsage   `json:"usage,omitempty"`
}

type LMStudioUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type MockProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

type MockProviderOption func(*MockProvider)

type OllamaChatRequest struct {
	Model    string          `json:"model"`
	Messages []OllamaMessage `json:"messages"`
	Stream   bool            `json:"stream"`
	Options  *OllamaOptions  `json:"options,omitempty"`
}

type OllamaChatResponse struct {
	Model     string        `json:"model"`
	CreatedAt string        `json:"created_at"`
	Message   OllamaMessage `json:"message"`
	Done      bool          `json:"done"`
}

type OllamaMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type OllamaOptions struct {
	Temperature float64 `json:"temperature,omitempty"`
	NumPredict  int     `json:"num_predict,omitempty"`
}

type OllamaProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

type OpenAIProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

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

	// AnalyzeStructured sends a prompt and parses response into a struct
	AnalyzeStructured(ctx context.Context, prompt string, result interface{}) error

	// AnalyzeWithSystem sends a prompt with a system message
	AnalyzeWithSystem(ctx context.Context, system, prompt string) (string, error)

	// Name returns the provider name
	Name() string

	// Model returns the model being used
	Model() string
}

type RateLimitConfig struct {
	RequestsPerMinute int           // Max requests per minute
	MinInterval       time.Duration // Minimum time between requests
	BurstSize         int           // Maximum burst size
}

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

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

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

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

type UsageStats struct {
	TotalRequests      int           `json:"total_requests"`
	SuccessfulRequests int           `json:"successful_requests"`
	FailedRequests     int           `json:"failed_requests"`
	TotalPromptChars   int64         `json:"total_prompt_chars"`
	TotalResponseChars int64         `json:"total_response_chars"`
	EstimatedTokens    int64         `json:"estimated_tokens"`
	AverageLatency     time.Duration `json:"average_latency"`
	TotalLatency       time.Duration `json:"total_latency"`
	RateLimitHits      int           `json:"rate_limit_hits"`
	StartTime          time.Time     `json:"start_time"`
	LastRequestTime    time.Time     `json:"last_request_time"`
}

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