@polka-codes/cli 0.8.12 → 0.8.14

package/dist/index.js

@@ -31615,17 +31615,17 @@ var require_visit3 = __commonJS((exports) => {
   visit2.BREAK = BREAK;
   visit2.SKIP = SKIP;
   visit2.REMOVE = REMOVE;
-  function visit_(key2, node, visitor, path2) {
-    const ctrl = callVisitor(key2, node, visitor, path2);
+  function visit_(key2, node, visitor, path) {
+    const ctrl = callVisitor(key2, node, visitor, path);
     if (identity2.isNode(ctrl) || identity2.isPair(ctrl)) {
-      replaceNode(key2, path2, ctrl);
-      return visit_(key2, ctrl, visitor, path2);
+      replaceNode(key2, path, ctrl);
+      return visit_(key2, ctrl, visitor, path);
     }
     if (typeof ctrl !== "symbol") {
       if (identity2.isCollection(node)) {
-        path2 = Object.freeze(path2.concat(node));
+        path = Object.freeze(path.concat(node));
         for (let i2 = 0;i2 < node.items.length; ++i2) {
-          const ci = visit_(i2, node.items[i2], visitor, path2);
+          const ci = visit_(i2, node.items[i2], visitor, path);
           if (typeof ci === "number")
             i2 = ci - 1;
           else if (ci === BREAK)
@@ -31636,13 +31636,13 @@ var require_visit3 = __commonJS((exports) => {
           }
         }
       } else if (identity2.isPair(node)) {
-        path2 = Object.freeze(path2.concat(node));
-        const ck = visit_("key", node.key, visitor, path2);
+        path = Object.freeze(path.concat(node));
+        const ck = visit_("key", node.key, visitor, path);
         if (ck === BREAK)
           return BREAK;
         else if (ck === REMOVE)
           node.key = null;
-        const cv = visit_("value", node.value, visitor, path2);
+        const cv = visit_("value", node.value, visitor, path);
         if (cv === BREAK)
           return BREAK;
         else if (cv === REMOVE)
@@ -31663,17 +31663,17 @@ var require_visit3 = __commonJS((exports) => {
   visitAsync.BREAK = BREAK;
   visitAsync.SKIP = SKIP;
   visitAsync.REMOVE = REMOVE;
-  async function visitAsync_(key2, node, visitor, path2) {
-    const ctrl = await callVisitor(key2, node, visitor, path2);
+  async function visitAsync_(key2, node, visitor, path) {
+    const ctrl = await callVisitor(key2, node, visitor, path);
     if (identity2.isNode(ctrl) || identity2.isPair(ctrl)) {
-      replaceNode(key2, path2, ctrl);
-      return visitAsync_(key2, ctrl, visitor, path2);
+      replaceNode(key2, path, ctrl);
+      return visitAsync_(key2, ctrl, visitor, path);
     }
     if (typeof ctrl !== "symbol") {
       if (identity2.isCollection(node)) {
-        path2 = Object.freeze(path2.concat(node));
+        path = Object.freeze(path.concat(node));
         for (let i2 = 0;i2 < node.items.length; ++i2) {
-          const ci = await visitAsync_(i2, node.items[i2], visitor, path2);
+          const ci = await visitAsync_(i2, node.items[i2], visitor, path);
           if (typeof ci === "number")
             i2 = ci - 1;
           else if (ci === BREAK)
@@ -31684,13 +31684,13 @@ var require_visit3 = __commonJS((exports) => {
           }
         }
       } else if (identity2.isPair(node)) {
-        path2 = Object.freeze(path2.concat(node));
-        const ck = await visitAsync_("key", node.key, visitor, path2);
+        path = Object.freeze(path.concat(node));
+        const ck = await visitAsync_("key", node.key, visitor, path);
         if (ck === BREAK)
           return BREAK;
         else if (ck === REMOVE)
           node.key = null;
-        const cv = await visitAsync_("value", node.value, visitor, path2);
+        const cv = await visitAsync_("value", node.value, visitor, path);
         if (cv === BREAK)
           return BREAK;
         else if (cv === REMOVE)
@@ -31717,23 +31717,23 @@ var require_visit3 = __commonJS((exports) => {
     }
     return visitor;
   }
-  function callVisitor(key2, node, visitor, path2) {
+  function callVisitor(key2, node, visitor, path) {
     if (typeof visitor === "function")
-      return visitor(key2, node, path2);
+      return visitor(key2, node, path);
     if (identity2.isMap(node))
-      return visitor.Map?.(key2, node, path2);
+      return visitor.Map?.(key2, node, path);
     if (identity2.isSeq(node))
-      return visitor.Seq?.(key2, node, path2);
+      return visitor.Seq?.(key2, node, path);
     if (identity2.isPair(node))
-      return visitor.Pair?.(key2, node, path2);
+      return visitor.Pair?.(key2, node, path);
     if (identity2.isScalar(node))
-      return visitor.Scalar?.(key2, node, path2);
+      return visitor.Scalar?.(key2, node, path);
     if (identity2.isAlias(node))
-      return visitor.Alias?.(key2, node, path2);
+      return visitor.Alias?.(key2, node, path);
     return;
   }
-  function replaceNode(key2, path2, node) {
-    const parent = path2[path2.length - 1];
+  function replaceNode(key2, path, node) {
+    const parent = path[path.length - 1];
     if (identity2.isCollection(parent)) {
       parent.items[key2] = node;
     } else if (identity2.isPair(parent)) {
@@ -32280,10 +32280,10 @@ var require_Collection3 = __commonJS((exports) => {
   var createNode = require_createNode3();
   var identity2 = require_identity3();
   var Node = require_Node3();
-  function collectionFromPath(schema, path2, value) {
+  function collectionFromPath(schema, path, value) {
     let v2 = value;
-    for (let i2 = path2.length - 1;i2 >= 0; --i2) {
-      const k2 = path2[i2];
+    for (let i2 = path.length - 1;i2 >= 0; --i2) {
+      const k2 = path[i2];
       if (typeof k2 === "number" && Number.isInteger(k2) && k2 >= 0) {
         const a2 = [];
         a2[k2] = v2;
@@ -32302,7 +32302,7 @@ var require_Collection3 = __commonJS((exports) => {
       sourceObjects: new Map
     });
   }
-  var isEmptyPath = (path2) => path2 == null || typeof path2 === "object" && !!path2[Symbol.iterator]().next().done;
+  var isEmptyPath = (path) => path == null || typeof path === "object" && !!path[Symbol.iterator]().next().done;
 
   class Collection extends Node.NodeBase {
     constructor(type, schema) {
@@ -32323,11 +32323,11 @@ var require_Collection3 = __commonJS((exports) => {
         copy.range = this.range.slice();
       return copy;
     }
-    addIn(path2, value) {
-      if (isEmptyPath(path2))
+    addIn(path, value) {
+      if (isEmptyPath(path))
         this.add(value);
       else {
-        const [key2, ...rest] = path2;
+        const [key2, ...rest] = path;
         const node = this.get(key2, true);
         if (identity2.isCollection(node))
           node.addIn(rest, value);
@@ -32337,8 +32337,8 @@ var require_Collection3 = __commonJS((exports) => {
           throw new Error(`Expected YAML collection at ${key2}. Remaining path: ${rest}`);
       }
     }
-    deleteIn(path2) {
-      const [key2, ...rest] = path2;
+    deleteIn(path) {
+      const [key2, ...rest] = path;
       if (rest.length === 0)
         return this.delete(key2);
       const node = this.get(key2, true);
@@ -32347,8 +32347,8 @@ var require_Collection3 = __commonJS((exports) => {
       else
         throw new Error(`Expected YAML collection at ${key2}. Remaining path: ${rest}`);
     }
-    getIn(path2, keepScalar) {
-      const [key2, ...rest] = path2;
+    getIn(path, keepScalar) {
+      const [key2, ...rest] = path;
       const node = this.get(key2, true);
       if (rest.length === 0)
         return !keepScalar && identity2.isScalar(node) ? node.value : node;
@@ -32363,15 +32363,15 @@ var require_Collection3 = __commonJS((exports) => {
         return n2 == null || allowScalar && identity2.isScalar(n2) && n2.value == null && !n2.commentBefore && !n2.comment && !n2.tag;
       });
     }
-    hasIn(path2) {
-      const [key2, ...rest] = path2;
+    hasIn(path) {
+      const [key2, ...rest] = path;
       if (rest.length === 0)
         return this.has(key2);
       const node = this.get(key2, true);
       return identity2.isCollection(node) ? node.hasIn(rest) : false;
     }
-    setIn(path2, value) {
-      const [key2, ...rest] = path2;
+    setIn(path, value) {
+      const [key2, ...rest] = path;
       if (rest.length === 0) {
         this.set(key2, value);
       } else {
@@ -34753,9 +34753,9 @@ var require_Document3 = __commonJS((exports) => {
       if (assertCollection(this.contents))
         this.contents.add(value);
     }
-    addIn(path2, value) {
+    addIn(path, value) {
       if (assertCollection(this.contents))
-        this.contents.addIn(path2, value);
+        this.contents.addIn(path, value);
     }
     createAlias(node, name) {
       if (!node.anchor) {
@@ -34804,30 +34804,30 @@ var require_Document3 = __commonJS((exports) => {
     delete(key2) {
       return assertCollection(this.contents) ? this.contents.delete(key2) : false;
     }
-    deleteIn(path2) {
-      if (Collection.isEmptyPath(path2)) {
+    deleteIn(path) {
+      if (Collection.isEmptyPath(path)) {
         if (this.contents == null)
           return false;
         this.contents = null;
         return true;
       }
-      return assertCollection(this.contents) ? this.contents.deleteIn(path2) : false;
+      return assertCollection(this.contents) ? this.contents.deleteIn(path) : false;
     }
     get(key2, keepScalar) {
       return identity2.isCollection(this.contents) ? this.contents.get(key2, keepScalar) : undefined;
     }
-    getIn(path2, keepScalar) {
-      if (Collection.isEmptyPath(path2))
+    getIn(path, keepScalar) {
+      if (Collection.isEmptyPath(path))
         return !keepScalar && identity2.isScalar(this.contents) ? this.contents.value : this.contents;
-      return identity2.isCollection(this.contents) ? this.contents.getIn(path2, keepScalar) : undefined;
+      return identity2.isCollection(this.contents) ? this.contents.getIn(path, keepScalar) : undefined;
     }
     has(key2) {
       return identity2.isCollection(this.contents) ? this.contents.has(key2) : false;
     }
-    hasIn(path2) {
-      if (Collection.isEmptyPath(path2))
+    hasIn(path) {
+      if (Collection.isEmptyPath(path))
         return this.contents !== undefined;
-      return identity2.isCollection(this.contents) ? this.contents.hasIn(path2) : false;
+      return identity2.isCollection(this.contents) ? this.contents.hasIn(path) : false;
     }
     set(key2, value) {
       if (this.contents == null) {
@@ -34836,13 +34836,13 @@ var require_Document3 = __commonJS((exports) => {
         this.contents.set(key2, value);
       }
     }
-    setIn(path2, value) {
-      if (Collection.isEmptyPath(path2)) {
+    setIn(path, value) {
+      if (Collection.isEmptyPath(path)) {
         this.contents = value;
       } else if (this.contents == null) {
-        this.contents = Collection.collectionFromPath(this.schema, Array.from(path2), value);
+        this.contents = Collection.collectionFromPath(this.schema, Array.from(path), value);
       } else if (assertCollection(this.contents)) {
-        this.contents.setIn(path2, value);
+        this.contents.setIn(path, value);
       }
     }
     setSchema(version2, options = {}) {
@@ -36734,9 +36734,9 @@ var require_cst_visit3 = __commonJS((exports) => {
   visit2.BREAK = BREAK;
   visit2.SKIP = SKIP;
   visit2.REMOVE = REMOVE;
-  visit2.itemAtPath = (cst2, path2) => {
+  visit2.itemAtPath = (cst2, path) => {
     let item = cst2;
-    for (const [field, index] of path2) {
+    for (const [field, index] of path) {
       const tok = item?.[field];
       if (tok && "items" in tok) {
         item = tok.items[index];
@@ -36745,23 +36745,23 @@ var require_cst_visit3 = __commonJS((exports) => {
     }
     return item;
   };
-  visit2.parentCollection = (cst2, path2) => {
-    const parent = visit2.itemAtPath(cst2, path2.slice(0, -1));
-    const field = path2[path2.length - 1][0];
+  visit2.parentCollection = (cst2, path) => {
+    const parent = visit2.itemAtPath(cst2, path.slice(0, -1));
+    const field = path[path.length - 1][0];
     const coll = parent?.[field];
     if (coll && "items" in coll)
       return coll;
     throw new Error("Parent collection not found");
   };
-  function _visit(path2, item, visitor) {
-    let ctrl = visitor(item, path2);
+  function _visit(path, item, visitor) {
+    let ctrl = visitor(item, path);
     if (typeof ctrl === "symbol")
       return ctrl;
     for (const field of ["key", "value"]) {
       const token = item[field];
       if (token && "items" in token) {
         for (let i2 = 0;i2 < token.items.length; ++i2) {
-          const ci = _visit(Object.freeze(path2.concat([[field, i2]])), token.items[i2], visitor);
+          const ci = _visit(Object.freeze(path.concat([[field, i2]])), token.items[i2], visitor);
           if (typeof ci === "number")
             i2 = ci - 1;
           else if (ci === BREAK)
@@ -36772,10 +36772,10 @@ var require_cst_visit3 = __commonJS((exports) => {
           }
         }
         if (typeof ctrl === "function" && field === "key")
-          ctrl = ctrl(item, path2);
+          ctrl = ctrl(item, path);
       }
     }
-    return typeof ctrl === "function" ? ctrl(item, path2) : ctrl;
+    return typeof ctrl === "function" ? ctrl(item, path) : ctrl;
   }
   exports.visit = visit2;
 });
@@ -38447,7 +38447,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.8.12";
+var version = "0.8.14";
 
 // ../core/src/AiService/AiServiceBase.ts
 class AiServiceBase {
@@ -48240,11 +48240,11 @@ var delegate_default = {
 // ../core/src/tools/executeCommand.ts
 var toolInfo4 = {
   name: "execute_command",
-  description: "Run a single CLI command. The command is always executed in the project-root working directory (regardless of earlier commands). Prefer one-off shell commands over wrapper scripts for flexibility. After an `execute_command` call, no other tool calls are allowed in the same assistant response.",
+  description: "Run a single CLI command. The command is always executed in the project-root working directory (regardless of earlier commands). Prefer one-off shell commands over wrapper scripts for flexibility. **IMPORTANT**: After an `execute_command` call, you MUST stop and NOT allowed to make further tool calls in the same message.",
   parameters: [
     {
       name: "command",
-      description: "The exact command to run (valid for the current OS). It must be correctly formatted and free of harmful instructions.",
+      description: "The exact command to run  (valid for the current OS). It must be correctly formatted and free of harmful instructions.",
       required: true,
       usageValue: "your-command-here"
     },
@@ -49520,7 +49520,7 @@ var toolUsePrompt = (tools, toolNamePrefix) => {
 
 TOOL USE
 
-You have access to a set of tools that are executed upon the user's approval. You can use one tool per message, and will receive the result of that tool use in the user's response. You use tools step-by-step to accomplish a given task, with each tool use informed by the result of the previous tool use.
+You have access to a set of tools that are executed upon the user's approval. You can use up to 5 tool calls per message, and will receive the results of those tool uses in the user's response. You use tools step-by-step to accomplish a given task, with each tool use informed by the result of the previous tool use.
 
 # Tool Use Formatting
 
@@ -49586,18 +49586,16 @@ ${tools.map((tool) => {
 
 1. **Outline Your Thought Process**
   - Before using a tool, wrap your reasoning inside \`<thinking>\` tags. Be concise—just enough to clarify your plan and the rationale behind selecting a specific tool.
-
 2. **Wait for Feedback**
   - After using a tool, wait for the user's response indicating success/failure or any output logs. Do not assume the result of a tool without explicit confirmation.
-
 3. **Error Handling**
   - If a tool fails or produces an unexpected result, analyze the error, decide on an alternative approach or tool, and proceed carefully.
-
 4. **Avoid Repetition**
   - Do not quote or repeat previous commands or prompts verbatim. Move the conversation forward by focusing on the latest required action.
-
 5. **No Unnecessary Re-invocations**
-  - Only invoke the same tool again if a genuine need arises (e.g., different parameters or updated context).`;
+  - Only invoke the same tool again if a genuine need arises (e.g., different parameters or updated context).
+6. **Tool Call Limit**
+  - Do not make more than 5 tool calls in a single message.`;
 };
 var agentsPrompt = (agents, name) => `
 ====
@@ -54728,6 +54726,65 @@ var KnowledgeManagementPolicy = (tools) => {
     }
   };
 };
+// ../core/src/Agent/policies/TruncateContext.ts
+var DEFAULT_MAX_TOKENS_ESTIMATE = 8000;
+function getMaxTokens(agent) {
+  const params = agent.parameters || {};
+  if (params.maxTokens && typeof params.maxTokens === "number" && params.maxTokens > 0) {
+    return params.maxTokens;
+  }
+  return DEFAULT_MAX_TOKENS_ESTIMATE;
+}
+function estimateTokens(text) {
+  return Math.ceil(text.length / 4);
+}
+var TruncateContextPolicy = (tools) => {
+  return {
+    name: "truncatecontext" /* TruncateContext */,
+    async onBeforeRequest(agent) {
+      const messages = agent.messages;
+      if (messages.length < 3) {
+        return;
+      }
+      let totalTokens = 0;
+      for (const msg of messages) {
+        if (typeof msg.content === "string") {
+          totalTokens += estimateTokens(msg.content);
+        } else if (Array.isArray(msg.content)) {
+          for (const block of msg.content) {
+            if (typeof block === "object" && "text" in block && typeof block.text === "string") {
+              totalTokens += estimateTokens(block.text);
+            }
+          }
+        }
+      }
+      const maxTokens = getMaxTokens(agent);
+      if (totalTokens <= maxTokens) {
+        return;
+      }
+      const totalMessages = messages.length;
+      const messagesToKeep = Math.ceil(totalMessages / 2);
+      const minKeep = Math.max(2, messagesToKeep);
+      if (minKeep >= totalMessages) {
+        return;
+      }
+      const keepFromStart = Math.floor(minKeep / 2);
+      const keepFromEnd = minKeep - keepFromStart;
+      const startMessages = messages.slice(0, keepFromStart);
+      const endMessages = messages.slice(-keepFromEnd);
+      const truncatedCount = totalMessages - minKeep;
+      const truncatedMessages = [
+        ...startMessages,
+        {
+          role: "user",
+          content: `Note: ${truncatedCount} messages were truncated from the middle to prevent context overflow.`
+        },
+        ...endMessages
+      ];
+      agent.setMessages(truncatedMessages);
+    }
+  };
+};
 // ../core/src/Agent/index.ts
 var allAgents = [architectAgentInfo, coderAgentInfo, analyzerAgentInfo, codeFixerAgentInfo];
 // ../core/src/AiTool/createNewProject.ts
@@ -60689,14 +60746,8 @@ async function listFiles(dirPath, recursive, maxCount, cwd, excludeFiles) {
 
 // ../cli-shared/src/utils/searchFiles.ts
 import { spawn } from "node:child_process";
-
-// ../../node_modules/@vscode/ripgrep/lib/index.js
-var __dirname = "/Users/xiliangchen/projects/polka-codes/node_modules/@vscode/ripgrep/lib";
-var path = __require("path");
-var $rgPath = path.join(__dirname, `../bin/rg${process.platform === "win32" ? ".exe" : ""}`);
-
-// ../cli-shared/src/utils/searchFiles.ts
-async function searchFiles(path2, regex, filePattern, cwd, excludeFiles) {
+import { rgPath } from "@vscode/ripgrep";
+async function searchFiles(path, regex, filePattern, cwd, excludeFiles) {
   const args = [
     "--line-number",
     "--context=5",
@@ -60717,10 +60768,10 @@ async function searchFiles(path2, regex, filePattern, cwd, excludeFiles) {
       args.push("--glob", `!${pattern}`);
     }
   }
-  args.push(regex, path2);
+  args.push(regex, path);
   return new Promise((resolve2, reject) => {
     const results = [];
-    const rg = spawn($rgPath, args, {
+    const rg = spawn(rgPath, args, {
       cwd,
       stdio: ["ignore", "pipe", "pipe"]
     });
@@ -60746,28 +60797,28 @@ async function searchFiles(path2, regex, filePattern, cwd, excludeFiles) {
 var getProvider = (agentName, config2, options = {}) => {
   const ig = import_ignore2.default().add(options.excludeFiles ?? []);
   const provider2 = {
-    readFile: async (path2) => {
-      if (ig.ignores(path2)) {
-        throw new Error(`Not allow to access file ${path2}`);
+    readFile: async (path) => {
+      if (ig.ignores(path)) {
+        throw new Error(`Not allow to access file ${path}`);
       }
       try {
-        return await readFile(path2, "utf8");
+        return await readFile(path, "utf8");
       } catch (_e2) {
         return;
       }
     },
-    writeFile: async (path2, content) => {
-      if (ig.ignores(path2)) {
-        throw new Error(`Not allow to access file ${path2}`);
+    writeFile: async (path, content) => {
+      if (ig.ignores(path)) {
+        throw new Error(`Not allow to access file ${path}`);
       }
-      await mkdir(dirname2(path2), { recursive: true });
-      return await writeFile(path2, content, "utf8");
+      await mkdir(dirname2(path), { recursive: true });
+      return await writeFile(path, content, "utf8");
     },
-    removeFile: async (path2) => {
-      if (ig.ignores(path2)) {
-        throw new Error(`Not allow to access file ${path2}`);
+    removeFile: async (path) => {
+      if (ig.ignores(path)) {
+        throw new Error(`Not allow to access file ${path}`);
       }
-      return await unlink(path2);
+      return await unlink(path);
     },
     renameFile: async (sourcePath, targetPath) => {
       if (ig.ignores(sourcePath) || ig.ignores(targetPath)) {
@@ -60775,11 +60826,11 @@ var getProvider = (agentName, config2, options = {}) => {
       }
       return await rename(sourcePath, targetPath);
     },
-    listFiles: async (path2, recursive, maxCount) => {
-      return await listFiles(path2, recursive, maxCount, process.cwd(), options.excludeFiles);
+    listFiles: async (path, recursive, maxCount) => {
+      return await listFiles(path, recursive, maxCount, process.cwd(), options.excludeFiles);
     },
-    searchFiles: async (path2, regex, filePattern) => {
-      return await searchFiles(path2, regex, filePattern, process.cwd(), options.excludeFiles);
+    searchFiles: async (path, regex, filePattern) => {
+      return await searchFiles(path, regex, filePattern, process.cwd(), options.excludeFiles);
     },
     executeCommand: (command, needApprove) => {
       return new Promise((resolve2, reject) => {
@@ -61503,6 +61554,10 @@ class Runner {
           this.#hasKnowledgeManagementPolicy = true;
           console.log("KnowledgeManagementPolicy enabled");
           break;
+        case "truncatecontext" /* TruncateContext */:
+          policies2.push(TruncateContextPolicy);
+          console.log("TruncateContextPolicy enabled");
+          break;
         default:
           console.log("Unknown policy:", policy);
           break;
@@ -63005,8 +63060,8 @@ function getErrorMap3() {
   return overrideErrorMap3;
 }
 var makeIssue3 = (params) => {
-  const { data, path: path2, errorMaps, issueData } = params;
-  const fullPath = [...path2, ...issueData.path || []];
+  const { data, path, errorMaps, issueData } = params;
+  const fullPath = [...path, ...issueData.path || []];
   const fullIssue = {
     ...issueData,
     path: fullPath
@@ -63133,11 +63188,11 @@ var _ZodEnum_cache3;
 var _ZodNativeEnum_cache3;
 
 class ParseInputLazyPath3 {
-  constructor(parent, value, path2, key2) {
+  constructor(parent, value, path, key2) {
     this._cachedPath = [];
     this.parent = parent;
     this.data = value;
-    this._path = path2;
+    this._path = path;
     this._key = key2;
   }
   get path() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/cli",
-  "version": "0.8.12",
+  "version": "0.8.14",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",
@@ -14,7 +14,7 @@
     }
   },
   "scripts": {
-    "build": "bun build src/index.ts --outdir dist --target node"
+    "build": "bun build src/index.ts --outdir dist --target node --external=@vscode/ripgrep"
   },
   "dependencies": {
     "@inquirer/prompts": "^7.2.3",