@polka-codes/core 0.6.5 → 0.6.6

package/dist/index.js

@@ -8237,6 +8237,25 @@ __export(exports_allTools, {
   askFollowupQuestion: () => askFollowupQuestion_default
 });
 
+// src/tool.ts
+var PermissionLevel;
+((PermissionLevel2) => {
+  PermissionLevel2[PermissionLevel2["None"] = 0] = "None";
+  PermissionLevel2[PermissionLevel2["Read"] = 1] = "Read";
+  PermissionLevel2[PermissionLevel2["Write"] = 2] = "Write";
+  PermissionLevel2[PermissionLevel2["Arbitrary"] = 3] = "Arbitrary";
+})(PermissionLevel ||= {});
+var ToolResponseType;
+((ToolResponseType2) => {
+  ToolResponseType2["Reply"] = "Reply";
+  ToolResponseType2["Exit"] = "Exit";
+  ToolResponseType2["Invalid"] = "Invalid";
+  ToolResponseType2["Error"] = "Error";
+  ToolResponseType2["Interrupted"] = "Interrupted";
+  ToolResponseType2["HandOver"] = "HandOver";
+  ToolResponseType2["Delegate"] = "Delegate";
+})(ToolResponseType ||= {});
+
 // src/tools/utils/replaceInFile.ts
 var replaceInFile = async (fileContent, diff) => {
   const blockPattern = /<<<<<+ SEARCH\s*\r?\n([\s\S]*?)\r?\n=======[ \t]*\r?\n([\s\S]*?)\r?\n?>>>>>+ REPLACE/g;
@@ -8385,7 +8404,8 @@ var toolInfo = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 0 /* None */
 };
 var handler = async (provider, args) => {
   if (!provider.askFollowupQuestion) {
@@ -8433,7 +8453,8 @@ var toolInfo2 = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 0 /* None */
 };
 var handler2 = async (provider, args) => {
   const result = getString(args, "result");
@@ -8509,7 +8530,8 @@ var toolInfo3 = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 0 /* None */
 };
 var handler3 = async (_provider, args) => {
   const agentName = getString(args, "agent_name");
@@ -8564,7 +8586,8 @@ var toolInfo4 = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 3 /* Arbitrary */
 };
 var handler4 = async (provider, args) => {
   if (!provider.executeCommand) {
@@ -8606,7 +8629,7 @@ var executeCommand_default = {
 // src/tools/listCodeDefinitionNames.ts
 var toolInfo5 = {
   name: "list_code_definition_names",
-  description: "Request to list definition names (classes, functions, methods, etc.) used in a file. This tool provides insights into the codebase structure and important constructs, encapsulating high-level concepts and relationships that are crucial for understanding the overall architecture.",
+  description: "Request to list definition names (classes, functions, methods, etc.) used for all files in a directory. This tool provides insights into the codebase structure and important constructs, encapsulating high-level concepts and relationships that are crucial for understanding the overall architecture.",
   parameters: [
     {
       name: "path",
@@ -8614,7 +8637,19 @@ var toolInfo5 = {
       required: true,
       usageValue: "Directory path here"
     }
-  ]
+  ],
+  examples: [
+    {
+      description: "Request to list code definition names in a directory",
+      parameters: [
+        {
+          name: "path",
+          value: "src/utils"
+        }
+      ]
+    }
+  ],
+  permissionLevel: 1 /* Read */
 };
 var handler5 = async (provider, args) => {
   if (!provider.listCodeDefinitionNames) {
@@ -8679,7 +8714,8 @@ var toolInfo6 = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 1 /* Read */
 };
 var handler6 = async (provider, args) => {
   if (!provider.listFiles) {
@@ -8741,7 +8777,8 @@ var toolInfo7 = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 1 /* Read */
 };
 var handler7 = async (provider, args) => {
   if (!provider.readFile) {
@@ -8858,7 +8895,8 @@ return (
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 2 /* Write */
 };
 var handler8 = async (provider, args) => {
   if (!provider.readFile || !provider.writeFile) {
@@ -8927,7 +8965,8 @@ var toolInfo9 = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 1 /* Read */
 };
 var handler9 = async (provider, args) => {
   if (!provider.searchFiles) {
@@ -9003,7 +9042,8 @@ export default App;
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 2 /* Write */
 };
 var handler10 = async (provider, args) => {
   if (!provider.writeFile) {
@@ -9080,7 +9120,8 @@ var toolInfo11 = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 0 /* None */
 };
 var handler11 = async (_provider, args) => {
   const agentName = getString(args, "agent_name");
@@ -9125,7 +9166,8 @@ var toolInfo12 = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 2 /* Write */
 };
 var handler12 = async (provider, args) => {
   if (!provider.removeFile) {
@@ -9181,7 +9223,8 @@ var toolInfo13 = {
         }
       ]
     }
-  ]
+  ],
+  permissionLevel: 2 /* Write */
 };
 var handler13 = async (provider, args) => {
   if (!provider.renameFile) {
@@ -9206,18 +9249,8 @@ var renameFile_default = {
   handler: handler13,
   isAvailable: isAvailable13
 };
-// src/tool.ts
-var ToolResponseType;
-((ToolResponseType2) => {
-  ToolResponseType2["Reply"] = "Reply";
-  ToolResponseType2["Exit"] = "Exit";
-  ToolResponseType2["Invalid"] = "Invalid";
-  ToolResponseType2["Error"] = "Error";
-  ToolResponseType2["Interrupted"] = "Interrupted";
-  ToolResponseType2["HandOver"] = "HandOver";
-  ToolResponseType2["Delegate"] = "Delegate";
-})(ToolResponseType ||= {});
-var getAvailableTools = (provider2, allTools2, hasAgent) => {
+// src/getAvailableTools.ts
+var getAvailableTools = (provider2, allTools2, hasAgent, permissionLevel) => {
   const tools = [];
   for (const tool of allTools2) {
     if (!hasAgent) {
@@ -9227,7 +9260,7 @@ var getAvailableTools = (provider2, allTools2, hasAgent) => {
           continue;
       }
     }
-    if (tool.isAvailable(provider2)) {
+    if (tool.isAvailable(provider2) && tool.permissionLevel <= permissionLevel) {
       tools.push(tool);
     }
   }
@@ -9561,6 +9594,9 @@ ${agents}`;
     return await this.#processLoop(userMessage);
   }
   async#request(userMessage) {
+    if (!userMessage) {
+      throw new Error("userMessage is missing");
+    }
     await this.#callback({ kind: "StartRequest" /* StartRequest */, agent: this, userMessage });
     this.messages.push({
       role: "user",
@@ -9609,6 +9645,12 @@ ${agents}`;
           case "text":
             break;
           case "tool_use": {
+            if (toolReponses.length > 0) {
+              const toolLevel = this.handlers[content.name].permissionLevel;
+              if (toolLevel > 1 /* Read */) {
+                break outer;
+              }
+            }
             await this.#callback({ kind: "ToolUse" /* ToolUse */, agent: this, tool: content.name });
             const toolResp = await this.#invokeTool(content.name, content.params);
             switch (toolResp.type) {
@@ -9631,7 +9673,7 @@ ${agents}`;
                 return { exit: toolResp };
               case "HandOver" /* HandOver */:
                 if (toolReponses.length > 0) {
-                  continue;
+                  break outer;
                 }
                 await this.#callback({
                   kind: "ToolHandOver" /* ToolHandOver */,
@@ -9662,7 +9704,7 @@ ${agents}`;
           }
         }
       }
-    if (toolReponses.length === 0 && !this.config.interactive) {
+    if (toolReponses.length === 0) {
       return { replay: responsePrompts.requireUseTool };
     }
     const finalResp = toolReponses.map(({ tool, response: response2 }) => responsePrompts.toolResults(tool, response2)).join(`
@@ -9734,18 +9776,8 @@ ${interactiveMode(interactive)}
 // src/Agent/AnalyzerAgent/index.ts
 class AnalyzerAgent extends AgentBase {
   constructor(options) {
-    const agentTools = [
-      ...options.additionalTools ?? [],
-      askFollowupQuestion_default,
-      attemptCompletion_default,
-      handOver_default,
-      delegate_default,
-      listCodeDefinitionNames_default,
-      listFiles_default,
-      readFile_default,
-      searchFiles_default
-    ];
-    const tools = getAvailableTools(options.provider, agentTools, (options.agents?.length ?? 0) > 0);
+    const combinedTools = [...options.additionalTools ?? [], ...Object.values(exports_allTools)];
+    const tools = getAvailableTools(options.provider, combinedTools, (options.agents?.length ?? 0) > 0, 1 /* Read */);
     const toolNamePrefix = "tool_";
     const systemPrompt = fullSystemPrompt({
       os: options.os
@@ -9834,18 +9866,8 @@ ${interactiveMode(interactive)}
 // src/Agent/ArchitectAgent/index.ts
 class ArchitectAgent extends AgentBase {
   constructor(options) {
-    const agentTools = [
-      ...options.additionalTools ?? [],
-      askFollowupQuestion_default,
-      attemptCompletion_default,
-      handOver_default,
-      delegate_default,
-      listCodeDefinitionNames_default,
-      listFiles_default,
-      readFile_default,
-      searchFiles_default
-    ];
-    const tools = getAvailableTools(options.provider, agentTools, (options.agents?.length ?? 0) > 0);
+    const combinedTools = [...options.additionalTools ?? [], ...Object.values(exports_allTools)];
+    const tools = getAvailableTools(options.provider, combinedTools, (options.agents?.length ?? 0) > 0, 1 /* Read */);
     const toolNamePrefix = "tool_";
     const systemPrompt = fullSystemPrompt2({
       os: options.os
@@ -9958,7 +9980,7 @@ class CodeFixerAgent extends AgentBase {
   #retryCount = 0;
   constructor(options) {
     const combinedTools = [...options.additionalTools ?? [], ...Object.values(exports_allTools)];
-    const tools = getAvailableTools(options.provider, combinedTools, (options.agents?.length ?? 0) > 0);
+    const tools = getAvailableTools(options.provider, combinedTools, (options.agents?.length ?? 0) > 0, 3 /* Arbitrary */);
     const toolNamePrefix = "tool_";
     const systemPrompt = fullSystemPrompt3({
       os: options.os
@@ -10146,7 +10168,7 @@ ${interactiveMode(interactive)}
 class CoderAgent extends AgentBase {
   constructor(options) {
     const combinedTools = [...options.additionalTools ?? [], ...Object.values(exports_allTools)];
-    const tools = getAvailableTools(options.provider, combinedTools, (options.agents?.length ?? 0) > 0);
+    const tools = getAvailableTools(options.provider, combinedTools, (options.agents?.length ?? 0) > 0, 3 /* Arbitrary */);
     const toolNamePrefix = "tool_";
     const systemPrompt = fullSystemPrompt4({
       os: options.os
@@ -10685,6 +10707,7 @@ export {
   UsageMeter,
   ToolResponseType,
   TaskEventKind,
+  PermissionLevel,
   MultiAgent,
   MockProvider,
   CoderAgent,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.6.5",
+  "version": "0.6.6",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",