router

type ABMetrics struct {
	VariantName    string
	TotalRequests  int64
	SuccessCount   int64
	FailureCount   int64
	TotalLatencyMs int64
	TotalCost      float64
	QualityScores  []float64
	// contains filtered or unexported fields
}

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

type ABTestConfig struct {
	// 名称表示此测试 。
	Name string
	// 变体列出了测试变体.
	Variants []ABVariant
	// 粘接可以确定同一用户/会话的路径。
	StickyRouting bool
	// 粘接Key选择请求字段使用的:"user id","session id",或"tenant id".
	StickyKey string
	// 开始时间是测试开始的时候.
	StartTime time.Time
	// EndTime是测试结束的时候(零值表示无限期).
	EndTime time.Time
}

type ABVariant struct {
	// 名称是变体标识符(例如"control","experiment a").
	Name string
	// 提供方是这个变体所使用的LLM提供者.
	Provider llmpkg.Provider
	// 重量为交通重量(0-100). 所有变相权重必须相加为100.
	Weight int
	// 元数据为这个变体持有任意的密钥值对.
	Metadata map[string]string
}

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

type APIKeySelectionStrategy string

type APIKeyStats struct {
	KeyID          uint       `json:"key_id"`
	Label          string     `json:"label"`
	BaseURL        string     `json:"base_url"`
	Enabled        bool       `json:"enabled"`
	IsHealthy      bool       `json:"is_healthy"`
	TotalRequests  int64      `json:"total_requests"`
	FailedRequests int64      `json:"failed_requests"`
	SuccessRate    float64    `json:"success_rate"`
	CurrentRPM     int        `json:"current_rpm"`
	CurrentRPD     int        `json:"current_rpd"`
	LastUsedAt     *time.Time `json:"last_used_at"`
	LastErrorAt    *time.Time `json:"last_error_at"`
	LastError      string     `json:"last_error"`
}

type CanaryConfig = llmcore.CanaryConfig

type ChannelModelMapping struct {
	MappingID   string            `json:"mapping_id,omitempty"`
	ChannelID   string            `json:"channel_id,omitempty"`
	Provider    string            `json:"provider,omitempty"`
	PublicModel string            `json:"public_model,omitempty"`
	RemoteModel string            `json:"remote_model,omitempty"`
	BaseURL     string            `json:"base_url,omitempty"`
	Region      string            `json:"region,omitempty"`
	Priority    int               `json:"priority,omitempty"`
	Weight      int               `json:"weight,omitempty"`
	Metadata    map[string]string `json:"metadata,omitempty"`
}

type ChannelProviderConfig struct {
	Provider string            `json:"provider,omitempty"`
	BaseURL  string            `json:"base_url,omitempty"`
	Region   string            `json:"region,omitempty"`
	Model    string            `json:"model,omitempty"`
	Metadata map[string]string `json:"metadata,omitempty"`
	Extra    map[string]any    `json:"extra,omitempty"`
}

type ChannelRouteCallbacks struct {
	OnKeySelected   func(ctx context.Context, selection *ChannelSelection)
	OnModelRemapped func(ctx context.Context, event *ModelRemapEvent)
	OnUsageRecorded func(ctx context.Context, usage *ChannelUsageRecord, err error)
}

type ChannelRouteRequest struct {
	Capability         RouteCapability   `json:"capability"`
	Mode               RouteMode         `json:"mode"`
	Attempt            int               `json:"attempt,omitempty"`
	TraceID            string            `json:"trace_id,omitempty"`
	RequestedModel     string            `json:"requested_model,omitempty"`
	ProviderHint       string            `json:"provider_hint,omitempty"`
	RoutePolicy        string            `json:"route_policy,omitempty"`
	Region             string            `json:"region,omitempty"`
	Metadata           map[string]string `json:"metadata,omitempty"`
	Tags               []string          `json:"tags,omitempty"`
	ExcludedChannelIDs []string          `json:"excluded_channel_ids,omitempty"`
	ExcludedKeyIDs     []string          `json:"excluded_key_ids,omitempty"`
}

