RAD Security

Instructions

Implementation Reference

• src/index.ts:1469-1487 (handler)

  MCP server CallToolRequest handler case that executes the search_knowledge_base tool by parsing arguments, calling the underlying searchKnowledgeBase function, and formatting the response as MCP content.

  case "search_knowledge_base": {
    const args = knowledgeBase.SearchKnowledgeBaseSchema.parse(
      request.params.arguments
    );
    const response = await knowledgeBase.searchKnowledgeBase(
      client,
      args.query,
      args.top_k,
      args.min_score,
      args.thread_id,
      args.collections,
      args.document_ids
    );
    return {
      content: [
        { type: "text", text: JSON.stringify(response, null, 2) },
      ],
    };
  }

• src/operations/knowledge-base.ts:4-11 (schema)

  Zod schema definition for input validation of the search_knowledge_base tool.

  export const SearchKnowledgeBaseSchema = z.object({
    query: z.string().describe("Natural language question or keywords to search for across your uploaded knowledge base content. Can be a full question, technical terms, or key phrases."),
    top_k: z.number().optional().describe("Maximum number of most relevant document excerpts to return. Use higher values (10-20) for comprehensive research, lower values (3-5) for focused answers. Default: 5"),
    min_score: z.number().optional().describe("Minimum semantic similarity score threshold (0.0 to 1.0). Higher values (0.8-1.0) return only highly relevant matches, lower values (0.5-0.7) include broader context. Default: 0.5"),
    thread_id: z.string().optional().describe("Thread identifier for the current conversation or session. IMPORTANT: If a thread_id is available in your context, you MUST provide it to include thread-specific documents alongside general knowledge base content. Only omit if no thread context exists."),
    collections: z.array(z.string()).optional().describe("Optional list of collection names to filter search results. Only documents tagged with these collections will be searched. Cannot be used with document_ids."),
    document_ids: z.array(z.string()).optional().describe("Optional list of specific document IDs to search within. Use this to restrict search to known documents. Cannot be used with collections."),
  });

• src/operations/knowledge-base.ts:29-70 (handler)

  Core handler function that constructs the request body and calls the RAD Security API endpoint for searching the knowledge base.

  export async function searchKnowledgeBase(
    client: RadSecurityClient,
    query: string,
    topK?: number,
    minScore?: number,
    threadId?: string,
    collections?: string[],
    documentIds?: string[],
  ): Promise<any> {
    const tenantId = await client.getTenantId();
  
    const body: Record<string, any> = { query };
  
    if (topK !== undefined) {
      body.top_k = topK;
    }
  
    if (minScore !== undefined) {
      body.min_score = minScore;
    }
  
    if (threadId !== undefined) {
      body.thread_id = threadId;
    }
  
    if (collections !== undefined) {
      body.collections = collections;
    }
  
    if (documentIds !== undefined) {
      body.document_ids = documentIds;
    }
  
    return client.makeRequest(
      `/tenants/${tenantId}/accounts/${client.getAccountId()}/knowledge_base/search`,
      {},
      {
        method: "POST",
        body: body,
      }
    );
  }

• src/index.ts:548-555 (registration)

  Tool registration in the MCP ListToolsRequest handler, defining the tool name, description, and input schema.

  {
    name: "search_knowledge_base",
    description:
      "Search your organization's knowledge base to find relevant uploaded documents, procedures, reports, and other content using natural language queries",
    inputSchema: zodToJsonSchema(
      knowledgeBase.SearchKnowledgeBaseSchema
    ),
  },