feat: implement AI service with enhanced error handling, rate limiting, and conversation context management

src/commands/messages/AI/chat.ts

@@ -1,56 +1,89 @@
 import logger from "../../../core/lib/logger";
-import { GoogleGenAI } from "@google/genai";
-import {CommandMessage} from "../../../core/types/commands";
-import { TextChannel, DMChannel, NewsChannel, ThreadChannel } from "discord.js";
+import { CommandMessage } from "../../../core/types/commands";
+import { TextChannel, DMChannel, NewsChannel, ThreadChannel, EmbedBuilder } from "discord.js";
+import { aiService } from "../../../core/services/AIService";
 
-// Función para estimar tokens aproximadamente (1 token ≈ 4 caracteres para texto en español/inglés)
-function estimateTokens(text: string): number {
-    return Math.ceil(text.length / 4);
+/**
+ * Dividir texto de forma inteligente preservando markdown
+ */
+function smartChunkText(text: string, maxLength: number): string[] {
+    if (text.length <= maxLength) return [text];
+
+    const chunks: string[] = [];
+    let currentChunk = '';
+    const lines = text.split('\n');
+
+    for (const line of lines) {
+        // Si agregar esta línea excede el límite
+        if (currentChunk.length + line.length + 1 > maxLength) {
+            if (currentChunk) {
+                chunks.push(currentChunk.trim());
+                currentChunk = '';
+            }
+
+            // Si la línea misma es muy larga, dividirla por palabras
+            if (line.length > maxLength) {
+                const words = line.split(' ');
+                let wordChunk = '';
+
+                for (const word of words) {
+                    if (wordChunk.length + word.length + 1 > maxLength) {
+                        if (wordChunk) {
+                            chunks.push(wordChunk.trim());
+                            wordChunk = '';
+                        }
+                    }
+                    wordChunk += (wordChunk ? ' ' : '') + word;
+                }
+
+                if (wordChunk) {
+                    currentChunk = wordChunk;
+                }
+            } else {
+                currentChunk = line;
+            }
+        } else {
+            currentChunk += (currentChunk ? '\n' : '') + line;
+        }
+    }
+
+    if (currentChunk) {
+        chunks.push(currentChunk.trim());
+    }
+
+    return chunks;
 }
 
-// Límites de tokens según Gemini 2.5 Flash
-const MAX_INPUT_TOKENS = 1048576;  // 1M tokens de entrada
-const MAX_OUTPUT_TOKENS = 65536;   // 64K tokens de salida
-const TOKEN_RESET_THRESHOLD = 0.85; // Resetear cuando esté al 85% del límite
-
-// Estado de conversación por usuario (memoria simple en memoria)
-const conversationHistory = new Map<string, {
-    messages: string[],
-    totalTokens: number,
-    imageCount: number
-}>();
-
 export const command: CommandMessage = {
     name: 'ai',
     type: "message",
     aliases: ['chat', 'gemini'],
-    cooldown: 5,
-    description: 'Chatea con la IA (Gemini) directamente desde Discord.',
+    cooldown: 2, // Reducido porque el servicio maneja su propio rate limiting
+    description: 'Chatea con la IA (Gemini) de forma estable y escalable.',
     category: 'IA',
     usage: 'ai <mensaje>',
     run: async (message, args) => {
-        // Verificar que se proporcione un prompt
+        // Validaciones básicas
         if (!args || args.length === 0) {
-            await message.reply({
-                content: "❌ **Error:** Necesitas proporcionar un mensaje para la IA.\n" +
-                        "**Uso:** `ai <tu mensaje>`\n" +
-                        "**Ejemplo:** `ai ¿Cómo funciona JavaScript?`"
-            });
+            const helpEmbed = new EmbedBuilder()
+                .setColor(0xFF69B4)
+                .setTitle('❌ Error: Mensaje requerido')
+                .setDescription(
+                    '**Uso:** `ai <tu mensaje>`\n' +
+                    '**Ejemplo:** `ai ¿Cómo funciona JavaScript?`\n' +
+                    '**Límite:** 4000 caracteres máximo'
+                )
+                .setFooter({ text: 'AI Chat mejorado con Gemini 2.5 Flash' });
+
+            await message.reply({ embeds: [helpEmbed] });
             return;
         }
 
         const prompt = args.join(' ');
         const userId = message.author.id;
+        const guildId = message.guild?.id;
 
-        // Validar longitud del prompt
-        if (prompt.length > 4000) {
-            await message.reply({
-                content: "❌ **Error:** Tu mensaje es demasiado largo. El límite es de 4000 caracteres."
-            });
-            return;
-        }
-
-        // Verificar que el canal sea de texto
+        // Verificar tipo de canal
         const channel = message.channel as TextChannel | DMChannel | NewsChannel | ThreadChannel;
         if (!channel || !('send' in channel)) {
             await message.reply({
@@ -59,228 +92,94 @@ export const command: CommandMessage = {
             return;
         }
 
+        // Indicador de escritura mejorado
+        const typingInterval = setInterval(() => {
+            channel.sendTyping().catch(() => {});
+        }, 5000);
+
         try {
-            // Inicializar Google Gemini con configuración desde variables de entorno
-            const genAI = new GoogleGenAI({
-                apiKey: process.env.GOOGLE_AI_API_KEY
-            });
+            // Usar el servicio mejorado con manejo de prioridad
+            const priority = message.member?.permissions.has('Administrator') ? 'high' : 'normal';
 
-            // Obtener o inicializar historial de conversación del usuario
-            let userHistory = conversationHistory.get(userId);
-            if (!userHistory) {
-                userHistory = { messages: [], totalTokens: 0, imageCount: 0 };
-                conversationHistory.set(userId, userHistory);
-            }
-
-            // Enviar mensaje de "escribiendo..."
-            await channel.sendTyping();
-
-            const USERNAME = message.author.username;
-            const CURRENT_DATETIME = new Date().toLocaleString('es-ES', {
-                timeZone: 'America/Monterrey',
-                dateStyle: 'full',
-                timeStyle: 'long'
-            });
-
-            // Detectar si el usuario quiere generar una imagen
-            const imageKeywords = ['imagen', 'image', 'dibujo', 'draw', 'generar imagen', 'create image', 'picture', 'foto'];
-            const isImageRequest = imageKeywords.some(keyword =>
-                prompt.toLowerCase().includes(keyword.toLowerCase())
+            const aiResponse = await aiService.processAIRequest(
+                userId,
+                prompt,
+                guildId,
+                priority
             );
 
-            // Construir el prompt del sistema más natural y menos saturado de emojis
-            const baseSystemPrompt = `Eres una hermana mayor kawaii y cariñosa que habla por Discord. Responde de manera natural y útil, pero con personalidad tierna.
+            // Crear embed de respuesta mejorado
+            const embed = new EmbedBuilder()
+                .setColor(0xFF69B4)
+                .setTitle('🌸 Gemini-chan')
+                .setDescription(aiResponse)
+                .setFooter({
+                    text: `Solicitado por ${message.author.username}`,
+                    iconURL: message.author.displayAvatarURL({ forceStatic: false })
+                })
+                .setTimestamp();
 
-## Información del usuario:
-- Username: ${USERNAME}
-- Fecha actual: ${CURRENT_DATETIME}
-
-## Reglas importantes para Discord:
-- NUNCA uses LaTeX ($$), solo usa **markdown normal de Discord**
-- Para matemáticas usa: **negrita**, *cursiva*, \`código\` y bloques de código
-- NO uses emojis excesivamente, máximo 2-3 por respuesta
-- Para tablas usa formato simple de Discord con backticks
-- Mantén las respuestas claras y legibles en Discord
-
-## Ejemplos de formato correcto:
-- Matemáticas: "La raíz cuadrada de 16 es **4**"
-- Código: \`\`\`javascript\nfunction ejemplo() {}\`\`\`
-- Énfasis: **importante** o *destacado*
-
-${isImageRequest ? `
-## Generación de imágenes:
-- El usuario está pidiendo una imagen
-- Gemini 2.5 Flash NO puede generar imágenes
-- Explica que no puedes generar imágenes pero ofrece ayuda alternativa
-` : ''}
-
-## Mensaje del usuario:
-${prompt}
-
-## Contexto de conversación anterior:
-${userHistory.messages.slice(-3).join('\n')}`;
-
-            // Verificar límites de tokens de entrada
-            const estimatedInputTokens = estimateTokens(baseSystemPrompt);
-
-            // Verificar si necesitamos resetear la conversación
-            if (userHistory.totalTokens > MAX_INPUT_TOKENS * TOKEN_RESET_THRESHOLD) {
-                userHistory.messages = [];
-                userHistory.totalTokens = 0;
-                await message.reply({
-                    content: "🔄 **Conversación reseteada** - Límite de tokens alcanzado, empezamos de nuevo."
-                });
-            }
-
-            // Verificar si necesitamos resetear por imágenes
-            if (isImageRequest && userHistory.imageCount >= 5) {
-                userHistory.messages = [];
-                userHistory.totalTokens = 0;
-                userHistory.imageCount = 0;
-                await message.reply({
-                    content: "🔄 **Conversación reseteada** - Límite de solicitudes de imagen alcanzado (5), empezamos de nuevo."
-                });
-            }
-
-            if (estimatedInputTokens > MAX_INPUT_TOKENS) {
-                await message.reply({
-                    content: `❌ **Error:** Tu mensaje es demasiado largo para procesar.\n` +
-                            `**Tokens estimados:** ${estimatedInputTokens.toLocaleString()}\n` +
-                            `**Límite máximo:** ${MAX_INPUT_TOKENS.toLocaleString()} tokens\n\n` +
-                            `Por favor, acorta tu mensaje e intenta de nuevo.`
-                });
-                return;
-            }
-
-            // Calcular tokens de salida apropiados
-            const dynamicOutputTokens = Math.min(
-                Math.max(1024, Math.floor(estimatedInputTokens * 0.3)), // Mínimo 1024, máximo 30% del input
-                MAX_OUTPUT_TOKENS
-            );
-
-            // Generar respuesta
-            const response = await genAI.models.generateContent({
-                model: "gemini-2.5-flash",
-                contents: baseSystemPrompt,
-                // @ts-ignore
-                generationConfig: {
-                    maxOutputTokens: dynamicOutputTokens,
-                    temperature: 0.7, // Reducido para respuestas más consistentes
-                    topP: 0.8,
-                    topK: 30,
-                }
-            });
-
-            // Extraer el texto de la respuesta
-            const aiResponse = response.text;
-
-            // Verificar si la respuesta está vacía
-            if (!aiResponse || aiResponse.trim().length === 0) {
-                await message.reply({
-                    content: "❌ **Error:** La IA no pudo generar una respuesta. Intenta reformular tu pregunta."
-                });
-                return;
-            }
-
-            // Actualizar historial y contadores
-            const estimatedOutputTokens = estimateTokens(aiResponse);
-            userHistory.messages.push(`Usuario: ${prompt}`);
-            userHistory.messages.push(`Asistente: ${aiResponse}`);
-            userHistory.totalTokens += estimatedInputTokens + estimatedOutputTokens;
-
-            if (isImageRequest) {
-                userHistory.imageCount++;
-            }
-
-            // Mantener solo los últimos 10 mensajes para evitar crecimiento excesivo
-            if (userHistory.messages.length > 10) {
-                userHistory.messages = userHistory.messages.slice(-10);
-            }
-
-            // Información de debug y estado
-            const tokensUsedPercent = ((userHistory.totalTokens / MAX_INPUT_TOKENS) * 100).toFixed(1);
-            const debugInfo = process.env.NODE_ENV === 'development' ?
-                `\n\n*Debug: Input ~${estimatedInputTokens} tokens, Output ~${estimatedOutputTokens} tokens | Total: ${userHistory.totalTokens} (${tokensUsedPercent}%) | Imágenes: ${userHistory.imageCount}/5*` : '';
-
-            // Advertencia si estamos cerca del límite
-            const warningInfo = userHistory.totalTokens > MAX_INPUT_TOKENS * 0.7 ?
-                `\n\n⚠️ *Nota: Conversación larga detectada (${tokensUsedPercent}% del límite). Se reseteará pronto.*` : '';
-
-            // Dividir respuesta si es muy larga para Discord (límite de 2000 caracteres)
-            if (aiResponse.length > 1900) {
-                const chunks = aiResponse.match(/.{1,1900}/gs) || [];
+            // Manejar respuestas largas de forma inteligente
+            if (aiResponse.length > 4000) {
+                // Dividir en chunks preservando markdown
+                const chunks = smartChunkText(aiResponse, 4000);
 
                 for (let i = 0; i < chunks.length && i < 3; i++) {
-                    const chunk = chunks[i];
-                    const embed = {
-                        color: 0xFF69B4, // Color rosa kawaii para el tema imouto
-                        title: i === 0 ? '🌸 Respuesta de Gemini-chan' : `🌸 Respuesta de Gemini-chan (${i + 1}/${chunks.length})`,
-                        description: chunk + (i === chunks.length - 1 ? debugInfo : ''),
-                        footer: {
-                            text: `Solicitado por ${message.author.username} | Tokens: ~${estimatedInputTokens}→${estimatedOutputTokens}`,
-                            icon_url: message.author.displayAvatarURL({ forceStatic: false })
-                        },
-                        timestamp: new Date().toISOString()
-                    };
+                    const chunkEmbed = new EmbedBuilder()
+                        .setColor(0xFF69B4)
+                        .setTitle(i === 0 ? '🌸 Gemini-chan' : `🌸 Gemini-chan (${i + 1}/${chunks.length})`)
+                        .setDescription(chunks[i])
+                        .setFooter({
+                            text: `Solicitado por ${message.author.username} | Parte ${i + 1}`,
+                            iconURL: message.author.displayAvatarURL({ forceStatic: false })
+                        })
+                        .setTimestamp();
 
                     if (i === 0) {
-                        await message.reply({ embeds: [embed] });
+                        await message.reply({ embeds: [chunkEmbed] });
                     } else {
-                        await channel.send({ embeds: [embed] });
-                    }
-
-                    // Pequeña pausa entre mensajes
-                    if (i < chunks.length - 1) {
+                        await channel.send({ embeds: [chunkEmbed] });
+                        // Pausa entre mensajes para evitar rate limits
                         await new Promise(resolve => setTimeout(resolve, 1000));
                     }
                 }
 
                 if (chunks.length > 3) {
                     await channel.send({
-                        content: "⚠️ **Nota:** La respuesta fue truncada por ser demasiado larga. Intenta hacer preguntas más específicas."
+                        content: "⚠️ **Nota:** La respuesta fue truncada. Intenta preguntas más específicas."
                     });
                 }
             } else {
-                // Respuesta normal
-                const embed = {
-                    color: 0xFF69B4, // Color rosa kawaii
-                    title: '🌸 Respuesta de Gemini-chan',
-                    description: aiResponse + debugInfo + warningInfo,
-                    footer: {
-                        text: `Solicitado por ${message.author.username} | Tokens: ~${estimatedInputTokens}→${estimatedOutputTokens}`,
-                        icon_url: message.author.displayAvatarURL({ forceStatic: false })
-                    },
-                    timestamp: new Date().toISOString()
-                };
-
                 await message.reply({ embeds: [embed] });
             }
 
-        } catch (error: any) {
-            logger.error('Error en comando AI:', error);
-
-            // Manejar errores específicos incluyendo límites de tokens
-            let errorMessage = "❌ **Error:** Ocurrió un problema al comunicarse con la IA.";
-
-            if (error?.message?.includes('API key') || error?.message?.includes('Invalid API') || error?.message?.includes('authentication')) {
-                errorMessage = "❌ **Error:** Token de API inválido o no configurado.";
-            } else if (error?.message?.includes('quota') || error?.message?.includes('exceeded') || error?.message?.includes('rate limit')) {
-                errorMessage = "❌ **Error:** Se ha alcanzado el límite de uso de la API.";
-            } else if (error?.message?.includes('safety') || error?.message?.includes('blocked')) {
-                errorMessage = "❌ **Error:** Tu mensaje fue bloqueado por las políticas de seguridad de la IA.";
-            } else if (error?.message?.includes('timeout')) {
-                errorMessage = "❌ **Error:** La solicitud tardó demasiado tiempo. Intenta de nuevo.";
-            } else if (error?.message?.includes('model not found') || error?.message?.includes('not available')) {
-                errorMessage = "❌ **Error:** El modelo de IA no está disponible en este momento.";
-            } else if (error?.message?.includes('token') || error?.message?.includes('length')) {
-                errorMessage = "❌ **Error:** El mensaje excede los límites de tokens permitidos. Intenta con un mensaje más corto.";
-            } else if (error?.message?.includes('context_length')) {
-                errorMessage = "❌ **Error:** El contexto de la conversación es demasiado largo. La conversación se ha reiniciado.";
+            // Log para monitoreo (solo en desarrollo)
+            if (process.env.NODE_ENV === 'development') {
+                const stats = aiService.getStats();
+                logger.info(`AI Request completado - Usuario: ${userId}, Queue: ${stats.queueLength}, Conversaciones activas: ${stats.activeConversations}`);
             }
 
-            await message.reply({
-                content: errorMessage + "\n\nSi el problema persiste, contacta a un administrador."
-            });
+        } catch (error: any) {
+            logger.error(`Error en comando AI para usuario ${userId}:`, error);
+
+            // Crear embed de error informativo
+            const errorEmbed = new EmbedBuilder()
+                .setColor(0xFF4444)
+                .setTitle('❌ Error del Servicio de IA')
+                .setDescription(error.message || 'Error desconocido del servicio')
+                .addFields({
+                    name: '💡 Consejos',
+                    value: '• Verifica que tu mensaje no sea demasiado largo\n' +
+                           '• Espera unos segundos entre consultas\n' +
+                           '• Evita contenido inapropiado'
+                })
+                .setFooter({ text: 'Si el problema persiste, contacta a un administrador' })
+                .setTimestamp();
+
+            await message.reply({ embeds: [errorEmbed] });
+        } finally {
+            // Limpiar indicador de escritura
+            clearInterval(typingInterval);
         }
     }
 }