RAG MCP Server

// src/tools/graph/register-graph-tools-batch2.ts // Enregistrement des outils graph Batch 2 dans le ToolRegistry import { toolRegistry } from "../../core/tool-registry.js"; import { deleteEntitiesHandler, deleteEntitiesTool } from "./delete-entities.js"; import { deleteObservationsHandler, deleteObservationsTool } from "./delete-observations.js"; import { deleteRelationsHandler, deleteRelationsTool } from "./delete-relations.js"; /** * Enregistre tous les outils graph Batch 2 dans le ToolRegistry */ export function registerGraphToolsBatch2() { console.log("📝 Enregistrement des outils graph Batch 2..."); // Enregistrer delete_entities try { toolRegistry.register(deleteEntitiesTool, deleteEntitiesHandler); console.log(`✅ Outil enregistré: ${deleteEntitiesTool.name}`); } catch (error) { console.error(`❌ Erreur lors de l'enregistrement de ${deleteEntitiesTool.name}:`, error); } // Enregistrer delete_observations try { toolRegistry.register(deleteObservationsTool, deleteObservationsHandler); console.log(`✅ Outil enregistré: ${deleteObservationsTool.name}`); } catch (error) { console.error(`❌ Erreur lors de l'enregistrement de ${deleteObservationsTool.name}:`, error); } // Enregistrer delete_relations try { toolRegistry.register(deleteRelationsTool, deleteRelationsHandler); console.log(`✅ Outil enregistré: ${deleteRelationsTool.name}`); } catch (error) { console.error(`❌ Erreur lors de l'enregistrement de ${deleteRelationsTool.name}:`, error); } console.log(`🎉 Batch 2 terminé. Outils enregistrés au total: ${toolRegistry.size()}`); } /** * Vérifie que tous les outils Batch 2 sont correctement enregistrés */ export function verifyGraphToolsBatch2() { const expectedTools = [ "delete_entities", "delete_observations", "delete_relations" ]; console.log("🔍 Vérification des outils graph Batch 2..."); 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 2 sont correctement enregistrés !"); } else { console.log("⚠️ Certains outils Batch 2 ne sont pas enregistrés"); } return allRegistered; } /** * Test d'intégration des outils Batch 2 */ export async function testGraphToolsBatch2() { console.log("🧪 Test d'intégration des outils graph Batch 2..."); // Préparer des données de test const testEntity1 = "Test Entity for Batch 2 Delete"; const testEntity2 = "Test Entity B for Relations"; 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: ["Observation 1", "Observation 2", "Observation 3"] }, { name: testEntity2, entityType: "Test", observations: ["Entity B for relation tests"] } ] }); // Créer une relation pour le test delete_relations await toolRegistry.execute("create_relations", { relations: [{ from: testEntity1, to: testEntity2, relationType: "TEST_RELATION_BATCH2" }] }); 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: delete_observations try { console.log("\nTest 1: delete_observations"); const result = await toolRegistry.execute("delete_observations", { deletions: [{ entityName: testEntity1, observations: ["Observation 2"] }] }); console.log("✅ delete_observations fonctionne:", result ? "Oui" : "Non"); } catch (error) { console.error("❌ delete_observations a échoué:", error instanceof Error ? error.message : String(error)); return false; } // Test 2: delete_relations try { console.log("\nTest 2: delete_relations"); const result = await toolRegistry.execute("delete_relations", { relations: [{ from: testEntity1, to: testEntity2, relationType: "TEST_RELATION_BATCH2" }] }); console.log("✅ delete_relations fonctionne:", result ? "Oui" : "Non"); } catch (error) { console.error("❌ delete_relations a échoué:", error instanceof Error ? error.message : String(error)); return false; } // Test 3: delete_entities try { console.log("\nTest 3: delete_entities"); const result = await toolRegistry.execute("delete_entities", { entityNames: [testEntity1, testEntity2] }); console.log("✅ delete_entities fonctionne:", result ? "Oui" : "Non"); } catch (error) { console.error("❌ delete_entities a échoué:", error instanceof Error ? error.message : String(error)); return false; } console.log("\n🎉 Tous les tests d'intégration Batch 2 ont réussi !"); return true; } /** * Fonction principale pour initialiser les outils graph Batch 2 */ export function initializeGraphToolsBatch2() { console.log("🚀 Initialisation des outils graph Batch 2..."); // Enregistrer les outils registerGraphToolsBatch2(); // Vérifier l'enregistrement const registered = verifyGraphToolsBatch2(); if (registered) { console.log("✅ Initialisation Batch 2 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 2"); } } // Exécution automatique si ce fichier est exécuté directement if (import.meta.url === `file://${process.argv[1]}`) { initializeGraphToolsBatch2(); // Exécuter les tests si demandé if (process.argv.includes("--test")) { testGraphToolsBatch2().then(success => { if (success) { console.log("🎉 Tous les tests Batch 2 ont réussi !"); process.exit(0); } else { console.error("❌ Certains tests Batch 2 ont échoué"); process.exit(1); } }); } }