fix(umind): la config de IA de un cliente podía atender tareas del sistema
Auditando el aislamiento apareció el agujero al revés del que se buscaba: no un cliente leyendo datos de otro, sino la cuenta de IA de un cliente pagando trabajo nuestro. GetAiConfigForService recorre las configs activas y devuelve la primera sin módulo asignado. Las configs de cliente no llevan módulo — ninguna lo lleva, es parte del diseño — así que caían justo en ese fallback. Con un cliente que hubiera conectado su cuenta, su clave terminaba clasificando correos de soporte, importando plantillas o atendiendo la vCard. Ninguno de los dos se enteraba: la respuesta llegaba igual y la factura le llegaba a él. Todos los resolvedores globales filtran ahora tenant_id IS NULL. Un test lo verifica sobre el código de cada uno, porque son consultas a base y acá no hay una. El de embeddings además no podía ser de cliente por otra razón: los vectores de todos los agentes tienen que salir del mismo modelo o la similitud coseno entre ellos no significa nada. Un cliente con su propio modelo de embeddings rompía su propia búsqueda sin un solo error visible. Del alcance entre clientes, que era lo que se auditaba: los 39 handlers de uMind validan, y el CRUD de espacios y planes ni siquiera se monta en las rutas del portal. Lo que faltaba era prueba: UmindScopeDe distingue "staff" de "cliente sin espacios" por nil contra slice vacío, y esa diferencia no tenía un solo test. Ahora la cubre uno que además falla si se invierte el fail-closed de una ruta sin scope — probado inyectando las dos fugas. Y dos cosas que quedaban colgando: En /app/ai-config toda config de cliente se mostraba como "Global", que es justo lo que no es. Ahora dice de qué espacio es, por nombre. El consumo de una cuenta propia se registraba con el costo del plan. Se sigue midiendo —el cliente quiere ver cuánto usa su asistente— pero con costo cero y marcado como cuenta propia: cobrarlo también sería cobrar dos veces lo mismo. En la pantalla de consumo aparece "va por tu cuenta de IA" en vez de un "$0" que parecería un error. El OCR y la transcripción siguen costando: son servicios nuestros, los use quien los use. Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 5
parent
926bbde9fb
commit
b125ceef95
+19
-6
@@ -189,7 +189,7 @@ func QuitarAgenteBotSalvo(id uint) {
|
||||
// GetAgenteBotConfig retorna la config marcada como agente Telegram, con su TelegramConfig cargada.
|
||||
func GetAgenteBotConfig() (*AiConfig, *TelegramConfig, error) {
|
||||
var ai AiConfig
|
||||
if err := app.Http.Database.DB.Where("es_agente_bot = ? AND is_active = ?", true, true).First(&ai).Error; err != nil {
|
||||
if err := app.Http.Database.DB.Where("es_agente_bot = ? AND is_active = ? AND tenant_id IS NULL", true, true).First(&ai).Error; err != nil {
|
||||
return nil, nil, fmt.Errorf("no hay agente bot configurado: %w", err)
|
||||
}
|
||||
if ai.TelegramConfigID == nil {
|
||||
@@ -207,7 +207,7 @@ func GetAgenteBotConfig() (*AiConfig, *TelegramConfig, error) {
|
||||
// asignado. La usa el chat propio del dashboard para compartir el mismo motor.
|
||||
func GetAgenteBotAiConfig() (*AiConfig, error) {
|
||||
var ai AiConfig
|
||||
if err := app.Http.Database.DB.Where("es_agente_bot = ? AND is_active = ?", true, true).First(&ai).Error; err != nil {
|
||||
if err := app.Http.Database.DB.Where("es_agente_bot = ? AND is_active = ? AND tenant_id IS NULL", true, true).First(&ai).Error; err != nil {
|
||||
return nil, fmt.Errorf("no hay agente configurado: %w", err)
|
||||
}
|
||||
return &ai, nil
|
||||
@@ -220,7 +220,15 @@ func GetAgenteBotAiConfig() (*AiConfig, error) {
|
||||
// 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 {
|
||||
// tenant_id IS NULL: las configs de un cliente son SUYAS y solo las usa su
|
||||
// agente. Sin este filtro, la del cliente que no tiene módulo asignado
|
||||
// —ninguna lo tiene— caía en el fallback global y terminaba atendiendo
|
||||
// tareas nuestras: clasificar correos de soporte, importar plantillas, la
|
||||
// vCard. Es decir, su cuenta pagando nuestro trabajo, sin que se entere
|
||||
// ninguno de los dos.
|
||||
if err := app.Http.Database.DB.
|
||||
Where("is_active = ? AND tenant_id IS NULL", true).
|
||||
Order("id ASC").Find(&items).Error; err != nil {
|
||||
return nil, fmt.Errorf("error leyendo ai_configs: %w", err)
|
||||
}
|
||||
if len(items) == 0 {
|
||||
@@ -276,7 +284,7 @@ func esConfigDeUsoEspecial(modulo string) bool {
|
||||
// tiene fallback a la global y a cualquier activa.
|
||||
func HayAiConfigParaModulo(modulo string) bool {
|
||||
var items []AiConfig
|
||||
if err := app.Http.Database.DB.Where("is_active = ?", true).Find(&items).Error; err != nil {
|
||||
if err := app.Http.Database.DB.Where("is_active = ? AND tenant_id IS NULL", true).Find(&items).Error; err != nil {
|
||||
return false
|
||||
}
|
||||
for i := range items {
|
||||
@@ -296,7 +304,8 @@ func HayAiConfigParaModulo(modulo string) bool {
|
||||
// transcribir contra un proveedor que no soporta ese endpoint (ej. Anthropic).
|
||||
func GetWhisperConfig() (*AiConfig, error) {
|
||||
var items []AiConfig
|
||||
if err := app.Http.Database.DB.Where("is_active = ?", true).Find(&items).Error; err != nil {
|
||||
// Solo configs del staff, por lo mismo que GetAiConfigForService.
|
||||
if err := app.Http.Database.DB.Where("is_active = ? AND tenant_id IS NULL", true).Find(&items).Error; err != nil {
|
||||
return nil, fmt.Errorf("error leyendo ai_configs: %w", err)
|
||||
}
|
||||
for i := range items {
|
||||
@@ -319,7 +328,11 @@ func GetWhisperConfig() (*AiConfig, error) {
|
||||
// en vez de caer a cualquier config activa.
|
||||
func GetUmindEmbeddingsConfig() (*AiConfig, error) {
|
||||
var items []AiConfig
|
||||
if err := app.Http.Database.DB.Where("is_active = ?", true).Find(&items).Error; err != nil {
|
||||
// Global obligatoriamente: los vectores de todos los agentes tienen que
|
||||
// salir del mismo modelo o la similitud coseno entre ellos no significa
|
||||
// nada. Un cliente con su propio modelo de embeddings rompería su propia
|
||||
// búsqueda sin ningún error visible.
|
||||
if err := app.Http.Database.DB.Where("is_active = ? AND tenant_id IS NULL", true).Find(&items).Error; err != nil {
|
||||
return nil, fmt.Errorf("error leyendo ai_configs: %w", err)
|
||||
}
|
||||
for i := range items {
|
||||
|
||||
@@ -1,6 +1,10 @@
|
||||
package models
|
||||
|
||||
import "testing"
|
||||
import (
|
||||
"os"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// El fallback de GetAiConfigForService puede terminar usando una config que no
|
||||
// declara el servicio pedido. Lo que no puede es agarrar una dedicada a
|
||||
@@ -21,3 +25,44 @@ func TestConfigsDeUsoEspecialNoSirvenDeComodin(t *testing.T) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// La consulta de todo resolvedor global tiene que excluir las configs de
|
||||
// cliente. Sin ese filtro, la config que carga un cliente —que no lleva módulo,
|
||||
// ninguna lo lleva— cae en el fallback y pasa a atender tareas nuestras:
|
||||
// clasificar correos de soporte, importar plantillas, la vCard. Su cuenta
|
||||
// pagando nuestro trabajo, y sin que se entere ninguno de los dos.
|
||||
//
|
||||
// Se verifica sobre el código porque son consultas a base: acá no hay una.
|
||||
func TestLosResolvedoresGlobalesExcluyenConfigsDeCliente(t *testing.T) {
|
||||
fuente, err := os.ReadFile("ai_config.go")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
texto := string(fuente)
|
||||
|
||||
// Toda función que resuelve una config para uso del sistema.
|
||||
resolvedores := []string{
|
||||
"func GetAiConfigForService(",
|
||||
"func GetWhisperConfig(",
|
||||
"func GetUmindEmbeddingsConfig(",
|
||||
"func GetAgenteBotAiConfig(",
|
||||
"func GetAgenteBotConfig(",
|
||||
"func HayAiConfigParaModulo(",
|
||||
}
|
||||
|
||||
for _, firma := range resolvedores {
|
||||
i := strings.Index(texto, firma)
|
||||
if i < 0 {
|
||||
t.Errorf("no encontré %s — ¿se renombró?", firma)
|
||||
continue
|
||||
}
|
||||
// El cuerpo hasta la próxima función de nivel superior.
|
||||
resto := texto[i+len(firma):]
|
||||
if j := strings.Index(resto, "\nfunc "); j > 0 {
|
||||
resto = resto[:j]
|
||||
}
|
||||
if !strings.Contains(resto, "tenant_id IS NULL") {
|
||||
t.Errorf("%s no filtra tenant_id IS NULL: puede devolver la config de un cliente para una tarea del sistema", firma)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -150,6 +150,21 @@ type UmindDocumento struct {
|
||||
AutoActualizar bool `json:"auto_actualizar" gorm:"column:auto_actualizar;default:false"`
|
||||
}
|
||||
|
||||
// NombresDeTenantsUmind devuelve id → nombre para poder etiquetar cosas que
|
||||
// solo guardan el id. Una consulta y no una por fila: son pocos y se usan para
|
||||
// pintar una lista entera.
|
||||
func NombresDeTenantsUmind() map[uint]string {
|
||||
var filas []UmindTenant
|
||||
out := map[uint]string{}
|
||||
if err := app.Http.Database.DB.Select("id, nombre").Find(&filas).Error; err != nil {
|
||||
return out
|
||||
}
|
||||
for _, f := range filas {
|
||||
out[f.ID] = f.Nombre
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func (UmindDocumento) TableName() string { return "umind_documentos" }
|
||||
|
||||
func CreateUmindDocumento(d *UmindDocumento) error {
|
||||
|
||||
+30
-2
@@ -33,6 +33,11 @@ type UmindUso struct {
|
||||
Moneda string `json:"moneda" gorm:"column:moneda;size:3"`
|
||||
// FacturadoAt null = pendiente de cobrar en el próximo ciclo.
|
||||
FacturadoAt *time.Time `json:"facturado_at" gorm:"column:facturado_at;index"`
|
||||
// CuentaPropia marca el consumo que salió por la cuenta de IA del propio
|
||||
// cliente. Se sigue midiendo —quiere ver cuánto usa su asistente— pero con
|
||||
// costo cero: ya se lo factura su proveedor, y cobrárselo también sería
|
||||
// cobrar dos veces por lo mismo.
|
||||
CuentaPropia bool `json:"cuenta_propia" gorm:"column:cuenta_propia;default:false;index"`
|
||||
}
|
||||
|
||||
func (UmindUso) TableName() string { return "umind_uso" }
|
||||
@@ -55,12 +60,22 @@ func RegistrarUsoUmind(agenteID uint, tipo string, cantidad float64, unidad stri
|
||||
if plan != nil {
|
||||
moneda = plan.Moneda
|
||||
}
|
||||
costo := costoDeUso(plan, tipo, cantidad)
|
||||
|
||||
// Si el agente corre sobre la cuenta de IA del propio cliente, el consumo
|
||||
// se registra igual pero no se le cobra: su proveedor ya se lo factura.
|
||||
// Solo aplica a los tokens de IA — el OCR y la transcripción son nuestros
|
||||
// servicios, los use quien los use.
|
||||
propia := tipo == UsoTipoIA && agenteUsaCuentaPropia(agente)
|
||||
|
||||
costo := 0.0
|
||||
if !propia {
|
||||
costo = costoDeUso(plan, tipo, cantidad)
|
||||
}
|
||||
|
||||
uso := &UmindUso{
|
||||
TenantID: agente.TenantID, AgenteID: agenteID,
|
||||
Tipo: tipo, Cantidad: cantidad, Unidad: unidad,
|
||||
Costo: costo, Moneda: moneda,
|
||||
Costo: costo, Moneda: moneda, CuentaPropia: propia,
|
||||
}
|
||||
if err := app.Http.Database.DB.Create(uso).Error; err != nil {
|
||||
log.Printf("[UMIND_USO] no se pudo registrar consumo del agente %d (%s %.2f %s): %v", agenteID, tipo, cantidad, unidad, err)
|
||||
@@ -204,3 +219,16 @@ func MarcarUsoFacturadoPorCliente(clienteID uint) {
|
||||
log.Printf("[UMIND_USO] no se pudo marcar como facturado el consumo del cliente %d: %v", clienteID, err)
|
||||
}
|
||||
}
|
||||
|
||||
// agenteUsaCuentaPropia dice si el agente apunta a una config de IA cargada por
|
||||
// el cliente (con tenant), en vez de a una nuestra.
|
||||
func agenteUsaCuentaPropia(agente *UmindAgente) bool {
|
||||
if agente == nil || agente.AiConfigID == nil {
|
||||
return false
|
||||
}
|
||||
var cfg AiConfig
|
||||
if err := GetAiConfigByID(*agente.AiConfigID, &cfg); err != nil {
|
||||
return false
|
||||
}
|
||||
return cfg.TenantID != nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user