@sanderkooger/mcp-server-ragdocs

by sanderkooger
Verified
import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js' import { BaseHandler } from './base-handler.js' import type { McpToolResponse } from '../types.js' const COLLECTION_NAME = 'documentation' export class RemoveDocumentationHandler extends BaseHandler { async handle(args: any): Promise<McpToolResponse> { if (!args.urls || !Array.isArray(args.urls) || args.urls.length === 0) { throw new McpError( ErrorCode.InvalidParams, 'urls must be a non-empty array' ) } if (!args.urls.every((url: string) => typeof url === 'string')) { throw new McpError(ErrorCode.InvalidParams, 'All URLs must be strings') } try { // Delete using filter to match any of the provided URLs const result = await this.apiClient.qdrantClient.delete(COLLECTION_NAME, { filter: { should: args.urls.map((url: string) => ({ key: 'url', match: { value: url } })) }, wait: true // Ensure deletion is complete before responding }) if (!['acknowledged', 'completed'].includes(result.status)) { throw new Error('Delete operation failed') } return { content: [ { type: 'text', text: `Successfully removed documentation from ${args.urls.length} source${args.urls.length > 1 ? 's' : ''}: ${args.urls.join(', ')}` } ] } } catch (error) { if (error instanceof Error) { if (error.message.includes('unauthorized')) { throw new McpError( ErrorCode.InvalidRequest, 'Failed to authenticate with Qdrant cloud while removing documentation' ) } else if ( error.message.includes('ECONNREFUSED') || error.message.includes('ETIMEDOUT') ) { throw new McpError( ErrorCode.InternalError, 'Connection to Qdrant cloud failed while removing documentation' ) } } return { content: [ { type: 'text', text: `Failed to remove documentation: ${error}` } ], isError: true } } } }