package models import ( "fmt" "log" "strings" "github.com/sujit-baniya/fiber-boilerplate/app" "gorm.io/gorm" ) // AiConfig almacena las configuraciones de proveedores de IA (Qwen, OpenAI, etc.) // que son usadas por el Landing Generator y otros módulos. type AiConfig struct { gorm.Model Nombre string `gorm:"size:100;not null" json:"nombre"` // Alias amigable, ej: "Qwen 2.5 Producción" Provider string `gorm:"size:50;not null" json:"provider"` // qwen | openai | anthropic | etc ApiKey string `gorm:"type:text;not null" json:"api_key"` // Clave de API BaseURL string `gorm:"type:text" json:"base_url"` // URL base (override), vacío = default del provider ModelName string `gorm:"size:100" json:"model_name"` // ej: qwen2.5-72b-instruct IsActive bool `gorm:"default:true" json:"is_active"` // Solo uno activo a la vez Notes string `gorm:"type:text" json:"notes"` // Modulo indica a qué servicio pertenece esta config. // "" = global (disponible para todos como fallback) // "landing" = exclusivo para Landing Generator // "query_runner" = exclusivo para Query Runner SQL Modulo string `gorm:"size:50;default:''" json:"modulo"` } func (AiConfig) TableName() string { return "ai_configs" } func GetAllAiConfigs(limit, offset int, search string) ([]AiConfig, int64, error) { var items []AiConfig var total int64 db := app.Http.Database.DB.Model(&AiConfig{}) if search != "" { db = db.Where("nombre LIKE ? OR provider LIKE ?", "%"+search+"%", "%"+search+"%") } if err := db.Count(&total).Error; err != nil { return nil, 0, err } if err := db.Order("id DESC").Limit(limit).Offset(offset).Find(&items).Error; err != nil { return nil, 0, err } return items, total, nil } func CreateAiConfig(item *AiConfig) error { return app.Http.Database.DB.Create(item).Error } func UpdateAiConfig(id uint, updates map[string]interface{}) error { return app.Http.Database.DB.Model(&AiConfig{}).Where("id = ?", id).Updates(updates).Error } func DeleteAiConfig(id uint) error { return app.Http.Database.DB.Delete(&AiConfig{}, id).Error } // GetActiveAiConfig retorna la primera configuración activa del provider indicado. // Si provider está vacío, retorna cualquier config activa. func GetActiveAiConfig(provider string) (*AiConfig, error) { var item AiConfig db := app.Http.Database.DB.Where("is_active = ?", true) if provider != "" { db = db.Where("provider = ?", provider) } if err := db.First(&item).Error; err != nil { log.Printf("[AI_CONFIG] No se encontró config activa para provider '%s': %v", provider, err) return nil, err } return &item, nil } // SplitModulos parte el campo Modulo (comma-separated) en un slice limpio. // "" → [] (config global), "landing,query_runner" → ["landing","query_runner"] func SplitModulos(modulo string) []string { var out []string for _, s := range strings.Split(modulo, ",") { if s = strings.TrimSpace(s); s != "" { out = append(out, s) } } return out } // JoinModulos normaliza y une un slice de módulos en comma-separated. func JoinModulos(modules []string) string { var clean []string for _, s := range modules { if s = strings.TrimSpace(s); s != "" { clean = append(clean, s) } } return strings.Join(clean, ",") } // GetAiConfigForService retorna la config activa asignada al módulo indicado. // Lógica de prioridad: // 1. Config activa con modulo conteniendo service (puede ser comma-separated) // 2. Config activa con modulo == "" (global, fallback) // 3. Cualquier config activa (último recurso) func GetAiConfigForService(service string) (*AiConfig, error) { var items []AiConfig if err := app.Http.Database.DB.Where("is_active = ?", true).Order("id ASC").Find(&items).Error; err != nil { return nil, fmt.Errorf("error leyendo ai_configs: %w", err) } if len(items) == 0 { return nil, fmt.Errorf("no hay configs de IA activas") } // 1. Config específica para el servicio (puede tener varios módulos) if service != "" { for i := range items { for _, m := range SplitModulos(items[i].Modulo) { if m == service { return &items[i], nil } } } } // 2. Config global (modulo vacío = sin restricción de servicio) for i := range items { if strings.TrimSpace(items[i].Modulo) == "" { return &items[i], nil } } // 3. Cualquier config activa como último recurso log.Printf("[AI_CONFIG] No se encontró config para servicio '%s', usando cualquier activa", service) return &items[0], nil }