@polka-codes/cli 0.8.25 → 0.8.27

package/dist/index.js

@@ -38447,7 +38447,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.8.25";
+var version = "0.8.27";
 
 // ../core/src/AiService/AiServiceBase.ts
 class AiServiceBase {
@@ -48629,14 +48629,18 @@ var handler8 = async (provider, args) => {
   if (result.status === "no_diff_applied") {
     return {
       type: "Error" /* Error */,
-      message: `<replace_in_file_result path="${path}" status="failed" message="Unable to apply changes" />`
+      message: `<replace_in_file_result path="${path}" status="failed" message="Unable to apply changes">
+  <file_content path="${path}">${fileContent}</file_content>
+</replace_in_file_result>`
     };
   }
   await provider.writeFile(path, result.content);
   if (result.status === "some_diff_applied") {
     return {
       type: "Reply" /* Reply */,
-      message: `<replace_in_file_result path="${path}" status="some_diff_applied" applied_count="${result.appliedCount}" total_count="${result.totalCount}" />`
+      message: `<replace_in_file_result path="${path}" status="some_diff_applied" applied_count="${result.appliedCount}" total_count="${result.totalCount}">
+  <file_content path="${path}">${result.content}</file_content>
+</replace_in_file_result>`
     };
   }
   return {
@@ -49800,12 +49804,27 @@ ${instance.prompt}`;
       }
     }
     let currentAssistantMessage = "";
-    const retryCount = 5;
+    const retryCount = this.config.retryCount ?? 5;
+    const requestTimeoutSeconds = this.config.requestTimeoutSeconds ?? 10;
     for (let i2 = 0;i2 < retryCount; i2++) {
       currentAssistantMessage = "";
+      let timeout;
+      const resetTimeout = () => {
+        if (timeout) {
+          clearTimeout(timeout);
+        }
+        if (requestTimeoutSeconds > 0) {
+          timeout = setTimeout(() => {
+            console.debug(`No data received for ${requestTimeoutSeconds} seconds. Aborting request.`);
+            this.ai.abort();
+          }, requestTimeoutSeconds * 1000);
+        }
+      };
       const stream = this.ai.send(this.config.systemPrompt, this.#messages);
       try {
+        resetTimeout();
         for await (const chunk of stream) {
+          resetTimeout();
           switch (chunk.type) {
             case "usage":
               await this.#callback({ kind: "Usage" /* Usage */, agent: this });
@@ -49824,6 +49843,10 @@ ${instance.prompt}`;
           break;
         }
         console.error("Error in stream:", error);
+      } finally {
+        if (timeout) {
+          clearTimeout(timeout);
+        }
       }
       if (currentAssistantMessage) {
         break;
@@ -54575,7 +54598,9 @@ var agentSchema = providerModelSchema.extend({
   initialContext: z2.object({
     maxFileCount: z2.number().int().positive().optional(),
     excludes: z2.array(z2.string()).optional()
-  }).optional()
+  }).optional(),
+  retryCount: z2.number().int().min(0).optional(),
+  requestTimeoutSeconds: z2.number().int().positive().optional()
 });
 var configSchema = z2.object({
   agent: z2.string().optional(),
@@ -54595,6 +54620,8 @@ var configSchema = z2.object({
   defaultParameters: z2.record(z2.string(), z2.any()).optional(),
   maxMessageCount: z2.number().int().positive().optional(),
   budget: z2.number().positive().optional(),
+  retryCount: z2.number().int().min(0).optional(),
+  requestTimeoutSeconds: z2.number().int().positive().optional(),
   scripts: z2.record(z2.string(), z2.string().or(z2.object({
     command: z2.string(),
     description: z2.string()
@@ -55076,64 +55103,49 @@ ${output}`);
 };
 
 // ../core/src/AiTool/generateProjectConfig.ts
-var prompt5 = `You are an analyzer agent responsible for examining project files and generating appropriate polkacodes configuration. Your task is to:
-
-1. Read and analyze the provided files using tool_read_file to understand:
-   - Build tools and package manager (e.g., bun, npm)
-   - Testing frameworks and patterns
-   - Code style tools and rules
-   - Project structure and conventions
-   - Common development workflows
-
-2. Generate a YAML configuration that captures:
-   - scripts section based on package.json scripts and CI workflows. If applicable, generate following scripts:
-     - format: Format the code using a code formatter tool
-     - check: Check the code for errors using a linter tool
-     - test: Run tests using a test runner tool
-   - include other relevant scripts based on project conventions, tools, and patterns
-   - rules section based on project conventions, tools, and patterns
-   - excludeFiles section for sensitive files
-
-3. Use tool_attempt_completion to return the final configuration in this format:
+var prompt5 = `
+Role: Analyzer agent
+Goal: Produce a valid polkacodes YAML configuration for the project.
 
-<tool_attempt_completion>
-<tool_parameter_result>
-scripts:
-  test:
-    command: "bun test"
-    description: "Run tests with bun:test"
-  lint:
-    command: "biome check ."
-    description: "Check code style with Biome"
+Workflow
+1. Scan project files with tool_read_file and identify:
+   - Package/build tool (npm, bun, pnpm, etc.)
+   - Test framework and patterns (snapshot tests, coverage, etc.)
+   - Formatter / linter and their rules
+   - Folder structure and naming conventions
+   - CI / development workflows
 
-rules:
-  - "Use \`bun\` as package manager"
-  - "Write tests using bun:test with snapshots"
-  - "Follow Biome code style"
+2. Build a YAML config with three root keys:
 
-excludeFiles:
-  # Sensitive files
+\`\`\`yaml
+scripts:          # derive from package.json and CI
+  format:        # code formatter
+    command: "<formatter cmd>"
+    description: "Format code"
+  check:         # linter / type checker
+    command: "<linter cmd>"
+    description: "Static checks"
+  test:          # test runner
+    command: "<test cmd>"
+    description: "Run tests"
+  # add any other meaningful project scripts
+
+rules:            # bullet list of key conventions/tools
+
+excludeFiles:     # only files likely to hold secrets
   - ".env"
   - ".env.*"
   - ".npmrc"
+  # do NOT list build artifacts, lockfiles, or paths already in .gitignore
+\`\`\`
+
+3. Return the YAML exactly once, wrapped like:
+
+<tool_attempt_completion>
+<tool_parameter_result>
+# YAML (2-space indents, double-quoted commands)
 </tool_parameter_result>
 </tool_attempt_completion>
-
-Focus on:
-- Package manager and dependency management
-- Testing frameworks and patterns
-- Code style and linting rules
-- File organization and naming conventions
-- Build and development workflows
-- Sensitive files that should not be committed:
-  - Environment files (.env*)
-  - Configuration files with potential secrets
-- Generated files to exclude:
-  - Lock files from package managers
-  - Build artifacts and outputs
-  - Generated file that are not ignored by .gitignore
-
-The configuration should accurately reflect the project's structure, tools, and conventions.
 `;
 var generateProjectConfig_default = {
   name: "generateProjectConfig",
@@ -61682,6 +61694,9 @@ class Runner {
     this.multiAgent = new MultiAgent({
       createAgent: async (name) => {
         const agentName = name.trim().toLowerCase();
+        const agentConfig = this.#options.config.agents?.[name] ?? this.#options.config.agents?.default ?? {};
+        const retryCount = agentConfig.retryCount ?? this.#options.config.retryCount;
+        const requestTimeoutSeconds = agentConfig.requestTimeoutSeconds ?? this.#options.config.requestTimeoutSeconds;
         const args = {
           ai: getOrCreateService(agentName),
           os: platform,
@@ -61690,7 +61705,9 @@ class Runner {
           interactive: options.interactive,
           agents: this.#options.availableAgents ?? allAgents,
           callback,
-          policies: policies2
+          policies: policies2,
+          retryCount,
+          requestTimeoutSeconds
         };
         switch (agentName) {
           case coderAgentInfo.name:
@@ -61933,7 +61950,7 @@ function getEnv(override) {
 
 // src/options.ts
 function addSharedOptions(command) {
-  return command.option("-c --config <paths>", "Path to config file(s)", (value, prev) => prev.concat(value), []).option("--api-provider <provider>", "API provider").option("--model <model>", "Model ID").option("--api-key <key>", "API key").option("--max-messages <iterations>", "Maximum number of messages to send. Default to 50", Number.parseInt, 50).option("--budget <budget>", "Budget for the AI service. Default to $10", Number.parseFloat).option("-v --verbose", "Enable verbose output. Use -v for level 1, -vv for level 2", (value, prev) => prev + 1, 0).option("-d --base-dir <path>", "Base directory to run commands in").option("--agent <agent>", "Initial agent to use (default: architect)");
+  return command.option("-c --config <paths>", "Path to config file(s)", (value, prev) => prev.concat(value), []).option("--api-provider <provider>", "API provider").option("--model <model>", "Model ID").option("--api-key <key>", "API key").option("--max-messages <iterations>", "Maximum number of messages to send. Default to 50", Number.parseInt, 50).option("--budget <budget>", "Budget for the AI service. Default to $10", Number.parseFloat).option("--retry-count <count>", "Number of retries for failed requests. Default to 5", Number.parseInt, 5).option("--request-timeout-seconds <seconds>", "Request timeout in seconds. Default to 10", Number.parseInt, 10).option("-v --verbose", "Enable verbose output. Use -v for level 1, -vv for level 2", (value, prev) => prev + 1, 0).option("-d --base-dir <path>", "Base directory to run commands in").option("--agent <agent>", "Initial agent to use (default: architect)");
 }
 function parseOptions(options, cwdArg, home = os3.homedir(), env2 = getEnv()) {
   let cwd = cwdArg;
@@ -61945,6 +61962,12 @@ function parseOptions(options, cwdArg, home = os3.homedir(), env2 = getEnv()) {
     cwd = process.cwd();
   }
   const config3 = loadConfig(options.config, cwd, home) ?? {};
+  if (options.retryCount !== undefined) {
+    config3.retryCount = options.retryCount;
+  }
+  if (options.requestTimeoutSeconds !== undefined) {
+    config3.requestTimeoutSeconds = options.requestTimeoutSeconds;
+  }
   const defaultProvider = options.apiProvider || env2.POLKA_API_PROVIDER || config3.defaultProvider;
   const defaultModel = options.model || env2.POLKA_MODEL || config3.defaultModel;
   if (defaultProvider && defaultModel) {
@@ -66964,7 +66987,15 @@ async function runTask(taskArg, _options, command) {
     verbose,
     enableCache: true
   });
+  const sigintHandler = () => {
+    runner.abort();
+    console.log();
+    runner.printUsage();
+    process.exit(0);
+  };
+  process.on("SIGINT", sigintHandler);
   await runner.startTask(task, agent);
+  process.off("SIGINT", sigintHandler);
   runner.printUsage();
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/cli",
-  "version": "0.8.25",
+  "version": "0.8.27",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",