@polka-codes/core 0.5.2 → 0.5.4

package/dist/index.js

@@ -8207,7 +8207,7 @@ class MockProvider {
     return ["mock-file.txt"];
   }
   async listCodeDefinitionNames(path) {
-    return ["mockDefinition"];
+    return "mockDefinition";
   }
   async executeCommand(command, needApprove) {
     return { stdout: "mock output", stderr: "", exitCode: 0 };
@@ -8624,14 +8624,13 @@ var handler5 = async (provider, args) => {
     };
   }
   const path = getString(args, "path");
-  const files = await provider.listCodeDefinitionNames(path);
+  const result = await provider.listCodeDefinitionNames(path);
   return {
     type: "Reply" /* Reply */,
     message: `<list_code_definition_names_path>${path}</list_code_definition_names_path>
-<list_code_definition_names_files>
-${files.join(`
-`)}
-</list_code_definition_names_files>`
+<list_code_definition_names_result>
+${result}
+</list_code_definition_names_result>`
   };
 };
 var isAvailable5 = (provider) => {
@@ -9570,11 +9569,9 @@ ${agents}`;
     let currentAssistantMessage = "";
     const retryCount = 3;
     for (let i = 0;i < retryCount; i++) {
-      let gotAnything = false;
       currentAssistantMessage = "";
       const stream = this.ai.send(this.config.systemPrompt, this.messages);
       for await (const chunk of stream) {
-        gotAnything = true;
         switch (chunk.type) {
           case "usage":
             await this.#callback({ kind: "Usage" /* Usage */, agent: this });
@@ -9591,9 +9588,10 @@ ${agents}`;
       if (currentAssistantMessage) {
         break;
       }
-      if (gotAnything) {
-        throw new Error("No assistant message received");
-      }
+      console.debug(`Retrying request ${i + 1} of ${retryCount}`);
+    }
+    if (!currentAssistantMessage) {
+      throw new Error("No assistant message received");
     }
     this.messages.push({
       role: "assistant",
@@ -9676,7 +9674,7 @@ ${agents}`;
           canRetry: false
         };
       }
-      const resp = await this.onBeforeInvokeTool(name, args);
+      const resp = await this.onBeforeInvokeTool(this.handlers[name].name, args);
       if (resp) {
         return resp;
       }
@@ -9871,8 +9869,161 @@ var architectAgentInfo = {
   ]
 };
 
+// src/Agent/CodeFixerAgent/prompts.ts
+var basePrompt = `You are a highly skilled software engineer specializing in debugging and fixing code issues. You have extensive experience with:
+- Type systems and type checking
+- Test frameworks and debugging test failures
+- Code quality tools and best practices
+- Systematic debugging approaches`;
+var codeFixingStrategies = `
+====
+
+CODE FIXING STRATEGIES
+
+1. Type Errors
+   - Analyze type error messages carefully
+   - Check type definitions and imports
+   - Consider type assertions only as a last resort
+   - Verify type compatibility across function boundaries
+   - Look for null/undefined handling issues
+
+2. Test Failures
+   - Examine test output and error messages
+   - Check test setup and fixtures
+   - Verify assertions and expectations
+   - Look for async/timing issues
+   - Consider edge cases and input validation
+
+3. Code Quality Issues
+   - Follow project's coding standards
+   - Address linter warnings systematically
+   - Improve code readability
+   - Fix potential runtime issues
+   - Consider performance implications
+
+4. General Approach
+   - Start with the most critical issues
+   - Make minimal necessary changes
+   - Verify fixes don't introduce new issues
+   - Document complex fixes with comments
+   - Track attempted solutions for each issue`;
+var retryGuidelines = `
+====
+
+RETRY GUIDELINES
+
+1. Before Retrying
+   - Analyze previous attempt's failure
+   - Consider alternative approaches
+   - Check if similar issues were fixed
+   - Verify no new issues were introduced
+
+2. When to Retry
+   - Error message changed but issue persists
+   - New information available about the root cause
+   - Different fixing strategy available
+   - Previous attempt partially successful
+
+3. When to Stop
+   - Maximum retry limit reached
+   - Same error occurs repeatedly
+   - Fix would require major refactoring
+   - Issue requires human intervention
+
+4. After Maximum Retries
+   - Document attempted solutions
+   - Explain why the issue remains
+   - Suggest manual intervention steps
+   - Report any partial improvements`;
+var fullSystemPrompt3 = (info, tools, toolNamePrefix, instructions, scripts, interactive) => `
+${basePrompt}
+${toolUsePrompt(tools, toolNamePrefix)}
+${codeFixingStrategies}
+${retryGuidelines}
+${capabilities(toolNamePrefix)}
+${systemInformation(info)}
+${customInstructions(instructions)}
+${customScripts(scripts)}
+${interactiveMode(interactive)}`;
+
+// src/Agent/CodeFixerAgent/index.ts
+class CodeFixerAgent extends AgentBase {
+  #maxRetries;
+  #retryCount = 0;
+  constructor(options) {
+    const combinedTools = [...options.additionalTools ?? [], ...Object.values(exports_allTools)];
+    const tools = getAvailableTools(options.provider, combinedTools, (options.agents?.length ?? 0) > 0);
+    const toolNamePrefix = "tool_";
+    const systemPrompt = fullSystemPrompt3({
+      os: options.os
+    }, tools, toolNamePrefix, options.customInstructions ?? [], options.scripts ?? {}, options.interactive);
+    super(codeFixerAgentInfo.name, options.ai, {
+      systemPrompt,
+      tools,
+      toolNamePrefix,
+      provider: options.provider,
+      interactive: options.interactive,
+      agents: options.agents,
+      scripts: options.scripts,
+      callback: options.callback
+    });
+    this.#maxRetries = options.maxRetries ?? 5;
+  }
+  async onBeforeInvokeTool(name, args) {
+    if (name === attemptCompletion_default.name) {
+      if (this.#retryCount > this.#maxRetries) {
+        return;
+      }
+      this.#retryCount++;
+      const executeCommand = this.config.provider.executeCommand;
+      if (!executeCommand) {
+        return;
+      }
+      const check = this.config.scripts?.check;
+      const checkCommand = typeof check === "string" ? check : check?.command;
+      if (checkCommand) {
+        try {
+          const { exitCode, stdout, stderr } = await executeCommand(checkCommand, false);
+          if (exitCode !== 0) {
+            return {
+              type: "Reply" /* Reply */,
+              message: responsePrompts.commandResult(checkCommand, exitCode, stdout, stderr)
+            };
+          }
+        } catch (error) {
+          console.warn(`Failed to check code using command: ${checkCommand}`, error);
+        }
+      }
+      const test = this.config.scripts?.test;
+      const testCommand = typeof test === "string" ? test : test?.command;
+      if (testCommand) {
+        try {
+          const { exitCode, stdout, stderr } = await executeCommand(testCommand, false);
+          if (exitCode !== 0) {
+            return {
+              type: "Reply" /* Reply */,
+              message: responsePrompts.commandResult(testCommand, exitCode, stdout, stderr)
+            };
+          }
+        } catch (error) {
+          console.warn(`Failed to test code using command: ${testCommand}`, error);
+        }
+      }
+    }
+  }
+}
+var codeFixerAgentInfo = {
+  name: "codefixer",
+  responsibilities: [
+    "Fixing type errors and type-related issues",
+    "Resolving failing tests",
+    "Addressing code quality issues",
+    "Tracking and reporting unfixed issues"
+  ]
+};
+
 // src/Agent/CoderAgent/prompts.ts
-var basePrompt = "You are a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.";
+var basePrompt2 = "You are a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.";
 var editingFilesPrompt = (toolNamePrefix) => `
 ====
 
@@ -9972,8 +10123,8 @@ You accomplish a given task iteratively, breaking it down into clear steps and w
 3. Remember, you have extensive capabilities with access to a wide range of tools that can be used in powerful and clever ways as necessary to accomplish each goal. Before calling a tool, do some analysis within <thinking></thinking> tags. First, analyze the file structure provided in environment_details to gain context and insights for proceeding effectively. Then, think about which of the provided tools is the most relevant tool to accomplish the user's task. Next, go through each of the required parameters of the relevant tool and determine if the user has directly provided or given enough information to infer a value. When deciding if the parameter can be inferred, carefully consider all the context to see if it supports a specific value. If all of the required parameters are present or can be reasonably inferred, close the thinking tag and proceed with the tool use.
 4. Once you've completed the user's task, you must use the ${toolNamePrefix}attempt_completion tool to present the result of the task to the user.
 5. The user may provide feedback, which you can use to make improvements and try again. But DO NOT continue in pointless back and forth conversations, i.e. don't end your responses with questions or offers for further assistance.`;
-var fullSystemPrompt3 = (info, tools, toolNamePrefix, instructions, scripts, interactive) => `
-${basePrompt}
+var fullSystemPrompt4 = (info, tools, toolNamePrefix, instructions, scripts, interactive) => `
+${basePrompt2}
 ${toolUsePrompt(tools, toolNamePrefix)}
 ${editingFilesPrompt(toolNamePrefix)}
 ${capabilities(toolNamePrefix)}
@@ -9991,7 +10142,7 @@ class CoderAgent extends AgentBase {
     const combinedTools = [...options.additionalTools ?? [], ...Object.values(exports_allTools)];
     const tools = getAvailableTools(options.provider, combinedTools, (options.agents?.length ?? 0) > 0);
     const toolNamePrefix = "tool_";
-    const systemPrompt = fullSystemPrompt3({
+    const systemPrompt = fullSystemPrompt4({
       os: options.os
     }, tools, toolNamePrefix, options.customInstructions ?? [], options.scripts ?? {}, options.interactive);
     super(coderAgentInfo.name, options.ai, {
@@ -10006,6 +10157,9 @@ class CoderAgent extends AgentBase {
     });
   }
   async onBeforeInvokeTool(name, args) {
+    if (name !== attemptCompletion_default.name) {
+      return;
+    }
     const executeCommand = this.config.provider.executeCommand;
     if (!executeCommand) {
       return;
@@ -10122,7 +10276,7 @@ class MultiAgent {
 }
 
 // src/Agent/index.ts
-var allAgents = [architectAgentInfo, coderAgentInfo, analyzerAgentInfo];
+var allAgents = [architectAgentInfo, coderAgentInfo, analyzerAgentInfo, codeFixerAgentInfo];
 // src/AiTool/createNewProject.ts
 var prompt = `You are an AiTool designed to assist users in creating new projects. Follow these guidelines:
 
@@ -10515,6 +10669,7 @@ export {
   createService,
   createNewProject,
   coderAgentInfo,
+  codeFixerAgentInfo,
   attemptCompletion_default as attemptCompletion,
   askFollowupQuestion_default as askFollowupQuestion,
   architectAgentInfo,
@@ -10529,6 +10684,7 @@ export {
   MultiAgent,
   MockProvider,
   CoderAgent,
+  CodeFixerAgent,
   ArchitectAgent,
   AnalyzerAgent,
   AiServiceProvider,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.5.2",
+  "version": "0.5.4",
   "license": "AGPL-3.0",
   "type": "module",
   "exports": {