RAG MCP Server

// src/tools/rag/manage-projects.ts // Outil: manage_projects - Gérer et lister les projets indexés import { ToolDefinition, ToolHandler } from "../../core/tool-registry.js"; import { getProjects, getProjectStatistics } from "../../rag/searcher.js"; /** * Définition de l'outil manage_projects */ export const manageProjectsTool: ToolDefinition = { name: "manage_projects", description: "Gérer et lister les projets indexés", inputSchema: { type: "object", properties: { action: { type: "string", description: "Action à effectuer", enum: ["list", "stats"], default: "list" }, project_path: { type: "string", description: "Chemin du projet pour les statistiques (requis pour 'stats')" } } }, }; /** * Handler pour l'outil manage_projects */ export const manageProjectsHandler: ToolHandler = async (args) => { const action = args.action || "list"; try { if (action === "stats") { if (!args.project_path || typeof args.project_path !== 'string') { throw new Error("The 'project_path' parameter is required for 'stats' action"); } const stats = await getProjectStatistics(args.project_path); return { content: [{ type: "text", text: JSON.stringify(stats, null, 2) }] }; } else { const projects = await getProjects(); return { content: [{ type: "text", text: JSON.stringify(projects, null, 2) }] }; } } catch (error) { console.error("Error in manage_projects tool:", error); throw error; } }; /** * Test de l'outil (pour usage en développement) */ export async function testManageProjects() { console.log("Testing manage_projects tool..."); try { // D'abord indexer un projet de test const testProjectPath = "/tmp/test-project-manage"; const fs = await import('fs'); const path = await import('path'); if (!fs.existsSync(testProjectPath)) { fs.mkdirSync(testProjectPath, { recursive: true }); } // Créer un fichier de test const testFile = path.join(testProjectPath, "manage-test.js"); fs.writeFileSync(testFile, "// Test file for manage_projects\nconsole.log('Manage test');"); // Indexer le projet const { indexProjectHandler } = await import("./index-project.js"); await indexProjectHandler({ project_path: testProjectPath, file_patterns: ["**/*.js"], recursive: true, embedding_provider: "fake" }); console.log(`✅ Indexed test project at: ${testProjectPath}`); // Tester l'action 'list' console.log("Test 1: Action 'list'"); const listResult = await manageProjectsHandler({ action: "list" }); console.log("✅ manage_projects 'list' fonctionne:", listResult ? "Oui" : "Non"); // Tester l'action 'stats' console.log("Test 2: Action 'stats'"); const statsResult = await manageProjectsHandler({ action: "stats", project_path: testProjectPath }); console.log("✅ manage_projects 'stats' fonctionne:", statsResult ? "Oui" : "Non"); // Nettoyer fs.rmSync(testProjectPath, { recursive: true, force: true }); return { listResult, statsResult }; } catch (error) { console.error("❌ Test failed:", error); throw error; } }