type ChannelRouteRetryPolicy struct {
	MaxAttempts          int
	ExcludeFailedChannel bool
	ShouldRetry          func(ctx context.Context, err error, selection *ChannelSelection) bool
}

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

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

type ChannelRoutedProviderConfig struct {
	Name string
	// RetryPolicy controls route retry behavior after a selected key/channel
	// fails. Zero values default to the provider's built-in single-attempt
	// policy.
	RetryPolicy ChannelRouteRetryPolicy

	ModelResolver        ModelResolver
	ModelMappingResolver ModelMappingResolver
	ChannelSelector      ChannelSelector
	SecretResolver       SecretResolver
	UsageRecorder        UsageRecorder
	CooldownController   CooldownController
	QuotaPolicy          QuotaPolicy
	ProviderConfigSource ProviderConfigSource

	ChatProviderFactory   ChatProviderFactory
	LegacyProviderFactory ProviderFactory
	ProviderTimeout       time.Duration

	Callbacks ChannelRouteCallbacks
	Logger    *zap.Logger
}

type ChannelRoutedProviderOptions struct {
	Name                 string
	MaxAttempts          int
	ModelResolver        ModelResolver
	ModelMappingResolver ModelMappingResolver
	ChannelSelector      ChannelSelector
	SecretResolver       SecretResolver
	UsageRecorder        UsageRecorder
	CooldownController   CooldownController
	QuotaPolicy          QuotaPolicy
	ProviderConfigSource ProviderConfigSource
	Factory              ChatProviderFactory
	RetryPolicy          ChannelRouteRetryPolicy
	Callbacks            ChannelRouteCallbacks
	Logger               *zap.Logger
}

type ChannelSecret struct {
	APIKey    string            `json:"api_key,omitempty"`
	SecretKey string            `json:"secret_key,omitempty"`
	Headers   map[string]string `json:"headers,omitempty"`
	Metadata  map[string]string `json:"metadata,omitempty"`
}

type ChannelSelection struct {
	MappingID   string            `json:"mapping_id,omitempty"`
	ChannelID   string            `json:"channel_id,omitempty"`
	KeyID       string            `json:"key_id,omitempty"`
	Provider    string            `json:"provider,omitempty"`
	RemoteModel string            `json:"remote_model,omitempty"`
	BaseURL     string            `json:"base_url,omitempty"`
	Region      string            `json:"region,omitempty"`
	Priority    int               `json:"priority,omitempty"`
	Weight      int               `json:"weight,omitempty"`
	Metadata    map[string]string `json:"metadata,omitempty"`
}

type ChannelSelector interface {
	SelectChannel(ctx context.Context, request *ChannelRouteRequest, resolution *ModelResolution, mappings []ChannelModelMapping) (*ChannelSelection, error)
}

type ChannelUsageRecord struct {
	Capability     RouteCapability   `json:"capability"`
	Mode           RouteMode         `json:"mode"`
	Attempt        int               `json:"attempt,omitempty"`
	TraceID        string            `json:"trace_id,omitempty"`
	ChannelID      string            `json:"channel_id,omitempty"`
	KeyID          string            `json:"key_id,omitempty"`
	Provider       string            `json:"provider,omitempty"`
	RequestedModel string            `json:"requested_model,omitempty"`
	RemoteModel    string            `json:"remote_model,omitempty"`
	BaseURL        string            `json:"base_url,omitempty"`
	Success        bool              `json:"success"`
	ErrorMessage   string            `json:"error_message,omitempty"`
	LatencyMS      int64             `json:"latency_ms,omitempty"`
	Usage          *ChatUsage        `json:"usage,omitempty"`
	Metadata       map[string]string `json:"metadata,omitempty"`
}

type ChatChoice = llmcore.ChatChoice

type ChatProviderFactory interface {
	CreateChatProvider(ctx context.Context, config ChannelProviderConfig, secret ChannelSecret) (Provider, error)
}

type ChatProviderFactoryFunc func(ctx context.Context, config ChannelProviderConfig, secret ChannelSecret) (Provider, error)

type ChatRequest = llmcore.ChatRequest

type ChatResponse = llmcore.ChatResponse

type ChatUsage = llmcore.ChatUsage

