@crowley/rag-mcp 1.0.5 → 1.1.0

package/dist/annotations.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Tool annotations map for all MCP tools.
+ *
+ * Annotations help clients understand tool behaviour:
+ *   readOnlyHint   – tool does NOT modify its environment
+ *   destructiveHint – tool may delete or irreversibly change data
+ *   idempotentHint  – repeated calls with same args have no extra effect
+ *   openWorldHint   – tool interacts with external entities (RAG API)
+ */
+export interface ToolAnnotations {
+    readOnlyHint?: boolean;
+    destructiveHint?: boolean;
+    idempotentHint?: boolean;
+    openWorldHint?: boolean;
+}
+export declare const TOOL_ANNOTATIONS: Record<string, ToolAnnotations>;

package/dist/annotations.js

@@ -0,0 +1,158 @@
+/**
+ * Tool annotations map for all MCP tools.
+ *
+ * Annotations help clients understand tool behaviour:
+ *   readOnlyHint   – tool does NOT modify its environment
+ *   destructiveHint – tool may delete or irreversibly change data
+ *   idempotentHint  – repeated calls with same args have no extra effect
+ *   openWorldHint   – tool interacts with external entities (RAG API)
+ */
+// ── Shorthand presets ───────────────────────────────────────
+/** Read-only, idempotent – the vast majority of tools */
+const RO = {
+    readOnlyHint: true,
+    destructiveHint: false,
+    idempotentHint: true,
+    openWorldHint: true,
+};
+/** Mutating but non-destructive (create / append) */
+const MUT = {
+    readOnlyHint: false,
+    destructiveHint: false,
+    idempotentHint: false,
+    openWorldHint: true,
+};
+/** Idempotent upsert (record_*, setup_project, validate, promote) */
+const UPSERT = {
+    readOnlyHint: false,
+    destructiveHint: false,
+    idempotentHint: true,
+    openWorldHint: true,
+};
+/** Destructive – may delete or irreversibly change data */
+const DESTRUCT = {
+    readOnlyHint: false,
+    destructiveHint: true,
+    idempotentHint: false,
+    openWorldHint: true,
+};
+// ── Per-tool annotations ────────────────────────────────────
+export const TOOL_ANNOTATIONS = {
+    // ── search (8) ──────────────────────────────────────────
+    search_codebase: RO,
+    search_similar: RO,
+    grouped_search: RO,
+    hybrid_search: RO,
+    search_docs: RO,
+    get_project_stats: RO,
+    find_symbol: RO,
+    search_graph: RO,
+    // ── ask (5) ─────────────────────────────────────────────
+    ask_codebase: RO,
+    explain_code: RO,
+    find_feature: RO,
+    analyze_conversation: RO,
+    auto_remember: MUT, // creates memory from conversation
+    // ── indexing (4) ────────────────────────────────────────
+    index_codebase: MUT,
+    get_index_status: RO,
+    reindex_zero_downtime: MUT,
+    list_aliases: RO,
+    // ── memory (11) ────────────────────────────────────────
+    remember: MUT,
+    recall: RO,
+    list_memories: RO,
+    forget: DESTRUCT,
+    update_todo: MUT,
+    batch_remember: MUT,
+    validate_memory: UPSERT, // idempotent status update
+    review_memories: RO,
+    promote_memory: UPSERT, // idempotent promotion
+    run_quality_gates: MUT, // may update memory status
+    memory_maintenance: DESTRUCT, // prunes old memories
+    // ── architecture (9) ───────────────────────────────────
+    record_adr: UPSERT,
+    get_adrs: RO,
+    record_pattern: UPSERT,
+    get_patterns: RO,
+    check_architecture: RO,
+    suggest_architecture: RO,
+    record_tech_debt: UPSERT,
+    get_tech_debt: RO,
+    analyze_project_structure: RO,
+    // ── database (8) ───────────────────────────────────────
+    record_table: UPSERT,
+    get_table_info: RO,
+    record_db_rule: UPSERT,
+    get_db_rules: RO,
+    record_enum: UPSERT,
+    get_enums: RO,
+    check_db_schema: RO,
+    suggest_db_schema: RO,
+    // ── confluence (4) ─────────────────────────────────────
+    search_confluence: RO,
+    index_confluence: MUT,
+    get_confluence_status: RO,
+    list_confluence_spaces: RO,
+    // ── pm (7) ─────────────────────────────────────────────
+    search_requirements: RO,
+    analyze_requirements: RO,
+    estimate_feature: RO,
+    get_feature_status: RO,
+    list_requirements: RO,
+    ask_pm: RO,
+    generate_spec: RO,
+    // ── review (3) ─────────────────────────────────────────
+    review_code: RO, // generates review, no side-effects
+    generate_tests: RO, // generates test suggestions
+    analyze_tests: RO,
+    // ── analytics (8) ──────────────────────────────────────
+    get_tool_analytics: RO,
+    get_knowledge_gaps: RO,
+    get_analytics: RO,
+    backup_collection: MUT, // creates a backup snapshot
+    list_backups: RO,
+    enable_quantization: DESTRUCT, // irreversible vector re-encoding
+    get_platform_stats: RO,
+    get_prediction_stats: RO,
+    // ── clustering (4) ─────────────────────────────────────
+    cluster_code: RO, // computes clusters, no persistence
+    find_duplicates: RO,
+    recommend_similar: RO,
+    extract_learnings: RO,
+    // ── session (7) ────────────────────────────────────────
+    summarize_context: RO,
+    summarize_changes: RO,
+    analyze_usage_patterns: RO,
+    get_developer_profile: RO,
+    start_session: MUT,
+    get_session_context: RO,
+    end_session: MUT,
+    // ── feedback (4) ───────────────────────────────────────
+    feedback_search: MUT, // stores search feedback
+    feedback_memory: MUT, // stores memory feedback
+    suggest_better_query: RO,
+    get_quality_metrics: RO,
+    // ── suggestions (7) ────────────────────────────────────
+    context_briefing: RO,
+    get_contextual_suggestions: RO,
+    suggest_related_code: RO,
+    suggest_implementation: RO,
+    suggest_tests: RO,
+    get_code_context: RO,
+    setup_project: UPSERT, // writes config (idempotent)
+    // ── cache (2) ──────────────────────────────────────────
+    get_cache_stats: RO,
+    warm_cache: MUT,
+    // ── guidelines (1) ─────────────────────────────────────
+    get_rag_guidelines: RO,
+    // ── advanced (5) ───────────────────────────────────────
+    merge_memories: DESTRUCT, // merges → deletes originals
+    get_completion_context: RO,
+    get_import_suggestions: RO,
+    get_type_context: RO,
+    get_behavior_patterns: RO,
+    // ── agents (2) ─────────────────────────────────────────
+    run_agent: MUT, // side-effects depend on agent type
+    get_agent_types: RO,
+};

