request

type APIKeyAuthRequest struct {
	APIKey        string `json:"api_key,omitempty"` // #nosec G117
	HeaderName    string `json:"header_name,omitempty"`
	HeaderValue   string `json:"header_value,omitempty"`
	ParamName     string `json:"param_name,omitempty"`
	ParamValue    string `json:"param_value,omitempty"`
	ParamLocation string `json:"param_location,omitempty"`
}

type AttachRegistryRequest struct {
	// Weight is the relative weighted-round-robin share on a 1..100 scale.
	Weight *int `json:"weight,omitempty" example:"1" minimum:"1" maximum:"100"`
}

type CreateConsumerRequest struct {
	Name          string                   `json:"name"`
	Type          string                   `json:"type,omitempty"`
	RoutingMode   string                   `json:"routing_mode,omitempty"`
	LBConfig      *LBConfigRequest         `json:"lb_config,omitempty"`
	Headers       map[string]string        `json:"headers,omitempty"`
	Active        *bool                    `json:"active,omitempty"`
	Fallback      *FallbackRequest         `json:"fallback,omitempty"`
	Registries    []RegistryBindingRequest `json:"registries,omitempty"`
	Roles         []string                 `json:"roles,omitempty"`
	ModelPolicies []ModelPolicyRequest     `json:"model_policies,omitempty"`
	Toolkit       []ToolkitEntryRequest    `json:"toolkit,omitempty"`
	FailMode      string                   `json:"fail_mode,omitempty"`
}

type EmbeddingConfigRequest struct {
	Provider string             `json:"provider"`
	Model    string             `json:"model"`
	Auth     *APIKeyAuthRequest `json:"auth,omitempty"`
}

type FallbackBudgetRequest struct {
	MaxAttempts       int `json:"max_attempts,omitempty"`
	MaxTotalLatencyMs int `json:"max_total_latency_ms,omitempty"`
}

type FallbackRequest struct {
	Enabled  bool                   `json:"enabled"`
	Triggers []string               `json:"triggers,omitempty"`
	Budget   *FallbackBudgetRequest `json:"budget,omitempty"`
	Chain    []string               `json:"chain,omitempty"`
}

type LBConfigRequest struct {
	Enabled         bool                    `json:"enabled"`
	Algorithm       string                  `json:"algorithm,omitempty"`
	PoolAlias       string                  `json:"pool_alias,omitempty"`
	Members         []LBPoolMemberRequest   `json:"members,omitempty"`
	EmbeddingConfig *EmbeddingConfigRequest `json:"embedding_config,omitempty"`
}

type LBPoolMemberRequest struct {
	RegistryID string   `json:"registry_id"`
	Models     []string `json:"models,omitempty"`
}

type ListConsumerRequest struct {
	Name string
	Page int
	Size int
}

type ModelPolicyRequest struct {
	RegistryID string   `json:"registry_id"`
	Allowed    []string `json:"allowed,omitempty"`
	Default    string   `json:"default,omitempty"`
}

type RegistryBindingRequest struct {
	ID string `json:"id"`
	// Weight is the relative weighted-round-robin share on a 1..100 scale.
	Weight        *int                        `json:"weight,omitempty" example:"1" minimum:"1" maximum:"100"`
	ModelPolicies *RegistryModelPolicyRequest `json:"model_policies,omitempty"`
}

type RegistryModelPolicyRequest struct {
	Allowed []string `json:"allowed,omitempty"`
	Default string   `json:"default,omitempty"`
}

type ToolkitEntryRequest struct {
	RegistryID string `json:"registry_id"`
	Tool       string `json:"tool,omitempty"`
	Prompt     string `json:"prompt,omitempty"`
	Resource   string `json:"resource,omitempty"`
	ExposeAs   string `json:"expose_as,omitempty"`
}

type UpdateConsumerRequest struct {
	Name          *string                `json:"name,omitempty"`
	Type          *string                `json:"type,omitempty"`
	RoutingMode   *string                `json:"routing_mode,omitempty"`
	LBConfig      *LBConfigRequest       `json:"lb_config,omitempty"`
	Headers       *map[string]string     `json:"headers,omitempty"`
	Active        *bool                  `json:"active,omitempty"`
	Fallback      *FallbackRequest       `json:"fallback,omitempty"`
	ModelPolicies *[]ModelPolicyRequest  `json:"model_policies,omitempty"`
	Toolkit       *[]ToolkitEntryRequest `json:"toolkit,omitempty"`
	FailMode      *string                `json:"fail_mode,omitempty"`
}