lizandrogd/soft_usite
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(umind): la config de IA de un cliente podía atender tareas del sistema

Browse Source 
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:
Lizandro Guarnizo
2026-08-20 23:15:28 -05:00
co-authored by Claude Opus 5
parent 926bbde9fb
commit b125ceef95
11 changed files with 221 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rest/controllers/ai_config_controller.go
+10
View File
@@ -50,7 +50,13 @@ func GetAiConfigs(c *fiber.Ctx) error {
Modulo string `json:"modulo"`
EsAgenteBot bool `json:"es_agente_bot"`
TelegramConfigID *uint `json:"telegram_config_id"`
TenantID *uint `json:"tenant_id"`
TenantNombre string `json:"tenant_nombre"`
}
// Los nombres de los espacios, en una consulta: sin esto la pantalla
// mostraría "tenant 7" y nadie sabe de quién es el 7.
nombresTenant := models.NombresDeTenantsUmind()
safeItems := make([]safe, len(items))
for i, it := range items {
// El hint sale de la clave en claro: sobre el ciphertext mostraría
@@ -72,6 +78,10 @@ func GetAiConfigs(c *fiber.Ctx) error {
Modulo: it.Modulo,
EsAgenteBot: it.EsAgenteBot,
TelegramConfigID: it.TelegramConfigID,
TenantID: it.TenantID,
}
if it.TenantID != nil {
safeItems[i].TenantNombre = nombresTenant[*it.TenantID]
}
}
rest/middlewares/umind_scope_test.go
+79 -42
View File
@@ -1,63 +1,100 @@
package middlewares
import (
"net/http/httptest"
"testing"
"github.com/gofiber/fiber/v2"
"github.com/valyala/fasthttp"
)
// El aislamiento entre clientes es la única lógica nueva donde un bug
// significa que un cliente vea los datos de otro. Estos casos cubren el
// contrato de UmindScopeDe/UmindPuedeVerTenant sin necesidad de base de datos.
func TestUmindPuedeVerTenant(t *testing.T) {
app := fiber.New()
// Este archivo prueba la única pieza que separa los datos de un cliente de los
// de otro. Un fallo acá no es un bug de pantalla: es el cliente A leyendo las
// conversaciones, el conocimiento y las claves del cliente B.
// Se prueba a través de una petición real en vez de fabricar un contexto a
// mano: es el mismo camino que recorre el middleware en producción.
func permiteVer(t *testing.T, scope interface{}, poner bool, tenantID uint) bool {
t.Helper()
app := fiber.New()
var permitido bool
app.Get("/x", func(c *fiber.Ctx) error {
if poner {
c.Locals(umindScopeKey, scope)
}
permitido = UmindPuedeVerTenant(c, tenantID)
return c.SendStatus(200)
})
req := httptest.NewRequest("GET", "/x", nil)
if _, err := app.Test(req); err != nil {
t.Fatal(err)
}
return permitido
}
func scopeResuelto(t *testing.T, scope interface{}) ([]uint, bool) {
t.Helper()
app := fiber.New()
var ids []uint
var esStaff bool
app.Get("/x", func(c *fiber.Ctx) error {
c.Locals(umindScopeKey, scope)
ids, esStaff = UmindScopeDe(c)
return c.SendStatus(200)
})
if _, err := app.Test(httptest.NewRequest("GET", "/x", nil)); err != nil {
t.Fatal(err)
}
return ids, esStaff
}
func TestUnClienteNoVeElEspacioDeOtro(t *testing.T) {
casos := []struct {
nombre string
scope interface{} // lo que quedó en Locals
tenantID uint
esperado bool
scope interface{}
poner bool
tenant uint
permitir bool
}{
{"staff ve cualquier tenant", []uint(nil), 42, true},
{"cliente ve el suyo", []uint{7, 9}, 7, true},
{"cliente ve el otro suyo", []uint{7, 9}, 9, true},
{"cliente NO ve el de otro", []uint{7, 9}, 8, false},
{"cliente sin tenants no ve nada", []uint{}, 1, false},
{"ruta sin middleware de scope no ve nada", nil, 1, false},
{"scope de tipo inesperado no ve nada", "todos", 1, false},
// El caso que importa: alcance sobre 1 y 7, pregunta por el 9.
{"espacio ajeno", []uint{1, 7}, true, 9, false},
{"espacio propio", []uint{1, 7}, true, 7, true},
{"otro espacio propio", []uint{1, 7}, true, 1, true},
// Un cliente sin espacios no ve absolutamente nada.
{"cliente sin espacios", []uint{}, true, 1, false},
// nil = staff, ve todo. Es la distinción más delicada del archivo:
// confundir "lista vacía" con nil le daría acceso total a un cliente.
{"staff", []uint(nil), true, 999, true},
// Ruta sin middleware de scope: fail-closed. Que se rompa visible en
// vez de servir los datos de todos por omisión.
{"sin scope en el contexto", nil, false, 1, false},
// Un valor de otro tipo en la clave tampoco puede abrir la puerta.
{"scope corrupto", "todos", true, 1, false},
}
for _, cas := range casos {
t.Run(cas.nombre, func(t *testing.T) {
c := app.AcquireCtx(&fasthttp.RequestCtx{})
defer app.ReleaseCtx(c)
if cas.scope != nil {
c.Locals(umindScopeKey, cas.scope)
}
if got := UmindPuedeVerTenant(c, cas.tenantID); got != cas.esperado {
t.Errorf("UmindPuedeVerTenant(tenant %d) = %v, esperaba %v", cas.tenantID, got, cas.esperado)
}
})
for _, cs := range casos {
got := permiteVer(t, cs.scope, cs.poner, cs.tenant)
if got != cs.permitir {
t.Errorf("%s: UmindPuedeVerTenant(%d) = %v, esperaba %v", cs.nombre, cs.tenant, got, cs.permitir)
}
}
}
// Un scope nil (staff) y uno vacío (cliente sin tenants) NO deben confundirse:
// tratar el vacío como staff le daría acceso total a un cliente sin tenants.
func TestScopeVacioNoEsStaff(t *testing.T) {
app := fiber.New()
c := app.AcquireCtx(&fasthttp.RequestCtx{})
defer app.ReleaseCtx(c)
c.Locals(umindScopeKey, []uint{})
if _, esStaff := UmindScopeDe(c); esStaff {
t.Fatal("un scope vacío se está tratando como staff: un cliente sin tenants vería todo")
// La diferencia entre nil y slice vacío es la que decide si alguien es staff.
// Si un cliente sin espacios quedara como nil, pasaría a ver todo.
func TestSliceVacioNoEsStaff(t *testing.T) {
ids, esStaff := scopeResuelto(t, []uint{})
if esStaff {
t.Error("un cliente sin espacios NO puede resolver como staff")
}
if len(ids) != 0 {
t.Errorf("esperaba lista vacía, dio %v", ids)
}
c2 := app.AcquireCtx(&fasthttp.RequestCtx{})
defer app.ReleaseCtx(c2)
c2.Locals(umindScopeKey, []uint(nil))
if _, esStaff := UmindScopeDe(c2); !esStaff {
t.Fatal("un scope nil debería ser staff")
if _, esStaff := scopeResuelto(t, []uint(nil)); !esStaff {
t.Error("nil tiene que resolver como staff: es como lo deja UmindScopeStaff")
}
}