package/dist/context-enrichment.js

@@ -101,11 +101,18 @@ export class ContextEnricher {
                 !result.includes("No results") &&
                 !result.includes("not found") &&
                 !result.includes("No relevant context found")) {
+                // Count approximate results for weighted feedback
+                const numbered = result.match(/^\d+\./gm);
+                const bullets = result.match(/^[-*] /gm);
+                const resultCount = numbered?.length ?? bullets?.length ?? 1;
                 ctx.api
                     .post("/api/feedback/search", {
                     projectName: ctx.projectName,
                     query,
                     feedbackType: "helpful",
+                    toolName: name,
+                    resultCount,
+                    sessionId: ctx.activeSessionId,
                 })
                     .catch(() => { });
             }

package/dist/formatters.d.ts

@@ -39,6 +39,8 @@ export declare function formatNavigationResults(results: Array<{
     score: number;
     graphExpanded?: boolean;
 }>): string;
+/** Format pagination footer for list tools */
+export declare function paginationFooter(count: number, limit: number, offset: number): string;
 /** Format a simple list of files with scores */
 export declare function formatFileList(files: Array<{
     file: string;

package/dist/formatters.js

@@ -80,6 +80,18 @@ export function formatNavigationResults(results) {
         return out;
     }).join('\n\n');
 }
