policy

type MemoryConfig struct {
	Enabled     bool   `json:"enabled,omitempty"`
	FilePath    string `json:"file_path,omitempty"`    // Directory for per-session files, or single file path when FileName is empty
	FileName    string `json:"file_name,omitempty"`    // Per-session file name pattern, e.g. "{token_hash}.md" or "{date}/{token_hash}.md"
	Redis       bool   `json:"redis,omitempty"`        // Push to Redis collection by session
	MaxSizeMB   int    `json:"max_size_mb,omitempty"`  // Max file size before rotation (0 = unlimited, default: 100 MB)
	CompressOld bool   `json:"compress_old,omitempty"` // Gzip compress rotated files (default: true)
}

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

type PIIMatch struct {
	Type  PIIType `json:"type"`
	Label string  `json:"label"`
	Value string  `json:"value"`
}

type PIIType string

type Policy struct {
	// Global policy fields — applied first to every request
	BaseKeyEnv  string    `json:"base_key_env,omitempty"`
	UpstreamURL string    `json:"upstream_url,omitempty"`
	MaxTokens   int64     `json:"max_tokens,omitempty"`
	Model       string    `json:"model,omitempty"`
	ModelRegex  string    `json:"model_regex,omitempty"`
	Prompts     []Message `json:"prompts,omitempty"`
	Rules       RuleList  `json:"rules,omitempty"`

	// Per-provider policy arrays keyed by provider name.
	// Each provider has an array of policies — one per model or model pattern.
	// Example: {"openai": [{model:"gpt-4o",...}, {model_regex:"^gpt-3.*",...}]}
	Providers map[string][]*ProviderPolicy `json:"providers,omitempty"`

	// DefaultProvider specifies which provider to use when ResolveForModel
	// returns a policy with no explicit ProviderName (i.e., when using global settings).
	// Used to route requests to the correct provider's upstream URL and auth scheme.
	DefaultProvider string `json:"default_provider,omitempty"`

	// Session memory configuration
	Memory MemoryConfig `json:"memory,omitempty"`

	// Rate limiting
	RateLimit *RateLimitConfig `json:"rate_limit,omitempty"`

	// Retry and fallback
	Retry *RetryConfig `json:"retry,omitempty"`

	// Per-request timeout in seconds (0 = use default)
	Timeout int `json:"timeout,omitempty"`

	// Metadata tags for analytics and cost attribution
	Metadata map[string]string `json:"metadata,omitempty"`
	// contains filtered or unexported fields
}

type ProviderPolicy struct {
	BaseKeyEnv  string    `json:"base_key_env"`
	UpstreamURL string    `json:"upstream_url,omitempty"`
	MaxTokens   int64     `json:"max_tokens,omitempty"`
	Model       string    `json:"model,omitempty"`
	ModelRegex  string    `json:"model_regex,omitempty"`
	Prompts     []Message `json:"prompts,omitempty"`
	Rules       RuleList  `json:"rules,omitempty"`
	Timeout     int       `json:"timeout,omitempty"` // Per-request timeout in seconds
	// contains filtered or unexported fields
}

type RateLimitConfig struct {
	Rules       []RateLimitRule `json:"rules,omitempty"`
	MaxParallel int             `json:"max_parallel,omitempty"` // Max concurrent requests
}

type RateLimitRule struct {
	Requests int    `json:"requests,omitempty"` // Max requests per window (0 = unlimited)
	Tokens   int    `json:"tokens,omitempty"`   // Max tokens per window (0 = unlimited)
	Window   string `json:"window,omitempty"`   // Duration: "1s", "1m" (default), "1h", "24h"
	Strategy string `json:"strategy,omitempty"` // "sliding" (default) or "fixed"
}

type ResolvedPolicy struct {
	BaseKeyEnv   string
	UpstreamURL  string
	MaxTokens    int64
	Model        string
	ModelRegex   string
	ProviderName string // Which provider was matched (key from Providers map)
	Prompts      []Message
	Rules        []Rule
	Memory       MemoryConfig
	RateLimit    *RateLimitConfig
	Retry        *RetryConfig
	Timeout      int
	Metadata     map[string]string
	// contains filtered or unexported fields
}

type RetryConfig struct {
	MaxRetries int      `json:"max_retries,omitempty"` // Max retry attempts (default 0)
	Fallbacks  []string `json:"fallbacks,omitempty"`   // Fallback model names to try in order
	RetryOn    []int    `json:"retry_on,omitempty"`    // HTTP status codes that trigger retry (default: 429,500,502,503)
}

type Rule struct {
	Name     string   `json:"name,omitempty"`     // Human-readable rule name
	Type     string   `json:"type"`               // "regex", "keyword", "pii"
	Pattern  string   `json:"pattern,omitempty"`  // For regex type
	Keywords []string `json:"keywords,omitempty"` // For keyword type
	Detect   []string `json:"detect,omitempty"`   // For pii type
	Action   string   `json:"action"`             // "fail", "warn", "log", "mask"
	Scope    string   `json:"scope,omitempty"`    // "input" (default), "output", "both"
	// contains filtered or unexported fields
}

type RuleList []Rule

type RuleMatch struct {
	Rule    *Rule  `json:"-"`
	Name    string `json:"name,omitempty"`
	Action  string `json:"action"`
	Message string `json:"message"`
}

type RuleSet []Rule