UNPKG

@sanderkooger/mcp-server-ragdocs

Version:

An MCP server for semantic documentation search and retrieval using vector databases to augment LLM capabilities.

66 lines (65 loc) 2.1 kB
import { BaseTool } from './base-tool.js'; import fs from 'fs/promises'; import path from 'path'; import { fileURLToPath } from 'url'; // Get current directory in ES modules const __filename = fileURLToPath(import.meta.url); const __dirname = path.dirname(__filename); const QUEUE_FILE = path.join(__dirname, '..', '..', 'queue.txt'); export class ClearQueueTool extends BaseTool { get definition() { return { name: 'clear_queue', description: 'Clear all URLs from the queue', inputSchema: { type: 'object', properties: {}, required: [] } }; } async execute(_args) { try { // Check if queue file exists try { await fs.access(QUEUE_FILE); } catch { return { content: [ { type: 'text', text: 'Queue is already empty (queue file does not exist)' } ] }; } // Read current queue to get count of URLs being cleared const content = await fs.readFile(QUEUE_FILE, 'utf-8'); const urlCount = content .split('\n') .filter((url) => url.trim() !== '').length; // Clear the queue by emptying the file await fs.writeFile(QUEUE_FILE, ''); return { content: [ { type: 'text', text: `Queue cleared successfully. Removed ${urlCount} URL${urlCount === 1 ? '' : 's'} from the queue.` } ] }; } catch (error) { return { content: [ { type: 'text', text: `Failed to clear queue: ${error}` } ], isError: true }; } } }