Files
soft_usite/pkg/models/telegram_agent.go
T
Lizandro GuarnizoandClaude Opus 5 b2e2e83fe8 feat(umind): el agente programa recordatorios, y solo para su dueño
"Avisame el 15 de marzo que vence la póliza de Acme, y todos los años" queda
programado desde la conversación, visible y cancelable en el panel. Tres tools:
programar, listar, cancelar.

El aviso vuelve por donde se pidió — Telegram a ese chat, correo a esa casilla
— o al dueño del espacio por correo y campanita. Nunca a una dirección que
dicte la conversación: eso sería un cañón de spam con destinatario libre.

Y las tools de aviso solo se le OFRECEN al modelo cuando la sesión es interna:
el chat de prueba del panel (que corre autenticado) o un canal marcado como
línea privada del dueño. En el widget público escribe cualquiera, y cualquiera
no puede programarle recordatorios ni gastarle el plan a otro. El gate está en
dos capas: la tool no se declara, y si igual la pide, el ejecutor la rechaza.

Tres detalles que se pagan una sola vez:
- El cron corre en memoria del proceso, sin lock distribuido: con dos
  instancias cada aviso saldría dos veces. El reclamo es un UPDATE condicional
  — la base ya es el árbitro, no hace falta traer otro.
- Si el servidor estuvo caído, una repetición diaria se saltea los ciclos
  perdidos en vez de disparar diez avisos viejos de golpe.
- Un fallo de SMTP devuelve el aviso a pendiente: una caída de correo no puede
  perder un vencimiento de póliza.

Una fecha sin hora se entrega a las 9, no a medianoche, que es cuando nadie
mira el teléfono.

De paso: los archivos de los espacios uMind nunca se sirven por el estático de
/uploads. El guard genérico solo sabe si hay sesión de panel, no de quién es el
archivo — salen por su endpoint, que sí valida propiedad. Cerrado por
construcción y no por acordarse.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-08-24 21:32:48 -05:00

98 lines
3.3 KiB
Go

package models
import (
"fmt"
"github.com/sujit-baniya/fiber-boilerplate/app"
"gorm.io/gorm"
)
// TelegramAgentHistory guarda el historial de conversación por chat_id.
// Se usan los últimos N mensajes como contexto en cada llamada al AI.
type TelegramAgentHistory struct {
gorm.Model
ChatID int64 `json:"chat_id" gorm:"column:chat_id;index;not null"`
Role string `json:"role" gorm:"column:role;not null"` // user | assistant | tool
Content string `json:"content" gorm:"column:content;type:text;not null"`
// ToolName y ToolResult se usan para mensajes de tipo "tool"
ToolName string `json:"tool_name" gorm:"column:tool_name"`
ToolResult string `json:"tool_result" gorm:"column:tool_result;type:text"`
}
func (TelegramAgentHistory) TableName() string { return "telegram_agent_history" }
// TelegramAgentChatID guarda qué chat_ids están autorizados a usar el agente.
type TelegramAgentAuth struct {
gorm.Model
ChatID int64 `json:"chat_id" gorm:"column:chat_id;uniqueIndex;not null"`
Nombre string `json:"nombre" gorm:"column:nombre"`
Activo bool `json:"activo" gorm:"column:activo;default:true"`
}
func (TelegramAgentAuth) TableName() string { return "telegram_agent_auth" }
// GetAgentHistory retorna los últimos n mensajes del historial para un chat_id.
func GetAgentHistory(chatID int64, n int) ([]TelegramAgentHistory, error) {
var items []TelegramAgentHistory
err := app.Http.Database.DB.
Where("chat_id = ?", chatID).
Order("created_at DESC").
Limit(n).
Find(&items).Error
// Invertir para orden cronológico
for i, j := 0, len(items)-1; i < j; i, j = i+1, j-1 {
items[i], items[j] = items[j], items[i]
}
return items, err
}
func SaveAgentMessage(chatID int64, role, content, toolName, toolResult string) error {
msg := &TelegramAgentHistory{
ChatID: chatID,
Role: role,
Content: content,
ToolName: toolName,
ToolResult: toolResult,
}
return app.Http.Database.DB.Create(msg).Error
}
// ClearAgentHistory borra el historial de un chat (comando /reset).
func ClearAgentHistory(chatID int64) error {
return app.Http.Database.DB.Where("chat_id = ?", chatID).Delete(&TelegramAgentHistory{}).Error
}
// IsAgentAuthChat verifica si un chat_id está autorizado.
func IsAgentAuthChat(chatID int64) bool {
var auth TelegramAgentAuth
err := app.Http.Database.DB.Where("chat_id = ? AND activo = ?", chatID, true).First(&auth).Error
return err == nil
}
func GetAllAgentAuth() ([]TelegramAgentAuth, error) {
var items []TelegramAgentAuth
err := app.Http.Database.DB.Order("created_at DESC").Find(&items).Error
return items, err
}
func CreateAgentAuth(chatID int64, nombre string) error {
auth := &TelegramAgentAuth{ChatID: chatID, Nombre: nombre, Activo: true}
return app.Http.Database.DB.Create(auth).Error
}
func DeleteAgentAuth(id uint) error {
return app.Http.Database.DB.Delete(&TelegramAgentAuth{}, id).Error
}
// GetAgentTelegramConfig retorna el TelegramConfig vinculado al agente bot activo.
func GetAgentTelegramConfig() (*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 {
return nil, fmt.Errorf("no hay agente bot configurado")
}
if ai.TelegramConfigID == nil {
return nil, fmt.Errorf("el agente no tiene Telegram configurado")
}
return GetTelegramConfigByID(*ai.TelegramConfigID)
}