lib

type Action struct {
	Type ActionType `mapstructure:"type"`
}

type ActionType string

type CacheConfig struct {
	Enabled bool   `mapstructure:"enabled,default=false"`
	TTL     int    `mapstructure:"ttl,default=60"`
	Prefix  string `mapstructure:"prefix,default=openshield"`
}

type Config struct {
	PluginName string      `mapstructure:"plugin_name"`
	Relation   string      `mapstructure:"relation,omitempty"`
	Threshold  float64     `mapstructure:"threshold,omitempty"`
	Url        string      `mapstructure:"url,omitempty"`
	ApiKey     string      `mapstructure:"api_key,omitempty"`
	PIIService interface{} `mapstructure:"piiservice,omitempty"`
	Categories []string    `mapstructure:"categories,omitempty"`
}

type Configuration struct {
	Settings  Setting   `mapstructure:"settings"`
	Rules     Rules     `mapstructure:"rules"`
	Secrets   Secrets   `mapstructure:"secrets"`
	Providers Providers `mapstructure:"providers"`
	Services  Services  `mapstructure:"services"`
}

type ContextCache struct {
	Enabled bool   `mapstructure:"enabled,default=false"`
	URL     string `mapstructure:"url,default=http://localhost:8001"`
}

type Database struct {
	URL           string
	AutoMigration bool
}

type DatabaseConfig struct {
	URI           string `mapstructure:"uri"`
	AutoMigration bool   `mapstructure:"auto_migration,default=false"`
}

type FeatureToggle struct {
	Enabled bool `mapstructure:"enabled,default=false"`
}

type Log struct {
	DisableColor bool
	AuditLog     bool
	Usage        bool
}

type Network struct {
	Port int `mapstructure:"port,default=8080"`
}

type OpenAIRoutes struct {
	Models          Route
	Model           Route
	ChatCompletions Route
}

type Option = httprate.Option

type ProviderAnthropic struct {
	Enabled bool   `mapstructure:"enabled"`
	BaseUrl string `mapstructure:"url"`
}

type ProviderHuggingFace struct {
	Enabled bool   `mapstructure:"enabled"`
	BaseUrl string `mapstructure:"url"`
}

type ProviderNvidia struct {
	Enabled bool   `mapstructure:"enabled"`
	BaseUrl string `mapstructure:"url"`
}

type ProviderOpenAI struct {
	Enabled bool   `mapstructure:"enabled"`
	BaseUrl string `mapstructure:"url"`
}

type Providers struct {
	OpenAI      *ProviderOpenAI      `mapstructure:"openai"`
	Anthropic   *ProviderAnthropic   `mapstructure:"anthropic"`
	Nvidia      *ProviderNvidia      `mapstructure:"nvidia"`
	HuggingFace *ProviderHuggingFace `mapstructure:"huggingface"`
}

type RateLimiting struct {
	*FeatureToggle
	Window int `mapstructure:"window"`
	Max    int `mapstructure:"max"`
}

type Redis struct {
	Options *redis.Options
}

type RedisConfig struct {
	URI string `mapstructure:"uri"`
	SSL bool   `mapstructure:"ssl,default=false"`
}

type RedisSettings struct {
	URI string
	SSL bool
}

type Route struct {
	RateLimit RateLimiting
}

type RouteSettings struct {
	RateLimit *RateLimiting
	Redis     Redis
}

type Routes struct {
	OpenAI    OpenAIRoutes
	Tokenizer Route
	Storage   *redis.Client
}

type Rule struct {
	Enabled     bool   `mapstructure:"enabled,default=false"`
	Name        string `mapstructure:"name"`
	Type        string `mapstructure:"type"`
	Config      Config `mapstructure:"config"`
	Action      Action `mapstructure:"action"`
	OrderNumber int    `mapstructure:"order_number"`
}

type RuleServer struct {
	Url string `mapstructure:"url,default=http://localhost:8000"`
}

type Rules struct {
	Input  []Rule `mapstructure:"input,default=[]"`
	Output []Rule `mapstructure:"output,default=[]"`
}

type Secrets struct {
	OpenAIApiKey      string `mapstructure:"openai_api_key"`
	HuggingFaceApiKey string `mapstructure:"huggingface_api_key"`
	AnthropicApiKey   string `mapstructure:"anthropic_api_key"`
	NvidiaApiKey      string `mapstructure:"nvidia_api_key"`
}

type ServiceLlamaGuard struct {
	Enabled bool   `mapstructure:"enabled"`
	BaseUrl string `mapstructure:"url"`
}

type ServicePromptGuard struct {
	Enabled bool   `mapstructure:"enabled"`
	BaseUrl string `mapstructure:"url"`
}

type Services struct {
	LlamaGuard  *ServiceLlamaGuard  `mapstructure:"llamaguard"`
	PromptGuard *ServicePromptGuard `mapstructure:"promptguard"`
}

type Setting struct {
	Redis               *RedisConfig    `mapstructure:"redis"`
	Database            *DatabaseConfig `mapstructure:"database"`
	Cache               *CacheConfig    `mapstructure:"cache"`
	ContextCache        *ContextCache   `mapstructure:"context_cache"`
	AuditLogging        *FeatureToggle  `mapstructure:"audit_logging,default=false"`
	UsageLogging        *FeatureToggle  `mapstructure:"usage_logging,default=false"`
	Network             *Network        `mapstructure:"network"`
	RateLimit           *RateLimiting   `mapstructure:"rate_limiting"`
	RuleServer          *RuleServer     `mapstructure:"rule_server"`
	EnglishDetectionURL string          `mapstructure:"english_detection_url"`
}

type Settings struct {
	Log        Log
	OpenAI     openai
	OpenShield openShield
	Routes     Routes
	Database   Database
	Redis      RedisSettings
}