llm

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

type Config struct {
	Model            string
	Temperature      float32
	MaxTokens        int
	TopP             float32
	FrequencyPenalty float32
	PresencePenalty  float32
	Stop             []string
	APIKey           string
	BaseURL          string
	APIVersion       string
	Timeout          float32
	MaxRPM           int  // Maximum requests per minute
	EnableTokenCheck bool // Whether to enable token counting and limits
}

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

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

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

type Message struct {
	Role    Role
	Content string
}

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

type Provider interface {
	// Complete generates a completion for a prompt
	Complete(ctx context.Context, prompt string) (string, error)

	// CompleteWithFunctions generates a completion with function calling support
	CompleteWithFunctions(ctx context.Context, prompt string, functions []Function) (string, error)

	// Chat generates a response in a chat conversation
	Chat(ctx context.Context, messages []Message) (string, error)

	// GetAPIKey returns the API key for the provider
	GetAPIKey() string

	// GetConfig returns the current configuration
	GetConfig() *Config

	// SupportsFunction returns whether the provider supports function calling
	SupportsFunction() bool

	// GetContextWindowSize returns the context window size for the model
	GetContextWindowSize() int

	// GetTokenUsage returns the current token usage metrics
	GetTokenUsage() token.UsageMetrics

	// ResetTokenUsage resets the token usage metrics
	ResetTokenUsage()
}

type Role string