+/** Format pagination footer for list tools */
+export function paginationFooter(count, limit, offset) {
+    const hasMore = count >= limit;
+    if (!hasMore && offset === 0)
+        return "";
+    let footer = `\n---\n_Showing ${offset + 1}–${offset + count}`;
+    if (hasMore) {
+        footer += ` | More results available (offset: ${offset + limit})`;
+    }
+    footer += "_\n";
+    return footer;
+}
 /** Format a simple list of files with scores */
 export function formatFileList(files, emptyMessage = "No files found.") {
     if (!files || files.length === 0)

package/dist/index.js

@@ -10,12 +10,11 @@
  * - PROJECT_PATH: Path to project codebase for indexing
  * - RAG_API_URL: URL of the shared RAG API (default: http://localhost:3100)
  */
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
 import { createApiClient } from "./api-client.js";
-import { ToolRegistry } from "./tool-registry.js";
 import { ContextEnricher } from "./context-enrichment.js";
+import { wrapHandler } from "./tool-middleware.js";
 // Tool modules
 import { createSearchTools } from "./tools/search.js";
 import { createAskTools } from "./tools/ask.js";
@@ -35,6 +34,7 @@ import { createCacheTools } from "./tools/cache.js";
 import { createGuidelinesTools } from "./tools/guidelines.js";
 import { createAdvancedTools } from "./tools/advanced.js";
 import { createAgentTools } from "./tools/agents.js";
+import { createQualityTools } from "./tools/quality.js";
 // Configuration from environment
 const PROJECT_NAME = process.env.PROJECT_NAME || "default";
 const PROJECT_PATH = process.env.PROJECT_PATH || process.cwd();
@@ -51,54 +51,106 @@ const ctx = {
     collectionPrefix: COLLECTION_PREFIX,
     enrichmentEnabled: true,
 };
-// Build tool registry from modules
-const registry = new ToolRegistry();
-registry.register(createSearchTools(PROJECT_NAME));
-registry.register(createAskTools(PROJECT_NAME));
-registry.register(createIndexingTools(PROJECT_NAME));
-registry.register(createMemoryTools(PROJECT_NAME));
-registry.register(createArchitectureTools(PROJECT_NAME));
-registry.register(createDatabaseTools(PROJECT_NAME));
-registry.register(createConfluenceTools(PROJECT_NAME));
-registry.register(createPmTools(PROJECT_NAME));
-registry.register(createReviewTools(PROJECT_NAME));
-registry.register(createAnalyticsTools(PROJECT_NAME));
-registry.register(createClusteringTools(PROJECT_NAME));
-registry.register(createSessionTools(PROJECT_NAME, ctx));
-registry.register(createFeedbackTools(PROJECT_NAME));
-registry.register(createSuggestionTools(PROJECT_NAME));
-registry.register(createCacheTools(PROJECT_NAME));
-registry.register(createGuidelinesTools(PROJECT_NAME));
-registry.register(createAdvancedTools(PROJECT_NAME));
-registry.register(createAgentTools(PROJECT_NAME));
-// Initialize context enrichment middleware
+// Context enrichment middleware
 const enricher = new ContextEnricher({
     maxAutoRecall: 3,
     minRelevance: 0.6,
     timeoutMs: 2000,
 });
-registry.setEnricher(enricher);
-// MCP Server
-const server = new Server({
-    name: `${PROJECT_NAME}-rag`,
-    version: "1.0.0",
-}, {
-    capabilities: {
-        tools: {},
-    },
-});
-// List tools handler
-server.setRequestHandler(ListToolsRequestSchema, async () => ({
-    tools: registry.getTools(),
-}));
-// Call tool handler
-server.setRequestHandler(CallToolRequestSchema, async (request) => {
-    const { name, arguments: args } = request.params;
-    const result = await registry.handle(name, args || {}, ctx);
-    return {
-        content: [{ type: "text", text: result }],
-    };
-});
+// Collect all tool specs from modules
+const allSpecs = [
+    ...createSearchTools(PROJECT_NAME),
+    ...createAskTools(PROJECT_NAME),
+    ...createIndexingTools(PROJECT_NAME),
+    ...createMemoryTools(PROJECT_NAME),
+    ...createArchitectureTools(PROJECT_NAME),
+    ...createDatabaseTools(PROJECT_NAME),
+    ...createConfluenceTools(PROJECT_NAME),
+    ...createPmTools(PROJECT_NAME),
+    ...createReviewTools(PROJECT_NAME),
+    ...createAnalyticsTools(PROJECT_NAME),
+    ...createClusteringTools(PROJECT_NAME),
+    ...createSessionTools(PROJECT_NAME, ctx),
+    ...createFeedbackTools(PROJECT_NAME),
+    ...createSuggestionTools(PROJECT_NAME),
+    ...createCacheTools(PROJECT_NAME),
+    ...createGuidelinesTools(PROJECT_NAME),
+    ...createAdvancedTools(PROJECT_NAME),
+    ...createAgentTools(PROJECT_NAME),
+    ...createQualityTools(PROJECT_NAME),
+];
+// Core tools exposed directly to Claude (~35 tools).
+// Hidden tools remain accessible via run_agent (agent runtime calls API directly).
+const CORE_TOOLS = new Set([
+    // Search (6)
+    "search_codebase",
+    "hybrid_search",
+    "search_graph",
+    "find_symbol",
+    "search_docs",
+    "find_feature",
+    // Ask (2)
+    "ask_codebase",
+    "explain_code",
+    // Index (3)
+    "index_codebase",
+    "get_index_status",
+    "get_project_stats",
+    // Memory (7)
+    "remember",
+    "recall",
+    "list_memories",
+    "forget",
+    "batch_remember",
+    "promote_memory",
+    "review_memories",
+    // Architecture (6)
+    "record_adr",
+    "get_adrs",
+    "record_pattern",
+    "get_patterns",
+    "record_tech_debt",
+    "get_tech_debt",
+    // Context (3)
+    "context_briefing",
+    "smart_dispatch",
+    "setup_project",
+    // Session (2)
+    "start_session",
+    "end_session",
+    // Confluence (2)
+    "search_confluence",
+    "index_confluence",
+    // DB (4)
+    "record_table",
+    "get_table_info",
+    "check_db_schema",
+    "get_db_rules",
+    // Agents (1)
+    "run_agent",
+]);
+const coreSpecs = allSpecs.filter((s) => CORE_TOOLS.has(s.name));
+// MCP Server (modern McpServer API with native Zod validation)
+const server = new McpServer({ name: `${PROJECT_NAME}-rag`, version: "1.1.0" }, { capabilities: { tools: {} } });
+// Register core tools with McpServer using wrapHandler middleware
+for (const spec of coreSpecs) {
+    const wrapped = wrapHandler(spec.name, spec.handler, { enricher, ctx });
+    server.registerTool(spec.name, {
+        description: spec.description,
+        inputSchema: spec.schema,
+        ...(spec.outputSchema ? { outputSchema: spec.outputSchema } : {}),
+        annotations: spec.annotations,
+    }, async (args) => {
+        const result = await wrapped(args, ctx);
+        if (typeof result === "string") {
+            return { content: [{ type: "text", text: result }] };
+        }
+        return {
+            content: [{ type: "text", text: result.text }],
+            structuredContent: result.structured,
+        };
+    });
+}
 // Graceful shutdown: close active session on exit
 async function cleanup() {
     if (ctx.activeSessionId) {
@@ -122,6 +174,6 @@ async function main() {
     const transport = new StdioServerTransport();
     await server.connect(transport);
     console.error(`${PROJECT_NAME} RAG MCP server running (collection prefix: ${COLLECTION_PREFIX})`);
-    console.error(`Registered ${registry.getTools().length} tools from 18 modules`);
+    console.error(`Registered ${coreSpecs.length}/${allSpecs.length} core tools (${allSpecs.length - coreSpecs.length} hidden, accessible via run_agent)`);
 }
 main().catch(console.error);

package/dist/schemas.d.ts

@@ -0,0 +1,97 @@
+/**
+ * Common Zod schemas shared across tool modules.
+ *
+ * Import individual shapes into each tool module when migrating
+ * from raw JSON Schema objects to Zod-based inputSchema definitions.
+ */
+import { z } from "zod";
+import type { ToolInputSchema } from "./types.js";
+/**
+ * Convert a Zod object schema to the MCP ToolInputSchema format.
+ * Used during Phase 2 migration while ToolRegistry still expects raw JSON Schema.
+ * Phase 3 passes Zod schemas directly to McpServer.registerTool().
+ */
+export declare function zodToInputSchema(schema: z.ZodObject<z.ZodRawShape>): ToolInputSchema;
+export declare const QueryStr: z.ZodString;
+export declare const Limit: z.ZodDefault<z.ZodNumber>;
+export declare const Offset: z.ZodDefault<z.ZodNumber>;
+export declare const FilePath: z.ZodString;
+export declare const FilePaths: z.ZodArray<z.ZodString, "many">;
+export declare const Content: z.ZodString;
+export declare const CollectionSuffix: z.ZodString;
+export declare const MemoryType: z.ZodEnum<["decision", "insight", "pattern", "adr", "tech_debt", "todo", "architecture", "convention", "bug_fix", "optimization"]>;
+export declare const ResponseFormat: z.ZodDefault<z.ZodEnum<["json", "markdown"]>>;
+export declare const Importance: z.ZodDefault<z.ZodEnum<["low", "medium", "high", "critical"]>>;
+export declare const Priority: z.ZodEnum<["low", "medium", "high", "critical"]>;
+export declare const Severity: z.ZodEnum<["low", "medium", "high", "critical"]>;
+export declare const PaginationParams: z.ZodObject<{
+    limit: z.ZodOptional<z.ZodDefault<z.ZodNumber>>;
+    offset: z.ZodOptional<z.ZodDefault<z.ZodNumber>>;
+}, "strip", z.ZodTypeAny, {
+    limit?: number | undefined;
+    offset?: number | undefined;
+}, {
+    limit?: number | undefined;
+    offset?: number | undefined;
+}>;
+export declare const SearchFilters: z.ZodOptional<z.ZodObject<{
+    file_type: z.ZodOptional<z.ZodString>;
+    directory: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    file_type?: string | undefined;
+    directory?: string | undefined;
+}, {
+    file_type?: string | undefined;
+    directory?: string | undefined;
+}>>;
+export declare const Tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+export declare const Confidence: z.ZodOptional<z.ZodNumber>;
+/** Base shape for search tools: query + optional limit + optional filters */
+export declare const SearchInput: z.ZodObject<{
+    query: z.ZodString;
+    limit: z.ZodOptional<z.ZodDefault<z.ZodNumber>>;
+    filters: z.ZodOptional<z.ZodObject<{
+        file_type: z.ZodOptional<z.ZodString>;
+        directory: z.ZodOptional<z.ZodString>;
+    }, "strip", z.ZodTypeAny, {
+        file_type?: string | undefined;
+        directory?: string | undefined;
+    }, {
+        file_type?: string | undefined;
+        directory?: string | undefined;
+    }>>;
+}, "strip", z.ZodTypeAny, {
+    query: string;
+    limit?: number | undefined;
+    filters?: {
+        file_type?: string | undefined;
+        directory?: string | undefined;
+    } | undefined;
+}, {
+    query: string;
+    limit?: number | undefined;
+    filters?: {
+        file_type?: string | undefined;
+        directory?: string | undefined;
+    } | undefined;
+}>;
+/** Base shape for memory record tools */
+export declare const MemoryRecordInput: z.ZodObject<{
+    content: z.ZodString;
+    type: z.ZodEnum<["decision", "insight", "pattern", "adr", "tech_debt", "todo", "architecture", "convention", "bug_fix", "optimization"]>;
+    tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    importance: z.ZodOptional<z.ZodDefault<z.ZodEnum<["low", "medium", "high", "critical"]>>>;
+    context: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    type: "decision" | "insight" | "todo" | "adr" | "pattern" | "architecture" | "tech_debt" | "convention" | "bug_fix" | "optimization";
+    content: string;
+    context?: string | undefined;
+    tags?: string[] | undefined;
+    importance?: "low" | "medium" | "high" | "critical" | undefined;
+}, {
+    type: "decision" | "insight" | "todo" | "adr" | "pattern" | "architecture" | "tech_debt" | "convention" | "bug_fix" | "optimization";
+    content: string;
+    context?: string | undefined;
+    tags?: string[] | undefined;
+    importance?: "low" | "medium" | "high" | "critical" | undefined;
+}>;

package/dist/schemas.js

@@ -0,0 +1,128 @@
+/**
+ * Common Zod schemas shared across tool modules.
+ *
+ * Import individual shapes into each tool module when migrating
+ * from raw JSON Schema objects to Zod-based inputSchema definitions.
+ */
+import { z } from "zod";
+import { zodToJsonSchema } from "zod-to-json-schema";
+// ── JSON Schema conversion ──────────────────────────────────
+/**
+ * Convert a Zod object schema to the MCP ToolInputSchema format.
+ * Used during Phase 2 migration while ToolRegistry still expects raw JSON Schema.
+ * Phase 3 passes Zod schemas directly to McpServer.registerTool().
+ */
+export function zodToInputSchema(schema) {
+    const jsonSchema = zodToJsonSchema(schema, { target: "openApi3" });
+    return {
+        type: "object",
+        properties: (jsonSchema.properties ?? {}),
+        ...(Array.isArray(jsonSchema.required) && jsonSchema.required.length > 0
+            ? { required: jsonSchema.required }
+            : {}),
+    };
+}
+// ── Primitives ──────────────────────────────────────────────
+export const QueryStr = z
+    .string()
+    .min(1)
+    .describe("Search query or question");
+export const Limit = z
+    .number()
+    .int()
+    .min(1)
+    .max(100)
+    .default(20)
+    .describe("Maximum results to return");
+export const Offset = z
+    .number()
+    .int()
+    .min(0)
+    .default(0)
+    .describe("Pagination offset");
+export const FilePath = z
+    .string()
+    .min(1)
+    .describe("File path relative to project root");
+export const FilePaths = z
+    .array(FilePath)
+    .min(1)
+    .describe("List of file paths");
+export const Content = z
+    .string()
+    .min(1)
+    .describe("Text content");
+export const CollectionSuffix = z
+    .string()
+    .min(1)
+    .describe("Collection suffix (e.g. 'codebase', 'docs')");
+// ── Enums ───────────────────────────────────────────────────
+export const MemoryType = z.enum([
+    "decision",
+    "insight",
+    "pattern",
+    "adr",
+    "tech_debt",
+    "todo",
+    "architecture",
+    "convention",
+    "bug_fix",
+    "optimization",
+]);
+export const ResponseFormat = z
+    .enum(["json", "markdown"])
+    .default("markdown")
+    .describe("Output format");
+export const Importance = z
+    .enum(["low", "medium", "high", "critical"])
+    .default("medium")
+    .describe("Importance level");
+export const Priority = z
+    .enum(["low", "medium", "high", "critical"])
+    .describe("Priority level");
+export const Severity = z
+    .enum(["low", "medium", "high", "critical"])
+    .describe("Severity level");
+// ── Reusable object shapes ──────────────────────────────────
+export const PaginationParams = z.object({
+    limit: Limit.optional(),
+    offset: Offset.optional(),
+});
+export const SearchFilters = z
+    .object({
+    file_type: z
+        .string()
+        .optional()
+        .describe("Filter by file extension (e.g. 'ts', 'py')"),
+    directory: z
+        .string()
+        .optional()
+        .describe("Filter by directory prefix"),
+})
+    .optional()
+    .describe("Search filters");
+export const Tags = z
+    .array(z.string())
+    .optional()
+    .describe("Tags for categorization");
+export const Confidence = z
+    .number()
+    .min(0)
+    .max(1)
+    .optional()
+    .describe("Confidence score (0-1)");
+// ── Composite input shapes ──────────────────────────────────
+/** Base shape for search tools: query + optional limit + optional filters */
+export const SearchInput = z.object({
+    query: QueryStr,
+    limit: Limit.optional(),
+    filters: SearchFilters,
+});
+/** Base shape for memory record tools */
+export const MemoryRecordInput = z.object({
+    content: Content.describe("Content to remember"),
+    type: MemoryType,
+    tags: Tags,
+    importance: Importance.optional(),
+    context: z.string().optional().describe("Additional context"),
+});

package/dist/tool-middleware.d.ts

@@ -0,0 +1,40 @@
+/**
+ * Tool Middleware — standalone middleware functions for MCP tool handlers.
+ *
+ * Extracted from ToolRegistry.handle() so that McpServer.registerTool()
+ * (Phase 3) can reuse the same pipeline:
+ *   auto-session → enrichment.before → handler → enrichment.after → trackUsage
+ *
+ * During Phase 2 migration, ToolRegistry continues to use its own copy.
+ * Phase 3 replaces ToolRegistry with wrapHandler() + McpServer.registerTool().
+ */
+import type { ToolContext, ToolHandler } from "./types.js";
+import type { ContextEnricher } from "./context-enrichment.js";
+/** Per-tool timeout overrides (ms) */
+export declare const TOOL_TIMEOUTS: Record<string, number>;
+/** Tools excluded from usage tracking (meta/admin, avoid recursion) */
+export declare const TRACKING_EXCLUDE: Set<string>;
+/** Session management tools — skip auto-session to avoid recursion */
+export declare const SESSION_TOOLS: Set<string>;
+/** Summarize tool args into a short string for analytics */
+export declare function summarizeInput(name: string, args: Record<string, unknown>): string;
+/** Count results from a tool response string */
+export declare function countResults(result: string): number;
+/** Auto-start a session via the RAG API if none is active */
+export declare function ensureSession(ctx: ToolContext): Promise<void>;
+/** Fire-and-forget usage tracking */
+export declare function trackUsage(name: string, args: Record<string, unknown>, startTime: number, success: boolean, result: string, errorMessage: string | undefined, ctx: ToolContext): void;
+/** Format an error caught during tool execution */
+export declare function formatToolError(error: unknown, ctx: ToolContext): string;
+export interface MiddlewareDeps {
+    enricher?: ContextEnricher;
+    ctx: ToolContext;
+}
+/**
+ * Wrap a raw ToolHandler with the full middleware pipeline:
+ * auto-session → enrichment.before → handler → enrichment.after → trackUsage
+ *
+ * Returns a ToolHandler with the same signature, so it works both
+ * with the legacy ToolRegistry and the Phase 3 McpServer adapter.
+ */
+export declare function wrapHandler(name: string, handler: ToolHandler, deps: MiddlewareDeps): ToolHandler;