RAG MCP Server

// src/tools/graph/register-graph-tools-batch3.ts // Enregistrement des outils graph Batch 3 dans le ToolRegistry import { toolRegistry } from "../../core/tool-registry.js"; import { openNodesHandler, openNodesTool } from "./open-nodes.js"; import { readGraphHandler, readGraphTool } from "./read-graph.js"; import { searchNodesHandler, searchNodesTool } from "./search-nodes.js"; /** * Enregistre tous les outils graph Batch 3 dans le ToolRegistry */ export function registerGraphToolsBatch3() { console.log("📝 Enregistrement des outils graph Batch 3..."); // Enregistrer read_graph try { toolRegistry.register(readGraphTool, readGraphHandler); console.log(`✅ Outil enregistré: ${readGraphTool.name}`); } catch (error) { console.error(`❌ Erreur lors de l'enregistrement de ${readGraphTool.name}:`, error); } // Enregistrer search_nodes try { toolRegistry.register(searchNodesTool, searchNodesHandler); console.log(`✅ Outil enregistré: ${searchNodesTool.name}`); } catch (error) { console.error(`❌ Erreur lors de l'enregistrement de ${searchNodesTool.name}:`, error); } // Enregistrer open_nodes try { toolRegistry.register(openNodesTool, openNodesHandler); console.log(`✅ Outil enregistré: ${openNodesTool.name}`); } catch (error) { console.error(`❌ Erreur lors de l'enregistrement de ${openNodesTool.name}:`, error); } console.log(`🎉 Batch 3 terminé. Outils enregistrés au total: ${toolRegistry.size()}`); } /** * Vérifie que tous les outils Batch 3 sont correctement enregistrés */ export function verifyGraphToolsBatch3() { const expectedTools = [ "read_graph", "search_nodes", "open_nodes" ]; console.log("🔍 Vérification des outils graph Batch 3..."); let allRegistered = true; for (const toolName of expectedTools) { if (toolRegistry.hasTool(toolName)) { console.log(`✅ ${toolName} est enregistré`); } else { console.log(`❌ ${toolName} n'est PAS enregistré`); allRegistered = false; } } if (allRegistered) { console.log("🎉 Tous les outils Batch 3 sont correctement enregistrés !"); } else { console.log("⚠️ Certains outils Batch 3 ne sont pas enregistrés"); } return allRegistered; } /** * Test d'intégration des outils Batch 3 */ export async function testGraphToolsBatch3() { console.log("🧪 Test d'intégration des outils graph Batch 3..."); // Préparer des données de test const testEntity1 = "Test Entity for Batch 3 Read"; const testEntity2 = "Test Entity for Batch 3 Search"; try { // Créer des entités pour les tests console.log("Préparation: Création des entités de test..."); await toolRegistry.execute("create_entities", { entities: [ { name: testEntity1, entityType: "Test", observations: ["Entity for read_graph test"] }, { name: testEntity2, entityType: "SearchTest", observations: ["Entity for search_nodes test"] } ] }); console.log("✅ Données de test préparées"); } catch (error) { console.error("❌ Échec de la préparation des données de test:", error instanceof Error ? error.message : String(error)); return false; } // Test 1: read_graph try { console.log("\nTest 1: read_graph"); const result = await toolRegistry.execute("read_graph", {}); console.log("✅ read_graph fonctionne:", result ? "Oui" : "Non"); } catch (error) { console.error("❌ read_graph a échoué:", error instanceof Error ? error.message : String(error)); return false; } // Test 2: search_nodes try { console.log("\nTest 2: search_nodes"); const result = await toolRegistry.execute("search_nodes", { query: "SearchTest" }); console.log("✅ search_nodes fonctionne:", result ? "Oui" : "Non"); } catch (error) { console.error("❌ search_nodes a échoué:", error instanceof Error ? error.message : String(error)); return false; } // Test 3: open_nodes try { console.log("\nTest 3: open_nodes"); const result = await toolRegistry.execute("open_nodes", { names: [testEntity1] }); console.log("✅ open_nodes fonctionne:", result ? "Oui" : "Non"); } catch (error) { console.error("❌ open_nodes a échoué:", error instanceof Error ? error.message : String(error)); return false; } // Nettoyer les données de test try { console.log("\nNettoyage: Suppression des entités de test..."); await toolRegistry.execute("delete_entities", { entityNames: [testEntity1, testEntity2] }); console.log("✅ Données de test nettoyées"); } catch (error) { console.warn("⚠️ Échec du nettoyage des données de test:", error instanceof Error ? error.message : String(error)); } console.log("\n🎉 Tous les tests d'intégration Batch 3 ont réussi !"); return true; } /** * Fonction principale pour initialiser les outils graph Batch 3 */ export function initializeGraphToolsBatch3() { console.log("🚀 Initialisation des outils graph Batch 3..."); // Enregistrer les outils registerGraphToolsBatch3(); // Vérifier l'enregistrement const registered = verifyGraphToolsBatch3(); if (registered) { console.log("✅ Initialisation Batch 3 terminée avec succès"); console.log(`📊 Outils disponibles: ${toolRegistry.getToolNames().join(', ')}`); console.log(`📈 Total d'outils: ${toolRegistry.size()}`); } else { console.error("❌ Échec de l'initialisation Batch 3"); } } // Exécution automatique si ce fichier est exécuté directement if (import.meta.url === `file://${process.argv[1]}`) { initializeGraphToolsBatch3(); // Exécuter les tests si demandé if (process.argv.includes("--test")) { testGraphToolsBatch3().then(success => { if (success) { console.log("🎉 Tous les tests Batch 3 ont réussi !"); process.exit(0); } else { console.error("❌ Certains tests Batch 3 ont échoué"); process.exit(1); } }); } }