@crowley/rag-mcp 1.0.5 → 1.0.6

package/dist/tools/search.js

@@ -3,355 +3,280 @@
  * documentation search, and project statistics.
  */
 import { formatCodeResults, formatNavigationResults, truncate } from "../formatters.js";
+import { z } from "zod";
+import { TOOL_ANNOTATIONS } from "../annotations.js";
 /**
  * Create the search tools module with project-specific descriptions.
  */
 export function createSearchTools(projectName) {
-    const tools = [
+    return [
         {
             name: "search_codebase",
             description: `Search the ${projectName} codebase. Returns file locations, symbols, and graph connections. Use Read tool to view the actual code at returned locations.`,
-            inputSchema: {
-                type: "object",
-                properties: {
-                    query: {
-                        type: "string",
-                        description: "Search query for finding code",
-                    },
-                    limit: {
-                        type: "number",
-                        description: "Max results to return (default: 5)",
-                        default: 5,
-                    },
-                    language: {
-                        type: "string",
-                        description: "Filter by language (typescript, python, vue, etc.)",
-                    },
-                    path: {
-                        type: "string",
-                        description: "Filter by path pattern (e.g., 'src/modules/*')",
-                    },
-                    layer: {
-                        type: "string",
-                        description: "Filter by architectural layer (api, service, util, model, middleware, test, parser, types, config, other)",
-                    },
-                    service: {
-                        type: "string",
-                        description: "Filter by service/class name (e.g., 'EmbeddingService')",
-                    },
-                },
-                required: ["query"],
+            schema: z.object({
+                query: z.string().describe("Search query for finding code"),
+                limit: z.number().optional().describe("Max results to return (default: 5)"),
+                language: z.string().optional().describe("Filter by language (typescript, python, vue, etc.)"),
+                path: z.string().optional().describe("Filter by path pattern (e.g., 'src/modules/*')"),
+                layer: z.string().optional().describe("Filter by architectural layer (api, service, util, model, middleware, test, parser, types, config, other)"),
+                service: z.string().optional().describe("Filter by service/class name (e.g., 'EmbeddingService')"),
+            }),
+            annotations: TOOL_ANNOTATIONS["search_codebase"],
+            handler: async (args, ctx) => {
+                const { query, limit = 5, language, path, layer, service } = args;
+                const response = await ctx.api.post("/api/search", {
+                    collection: `${ctx.collectionPrefix}codebase`,
+                    query,
+                    limit,
+                    mode: "navigate",
+                    filters: { language, path, layer, service },
+                });
+                const results = response.data.results;
+                if (!results || results.length === 0) {
+                    return "No results found for this query.";
+                }
+                return formatNavigationResults(results);
             },
         },
         {
             name: "search_similar",
             description: "Find code similar to a given snippet.",
-            inputSchema: {
-                type: "object",
-                properties: {
-                    code: {
-                        type: "string",
-                        description: "Code snippet to find similar code for",
-                    },
-                    limit: {
-                        type: "number",
-                        description: "Max results (default: 5)",
-                        default: 5,
-                    },
-                },
-                required: ["code"],
+            schema: z.object({
+                code: z.string().describe("Code snippet to find similar code for"),
+                limit: z.number().optional().describe("Max results (default: 5)"),
+            }),
+            annotations: TOOL_ANNOTATIONS["search_similar"],
+            handler: async (args, ctx) => {
+                const { code, limit = 5 } = args;
+                const response = await ctx.api.post("/api/search-similar", {
+                    collection: `${ctx.collectionPrefix}codebase`,
+                    code,
+                    limit,
+                });
+                const results = response.data.results;
+                if (!results || results.length === 0) {
+                    return "No similar code found.";
+                }
+                return formatCodeResults(results, 400);
             },
         },
         {
             name: "grouped_search",
             description: `Search ${projectName} codebase grouped by file. Returns file locations with symbols and connections. Use Read tool to view the actual code.`,
-            inputSchema: {
-                type: "object",
-                properties: {
-                    query: {
-                        type: "string",
-                        description: "Search query",
-                    },
-                    groupBy: {
-                        type: "string",
-                        description: "Field to group by (default: 'file')",
-                        default: "file",
-                    },
-                    limit: {
-                        type: "number",
-                        description: "Max groups to return (default: 10)",
-                        default: 10,
-                    },
-                    language: {
-                        type: "string",
-                        description: "Filter by language",
-                    },
-                    layer: {
-                        type: "string",
-                        description: "Filter by architectural layer (api, service, util, etc.)",
-                    },
-                    service: {
-                        type: "string",
-                        description: "Filter by service/class name",
-                    },
-                },
-                required: ["query"],
+            schema: z.object({
+                query: z.string().describe("Search query"),
+                groupBy: z.string().optional().describe("Field to group by (default: 'file')"),
+                limit: z.number().optional().describe("Max groups to return (default: 10)"),
+                language: z.string().optional().describe("Filter by language"),
+                layer: z.string().optional().describe("Filter by architectural layer (api, service, util, etc.)"),
+                service: z.string().optional().describe("Filter by service/class name"),
+            }),
+            annotations: TOOL_ANNOTATIONS["grouped_search"],
+            handler: async (args, ctx) => {
+                const { query, groupBy = "file", limit = 10, language, layer, service } = args;
+                const filters = { language, layer, service };
+                const hasFilters = Object.values(filters).some(v => v !== undefined);
+                const response = await ctx.api.post("/api/search-grouped", {
+                    collection: `${ctx.collectionPrefix}codebase`,
+                    query,
+                    groupBy,
+                    limit,
+                    mode: "navigate",
+                    filters: hasFilters ? filters : undefined,
+                });
+                const groups = response.data.groups;
+                if (!groups || groups.length === 0) {
+                    return "No results found.";
+                }
+                const allResults = groups.flatMap((g) => g.results);
+                return formatNavigationResults(allResults);
             },
         },
         {
             name: "hybrid_search",
             description: `Hybrid search combining keyword matching and semantic similarity for ${projectName}. Returns file locations with symbols and connections. Use Read tool to view code.`,
-            inputSchema: {
-                type: "object",
-                properties: {
-                    query: {
-                        type: "string",
-                        description: "Search query",
-                    },
-                    limit: {
-                        type: "number",
-                        description: "Max results (default: 10)",
-                        default: 10,
-                    },
-                    semanticWeight: {
-                        type: "number",
-                        description: "Weight for semantic vs keyword (0-1, default: 0.7)",
-                        default: 0.7,
-                    },
-                    language: {
-                        type: "string",
-                        description: "Filter by language",
-                    },
-                    layer: {
-                        type: "string",
-                        description: "Filter by architectural layer (api, service, util, etc.)",
-                    },
-                    service: {
-                        type: "string",
-                        description: "Filter by service/class name",
-                    },
-                },
-                required: ["query"],
+            schema: z.object({
+                query: z.string().describe("Search query"),
+                limit: z.number().optional().describe("Max results (default: 10)"),
+                semanticWeight: z.number().optional().describe("Weight for semantic vs keyword (0-1, default: 0.7)"),
+                language: z.string().optional().describe("Filter by language"),
+                layer: z.string().optional().describe("Filter by architectural layer (api, service, util, etc.)"),
+                service: z.string().optional().describe("Filter by service/class name"),
+            }),
+            annotations: TOOL_ANNOTATIONS["hybrid_search"],
+            handler: async (args, ctx) => {
+                const { query, limit = 10, semanticWeight = 0.7, language, layer, service } = args;
+                const filters = { language, layer, service };
+                const hasFilters = Object.values(filters).some(v => v !== undefined);
+                const response = await ctx.api.post("/api/search-hybrid", {
+                    collection: `${ctx.collectionPrefix}codebase`,
+                    query,
+                    limit,
+                    semanticWeight,
+                    mode: "navigate",
+                    filters: hasFilters ? filters : undefined,
+                });
+                const results = response.data.results;
+                if (!results || results.length === 0) {
+                    return "No results found.";
+                }
+                return formatNavigationResults(results);
             },
         },
         {
             name: "search_docs",
             description: `Search documentation in the ${projectName} project.`,
-            inputSchema: {
-                type: "object",
-                properties: {
-                    query: {
-                        type: "string",
-                        description: "Search query",
-                    },
-                    limit: {
-                        type: "number",
-                        description: "Max results (default: 5)",
-                        default: 5,
-                    },
-                },
-                required: ["query"],
+            schema: z.object({
+                query: z.string().describe("Search query"),
+                limit: z.number().optional().describe("Max results (default: 5)"),
+            }),
+            annotations: TOOL_ANNOTATIONS["search_docs"],
+            handler: async (args, ctx) => {
+                const { query, limit = 5 } = args;
+                const response = await ctx.api.post("/api/search", {
+                    collection: `${ctx.collectionPrefix}docs`,
+                    query,
+                    limit,
+                });
+                const results = response.data.results;
+                if (!results || results.length === 0) {
+                    return "No documentation found for this query.";
+                }
+                return results
+                    .map((r) => `**${r.file}**\n` +
+                    truncate(r.content, 500))
+                    .join("\n\n---\n\n");
             },
         },
         {
             name: "get_project_stats",
             description: `Get statistics about the ${projectName} codebase.`,
-            inputSchema: {
-                type: "object",
-                properties: {},
+            schema: z.object({}),
+            outputSchema: z.object({
+                projectName: z.string(),
+                totalFiles: z.number(),
+                totalLines: z.number().optional(),
+                vectorCount: z.number(),
+                lastIndexed: z.string().optional(),
+                languages: z.record(z.string(), z.number()).optional(),
+            }),
+            annotations: TOOL_ANNOTATIONS["get_project_stats"],
+            handler: async (_args, ctx) => {
+                const response = await ctx.api.get(`/api/stats/${ctx.collectionPrefix}codebase`);
+                const stats = response.data;
+                let text = `**${ctx.projectName} Project Statistics**\n\n`;
+                text += `- Total Files: ${stats.totalFiles}\n`;
+                text += `- Total Lines: ${stats.totalLines?.toLocaleString() || "N/A"}\n`;
+                text += `- Vector Count: ${stats.vectorCount}\n`;
+                text += `- Last Indexed: ${stats.lastIndexed ? new Date(stats.lastIndexed).toLocaleString() : "Never"}\n`;
+                if (stats.languages) {
+                    text += `\n**Languages:**\n`;
+                    for (const [lang, count] of Object.entries(stats.languages)) {
+                        text += `- ${lang}: ${count} files\n`;
+                    }
+                }
+                return {
+                    text,
+                    structured: {
+                        projectName: ctx.projectName,
+                        totalFiles: stats.totalFiles,
+                        totalLines: stats.totalLines,
+                        vectorCount: stats.vectorCount,
+                        lastIndexed: stats.lastIndexed,
+                        languages: stats.languages,
+                    },
+                };
             },
         },
         {
             name: "find_symbol",
             description: `Find a function, class, type, or interface by name in ${projectName}. Fast symbol lookup without full-text search.`,
-            inputSchema: {
-                type: "object",
-                properties: {
-                    symbol: {
-                        type: "string",
-                        description: "Symbol name to find (function, class, type, etc.)",
-                    },
-                    kind: {
-                        type: "string",
-                        description: "Filter by kind: function, class, interface, type, enum, const",
-                    },
-                    limit: {
-                        type: "number",
-                        description: "Max results (default: 10)",
-                        default: 10,
+            schema: z.object({
+                symbol: z.string().describe("Symbol name to find (function, class, type, etc.)"),
+                kind: z.string().optional().describe("Filter by kind: function, class, interface, type, enum, const"),
+                limit: z.number().optional().describe("Max results (default: 10)"),
+            }),
+            outputSchema: z.object({
+                symbols: z.array(z.object({
+                    kind: z.string(),
+                    name: z.string(),
+                    file: z.string(),
+                    startLine: z.number(),
+                    endLine: z.number(),
+                    signature: z.string(),
+                    exported: z.boolean(),
+                })),
+            }),
+            annotations: TOOL_ANNOTATIONS["find_symbol"],
+            handler: async (args, ctx) => {
+                const { symbol, kind, limit = 10 } = args;
+                const response = await ctx.api.post("/api/find-symbol", {
+                    projectName: ctx.projectName,
+                    symbol,
+                    kind,
+                    limit,
+                });
+                const results = response.data.results;
+                if (!results || results.length === 0) {
+                    return `No symbol "${symbol}" found.`;
+                }
+                const text = results
+                    .map((r) => `**${r.kind} ${r.name}** in \`${r.file}\` (lines ${r.startLine}-${r.endLine})\n` +
+                    `\`${truncate(r.signature, 150)}\`` +
+                    (r.exports ? " _(exported)_" : ""))
+                    .join("\n\n");
+                return {
+                    text,
+                    structured: {
+                        symbols: results.map((r) => ({
+                            kind: r.kind,
+                            name: r.name,
+                            file: r.file,
+                            startLine: r.startLine,
+                            endLine: r.endLine,
+                            signature: r.signature,
+                            exported: !!r.exports,
+                        })),
                     },
-                },
-                required: ["symbol"],
+                };
             },
         },
         {
             name: "search_graph",
             description: `Search ${projectName} codebase with graph expansion. Returns file locations plus connected files via import/call relationships. Use Read tool to view code.`,
-            inputSchema: {
-                type: "object",
-                properties: {
-                    query: {
-                        type: "string",
-                        description: "Search query",
-                    },
-                    limit: {
-                        type: "number",
-                        description: "Max direct results (default: 5)",
-                        default: 5,
-                    },
-                    expandHops: {
-                        type: "number",
-                        description: "Number of graph hops to expand (default: 1)",
-                        default: 1,
-                    },
-                },
-                required: ["query"],
+            schema: z.object({
+                query: z.string().describe("Search query"),
+                limit: z.number().optional().describe("Max direct results (default: 5)"),
+                expandHops: z.number().optional().describe("Number of graph hops to expand (default: 1)"),
+            }),
+            annotations: TOOL_ANNOTATIONS["search_graph"],
+            handler: async (args, ctx) => {
+                const { query, limit = 5, expandHops = 1 } = args;
+                const response = await ctx.api.post("/api/search-graph", {
+                    collection: `${ctx.collectionPrefix}codebase`,
+                    query,
+                    limit,
+                    expandHops,
+                    mode: "navigate",
+                });
+                const { results, graphExpanded, expandedFiles } = response.data;
+                if ((!results || results.length === 0) && (!graphExpanded || graphExpanded.length === 0)) {
+                    return "No results found.";
+                }
+                let output = "";
+                if (results && results.length > 0) {
+                    output += "**Direct matches:**\n\n";
+                    output += formatNavigationResults(results);
+                }
+                if (graphExpanded && graphExpanded.length > 0) {
+                    output += "\n\n---\n\n**Graph-connected files:**\n\n";
+                    output += formatNavigationResults(graphExpanded);
+                }
+                if (expandedFiles && expandedFiles.length > 0) {
+                    output += `\n\n_Graph expanded to ${expandedFiles.length} additional files._`;
+                }
+                return output;
             },
         },
     ];
-    const handlers = {
-        search_codebase: async (args, ctx) => {
-            const { query, limit = 5, language, path, layer, service } = args;
-            const response = await ctx.api.post("/api/search", {
-                collection: `${ctx.collectionPrefix}codebase`,
-                query,
-                limit,
-                mode: "navigate",
-                filters: { language, path, layer, service },
-            });
-            const results = response.data.results;
-            if (!results || results.length === 0) {
-                return "No results found for this query.";
-            }
-            return formatNavigationResults(results);
-        },
-        search_similar: async (args, ctx) => {
-            const { code, limit = 5 } = args;
-            const response = await ctx.api.post("/api/search-similar", {
-                collection: `${ctx.collectionPrefix}codebase`,
-                code,
-                limit,
-            });
-            const results = response.data.results;
-            if (!results || results.length === 0) {
-                return "No similar code found.";
-            }
-            return formatCodeResults(results, 400);
-        },
-        grouped_search: async (args, ctx) => {
-            const { query, groupBy = "file", limit = 10, language, layer, service } = args;
-            const filters = { language, layer, service };
-            const hasFilters = Object.values(filters).some(v => v !== undefined);
-            const response = await ctx.api.post("/api/search-grouped", {
-                collection: `${ctx.collectionPrefix}codebase`,
-                query,
-                groupBy,
-                limit,
-                mode: "navigate",
-                filters: hasFilters ? filters : undefined,
-            });
-            const groups = response.data.groups;
-            if (!groups || groups.length === 0) {
-                return "No results found.";
-            }
-            const allResults = groups.flatMap((g) => g.results);
-            return formatNavigationResults(allResults);
-        },
-        hybrid_search: async (args, ctx) => {
-            const { query, limit = 10, semanticWeight = 0.7, language, layer, service } = args;
-            const filters = { language, layer, service };
-            const hasFilters = Object.values(filters).some(v => v !== undefined);
-            const response = await ctx.api.post("/api/search-hybrid", {
-                collection: `${ctx.collectionPrefix}codebase`,
-                query,
-                limit,
-                semanticWeight,
-                mode: "navigate",
-                filters: hasFilters ? filters : undefined,
-            });
-            const results = response.data.results;
-            if (!results || results.length === 0) {
-                return "No results found.";
-            }
-            return formatNavigationResults(results);
-        },
-        search_docs: async (args, ctx) => {
-            const { query, limit = 5 } = args;
-            const response = await ctx.api.post("/api/search", {
-                collection: `${ctx.collectionPrefix}docs`,
-                query,
-                limit,
-            });
-            const results = response.data.results;
-            if (!results || results.length === 0) {
-                return "No documentation found for this query.";
-            }
-            return results
-                .map((r) => `**${r.file}**\n` +
-                truncate(r.content, 500))
-                .join("\n\n---\n\n");
-        },
-        find_symbol: async (args, ctx) => {
-            const { symbol, kind, limit = 10 } = args;
-            const response = await ctx.api.post("/api/find-symbol", {
-                projectName: ctx.projectName,
-                symbol,
-                kind,
-                limit,
-            });
-            const results = response.data.results;
-            if (!results || results.length === 0) {
-                return `No symbol "${symbol}" found.`;
-            }
-            return results
-                .map((r) => `**${r.kind} ${r.name}** in \`${r.file}\` (lines ${r.startLine}-${r.endLine})\n` +
-                `\`${truncate(r.signature, 150)}\`` +
-                (r.exports ? " _(exported)_" : ""))
-                .join("\n\n");
-        },
-        search_graph: async (args, ctx) => {
-            const { query, limit = 5, expandHops = 1 } = args;
-            const response = await ctx.api.post("/api/search-graph", {
-                collection: `${ctx.collectionPrefix}codebase`,
-                query,
-                limit,
-                expandHops,
-                mode: "navigate",
-            });
-            const { results, graphExpanded, expandedFiles } = response.data;
-            if ((!results || results.length === 0) && (!graphExpanded || graphExpanded.length === 0)) {
-                return "No results found.";
-            }
-            let output = "";
-            if (results && results.length > 0) {
-                output += "**Direct matches:**\n\n";
-                output += formatNavigationResults(results);
-            }
-            if (graphExpanded && graphExpanded.length > 0) {
-                output += "\n\n---\n\n**Graph-connected files:**\n\n";
-                output += formatNavigationResults(graphExpanded);
-            }
-            if (expandedFiles && expandedFiles.length > 0) {
-                output += `\n\n_Graph expanded to ${expandedFiles.length} additional files._`;
-            }
-            return output;
-        },
-        get_project_stats: async (_args, ctx) => {
-            const response = await ctx.api.get(`/api/stats/${ctx.collectionPrefix}codebase`);
-            const stats = response.data;
-            let result = `**${ctx.projectName} Project Statistics**\n\n`;
-            result += `- Total Files: ${stats.totalFiles}\n`;
-            result += `- Total Lines: ${stats.totalLines?.toLocaleString() || "N/A"}\n`;
-            result += `- Vector Count: ${stats.vectorCount}\n`;
-            result += `- Last Indexed: ${stats.lastIndexed ? new Date(stats.lastIndexed).toLocaleString() : "Never"}\n`;
-            if (stats.languages) {
-                result += `\n**Languages:**\n`;
-                for (const [lang, count] of Object.entries(stats.languages)) {
-                    result += `- ${lang}: ${count} files\n`;
-                }
-            }
-            return result;
-        },
-    };
-    return { tools, handlers };
 }

package/dist/tools/session.d.ts

@@ -2,9 +2,9 @@
  * Session tools module - context summarization, session lifecycle management,
  * change tracking, and usage pattern analysis.
  */
-import type { ToolModule, ToolContext } from "../types.js";
+import type { ToolSpec, ToolContext } from "../types.js";
 /**
  * Create the session tools module with project-specific descriptions.
  * Accepts a mutable ctx reference to update activeSessionId on start/end.
  */
-export declare function createSessionTools(projectName: string, sharedCtx?: ToolContext): ToolModule;
+export declare function createSessionTools(projectName: string, sharedCtx?: ToolContext): ToolSpec[];