@nxuss/lemma 1.1.0 → 1.2.0

package/dist/esm/mcp/tools.js

@@ -30,6 +30,7 @@ import { getSymbolSurgicalContext } from "../utils/SymbolSurgicalContext";
 import { compressWormhole } from "../utils/WormholeCompressor";
 import { generateExecutiveReport } from "../utils/ExecutiveReport";
 import { surgicalASTInsert } from "../utils/SurgicalASTInsert";
+import { getInfraToolDefinitions, getInfraToolHandlers, INFRA_TOOL_NAMES } from "../infra/mcp-tools";
 // ── Pro License Gate ──────────────────────────────────────────────────────────
 const LICENSE_FILE = path.join(os.homedir(), '.lemma-cache', 'license.json');
 let _proStatusCache = null;
@@ -60,6 +61,8 @@ const FREE_TOOLS = new Set([
     "read_token_budgeted", "bulk_file_digest", "auto_context_bundle", "repeat_guard",
     // Phase 3 session & automation tools
     "conversation_checkpoint", "test_autofix_interceptor", "prompt_pattern_cache",
+    // Infrastructure tools
+    ...INFRA_TOOL_NAMES,
 ]);
 const PRO_GATE_MESSAGE = [
     "🔒 This tool requires Lemma Pro.",
@@ -930,6 +933,10 @@ const toolDefinitions = [
         }
     }
 ];
+// ── Infrastructure Tool Definitions ──────────────────────────────────────────
+const _infraToolDefs = getInfraToolDefinitions();
+const allToolDefinitions = [...toolDefinitions, ..._infraToolDefs];
+const toolDefinitionsArray = _infraToolDefs.length > 0 ? allToolDefinitions : toolDefinitions;
 const toolHandlers = {
     scrub_privacy: handleScrubPrivacy,
     search_memory: handleSearchMemory,
@@ -999,10 +1006,11 @@ const toolHandlers = {
     conversation_checkpoint: handleConversationCheckpoint,
     test_autofix_interceptor: handleTestAutofixInterceptor,
     prompt_pattern_cache: handlePromptPatternCache,
+    ...getInfraToolHandlers(),
 };
 export function setupToolsHandlers(server, onToolCall) {
     server.setRequestHandler(ListToolsRequestSchema, async () => ({
-        tools: toolDefinitions,
+        tools: toolDefinitionsArray,
     }));
     server.setRequestHandler(CallToolRequestSchema, async (request) => {
         const { name, arguments: args } = request.params;