@ridit/lens 0.4.5 → 0.4.7

package/README.md

@@ -41,6 +41,7 @@ lens chat --dev                         output structured JSON (for SDK/tooling
 lens chat --single                      run one message then exit, resumes latest session
 lens chat --force-all                   auto-approve all tools including writes and shell
 lens chat --dev --prompt <text>         headless mode: JSON output, no UI
+lens chat --runtime-tools <path>        load extra tools from a JSON file at runtime
 
 lens provider                           configure AI providers (interactive)
 lens provider --list                    list configured providers
@@ -90,7 +91,32 @@ Once inside a `lens chat` session, use slash commands:
 
 ## Extending Lens
 
-Custom tools can be built and registered using [`@ridit/lens-sdk`](https://www.npmjs.com/package/@ridit/lens-sdk).
+### Runtime Tools
+
+Pass a JSON file to `--runtime-tools` to inject custom tools into any chat session without modifying Lens itself. Each tool declares a name, description, optional parameters, and an HTTP endpoint that Lens will POST to when the AI calls it.
+
+```json
+[
+  {
+    "name": "get_weather",
+    "description": "Returns current weather for a city",
+    "parameters": {
+      "city": { "type": "string", "description": "City name" }
+    },
+    "endpoint": "http://localhost:4242/get_weather"
+  }
+]
+```
+
+```bash
+lens chat --runtime-tools ./my-tools.json --prompt "What's the weather in London?"
+```
+
+Lens POSTs the tool arguments as JSON to the endpoint and returns the response body to the model. Runtime tools are always auto-approved in headless mode.
+
+### SDK
+
+Custom tools can also be built and registered using [`@ridit/lens-sdk`](https://www.npmjs.com/package/@ridit/lens-sdk).
 
 ## License
 

package/dist/index.mjs

@@ -68823,6 +68823,9 @@ function trimStartOfStream() {
 }
 var HANGING_STREAM_WARNING_TIME_MS = 15 * 1000;
 
+// ../core/src/agent/index.ts
+import { readFileSync as readFileSync5, existsSync as existsSync5 } from "fs";
+
 // ../core/src/config/index.ts
 import { join } from "path";
 import { homedir as homedir2 } from "os";
@@ -74705,8 +74708,43 @@ var scrape = tool({
 var tools = { read, write, bash, grep, ls, remember, del, search, scrape };
 
 // ../core/src/agent/index.ts
+function buildZodSchema(parameters) {
+  const shape = {};
+  for (const [key, val] of Object.entries(parameters)) {
+    let field;
+    if (val.type === "number")
+      field = exports_external.number();
+    else if (val.type === "boolean")
+      field = exports_external.boolean();
+    else
+      field = exports_external.string();
+    if (val.description)
+      field = field.describe(val.description);
+    shape[key] = field;
+  }
+  return exports_external.object(shape);
+}
 async function chat(options2) {
-  const activeTools = options2.onBeforeToolCall ? Object.fromEntries(Object.entries(tools).map(([name17, t]) => [
+  let extraTools = {};
+  if (options2.runtimeTools && existsSync5(options2.runtimeTools)) {
+    const raw = JSON.parse(readFileSync5(options2.runtimeTools, "utf-8"));
+    for (const t of raw) {
+      extraTools[t.name] = tool({
+        description: t.description,
+        parameters: buildZodSchema(t.parameters ?? {}),
+        execute: async (args) => {
+          const res = await fetch(t.endpoint, {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+            body: JSON.stringify(args)
+          });
+          return await res.json();
+        }
+      });
+    }
+  }
+  const allTools = { ...tools, ...extraTools };
+  const activeTools = options2.onBeforeToolCall ? Object.fromEntries(Object.entries(allTools).map(([name17, t]) => [
     name17,
     {
       ...t,
@@ -74717,7 +74755,7 @@ async function chat(options2) {
         return t.execute(args, opts);
       }
     }
-  ])) : tools;
+  ])) : allTools;
   const responseMessages = [];
   const providerSettings = (() => {
     try {
@@ -74732,8 +74770,12 @@ async function chat(options2) {
     messages: options2.messages,
     system: options2.system,
     maxSteps: options2.maxSteps ?? 50,
-    ...providerSettings?.maxTokens !== undefined && { maxTokens: providerSettings.maxTokens },
-    ...providerSettings?.temperature !== undefined && { temperature: providerSettings.temperature },
+    ...providerSettings?.maxTokens !== undefined && {
+      maxTokens: providerSettings.maxTokens
+    },
+    ...providerSettings?.temperature !== undefined && {
+      temperature: providerSettings.temperature
+    },
     onStepFinish: (step) => {
       responseMessages.push(...step.response.messages);
       for (const toolResult of step.toolResults) {
@@ -76188,7 +76230,8 @@ function ChatRunner({
   initialMessage,
   dev = false,
   single = false,
-  sessionId
+  sessionId,
+  runtimeTools
 }) {
   const [stage, setStage] = useState12("idle");
   const [showProvider, setShowProvider] = useState12(false);
@@ -76310,6 +76353,7 @@ function ChatRunner({
         await chat({
           messages: getMessages(sessionRef.current),
           system: getSystemPrompt(repoPath),
+          runtimeTools,
           onChunk: () => {},
           onToolCall: (tool2, args) => {
             devTools.push({ tool: tool2, args, result: null });
@@ -76359,6 +76403,7 @@ function ChatRunner({
       await chat({
         messages: getMessages(sessionRef.current),
         system: getSystemPrompt(repoPath),
+        runtimeTools,
         onBeforeToolCall: (tool2, args) => {
           if (forceApproveRef.current || SAFE_TOOLS.has(tool2))
             return Promise.resolve(true);
@@ -76622,7 +76667,8 @@ function ChatCommand({
   initialMessage,
   dev = false,
   single = false,
-  sessionId
+  sessionId,
+  runtimeTools
 }) {
   return /* @__PURE__ */ jsxDEV15(Box12, {
     flexDirection: "column",
@@ -76632,7 +76678,8 @@ function ChatCommand({
       initialMessage,
       dev,
       single,
-      sessionId
+      sessionId,
+      runtimeTools
     }, undefined, false, undefined, this)
   }, undefined, false, undefined, this);
 }
@@ -78158,24 +78205,59 @@ function getLastDeniedAction(messages) {
 async function runHeadless(opts) {
   const repoPath = opts.path;
   let session = opts.sessionId ? loadSession(opts.sessionId) ?? createSessionWithId(opts.sessionId, repoPath) : opts.single ? getLatestSession(repoPath) ?? createSession(repoPath) : createSession(repoPath);
-  let prompt = opts.prompt;
-  if (opts.forceAll && APPROVAL_WORDS.has(prompt.trim().toLowerCase())) {
-    const pending = getLastDeniedAction(getMessages(session));
-    if (pending) {
-      prompt = `Proceed with the previously denied operation: use the ${pending.tool} tool on "${pending.description}".`;
+  if (opts.resume) {
+    const msgs = getMessages(session);
+    let trimAt = -1;
+    for (let i = msgs.length - 1;i >= 0; i--) {
+      const msg = msgs[i];
+      if (!msg || msg.role !== "tool")
+        continue;
+      const content = Array.isArray(msg.content) ? msg.content : [];
+      const isDenied = content.some((p) => typeof p === "object" && p !== null && ("type" in p) && p.type === "tool-result" && ("result" in p) && typeof p.result === "string" && p.result.includes("Permission denied"));
+      if (isDenied) {
+        for (let j = i - 1;j >= 0; j--) {
+          if (msgs[j]?.role === "assistant") {
+            trimAt = j;
+            break;
+          }
+        }
+        break;
+      }
+    }
+    if (trimAt >= 0) {
+      session = { ...session, messages: msgs.slice(0, trimAt) };
     }
+    if (!opts.single || opts.sessionId)
+      saveSession(session);
+  } else {
+    let prompt = opts.prompt;
+    if (opts.forceAll && APPROVAL_WORDS.has(prompt.trim().toLowerCase())) {
+      const pending = getLastDeniedAction(getMessages(session));
+      if (pending) {
+        prompt = `Proceed with the previously denied operation: use the ${pending.tool} tool on "${pending.description}".`;
+      }
+    }
+    session = addMessage(session, "user", prompt);
+    if (!opts.single || opts.sessionId)
+      saveSession(session);
   }
-  session = addMessage(session, "user", prompt);
-  if (!opts.single || opts.sessionId)
-    saveSession(session);
   const toolLog = [];
   const denied = [];
+  const runtimeToolNames = new Set;
+  if (opts.runtimeTools) {
+    try {
+      const raw = JSON.parse(__require("fs").readFileSync(opts.runtimeTools, "utf-8"));
+      if (Array.isArray(raw))
+        raw.forEach((t) => runtimeToolNames.add(t.name));
+    } catch {}
+  }
   await chat({
     messages: getMessages(session),
     system: getSystemPrompt(repoPath),
+    runtimeTools: opts.runtimeTools,
     maxSteps: opts.forceAll ? 50 : 2,
     onBeforeToolCall: (tool2, args) => {
-      if (opts.forceAll || HEADLESS_SAFE_TOOLS.has(tool2))
+      if (opts.forceAll || HEADLESS_SAFE_TOOLS.has(tool2) || runtimeToolNames.has(tool2))
         return Promise.resolve(true);
       const a = args;
       const description = tool2 === "bash" ? String(a.command ?? a.cmd ?? "") : tool2 === "write" ? String(a.path ?? a.file_path ?? "") : String(a.path ?? a.file_path ?? "");
@@ -78208,10 +78290,10 @@ async function runHeadless(opts) {
   });
 }
 var program = new Command().enablePositionalOptions();
-program.command("chat").description("Chat with your codebase — ask questions or make changes").option("-p, --path <path>", "Path to the repo", ".").option("-d, --dev", "Output structured JSON (no UI)").option("--single", "Single-shot: run one message then exit").option("--session <id>", "Resume session by ID, or create one with that ID").option("--id <id>", "Alias for --session").option("--force-all", "Auto-approve all tools").option("--prompt <text>", "Run a prompt non-interactively").action((opts) => {
+program.command("chat").description("Chat with your codebase — ask questions or make changes").option("-p, --path <path>", "Path to the repo", ".").option("-d, --dev", "Output structured JSON (no UI)").option("--single", "Single-shot: run one message then exit").option("--session <id>", "Resume session by ID, or create one with that ID").option("--id <id>", "Alias for --session").option("--force-all", "Auto-approve all tools").option("--prompt <text>", "Run a prompt non-interactively").option("--resume", "Resume from last permission-denied tool call (no new prompt needed)").option("--runtime-tools <path>", "path to runtime tools JSON file").action((opts) => {
   const sessionId = opts.session ?? opts.id;
-  if (opts.prompt && (opts.dev || opts.single)) {
-    runHeadless({ path: opts.path, prompt: opts.prompt, sessionId, single: opts.single, forceAll: opts.forceAll });
+  if ((opts.prompt || opts.resume) && (opts.dev || opts.single)) {
+    runHeadless({ path: opts.path, prompt: opts.prompt, sessionId, single: opts.single, forceAll: opts.forceAll ?? opts.resume, runtimeTools: opts.runtimeTools, resume: opts.resume });
     return;
   }
   render(/* @__PURE__ */ jsxDEV21(ChatCommand, {
@@ -78220,7 +78302,8 @@ program.command("chat").description("Chat with your codebase — ask questions o
     dev: opts.dev ?? false,
     single: opts.single ?? false,
     sessionId,
-    initialMessage: opts.prompt
+    initialMessage: opts.prompt,
+    runtimeTools: opts.runtimeTools
   }, undefined, false, undefined, this));
 });
 program.command("commit [files...]").description("Generate a smart conventional commit message from staged changes").option("-p, --path <path>", "Path to the repo", ".").option("--auto", "Stage all changes and commit without confirmation").option("--push", "Push to remote after committing").action((files, opts) => {
@@ -78308,13 +78391,13 @@ program.command("run <cmd>").description("Run your dev server. Lens watches and
 });
 var firstArg = process.argv[2];
 if (!firstArg || firstArg.startsWith("-")) {
-  const defaultFlags = new Command().option("-p, --path <path>", "Path to the repo", ".").option("--session <id>", "Resume session by ID").option("--single", "Single-shot mode").option("--prompt <text>", "Run a prompt").option("-d, --dev", "Output JSON (no UI)").option("--force-all", "Auto-approve all tools").allowUnknownOption().exitOverride();
+  const defaultFlags = new Command().option("-p, --path <path>", "Path to the repo", ".").option("--session <id>", "Resume session by ID").option("--single", "Single-shot mode").option("--prompt <text>", "Run a prompt").option("-d, --dev", "Output JSON (no UI)").option("--force-all", "Auto-approve all tools").option("--resume", "Resume from last permission-denied tool call").allowUnknownOption().exitOverride();
   try {
     defaultFlags.parse(process.argv);
   } catch {}
   const opts = defaultFlags.opts();
-  if (opts.prompt && (opts.dev || opts.single)) {
-    runHeadless({ path: opts.path ?? ".", prompt: opts.prompt, sessionId: opts.session, single: opts.single, forceAll: opts.forceAll });
+  if ((opts.prompt || opts.resume) && (opts.dev || opts.single)) {
+    runHeadless({ path: opts.path ?? ".", prompt: opts.prompt, sessionId: opts.session, single: opts.single, forceAll: opts.forceAll ?? opts.resume, resume: opts.resume });
   } else {
     render(/* @__PURE__ */ jsxDEV21(ChatCommand, {
       path: opts.path ?? ".",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ridit/lens",
-  "version": "0.4.5",
+  "version": "0.4.7",
   "description": "Understand your codebase.",
   "author": "Ridit Jangra <riditjangra09@gmail.com> (https://ridit.space)",
   "license": "MIT",

package/src/commands/chat.tsx

@@ -9,6 +9,7 @@ export function ChatCommand({
   dev = false,
   single = false,
   sessionId,
+  runtimeTools,
 }: {
   path: string;
   autoForce?: boolean;
@@ -16,6 +17,7 @@ export function ChatCommand({
   dev?: boolean;
   single?: boolean;
   sessionId?: string;
+  runtimeTools?: string;
 }) {
   return (
     <Box flexDirection="column">
@@ -26,6 +28,7 @@ export function ChatCommand({
         dev={dev}
         single={single}
         sessionId={sessionId}
+        runtimeTools={runtimeTools}
       />
     </Box>
   );

package/src/components/chat/ChatView.tsx

@@ -128,6 +128,7 @@ export function ChatRunner({
   dev = false,
   single = false,
   sessionId,
+  runtimeTools,
 }: {
   repoPath: string;
   autoForce?: boolean;
@@ -135,6 +136,7 @@ export function ChatRunner({
   dev?: boolean;
   single?: boolean;
   sessionId?: string;
+  runtimeTools?: string;
 }) {
   const [stage, setStage] = useState<"idle" | "thinking">("idle");
   const [showProvider, setShowProvider] = useState(false);
@@ -292,6 +294,7 @@ export function ChatRunner({
         await chat({
           messages: getMessages(sessionRef.current),
           system: getSystemPrompt(repoPath),
+          runtimeTools,
           onChunk: () => {},
           onToolCall: (tool, args) => {
             devTools.push({ tool, args, result: null });
@@ -350,6 +353,7 @@ export function ChatRunner({
       await chat({
         messages: getMessages(sessionRef.current),
         system: getSystemPrompt(repoPath),
+        runtimeTools,
         onBeforeToolCall: (tool, args) => {
           if (forceApproveRef.current || SAFE_TOOLS.has(tool))
             return Promise.resolve(true);

package/src/index.tsx

@@ -72,10 +72,12 @@ function getLastDeniedAction(messages: ReturnType<typeof getMessages>): { tool:
 
 async function runHeadless(opts: {
   path: string;
-  prompt: string;
+  prompt?: string;
   sessionId?: string;
   single?: boolean;
   forceAll?: boolean;
+  runtimeTools?: string;
+  resume?: boolean;
 }) {
   const repoPath = opts.path;
 
@@ -85,29 +87,75 @@ async function runHeadless(opts: {
       ? (getLatestSession(repoPath) ?? createSession(repoPath))
       : createSession(repoPath);
 
-  // if user is approving a prior denial, make the intent unambiguous
-  let prompt = opts.prompt;
-  if (opts.forceAll && APPROVAL_WORDS.has(prompt.trim().toLowerCase())) {
-    const pending = getLastDeniedAction(getMessages(session));
-    if (pending) {
-      prompt = `Proceed with the previously denied operation: use the ${pending.tool} tool on "${pending.description}".`;
+  if (opts.resume) {
+    // Rewind the session: strip the last denied tool-call (assistant message),
+    // its "Permission denied" tool result, and the assistant's text response after it.
+    // This leaves the session ending at the last successful state so the agent
+    // can re-attempt the denied tool with forceAll: true.
+    const msgs = getMessages(session);
+    let trimAt = -1;
+    for (let i = msgs.length - 1; i >= 0; i--) {
+      const msg = msgs[i];
+      if (!msg || msg.role !== "tool") continue;
+      const content = Array.isArray(msg.content) ? msg.content : [];
+      const isDenied = content.some(
+        (p: unknown) =>
+          typeof p === "object" && p !== null &&
+          "type" in p && (p as { type: string }).type === "tool-result" &&
+          "result" in p && typeof (p as { result: unknown }).result === "string" &&
+          ((p as { result: string }).result).includes("Permission denied"),
+      );
+      if (isDenied) {
+        // Find the assistant message immediately before this tool result (the tool-call message)
+        for (let j = i - 1; j >= 0; j--) {
+          if (msgs[j]?.role === "assistant") {
+            trimAt = j;
+            break;
+          }
+        }
+        break;
+      }
+    }
+    if (trimAt >= 0) {
+      session = { ...session, messages: msgs.slice(0, trimAt) };
+    }
+    // Save trimmed session so context is clean for this run
+    if (!opts.single || opts.sessionId) saveSession(session);
+  } else {
+    // if user is approving a prior denial, make the intent unambiguous
+    let prompt = opts.prompt!;
+    if (opts.forceAll && APPROVAL_WORDS.has(prompt.trim().toLowerCase())) {
+      const pending = getLastDeniedAction(getMessages(session));
+      if (pending) {
+        prompt = `Proceed with the previously denied operation: use the ${pending.tool} tool on "${pending.description}".`;
+      }
     }
-  }
 
-  session = addMessage(session, "user", prompt);
-  // save now so context is available on follow-up messages even if we exit early
-  if (!opts.single || opts.sessionId) saveSession(session);
+    session = addMessage(session, "user", prompt);
+    // save now so context is available on follow-up messages even if we exit early
+    if (!opts.single || opts.sessionId) saveSession(session);
+  }
 
   const toolLog: { tool: string; args: unknown; result: unknown }[] = [];
   const denied: { tool: string; description: string }[] = [];
 
+  // runtime tools are explicitly user-provided — always approve them
+  const runtimeToolNames = new Set<string>();
+  if (opts.runtimeTools) {
+    try {
+      const raw = JSON.parse(require("fs").readFileSync(opts.runtimeTools, "utf-8"));
+      if (Array.isArray(raw)) raw.forEach((t: { name: string }) => runtimeToolNames.add(t.name));
+    } catch { /* ignore parse errors */ }
+  }
+
   await chat({
     messages: getMessages(session),
     system: getSystemPrompt(repoPath),
+    runtimeTools: opts.runtimeTools,
     // 2 steps: 1 tool attempt (or denial) + 1 text response
     maxSteps: opts.forceAll ? 50 : 2,
     onBeforeToolCall: (tool, args) => {
-      if (opts.forceAll || HEADLESS_SAFE_TOOLS.has(tool)) return Promise.resolve(true);
+      if (opts.forceAll || HEADLESS_SAFE_TOOLS.has(tool) || runtimeToolNames.has(tool)) return Promise.resolve(true);
       // record denial — model will respond naturally explaining what it needs
       const a = args as Record<string, unknown>;
       const description =
@@ -159,6 +207,8 @@ program
   .option("--id <id>", "Alias for --session")
   .option("--force-all", "Auto-approve all tools")
   .option("--prompt <text>", "Run a prompt non-interactively")
+  .option("--resume", "Resume from last permission-denied tool call (no new prompt needed)")
+  .option("--runtime-tools <path>", "path to runtime tools JSON file")
   .action(
     (opts: {
       path: string;
@@ -168,11 +218,13 @@ program
       id?: string;
       forceAll?: boolean;
       prompt?: string;
+      resume?: boolean;
+      runtimeTools?: string;
     }) => {
       const sessionId = opts.session ?? opts.id;
-      // headless: dev+prompt or single+prompt → no UI, output JSON and exit
-      if (opts.prompt && (opts.dev || opts.single)) {
-        runHeadless({ path: opts.path, prompt: opts.prompt, sessionId, single: opts.single, forceAll: opts.forceAll });
+      // headless: dev+prompt, single+prompt, or --resume → no UI, output JSON and exit
+      if ((opts.prompt || opts.resume) && (opts.dev || opts.single)) {
+        runHeadless({ path: opts.path, prompt: opts.prompt, sessionId, single: opts.single, forceAll: opts.forceAll ?? opts.resume, runtimeTools: opts.runtimeTools, resume: opts.resume });
         return;
       }
       render(
@@ -183,6 +235,7 @@ program
           single={opts.single ?? false}
           sessionId={sessionId}
           initialMessage={opts.prompt}
+          runtimeTools={opts.runtimeTools}
         />,
       );
     },
@@ -353,6 +406,7 @@ if (!firstArg || firstArg.startsWith("-")) {
     .option("--prompt <text>", "Run a prompt")
     .option("-d, --dev", "Output JSON (no UI)")
     .option("--force-all", "Auto-approve all tools")
+    .option("--resume", "Resume from last permission-denied tool call")
     .allowUnknownOption()
     .exitOverride();
 
@@ -365,10 +419,11 @@ if (!firstArg || firstArg.startsWith("-")) {
     prompt?: string;
     dev?: boolean;
     forceAll?: boolean;
+    resume?: boolean;
   }>();
 
-  if (opts.prompt && (opts.dev || opts.single)) {
-    runHeadless({ path: opts.path ?? ".", prompt: opts.prompt, sessionId: opts.session, single: opts.single, forceAll: opts.forceAll });
+  if ((opts.prompt || opts.resume) && (opts.dev || opts.single)) {
+    runHeadless({ path: opts.path ?? ".", prompt: opts.prompt, sessionId: opts.session, single: opts.single, forceAll: opts.forceAll ?? opts.resume, resume: opts.resume });
   } else {
     render(
       <ChatCommand