@polka-codes/cli-shared 0.8.25 → 0.8.27

package/dist/index.js

@@ -37654,14 +37654,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 {
@@ -38809,12 +38813,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 });
@@ -38833,6 +38852,10 @@ ${instance.prompt}`;
           break;
         }
         console.error("Error in stream:", error);
+      } finally {
+        if (timeout) {
+          clearTimeout(timeout);
+        }
       }
       if (currentAssistantMessage) {
         break;
@@ -43228,7 +43251,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(),
@@ -43248,6 +43273,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()
@@ -43530,64 +43557,49 @@ ${output}`);
 };
 
 // ../core/src/AiTool/generateProjectConfig.ts
-var prompt4 = `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:
-
-<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"
-
-rules:
-  - "Use \`bun\` as package manager"
-  - "Write tests using bun:test with snapshots"
-  - "Follow Biome code style"
-
-excludeFiles:
-  # Sensitive files
+var prompt4 = `
+Role: Analyzer agent
+Goal: Produce a valid polkacodes YAML configuration for the project.
+
+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
+
+2. Build a YAML config with three root keys:
+
+\`\`\`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",

package/package.json

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