pi-lean-ctx 3.4.1 → 3.4.3

package/extensions/index.ts

@@ -271,19 +271,37 @@ export default function (pi: ExtensionAPI) {
     },
   });
 
+  const rawBash = createBashToolDefinition(process.cwd());
+
+  const bashSchemaWithRaw = Type.Object({
+    command: Type.String({ description: "Bash command to execute" }),
+    timeout: Type.Optional(Type.Number({ description: "Timeout in seconds to prevent hanging commands" })),
+    raw: Type.Optional(Type.Boolean({ description: "Skip compression, return full uncompressed output" })),
+  });
+
   pi.registerTool({
     ...baseBashTool,
+    parameters: bashSchemaWithRaw,
     description:
-      "Execute a bash command through lean-ctx compression for 60-90% smaller output.",
-    promptSnippet: "Run shell commands through lean-ctx compression.",
+      "Execute a bash command. Output is auto-compressed by lean-ctx. "
+      + "IMPORTANT: Do NOT use bash to read files (cat/head/tail) — use the read tool instead. "
+      + "Do NOT use bash for grep/find/ls — use the dedicated tools. "
+      + "Set raw=true to skip compression when exact output matters. "
+      + "Use timeout (seconds) to prevent hanging commands.",
+    promptSnippet: "Run shell commands (not for file reading — use read tool)",
     promptGuidelines: [
-      "Use bash normally — commands are automatically routed through lean-ctx.",
-      "lean-ctx compresses verbose CLI output (git, cargo, npm, docker, kubectl, etc.) automatically.",
+      "Use bash only for commands with side effects: build, test, install, git, run scripts.",
     ],
     async execute(toolCallId, params, signal, onUpdate, ctx) {
+      const isRaw = !!params.raw;
+      const toolParams = { command: params.command, timeout: params.timeout };
+      const tool = isRaw ? rawBash : baseBashTool;
       try {
-        const result = await baseBashTool.execute(toolCallId, params, signal, onUpdate, ctx);
+        const result = await tool.execute(toolCallId, toolParams, signal, onUpdate, ctx);
         const text = result.content?.[0]?.type === "text" ? result.content[0].text : "";
+        if (isRaw) {
+          return { ...result, content: [{ type: "text", text }], details: { raw: true } };
+        }
         const decorated = withFooter(text, { always: true });
         return {
           ...result,
@@ -292,6 +310,7 @@ export default function (pi: ExtensionAPI) {
         };
       } catch (error) {
         if (error instanceof Error) {
+          if (isRaw) throw error;
           const decorated = withFooter(error.message, { always: true });
           throw new Error(decorated.text);
         }
@@ -306,11 +325,13 @@ export default function (pi: ExtensionAPI) {
     name: "read",
     label: "Read",
     description:
-      "Read file contents through lean-ctx with automatic mode selection (full/map/signatures) based on file type and size.",
-    promptSnippet: "Read files through lean-ctx compression with smart mode selection.",
+      "Read file contents. ALWAYS use this instead of cat/head/tail via bash. "
+      + "Auto-selects mode: configs (.yaml/.json/.toml/.env) are always full-read. "
+      + "Code files: full (<8KB), map (8-96KB), signatures (>96KB). "
+      + "Use offset and limit to read specific line ranges.",
+    promptSnippet: "Read file contents (always use instead of cat)",
     promptGuidelines: [
-      "Use read normally — lean-ctx automatically selects the optimal compression mode.",
-      "Small files get full reads, large code files get map/signatures mode.",
+      "Use read to inspect file contents instead of cat or less.",
     ],
     parameters: readSchema,
     renderCall(args, theme, context) {
@@ -410,9 +431,8 @@ export default function (pi: ExtensionAPI) {
   pi.registerTool({
     name: "ls",
     label: "ls",
-    description: "List directory contents through lean-ctx compression.",
-    promptSnippet: "List directory contents with token-optimized output.",
-    promptGuidelines: ["Use ls normally — output is automatically compressed by lean-ctx."],
+    description: "List directory contents. Use limit to reduce output size.",
+    promptSnippet: "List directory contents",
     parameters: lsSchema,
     async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
       const requestedPath = normalizePathArg(params.path || ".");
@@ -429,9 +449,8 @@ export default function (pi: ExtensionAPI) {
   pi.registerTool({
     name: "find",
     label: "find",
-    description: "Find files by glob pattern through lean-ctx compression.",
-    promptSnippet: "Find files with compressed output.",
-    promptGuidelines: ["Use find normally — output respects .gitignore and is compressed by lean-ctx."],
+    description: "Find files by glob pattern (respects .gitignore). Use limit to reduce output size.",
+    promptSnippet: "Find files by glob pattern",
     parameters: findSchema,
     async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
       const requestedPath = normalizePathArg(params.path || ".");
@@ -448,9 +467,8 @@ export default function (pi: ExtensionAPI) {
   pi.registerTool({
     name: "grep",
     label: "grep",
-    description: "Search file contents through ripgrep + lean-ctx compression.",
-    promptSnippet: "Search code with compressed, grouped results.",
-    promptGuidelines: ["Use grep normally — results are compressed and grouped by lean-ctx."],
+    description: "Search file contents with ripgrep. Use limit to cap matches and context for surrounding lines.",
+    promptSnippet: "Search file contents for patterns",
     parameters: grepSchema,
     async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
       const requestedPath = normalizePathArg(params.path || ".");

package/package.json

@@ -1,8 +1,14 @@
 {
   "name": "pi-lean-ctx",
-  "version": "3.4.1",
-  "description": "Pi Coding Agent extension with first-class MCP support — routes bash, read, grep, find, and ls through lean-ctx CLI, and exposes all 46 lean-ctx MCP tools (ctx_session, ctx_knowledge, ctx_semantic_search, ctx_impact, ctx_architecture, ctx_workflow, ctx_gain, etc.) natively in Pi",
-  "keywords": ["pi-package", "lean-ctx", "token-optimization", "compression", "mcp"],
+  "version": "3.4.3",
+  "description": "Pi Coding Agent extension with first-class MCP support \u2014 routes bash, read, grep, find, and ls through lean-ctx CLI, and exposes all 46 lean-ctx MCP tools (ctx_session, ctx_knowledge, ctx_semantic_search, ctx_impact, ctx_architecture, ctx_workflow, ctx_gain, etc.) natively in Pi",
+  "keywords": [
+    "pi-package",
+    "lean-ctx",
+    "token-optimization",
+    "compression",
+    "mcp"
+  ],
   "license": "Apache-2.0",
   "author": "Yves Gugger",
   "repository": {
@@ -22,7 +28,9 @@
     "@mariozechner/pi-tui": "*"
   },
   "pi": {
-    "extensions": ["./extensions"]
+    "extensions": [
+      "./extensions/index.ts"
+    ]
   },
   "files": [
     "extensions/",