109 lines
3.6 KiB
Go
109 lines
3.6 KiB
Go
package models
|
|
|
|
import (
|
|
"log"
|
|
|
|
"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
|
|
}
|
|
|
|
// GetAiConfigForService retorna la config activa asignada al módulo indicado.
|
|
// Lógica de prioridad:
|
|
// 1. Config activa con modulo == service (exclusiva del servicio)
|
|
// 2. Config activa con modulo == "" (global, fallback)
|
|
// 3. Cualquier config activa (último recurso)
|
|
func GetAiConfigForService(service string) (*AiConfig, error) {
|
|
var item AiConfig
|
|
|
|
// 1. Buscar config específica para el servicio
|
|
if service != "" {
|
|
err := app.Http.Database.DB.
|
|
Where("is_active = ? AND modulo = ?", true, service).
|
|
First(&item).Error
|
|
if err == nil {
|
|
return &item, nil
|
|
}
|
|
}
|
|
|
|
// 2. Buscar config global (modulo vacío)
|
|
err := app.Http.Database.DB.
|
|
Where("is_active = ? AND (modulo = '' OR modulo IS NULL)", true).
|
|
First(&item).Error
|
|
if err == nil {
|
|
return &item, nil
|
|
}
|
|
|
|
// 3. Cualquier config activa como último recurso
|
|
err = app.Http.Database.DB.
|
|
Where("is_active = ?", true).
|
|
First(&item).Error
|
|
if err != nil {
|
|
log.Printf("[AI_CONFIG] No se encontró config activa para servicio '%s': %v", service, err)
|
|
return nil, err
|
|
}
|
|
return &item, nil
|
|
}
|