type CooldownController interface {
	Allow(ctx context.Context, request *ChannelRouteRequest, selection *ChannelSelection) error
	RecordResult(ctx context.Context, usage *ChannelUsageRecord) error
}

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

type Error = llmcore.Error

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

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

type HealthStatus = llmcore.HealthStatus

type IntentClassification struct {
	Intent     IntentType        `json:"intent"`
	Confidence float64           `json:"confidence"`
	SubIntents []IntentType      `json:"sub_intents,omitempty"`
	Entities   map[string]string `json:"entities,omitempty"`
	Metadata   map[string]any    `json:"metadata,omitempty"`
}

type IntentType string

type LLMModel = llmcore.LLMModel

type LLMProvider = llmcore.LLMProvider

type LLMProviderAPIKey = llmcore.LLMProviderAPIKey

type LLMProviderModel = llmcore.LLMProviderModel

type Message = llmcore.Message

type Model = llmcore.Model

type ModelCandidate struct {
	ProviderCode   string
	ModelID        string
	ModelName      string
	Tags           []string
	PriceInput     float64
	PriceOutput    float64
	MaxTokens      int
	Weight         int
	CostWeight     float64
	LatencyWeight  float64
	QualityWeight  float64
	MaxCostPerReq  float64 // SLA: 单次请求最大成本
	MaxLatencyMs   int     // SLA: 最大延迟（毫秒）
	MinSuccessRate float64 // SLA: 最小成功率
	Health         *ModelHealth
	Enabled        bool
}

type ModelHealth struct {
	ModelID      string
	IsHealthy    bool
	SuccessRate  float64 // 成功率 (0-1)
	AvgLatencyMs int     // 平均延迟
	LastError    string
	LastErrorAt  *time.Time
	UpdatedAt    time.Time
}

type ModelMappingResolver interface {
	ResolveMappings(ctx context.Context, request *ChannelRouteRequest, resolution *ModelResolution) ([]ChannelModelMapping, error)
}

type ModelRemapEvent struct {
	RequestedModel string `json:"requested_model,omitempty"`
	ResolvedModel  string `json:"resolved_model,omitempty"`
	RemoteModel    string `json:"remote_model,omitempty"`
	Provider       string `json:"provider,omitempty"`
	ChannelID      string `json:"channel_id,omitempty"`
	KeyID          string `json:"key_id,omitempty"`
	BaseURL        string `json:"base_url,omitempty"`
}

type ModelResolution struct {
	RequestedModel string            `json:"requested_model,omitempty"`
	ResolvedModel  string            `json:"resolved_model,omitempty"`
	ProviderHint   string            `json:"provider_hint,omitempty"`
	Region         string            `json:"region,omitempty"`
	Metadata       map[string]string `json:"metadata,omitempty"`
}

type ModelResolver interface {
	ResolveModel(ctx context.Context, request *ChannelRouteRequest) (*ModelResolution, error)
}

type ModelRouter interface {
	// Select 选择最佳模型
	Select(ctx context.Context, req *RouteRequest) (*RouteResult, error)
	// UpdateHealth 更新模型健康状态
	UpdateHealth(modelID string, health *ModelHealth)
	// UpdateWeights 更新路由权重
	UpdateWeights(weights []config.RoutingWeight)
}

type ModelTier string

type MultiProviderRouter struct {
	*Router // 继承原有路由器
	// contains filtered or unexported fields
}

type NoopCooldownController struct{}

type NoopQuotaPolicy struct{}

type NoopUsageRecorder struct{}

type PassthroughModelResolver struct{}

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

type PrefixRule struct {
	Prefix   string // 模型 ID 前缀（如 "gpt-4o", "claude-3-5-sonnet"）
	Provider string // Provider 代码（如 "openai", "anthropic"）
}

type Provider = llmcore.Provider

type ProviderConfigSource interface {
	ResolveProviderConfig(ctx context.Context, request *ChannelRouteRequest, selection *ChannelSelection) (*ChannelProviderConfig, error)
}

type ProviderEndpoints = llmcore.ProviderEndpoints

type ProviderFactory interface {
	CreateProvider(providerCode string, apiKey string, baseURL string) (Provider, error)
}

