first commit
This commit is contained in:
Beyhan Oğur
187
framework/configstore/tables/clientconfig.go
Normal file
187
framework/configstore/tables/clientconfig.go
Normal file
@@ -0,0 +1,187 @@
|
||||
package tables
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// TableClientConfig represents global client configuration in the database
|
||||
type TableClientConfig struct {
|
||||
ID uint `gorm:"primaryKey;autoIncrement" json:"id"`
|
||||
DropExcessRequests bool `gorm:"default:false" json:"drop_excess_requests"`
|
||||
PrometheusLabelsJSON string `gorm:"type:text" json:"-"` // JSON serialized []string
|
||||
AllowedOriginsJSON string `gorm:"type:text" json:"-"` // JSON serialized []string
|
||||
AllowedHeadersJSON string `gorm:"type:text" json:"-"` // JSON serialized []string
|
||||
HeaderFilterConfigJSON string `gorm:"type:text" json:"-"` // JSON serialized GlobalHeaderFilterConfig
|
||||
InitialPoolSize int `gorm:"default:300" json:"initial_pool_size"`
|
||||
EnableLogging *bool `gorm:"default:true" json:"enable_logging"`
|
||||
DisableContentLogging bool `gorm:"default:false" json:"disable_content_logging"` // DisableContentLogging controls whether sensitive content (inputs, outputs, embeddings, etc.) is logged
|
||||
DisableDBPingsInHealth bool `gorm:"default:false" json:"disable_db_pings_in_health"`
|
||||
LogRetentionDays int `gorm:"default:365" json:"log_retention_days" validate:"min=1"` // Number of days to retain logs (minimum 1 day)
|
||||
EnforceAuthOnInference bool `gorm:"default:false" json:"enforce_auth_on_inference"`
|
||||
EnforceGovernanceHeader bool `gorm:"" json:"enforce_governance_header"`
|
||||
EnforceSCIMAuth bool `gorm:"default:false" json:"enforce_scim_auth"`
|
||||
AllowDirectKeys bool `gorm:"" json:"allow_direct_keys"`
|
||||
MaxRequestBodySizeMB int `gorm:"default:100" json:"max_request_body_size_mb"`
|
||||
MCPAgentDepth int `gorm:"default:10" json:"mcp_agent_depth"`
|
||||
MCPToolExecutionTimeout int `gorm:"default:30" json:"mcp_tool_execution_timeout"` // Timeout for individual tool execution in seconds (default: 30)
|
||||
MCPCodeModeBindingLevel string `gorm:"default:server" json:"mcp_code_mode_binding_level"` // How tools are exposed in VFS: "server" or "tool"
|
||||
MCPToolSyncInterval int `gorm:"default:10" json:"mcp_tool_sync_interval"` // Global tool sync interval in minutes (default: 10, 0 = disabled)
|
||||
MCPDisableAutoToolInject bool `gorm:"default:false" json:"mcp_disable_auto_tool_inject"` // When true, MCP tools are not injected into requests by default
|
||||
AsyncJobResultTTL int `gorm:"default:3600" json:"async_job_result_ttl"` // Default TTL for async job results in seconds (default: 3600 = 1 hour)
|
||||
RequiredHeadersJSON string `gorm:"type:text" json:"-"` // JSON serialized []string
|
||||
LoggingHeadersJSON string `gorm:"type:text" json:"-"` // JSON serialized []string
|
||||
HideDeletedVirtualKeysInFilters bool `gorm:"default:false" json:"hide_deleted_virtual_keys_in_filters"` // Hide deleted virtual keys in logs filter dropdowns
|
||||
RoutingChainMaxDepth int `gorm:"default:10" json:"routing_chain_max_depth"` // Maximum depth for routing rule chain evaluation (default: 10)
|
||||
WhitelistedRoutesJSON string `gorm:"type:text" json:"-"` // JSON serialized []string
|
||||
|
||||
// Compat plugin feature flags
|
||||
CompatConvertTextToChat bool `gorm:"column:compat_convert_text_to_chat;default:false" json:"-"`
|
||||
CompatConvertChatToResponses bool `gorm:"column:compat_convert_chat_to_responses;default:false" json:"-"`
|
||||
CompatShouldDropParams bool `gorm:"column:compat_should_drop_params;default:false" json:"-"`
|
||||
CompatShouldConvertParams bool `gorm:"column:compat_should_convert_params;default:false" json:"-"`
|
||||
|
||||
// Config hash is used to detect the changes synced from config.json file
|
||||
// Every time we sync the config.json file, we will update the config hash
|
||||
ConfigHash string `gorm:"type:varchar(255);null" json:"config_hash"`
|
||||
|
||||
CreatedAt time.Time `gorm:"index;not null" json:"created_at"`
|
||||
UpdatedAt time.Time `gorm:"index;not null" json:"updated_at"`
|
||||
|
||||
// Virtual fields for runtime use (not stored in DB)
|
||||
PrometheusLabels []string `gorm:"-" json:"prometheus_labels"`
|
||||
AllowedOrigins []string `gorm:"-" json:"allowed_origins,omitempty"`
|
||||
AllowedHeaders []string `gorm:"-" json:"allowed_headers,omitempty"`
|
||||
RequiredHeaders []string `gorm:"-" json:"required_headers,omitempty"`
|
||||
LoggingHeaders []string `gorm:"-" json:"logging_headers,omitempty"`
|
||||
WhitelistedRoutes []string `gorm:"-" json:"whitelisted_routes,omitempty"`
|
||||
HeaderFilterConfig *GlobalHeaderFilterConfig `gorm:"-" json:"header_filter_config,omitempty"`
|
||||
}
|
||||
|
||||
// TableName sets the table name for each model
|
||||
func (TableClientConfig) TableName() string { return "config_client" }
|
||||
|
||||
func (cc *TableClientConfig) BeforeSave(tx *gorm.DB) error {
|
||||
if cc.PrometheusLabels != nil {
|
||||
data, err := json.Marshal(cc.PrometheusLabels)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
cc.PrometheusLabelsJSON = string(data)
|
||||
} else {
|
||||
cc.PrometheusLabelsJSON = "[]"
|
||||
}
|
||||
|
||||
if cc.AllowedOrigins != nil {
|
||||
data, err := json.Marshal(cc.AllowedOrigins)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
cc.AllowedOriginsJSON = string(data)
|
||||
} else {
|
||||
cc.AllowedOriginsJSON = "[]"
|
||||
}
|
||||
|
||||
if cc.AllowedHeaders != nil {
|
||||
data, err := json.Marshal(cc.AllowedHeaders)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
cc.AllowedHeadersJSON = string(data)
|
||||
} else {
|
||||
cc.AllowedHeadersJSON = "[]"
|
||||
}
|
||||
|
||||
if cc.WhitelistedRoutes != nil {
|
||||
data, err := json.Marshal(cc.WhitelistedRoutes)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
cc.WhitelistedRoutesJSON = string(data)
|
||||
} else {
|
||||
cc.WhitelistedRoutesJSON = "[]"
|
||||
}
|
||||
|
||||
if cc.RequiredHeaders != nil {
|
||||
data, err := json.Marshal(cc.RequiredHeaders)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
cc.RequiredHeadersJSON = string(data)
|
||||
} else {
|
||||
cc.RequiredHeadersJSON = "[]"
|
||||
}
|
||||
|
||||
if cc.LoggingHeaders != nil {
|
||||
data, err := json.Marshal(cc.LoggingHeaders)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
cc.LoggingHeadersJSON = string(data)
|
||||
} else {
|
||||
cc.LoggingHeadersJSON = "[]"
|
||||
}
|
||||
|
||||
if cc.HeaderFilterConfig != nil {
|
||||
data, err := json.Marshal(cc.HeaderFilterConfig)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
cc.HeaderFilterConfigJSON = string(data)
|
||||
} else {
|
||||
cc.HeaderFilterConfigJSON = ""
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// AfterFind hooks for deserialization
|
||||
func (cc *TableClientConfig) AfterFind(tx *gorm.DB) error {
|
||||
if cc.PrometheusLabelsJSON != "" {
|
||||
if err := json.Unmarshal([]byte(cc.PrometheusLabelsJSON), &cc.PrometheusLabels); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if cc.AllowedOriginsJSON != "" {
|
||||
if err := json.Unmarshal([]byte(cc.AllowedOriginsJSON), &cc.AllowedOrigins); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if cc.AllowedHeadersJSON != "" {
|
||||
if err := json.Unmarshal([]byte(cc.AllowedHeadersJSON), &cc.AllowedHeaders); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if cc.WhitelistedRoutesJSON != "" {
|
||||
if err := json.Unmarshal([]byte(cc.WhitelistedRoutesJSON), &cc.WhitelistedRoutes); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if cc.RequiredHeadersJSON != "" {
|
||||
if err := json.Unmarshal([]byte(cc.RequiredHeadersJSON), &cc.RequiredHeaders); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if cc.LoggingHeadersJSON != "" {
|
||||
if err := json.Unmarshal([]byte(cc.LoggingHeadersJSON), &cc.LoggingHeaders); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if cc.HeaderFilterConfigJSON != "" {
|
||||
var headerFilterConfig GlobalHeaderFilterConfig
|
||||
if err := json.Unmarshal([]byte(cc.HeaderFilterConfigJSON), &headerFilterConfig); err != nil {
|
||||
return err
|
||||
}
|
||||
cc.HeaderFilterConfig = &headerFilterConfig
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user