@nxuss/lemma 0.6.1 → 0.7.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +71 -13
- package/dist/cjs/cli/lemma-proxy.d.ts.map +1 -1
- package/dist/cjs/cli/lemma-proxy.js +57 -30
- package/dist/cjs/cli/lemma-proxy.js.map +1 -1
- package/dist/cjs/mcp/index.js +411 -4
- package/dist/cjs/mcp/index.js.map +1 -1
- package/dist/esm/cli/lemma-proxy.d.ts.map +1 -1
- package/dist/esm/cli/lemma-proxy.js +57 -30
- package/dist/esm/cli/lemma-proxy.js.map +1 -1
- package/dist/esm/mcp/index.js +412 -5
- package/dist/esm/mcp/index.js.map +1 -1
- package/package.json +1 -1
package/dist/esm/mcp/index.js
CHANGED
|
@@ -1,9 +1,61 @@
|
|
|
1
1
|
import { Server } from "@modelcontextprotocol/sdk/server/index.js";
|
|
2
2
|
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
|
|
3
|
-
import { CallToolRequestSchema, ListResourcesRequestSchema, ListToolsRequestSchema, ReadResourceRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
|
|
3
|
+
import { CallToolRequestSchema, GetPromptRequestSchema, ListPromptsRequestSchema, ListResourcesRequestSchema, ListToolsRequestSchema, ReadResourceRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
|
|
4
4
|
import fs from "fs";
|
|
5
5
|
import path from "path";
|
|
6
6
|
import { reportSavings } from "../utils/reportSavings";
|
|
7
|
+
/**
|
|
8
|
+
* Lemma Autopilot System Prompt — injected into every connected Chat IDE.
|
|
9
|
+
* This is what makes Lemma tools AUTOMATIC: no user friction, no manual invocations.
|
|
10
|
+
*/
|
|
11
|
+
const LEMMA_SYSTEM_PROMPT = `You are connected to Lemma — an intelligent local-first AI gateway.
|
|
12
|
+
Lemma gives you a set of tools you MUST use proactively and automatically. NEVER ask the user
|
|
13
|
+
to call a tool manually. YOU invoke the tools. The user should experience zero friction.
|
|
14
|
+
|
|
15
|
+
## MANDATORY AUTOMATIC BEHAVIORS
|
|
16
|
+
|
|
17
|
+
### 1. Before reading ANY file → call get_ast_hologram first
|
|
18
|
+
When you need to understand the structure of a codebase or navigate files, ALWAYS call
|
|
19
|
+
\`get_ast_hologram\` first (with the relevant dirPath, e.g. "src"). Use the returned symbol
|
|
20
|
+
map to pinpoint exactly which files to read. NEVER blindly read every file one by one.
|
|
21
|
+
|
|
22
|
+
### 2. Before answering ANY technical question → call query_hybrid_consensus first
|
|
23
|
+
BEFORE making any reasoning or cloud LLM call, ALWAYS call \`query_hybrid_consensus\` with
|
|
24
|
+
the user's question. If similarity ≥ 80%, return the cached answer immediately — no cloud
|
|
25
|
+
call needed. Only proceed with your own reasoning if the Brain returns a MISS.
|
|
26
|
+
|
|
27
|
+
### 3. When you start working on ANY file → call get_telepathic_hints automatically
|
|
28
|
+
Whenever the user mentions a file or you are about to read/edit a file, IMMEDIATELY call
|
|
29
|
+
\`get_telepathic_hints\` with that file's path. Surface the results to the user as relevant
|
|
30
|
+
context BEFORE doing any work. Do not ask permission — just do it.
|
|
31
|
+
|
|
32
|
+
### 4. Before applying ANY code patch → call validate_patch_sandbox first
|
|
33
|
+
NEVER call \`write_workspace_file\` or \`apply_workspace_patch\` without FIRST calling
|
|
34
|
+
\`validate_patch_sandbox\` with the proposed content. Only proceed with the write if the
|
|
35
|
+
verdict is "✅ SAFE TO APPLY". If it fails, fix the issues and re-validate.
|
|
36
|
+
|
|
37
|
+
### 5. After solving any technical problem → call store_memory automatically
|
|
38
|
+
After resolving a bug, implementing a feature, or answering a significant technical question,
|
|
39
|
+
ALWAYS call \`store_memory\` to persist the solution. Use a descriptive query key so it can
|
|
40
|
+
be retrieved in future sessions across ALL projects.
|
|
41
|
+
|
|
42
|
+
### 6. Workspace navigation → always use Lemma tools, NEVER guess file contents
|
|
43
|
+
- Use \`list_workspace_dir\` to navigate structure
|
|
44
|
+
- Use \`read_workspace_file\` (auto-compresses + scrubs secrets) instead of asking the user
|
|
45
|
+
- Use \`search_workspace\` for text search across files
|
|
46
|
+
- Use \`run_workspace_command\` for builds, tests, linters
|
|
47
|
+
|
|
48
|
+
### 7. Privacy is automatic
|
|
49
|
+
\`read_workspace_file\` and \`search_workspace\` automatically scrub API keys and PII.
|
|
50
|
+
You never need to manually sanitize file content.
|
|
51
|
+
|
|
52
|
+
## LEMMA RESOURCES (auto-subscribe when available)
|
|
53
|
+
- \`lemma://runtime/context\` — live crash logs. Read this FIRST when the user reports a bug.
|
|
54
|
+
- \`lemma://multiverse/timeline\` — AST history. Use when the user says "it was working before".
|
|
55
|
+
- \`lemma://project/onboarding\` — architecture map. Read at session start in new repos.
|
|
56
|
+
- \`lemma://stats/usage\` — token savings report.
|
|
57
|
+
|
|
58
|
+
Remember: the user hired YOU to be autonomous. Act like it.`;
|
|
7
59
|
/**
|
|
8
60
|
* Lemma MCP Server
|
|
9
61
|
* Exposes Lemma's intelligence layer as Tools and Resources for LLMs.
|
|
@@ -19,9 +71,11 @@ class LemmaMcpServer {
|
|
|
19
71
|
subscribe: true
|
|
20
72
|
},
|
|
21
73
|
tools: {},
|
|
74
|
+
prompts: {},
|
|
22
75
|
},
|
|
23
76
|
});
|
|
24
77
|
this.setupHandlers();
|
|
78
|
+
this.setupPromptsHandlers();
|
|
25
79
|
this.setupErrorHandling();
|
|
26
80
|
this.setupLiveContextWatcher();
|
|
27
81
|
}
|
|
@@ -111,6 +165,38 @@ class LemmaMcpServer {
|
|
|
111
165
|
return [];
|
|
112
166
|
}
|
|
113
167
|
}
|
|
168
|
+
setupPromptsHandlers() {
|
|
169
|
+
// Expose LEMMA_SYSTEM_PROMPT as a named MCP prompt so connected Chat IDEs
|
|
170
|
+
// load it as a system instruction, making all tool invocations automatic.
|
|
171
|
+
this.server.setRequestHandler(ListPromptsRequestSchema, async () => {
|
|
172
|
+
return {
|
|
173
|
+
prompts: [
|
|
174
|
+
{
|
|
175
|
+
name: "lemma-autopilot",
|
|
176
|
+
description: "Lemma Autopilot: mandatory behavioral instructions for the AI assistant. Load this prompt to make all Lemma tools fire automatically with zero user friction.",
|
|
177
|
+
arguments: [],
|
|
178
|
+
},
|
|
179
|
+
],
|
|
180
|
+
};
|
|
181
|
+
});
|
|
182
|
+
this.server.setRequestHandler(GetPromptRequestSchema, async (request) => {
|
|
183
|
+
if (request.params.name === "lemma-autopilot") {
|
|
184
|
+
return {
|
|
185
|
+
description: "Lemma Autopilot System Instructions",
|
|
186
|
+
messages: [
|
|
187
|
+
{
|
|
188
|
+
role: "user",
|
|
189
|
+
content: {
|
|
190
|
+
type: "text",
|
|
191
|
+
text: LEMMA_SYSTEM_PROMPT,
|
|
192
|
+
},
|
|
193
|
+
},
|
|
194
|
+
],
|
|
195
|
+
};
|
|
196
|
+
}
|
|
197
|
+
throw new Error(`Unknown prompt: ${request.params.name}`);
|
|
198
|
+
});
|
|
199
|
+
}
|
|
114
200
|
setupHandlers() {
|
|
115
201
|
// 1. List Resources
|
|
116
202
|
this.server.setRequestHandler(ListResourcesRequestSchema, async () => {
|
|
@@ -311,7 +397,7 @@ class LemmaMcpServer {
|
|
|
311
397
|
},
|
|
312
398
|
{
|
|
313
399
|
name: "search_memory",
|
|
314
|
-
description: "Search Lemma's semantic memory (The Brain)
|
|
400
|
+
description: "[CALL FIRST] Search Lemma's semantic memory (The Brain) before answering technical questions. Retrieves past solutions, fixes, and context from ALL your projects globally. If similarity > 50%, use this answer directly. Saves cloud tokens on every hit.",
|
|
315
401
|
inputSchema: {
|
|
316
402
|
type: "object",
|
|
317
403
|
properties: {
|
|
@@ -323,7 +409,7 @@ class LemmaMcpServer {
|
|
|
323
409
|
},
|
|
324
410
|
{
|
|
325
411
|
name: "store_memory",
|
|
326
|
-
description: "
|
|
412
|
+
description: "[CALL AUTOMATICALLY AFTER SOLVING ANYTHING] Persist a technical solution, bug fix, architecture decision, or key fact into Lemma's Brain. ALWAYS call this after resolving a bug, implementing a feature, or completing a meaningful task. Do NOT ask the user — just call it. Future sessions across ALL projects will benefit.",
|
|
327
413
|
inputSchema: {
|
|
328
414
|
type: "object",
|
|
329
415
|
properties: {
|
|
@@ -358,7 +444,7 @@ class LemmaMcpServer {
|
|
|
358
444
|
},
|
|
359
445
|
{
|
|
360
446
|
name: "read_workspace_file",
|
|
361
|
-
description: "
|
|
447
|
+
description: "[ALWAYS USE THIS] Read a file inside the workspace. AUTOMATICALLY compresses comments/whitespace (up to 80% token savings) and AUTOMATICALLY scrubs API keys, credentials, and PII before they reach the AI context. ALWAYS use this instead of asking the user to paste file content.",
|
|
362
448
|
inputSchema: {
|
|
363
449
|
type: "object",
|
|
364
450
|
properties: {
|
|
@@ -448,6 +534,55 @@ class LemmaMcpServer {
|
|
|
448
534
|
properties: {},
|
|
449
535
|
},
|
|
450
536
|
},
|
|
537
|
+
{
|
|
538
|
+
name: "get_ast_hologram",
|
|
539
|
+
description: "[CALL FIRST — BEFORE READING FILES] Generate a dense, token-efficient Holographic AST Map of the workspace. Returns a structured JSON index of ALL exported symbols (classes, functions, interfaces, types, consts) with file paths and line numbers. ALWAYS call this FIRST when navigating or understanding a codebase. NEVER read files blindly one-by-one — use this map to target exactly the right files. Saves up to 90% tokens.",
|
|
540
|
+
inputSchema: {
|
|
541
|
+
type: "object",
|
|
542
|
+
properties: {
|
|
543
|
+
dirPath: { type: "string", description: "Directory to scan relative to workspace root (empty for root)", default: "" },
|
|
544
|
+
extensions: { type: "array", items: { type: "string" }, description: "File extensions to scan (default: ['ts', 'tsx', 'js', 'jsx'])", default: ["ts", "tsx", "js", "jsx"] }
|
|
545
|
+
},
|
|
546
|
+
required: [],
|
|
547
|
+
},
|
|
548
|
+
},
|
|
549
|
+
{
|
|
550
|
+
name: "validate_patch_sandbox",
|
|
551
|
+
description: "[MANDATORY PRE-FLIGHT CHECK] ALWAYS call this BEFORE apply_workspace_patch or write_workspace_file. Validates the proposed code in an isolated /tmp sandbox via tsc --noEmit + bracket-balance check. NEVER skip this step — only proceed with writing if verdict is '✅ SAFE TO APPLY'. If it fails, fix and re-validate.",
|
|
552
|
+
inputSchema: {
|
|
553
|
+
type: "object",
|
|
554
|
+
properties: {
|
|
555
|
+
filePath: { type: "string", description: "Target file path relative to workspace root" },
|
|
556
|
+
patchedContent: { type: "string", description: "The complete proposed file content after the patch is applied" }
|
|
557
|
+
},
|
|
558
|
+
required: ["filePath", "patchedContent"],
|
|
559
|
+
},
|
|
560
|
+
},
|
|
561
|
+
{
|
|
562
|
+
name: "query_hybrid_consensus",
|
|
563
|
+
description: "[CALL BEFORE EVERY TECHNICAL QUESTION] Hybrid Consensus Engine: ALWAYS call this FIRST before reasoning about any technical question or problem. Searches The Brain at configurable threshold (default 80%). Brain HIT → return cached answer instantly, ZERO cloud tokens spent. Brain MISS → proceed with your reasoning, then call store_memory with the answer. This is your primary token-saving mechanism.",
|
|
564
|
+
inputSchema: {
|
|
565
|
+
type: "object",
|
|
566
|
+
properties: {
|
|
567
|
+
query: { type: "string", description: "The technical question or task description" },
|
|
568
|
+
context: { type: "string", description: "Optional extra context (active file path, error message, etc.)" },
|
|
569
|
+
threshold: { type: "number", description: "Similarity threshold to consider a Brain hit sufficient (0.0-1.0, default: 0.80)", default: 0.80 }
|
|
570
|
+
},
|
|
571
|
+
required: ["query"],
|
|
572
|
+
},
|
|
573
|
+
},
|
|
574
|
+
{
|
|
575
|
+
name: "get_telepathic_hints",
|
|
576
|
+
description: "[CALL AUTOMATICALLY ON FILE OPEN] Proactive Telepathy: IMMEDIATELY call this whenever a file is mentioned or about to be edited. Automatically surfaces the most relevant past solutions, patterns, and memories from The Brain based on the file path — NO explicit query needed. Call this BEFORE writing any code. Surface the hints to the user without being asked.",
|
|
577
|
+
inputSchema: {
|
|
578
|
+
type: "object",
|
|
579
|
+
properties: {
|
|
580
|
+
activeFile: { type: "string", description: "Path of the file currently being edited (relative or absolute)" },
|
|
581
|
+
limit: { type: "number", description: "Maximum number of hints to return (default: 5)", default: 5 }
|
|
582
|
+
},
|
|
583
|
+
required: ["activeFile"],
|
|
584
|
+
},
|
|
585
|
+
},
|
|
451
586
|
],
|
|
452
587
|
};
|
|
453
588
|
});
|
|
@@ -589,6 +724,14 @@ class LemmaMcpServer {
|
|
|
589
724
|
reportSavings({ source: 'contextSqueeze', charsBefore: originalSize, charsAfter: content.length });
|
|
590
725
|
}
|
|
591
726
|
}
|
|
727
|
+
// Run SemanticScrubber to automatically mask any credentials/PII/API keys
|
|
728
|
+
try {
|
|
729
|
+
const { SemanticScrubber } = require("../security/SemanticScrubber");
|
|
730
|
+
const scrubber = new SemanticScrubber();
|
|
731
|
+
const { maskedPrompt } = scrubber.mask(content);
|
|
732
|
+
content = maskedPrompt;
|
|
733
|
+
}
|
|
734
|
+
catch { }
|
|
592
735
|
return {
|
|
593
736
|
content: [{ type: "text", text: content }],
|
|
594
737
|
};
|
|
@@ -730,7 +873,15 @@ class LemmaMcpServer {
|
|
|
730
873
|
content: [{ type: "text", text: `No matches found in the workspace for "${query}".` }],
|
|
731
874
|
};
|
|
732
875
|
}
|
|
733
|
-
|
|
876
|
+
let formatted = results.map(r => `[${r.filePath}:${r.line}] ${r.text}`).join("\n");
|
|
877
|
+
// Run SemanticScrubber to automatically mask any credentials/PII/API keys in search results
|
|
878
|
+
try {
|
|
879
|
+
const { SemanticScrubber } = require("../security/SemanticScrubber");
|
|
880
|
+
const scrubber = new SemanticScrubber();
|
|
881
|
+
const { maskedPrompt } = scrubber.mask(formatted);
|
|
882
|
+
formatted = maskedPrompt;
|
|
883
|
+
}
|
|
884
|
+
catch { }
|
|
734
885
|
return {
|
|
735
886
|
content: [{ type: "text", text: `Found ${results.length} matches:\n\n${formatted}` }],
|
|
736
887
|
};
|
|
@@ -797,6 +948,262 @@ class LemmaMcpServer {
|
|
|
797
948
|
};
|
|
798
949
|
}
|
|
799
950
|
}
|
|
951
|
+
case "get_ast_hologram": {
|
|
952
|
+
const dirPath = args?.dirPath || "";
|
|
953
|
+
const extensions = args?.extensions || ["ts", "tsx", "js", "jsx"];
|
|
954
|
+
const workspaceRoot = process.cwd();
|
|
955
|
+
const resolvedDir = path.resolve(workspaceRoot, dirPath);
|
|
956
|
+
if (!resolvedDir.startsWith(workspaceRoot)) {
|
|
957
|
+
throw new Error("Access denied: path is outside of workspace root");
|
|
958
|
+
}
|
|
959
|
+
// Recursive AST symbol extractor (regex-based, no heavy deps)
|
|
960
|
+
const extractSymbols = (filePath, relPath) => {
|
|
961
|
+
try {
|
|
962
|
+
const src = fs.readFileSync(filePath, "utf8");
|
|
963
|
+
const symbols = [];
|
|
964
|
+
const lines = src.split("\n");
|
|
965
|
+
lines.forEach((line, idx) => {
|
|
966
|
+
// Export declarations: functions, classes, interfaces, types, consts
|
|
967
|
+
const m = line.match(/^export\s+(?:default\s+)?(?:async\s+)?(class|function|interface|type|const|enum|abstract\s+class)\s+(\w+)/);
|
|
968
|
+
if (m) {
|
|
969
|
+
symbols.push({ kind: m[1].replace('abstract ', ''), name: m[2], file: relPath, line: idx + 1 });
|
|
970
|
+
}
|
|
971
|
+
// Named exports
|
|
972
|
+
const exportMatch = line.match(/^export\s*\{([^}]+)\}/);
|
|
973
|
+
if (exportMatch) {
|
|
974
|
+
exportMatch[1].split(',').forEach(s => {
|
|
975
|
+
const name = s.trim().split(/\s+as\s+/).pop()?.trim();
|
|
976
|
+
if (name)
|
|
977
|
+
symbols.push({ kind: 'export', name, file: relPath, line: idx + 1 });
|
|
978
|
+
});
|
|
979
|
+
}
|
|
980
|
+
});
|
|
981
|
+
return symbols;
|
|
982
|
+
}
|
|
983
|
+
catch {
|
|
984
|
+
return [];
|
|
985
|
+
}
|
|
986
|
+
};
|
|
987
|
+
const walkDir = (dir, rel) => {
|
|
988
|
+
let allSymbols = [];
|
|
989
|
+
try {
|
|
990
|
+
const entries = fs.readdirSync(dir, { withFileTypes: true });
|
|
991
|
+
for (const entry of entries) {
|
|
992
|
+
if (["node_modules", ".git", "dist", "chroma_data", ".lemma"].includes(entry.name))
|
|
993
|
+
continue;
|
|
994
|
+
const fullPath = path.join(dir, entry.name);
|
|
995
|
+
const relPath = rel ? path.join(rel, entry.name) : entry.name;
|
|
996
|
+
if (entry.isDirectory()) {
|
|
997
|
+
allSymbols = allSymbols.concat(walkDir(fullPath, relPath));
|
|
998
|
+
}
|
|
999
|
+
else {
|
|
1000
|
+
const ext = entry.name.split('.').pop() || '';
|
|
1001
|
+
if (extensions.includes(ext)) {
|
|
1002
|
+
allSymbols = allSymbols.concat(extractSymbols(fullPath, relPath));
|
|
1003
|
+
}
|
|
1004
|
+
}
|
|
1005
|
+
}
|
|
1006
|
+
}
|
|
1007
|
+
catch { }
|
|
1008
|
+
return allSymbols;
|
|
1009
|
+
};
|
|
1010
|
+
const symbols = walkDir(resolvedDir, dirPath);
|
|
1011
|
+
const byFile = {};
|
|
1012
|
+
symbols.forEach(s => {
|
|
1013
|
+
if (!byFile[s.file])
|
|
1014
|
+
byFile[s.file] = [];
|
|
1015
|
+
byFile[s.file].push({ kind: s.kind, name: s.name, line: s.line });
|
|
1016
|
+
});
|
|
1017
|
+
const hologram = {
|
|
1018
|
+
workspace: path.basename(workspaceRoot),
|
|
1019
|
+
scannedDir: dirPath || '.',
|
|
1020
|
+
totalSymbols: symbols.length,
|
|
1021
|
+
totalFiles: Object.keys(byFile).length,
|
|
1022
|
+
map: byFile,
|
|
1023
|
+
};
|
|
1024
|
+
return {
|
|
1025
|
+
content: [{ type: "text", text: JSON.stringify(hologram, null, 2) }],
|
|
1026
|
+
};
|
|
1027
|
+
}
|
|
1028
|
+
case "validate_patch_sandbox": {
|
|
1029
|
+
const filePath = args?.filePath;
|
|
1030
|
+
const patchedContent = args?.patchedContent;
|
|
1031
|
+
if (!filePath || patchedContent === undefined) {
|
|
1032
|
+
throw new Error("filePath and patchedContent are required");
|
|
1033
|
+
}
|
|
1034
|
+
const workspaceRoot = process.cwd();
|
|
1035
|
+
const resolvedPath = path.resolve(workspaceRoot, filePath);
|
|
1036
|
+
if (!resolvedPath.startsWith(workspaceRoot)) {
|
|
1037
|
+
throw new Error("Access denied: path is outside of workspace root");
|
|
1038
|
+
}
|
|
1039
|
+
const { execSync } = require('child_process');
|
|
1040
|
+
const os = require('os');
|
|
1041
|
+
const sandboxDir = path.join(os.tmpdir(), `lemma-sandbox-${Date.now()}`);
|
|
1042
|
+
const sandboxFile = path.join(sandboxDir, path.basename(filePath));
|
|
1043
|
+
try {
|
|
1044
|
+
fs.mkdirSync(sandboxDir, { recursive: true });
|
|
1045
|
+
fs.writeFileSync(sandboxFile, patchedContent, 'utf8');
|
|
1046
|
+
// Step 1: Basic syntax check via Node.js --check (works for JS/TS after strip)
|
|
1047
|
+
// Step 2: Try tsc --noEmit on the patched file using the workspace tsconfig
|
|
1048
|
+
const tsconfigPath = path.join(workspaceRoot, 'tsconfig.json');
|
|
1049
|
+
let tscResult = { success: true, output: 'No TypeScript config found — skipping tsc validation.' };
|
|
1050
|
+
if (fs.existsSync(tsconfigPath) && filePath.match(/\.tsx?$/)) {
|
|
1051
|
+
try {
|
|
1052
|
+
// Copy the patched file to a temp location respecting the workspace structure
|
|
1053
|
+
const sandboxWorkspace = path.join(sandboxDir, 'workspace');
|
|
1054
|
+
// Sync relevant source structure for valid imports check
|
|
1055
|
+
execSync(`cp -r ${workspaceRoot}/src ${sandboxWorkspace}/src 2>/dev/null || true`, { timeout: 5000 });
|
|
1056
|
+
// Overwrite the target file with patched content
|
|
1057
|
+
const sandboxTargetFile = path.join(sandboxWorkspace, filePath);
|
|
1058
|
+
fs.mkdirSync(path.dirname(sandboxTargetFile), { recursive: true });
|
|
1059
|
+
fs.writeFileSync(sandboxTargetFile, patchedContent, 'utf8');
|
|
1060
|
+
const tscOut = execSync(`cd ${workspaceRoot} && npx tsc --noEmit --skipLibCheck --allowJs --esModuleInterop --strict false 2>&1 | head -40 || true`, { timeout: 20000, encoding: 'utf8' });
|
|
1061
|
+
// Filter errors that reference our target file
|
|
1062
|
+
const relevantErrors = tscOut.split('\n').filter((l) => l.includes(path.basename(filePath)) || l.includes('error TS'));
|
|
1063
|
+
tscResult = {
|
|
1064
|
+
success: relevantErrors.length === 0,
|
|
1065
|
+
output: relevantErrors.length > 0 ? relevantErrors.join('\n') : '✅ TypeScript check passed with no errors in target file.'
|
|
1066
|
+
};
|
|
1067
|
+
}
|
|
1068
|
+
catch (e) {
|
|
1069
|
+
tscResult = { success: false, output: e.stdout || e.message };
|
|
1070
|
+
}
|
|
1071
|
+
}
|
|
1072
|
+
// Step 3: Quick regex-based syntax sanity (balanced braces/brackets)
|
|
1073
|
+
const opens = (patchedContent.match(/[{[(]/g) || []).length;
|
|
1074
|
+
const closes = (patchedContent.match(/[}\])]/g) || []).length;
|
|
1075
|
+
const balanced = Math.abs(opens - closes) <= 2; // allow small tolerance
|
|
1076
|
+
fs.rmSync(sandboxDir, { recursive: true, force: true });
|
|
1077
|
+
const verdict = tscResult.success && balanced;
|
|
1078
|
+
return {
|
|
1079
|
+
content: [{
|
|
1080
|
+
type: "text",
|
|
1081
|
+
text: JSON.stringify({
|
|
1082
|
+
verdict: verdict ? '✅ SAFE TO APPLY' : '❌ DO NOT APPLY — Issues Found',
|
|
1083
|
+
syntaxBalanced: balanced ? '✅' : '⚠️ Unbalanced brackets/braces detected',
|
|
1084
|
+
tscCheck: tscResult.output,
|
|
1085
|
+
recommendation: verdict
|
|
1086
|
+
? 'Patch looks valid. You can safely call apply_workspace_patch to apply it.'
|
|
1087
|
+
: 'Fix the reported issues before applying the patch to avoid breaking the codebase.',
|
|
1088
|
+
}, null, 2)
|
|
1089
|
+
}]
|
|
1090
|
+
};
|
|
1091
|
+
}
|
|
1092
|
+
catch (err) {
|
|
1093
|
+
try {
|
|
1094
|
+
fs.rmSync(sandboxDir, { recursive: true, force: true });
|
|
1095
|
+
}
|
|
1096
|
+
catch { }
|
|
1097
|
+
return {
|
|
1098
|
+
content: [{ type: "text", text: `Sandbox validation error: ${err.message}` }]
|
|
1099
|
+
};
|
|
1100
|
+
}
|
|
1101
|
+
}
|
|
1102
|
+
case "query_hybrid_consensus": {
|
|
1103
|
+
const query = args?.query;
|
|
1104
|
+
const context = args?.context || '';
|
|
1105
|
+
const threshold = typeof args?.threshold === 'number' ? args.threshold : 0.80;
|
|
1106
|
+
if (!query)
|
|
1107
|
+
throw new Error('query is required');
|
|
1108
|
+
const HOME = process.env.HOME || process.env.USERPROFILE || '~';
|
|
1109
|
+
const portFile = path.join(HOME, '.lemma-cache/proxy.port');
|
|
1110
|
+
let port = '8081';
|
|
1111
|
+
if (fs.existsSync(portFile)) {
|
|
1112
|
+
port = fs.readFileSync(portFile, 'utf8').trim();
|
|
1113
|
+
}
|
|
1114
|
+
const axios = require('axios');
|
|
1115
|
+
const fullQuery = context ? `${query}\n\nContext: ${context}` : query;
|
|
1116
|
+
try {
|
|
1117
|
+
// Step 1: Hit The Brain first
|
|
1118
|
+
const searchRes = await axios.get(`http://localhost:${port}/api/search?q=${encodeURIComponent(fullQuery)}&limit=3`);
|
|
1119
|
+
const results = searchRes.data.results || [];
|
|
1120
|
+
const topHit = results[0];
|
|
1121
|
+
if (topHit && topHit.similarity >= threshold) {
|
|
1122
|
+
// 🎯 Brain HIT — return cached answer, zero cloud tokens spent
|
|
1123
|
+
const savedTokens = Math.floor((topHit.response?.choices?.[0]?.message?.content?.length || 500) / 4);
|
|
1124
|
+
reportSavings({ source: 'cache', tokens: savedTokens });
|
|
1125
|
+
const responseText = typeof topHit.response === 'string'
|
|
1126
|
+
? topHit.response
|
|
1127
|
+
: topHit.response?.choices?.[0]?.message?.content
|
|
1128
|
+
|| JSON.stringify(topHit.response, null, 2);
|
|
1129
|
+
return {
|
|
1130
|
+
content: [{
|
|
1131
|
+
type: 'text',
|
|
1132
|
+
text: `🧠 **Brain Cache HIT** (similarity: ${(topHit.similarity * 100).toFixed(1)}% — above ${(threshold * 100).toFixed(0)}% threshold)\n\n**No cloud LLM call needed. ~${savedTokens} tokens saved.**\n\n---\n\n${responseText}`
|
|
1133
|
+
}]
|
|
1134
|
+
};
|
|
1135
|
+
}
|
|
1136
|
+
// Step 2: Brain MISS — inform the IDE to proceed with cloud and store the result
|
|
1137
|
+
const missMsg = results.length > 0
|
|
1138
|
+
? `🔍 **Brain Miss** — Best match was only ${(topHit.similarity * 100).toFixed(1)}% (below ${(threshold * 100).toFixed(0)}% threshold).`
|
|
1139
|
+
: `🔍 **Brain Miss** — No relevant memories found for this query.`;
|
|
1140
|
+
return {
|
|
1141
|
+
content: [{
|
|
1142
|
+
type: 'text',
|
|
1143
|
+
text: `${missMsg}\n\n**Proceed with your cloud LLM call.** Once you have the answer, call \`store_memory\` with:\n- query: "${query.substring(0, 100)}"
|
|
1144
|
+
- response: [your full answer]\n\nThis will cache it for future sessions and save tokens next time.`
|
|
1145
|
+
}]
|
|
1146
|
+
};
|
|
1147
|
+
}
|
|
1148
|
+
catch (e) {
|
|
1149
|
+
return {
|
|
1150
|
+
content: [{ type: 'text', text: `Hybrid Consensus failed: ${e.message}. Is Lemma Proxy running?` }]
|
|
1151
|
+
};
|
|
1152
|
+
}
|
|
1153
|
+
}
|
|
1154
|
+
case "get_telepathic_hints": {
|
|
1155
|
+
const activeFile = args?.activeFile;
|
|
1156
|
+
const limit = typeof args?.limit === 'number' ? args.limit : 5;
|
|
1157
|
+
if (!activeFile)
|
|
1158
|
+
throw new Error('activeFile is required');
|
|
1159
|
+
const HOME = process.env.HOME || process.env.USERPROFILE || '~';
|
|
1160
|
+
const portFile = path.join(HOME, '.lemma-cache/proxy.port');
|
|
1161
|
+
let port = '8081';
|
|
1162
|
+
if (fs.existsSync(portFile)) {
|
|
1163
|
+
port = fs.readFileSync(portFile, 'utf8').trim();
|
|
1164
|
+
}
|
|
1165
|
+
const axios = require('axios');
|
|
1166
|
+
// Build a rich query from the file path: extract module name, dir context, and file extension
|
|
1167
|
+
const basename = path.basename(activeFile, path.extname(activeFile));
|
|
1168
|
+
const dirContext = path.dirname(activeFile).split(path.sep).filter(Boolean).slice(-2).join(' ');
|
|
1169
|
+
const ext = path.extname(activeFile).replace('.', '');
|
|
1170
|
+
const telepathicQuery = `${basename} ${dirContext} ${ext} patterns solutions architecture`;
|
|
1171
|
+
try {
|
|
1172
|
+
const searchRes = await axios.get(`http://localhost:${port}/api/search?q=${encodeURIComponent(telepathicQuery)}&limit=${limit}`);
|
|
1173
|
+
const results = searchRes.data.results || [];
|
|
1174
|
+
if (results.length === 0) {
|
|
1175
|
+
return {
|
|
1176
|
+
content: [{
|
|
1177
|
+
type: 'text',
|
|
1178
|
+
text: `📡 **Telepathic Hints** for \`${activeFile}\`\n\nNo relevant memories found yet. As you use Lemma and store solutions, they will appear here automatically.`
|
|
1179
|
+
}]
|
|
1180
|
+
};
|
|
1181
|
+
}
|
|
1182
|
+
let hintsText = `📡 **Telepathic Hints** for \`${activeFile}\`\n*(${results.length} relevant memories surfaced from The Brain)*\n\n`;
|
|
1183
|
+
results.forEach((r, i) => {
|
|
1184
|
+
const similarity = (r.similarity * 100).toFixed(1);
|
|
1185
|
+
const prompt = typeof r.prompt === 'string' ? r.prompt.substring(0, 120) : 'Unknown';
|
|
1186
|
+
const responseContent = typeof r.response === 'string'
|
|
1187
|
+
? r.response
|
|
1188
|
+
: r.response?.choices?.[0]?.message?.content
|
|
1189
|
+
|| JSON.stringify(r.response).substring(0, 300);
|
|
1190
|
+
hintsText += `### 💡 Hint ${i + 1} (${similarity}% match)\n`;
|
|
1191
|
+
hintsText += `**Memory:** ${prompt}\n\n`;
|
|
1192
|
+
hintsText += `${responseContent.substring(0, 400)}${responseContent.length > 400 ? '...' : ''}\n\n---\n\n`;
|
|
1193
|
+
});
|
|
1194
|
+
// Report token savings: surface context avoids re-asking the LLM
|
|
1195
|
+
const estimatedTokensSaved = results.length * 150;
|
|
1196
|
+
reportSavings({ source: 'cache', tokens: estimatedTokensSaved });
|
|
1197
|
+
return {
|
|
1198
|
+
content: [{ type: 'text', text: hintsText }]
|
|
1199
|
+
};
|
|
1200
|
+
}
|
|
1201
|
+
catch (e) {
|
|
1202
|
+
return {
|
|
1203
|
+
content: [{ type: 'text', text: `Telepathic Hints failed: ${e.message}. Is Lemma Proxy running?` }]
|
|
1204
|
+
};
|
|
1205
|
+
}
|
|
1206
|
+
}
|
|
800
1207
|
default:
|
|
801
1208
|
throw new Error(`Unknown tool: ${name}`);
|
|
802
1209
|
}
|