type ProviderFactoryAdapter struct {
	Factory ProviderFactory
}

type ProviderHealthStats struct {
	ProviderCode string
	HealthScore  float64
	CurrentQPS   int
	ErrorRate    float64
	LatencyP95   time.Duration
	LastCheckAt  time.Time
}

type ProviderProbeResult struct {
	Healthy     bool
	Latency     time.Duration
	ErrorRate   float64
	LastError   string
	LastCheckAt time.Time
}

type ProviderSelection struct {
	Provider     Provider
	ProviderID   uint
	APIKeyID     uint
	ProviderCode string
	ModelID      uint
	ModelName    string
	RemoteModel  string
	BaseURL      string
	IsCanary     bool
	Strategy     RoutingStrategy
}

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

type QuotaPolicy interface {
	Allow(ctx context.Context, request *ChannelRouteRequest, selection *ChannelSelection) error
	RecordUsage(ctx context.Context, usage *ChannelUsageRecord) error
}

type RouteCapability string

type RouteConfig struct {
	Intent           IntentType `json:"intent"`
	PreferredModels  []string   `json:"preferred_models"`
	FallbackModels   []string   `json:"fallback_models,omitempty"`
	MaxTokens        int        `json:"max_tokens,omitempty"`
	Temperature      float32    `json:"temperature,omitempty"`
	RequiredFeatures []string   `json:"required_features,omitempty"` // e.g., "function_calling", "vision"
}

type RouteMode string

type RouteRequest struct {
	TaskType     string   // 任务类型：chat/completion/embedding
	TenantID     string   // 租户 ID
	Tags         []string // 期望的标签：jsonify/cheap/fast
	MaxCost      float64  // 最大成本预算
	MaxLatencyMs int      // 最大延迟要求
	PreferModel  string   // 优先模型（可选）
}

type RouteResult struct {
	ProviderCode string
	ModelName    string
	ModelID      string
	Score        float64
	Reason       string
}

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

type RoutedChatProviderOptions struct {
	DefaultStrategy RoutingStrategy
	Fallback        Provider
	Logger          *zap.Logger
	TierRouter      *TierRouter
}

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

type RouterOptions struct {
	HealthCheckInterval time.Duration
	HealthCheckTimeout  time.Duration
	Logger              *zap.Logger
}

type RoutingStrategy string

type ScoringWeights struct {
	MessageCount  int `json:"message_count" yaml:"message_count"`
	ContentLength int `json:"content_length" yaml:"content_length"`
	ToolCount     int `json:"tool_count" yaml:"tool_count"`
	SystemPrompt  int `json:"system_prompt" yaml:"system_prompt"`
}

type SecretResolver interface {
	ResolveSecret(ctx context.Context, request *ChannelRouteRequest, selection *ChannelSelection) (*ChannelSecret, error)
}

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

type SemanticRouterConfig struct {
	ClassifierModel      string                     `json:"classifier_model"`
	DefaultRoute         RouteConfig                `json:"default_route"`
	Routes               map[IntentType]RouteConfig `json:"routes"`
	CacheClassifications bool                       `json:"cache_classifications"`
	CacheTTL             time.Duration              `json:"cache_ttl"`
}

type StaticProviderConfigSource struct{}

type StreamChunk = llmcore.StreamChunk

type TierConfig struct {
	Enabled           bool           `json:"enabled" yaml:"enabled"`
	NanoModels        []string       `json:"nano_models" yaml:"nano_models"`
	StandardModels    []string       `json:"standard_models" yaml:"standard_models"`
	FrontierModels    []string       `json:"frontier_models" yaml:"frontier_models"`
	NanoThreshold     int            `json:"nano_threshold" yaml:"nano_threshold"`
	FrontierThreshold int            `json:"frontier_threshold" yaml:"frontier_threshold"`
	Weights           ScoringWeights `json:"weights" yaml:"weights"`
}

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

type UsageRecorder interface {
	RecordUsage(ctx context.Context, usage *ChannelUsageRecord) error
}

type VendorChatProviderFactory struct {
	Timeout time.Duration
	Logger  *zap.Logger
}

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