@oliverlessa/gemini-agent-lib/examples/exemplo-chat-agent-com-memoria.js

Uma biblioteca NodeJS para criar agentes de IA com Gemini LLM

/**
 * Exemplo de uso do ChatAgent com diferentes tipos de memória
 * 
 * Este exemplo demonstra como configurar e usar os diferentes tipos de memória
 * (ConversationMemory, FactMemory e SummaryMemory) com o ChatAgent.
 */

require('dotenv').config();
const { ChatAgent, VertexAILLM, memory } = require('../index');
const path = require('path');

// Caminho para o banco de dados SQLite (será criado se não existir)
const dbPath = path.join(__dirname, 'chat_memory.db');

async function main() {
    try {
        console.log("Iniciando exemplo de ChatAgent com memória persistente...");
        
        // Criar instâncias dos adaptadores de memória
        const conversationMemory = new memory.SQLiteConversationMemoryAdapter({
            dbConfig: { dbPath }
        });
        
        const factMemory = new memory.SQLiteFactMemoryAdapter({
            dbConfig: { dbPath }
        });
        
        const summaryMemory = new memory.SQLiteSummaryMemoryAdapter({
            dbConfig: { dbPath }
        });
        
        // Criar instância do LLM
        const llm = new VertexAILLM({
            projectId: process.env.GOOGLE_CLOUD_PROJECT_ID,
            credentialsPath: process.env.GOOGLE_APPLICATION_CREDENTIALS,
            modelName: "gemini-2.0-flash-001",
            mode: "chat",
            generationConfig: {
                maxOutputTokens: 2048,
                temperature: 0.2
            }
        });
        
        // Criar instância do ChatAgent com memória persistente
        const chatAgent = new ChatAgent({
            role: "Assistente Pessoal",
            objective: "Ajudar o usuário com suas tarefas diárias",
            context: `Você é um assistente pessoal amigável e prestativo.
Você deve manter um registro das preferências do usuário e usar essas informações para personalizar suas respostas.
Você também deve resumir periodicamente a conversa para manter o contexto.`,
            llm,
            conversationMemory,
            factMemory,
            summaryMemory
        });
        
        console.log(`ChatAgent criado com ID de conversa: ${chatAgent.chatId}`);
        
        // Simular uma conversa
        console.log("\n--- Iniciando conversa ---\n");
        
        // Primeira mensagem do usuário
        const mensagem1 = "Olá! Meu nome é Carlos e eu gosto de programação em JavaScript.";
        console.log(`Usuário: ${mensagem1}`);
        
        let resposta = await chatAgent.processUserMessage(mensagem1);
        console.log(`Assistente: ${resposta.text}`);
        
        // Armazenar fatos sobre o usuário
        await chatAgent.setFact("nome", "Carlos");
        await chatAgent.setFact("linguagem_preferida", "JavaScript");
        
        // Segunda mensagem do usuário
        const mensagem2 = "Você pode me recomendar alguns frameworks de JavaScript para desenvolvimento web?";
        console.log(`\nUsuário: ${mensagem2}`);
        
        resposta = await chatAgent.processUserMessage(mensagem2);
        console.log(`Assistente: ${resposta.text}`);
        
        // Gerar e armazenar um resumo da conversa
        const resumo = "O usuário se chama Carlos, gosta de JavaScript e pediu recomendações de frameworks para desenvolvimento web.";
        await chatAgent.addSummary(resumo);
        
        // Terceira mensagem do usuário
        const mensagem3 = "Qual é o meu nome e qual linguagem de programação eu gosto?";
        console.log(`\nUsuário: ${mensagem3}`);
        
        resposta = await chatAgent.processUserMessage(mensagem3);
        console.log(`Assistente: ${resposta.text}`);
        
        // Recuperar e mostrar fatos armazenados
        console.log("\n--- Fatos armazenados ---");
        const fatos = await chatAgent.getAllFacts();
        console.log(fatos);
        
        // Recuperar e mostrar o resumo mais recente
        console.log("\n--- Resumo mais recente ---");
        const ultimoResumo = await chatAgent.getLatestSummary();
        console.log(ultimoResumo);
        
        // Mostrar o histórico de conversa
        console.log("\n--- Histórico de conversa ---");
        console.log(chatAgent.conversationHistory);
        
        // Fechar as conexões com o banco de dados
        await conversationMemory.close();
        await factMemory.close();
        await summaryMemory.close();
        
        console.log("\nExemplo concluído com sucesso!");
    } catch (error) {
        console.error("Erro durante a execução do exemplo:", error);
    }
}

// Executar o exemplo
main();