import { z } from "zod";
import { getFrmrDocuments } from "../indexer.js";
import type { ToolDefinition } from "./base.js";
interface Definition {
id: string;
term: string;
definition: string;
alts?: string[];
note?: string;
}
const schema = z.object({
term: z.string().describe("Search term to find in definitions"),
limit: z.number().int().min(1).max(100).default(20),
});
export const searchDefinitionsTool: ToolDefinition<
typeof schema,
{ total: number; definitions: Definition[] }
> = {
name: "search_definitions",
description:
"Search FedRAMP definitions (FRD document) by term. Returns matching definitions with their full text and any alternate terms.",
schema,
execute: async (input) => {
const frdDoc = getFrmrDocuments().find((doc) => doc.type === "FRD");
if (!frdDoc || !frdDoc.raw) {
return { total: 0, definitions: [] };
}
const raw = frdDoc.raw as Record<string, unknown>;
const frd = raw.FRD as Record<string, unknown> | undefined;
const allDefs = (frd?.ALL as Definition[]) ?? [];
const searchLower = input.term.toLowerCase();
const matches = allDefs.filter((def) => {
if (def.term?.toLowerCase().includes(searchLower)) return true;
if (def.definition?.toLowerCase().includes(searchLower)) return true;
if (def.alts?.some((alt) => alt.toLowerCase().includes(searchLower)))
return true;
return false;
});
return {
total: matches.length,
definitions: matches.slice(0, input.limit),
};
},
};
