@polka-codes/core 0.8.26 → 0.8.27

package/dist/_tsup-dts-rollup.d.ts

@@ -54,6 +54,8 @@ declare type AgentBaseConfig = {
     }>;
     callback?: TaskEventCallback;
     policies: AgentPolicy[];
+    retryCount?: number;
+    requestTimeoutSeconds?: number;
 };
 export { AgentBaseConfig }
 export { AgentBaseConfig as AgentBaseConfig_alias_1 }
@@ -482,6 +484,8 @@ declare const configSchema: z.ZodObject<{
     defaultParameters: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodAny>>;
     maxMessageCount: z.ZodOptional<z.ZodNumber>;
     budget: z.ZodOptional<z.ZodNumber>;
+    retryCount: z.ZodOptional<z.ZodNumber>;
+    requestTimeoutSeconds: z.ZodOptional<z.ZodNumber>;
     scripts: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodString, z.ZodObject<{
         command: z.ZodString;
         description: z.ZodString;
@@ -507,6 +511,8 @@ declare const configSchema: z.ZodObject<{
             maxFileCount?: number | undefined;
             excludes?: string[] | undefined;
         }>>;
+        retryCount: z.ZodOptional<z.ZodNumber>;
+        requestTimeoutSeconds: z.ZodOptional<z.ZodNumber>;
     }>, "strip", z.ZodTypeAny, {
         provider?: string | undefined;
         model?: string | undefined;
@@ -515,6 +521,8 @@ declare const configSchema: z.ZodObject<{
             maxFileCount?: number | undefined;
             excludes?: string[] | undefined;
         } | undefined;
+        retryCount?: number | undefined;
+        requestTimeoutSeconds?: number | undefined;
     }, {
         provider?: string | undefined;
         model?: string | undefined;
@@ -523,6 +531,8 @@ declare const configSchema: z.ZodObject<{
             maxFileCount?: number | undefined;
             excludes?: string[] | undefined;
         } | undefined;
+        retryCount?: number | undefined;
+        requestTimeoutSeconds?: number | undefined;
     }>>>;
     commands: z.ZodOptional<z.ZodObject<{
         default: z.ZodOptional<z.ZodObject<{
@@ -607,6 +617,8 @@ declare const configSchema: z.ZodObject<{
     excludeFiles: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
     policies: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
 }, "strict", z.ZodTypeAny, {
+    retryCount?: number | undefined;
+    requestTimeoutSeconds?: number | undefined;
     agent?: string | undefined;
     prices?: Record<string, Record<string, {
         inputPrice?: number | undefined;
@@ -636,6 +648,8 @@ declare const configSchema: z.ZodObject<{
             maxFileCount?: number | undefined;
             excludes?: string[] | undefined;
         } | undefined;
+        retryCount?: number | undefined;
+        requestTimeoutSeconds?: number | undefined;
     }> | undefined;
     commands?: z.objectOutputType<{
         default: z.ZodOptional<z.ZodObject<{
@@ -668,6 +682,8 @@ declare const configSchema: z.ZodObject<{
     excludeFiles?: string[] | undefined;
     policies?: string[] | undefined;
 }, {
+    retryCount?: number | undefined;
+    requestTimeoutSeconds?: number | undefined;
     agent?: string | undefined;
     prices?: Record<string, Record<string, {
         inputPrice?: number | undefined;
@@ -697,6 +713,8 @@ declare const configSchema: z.ZodObject<{
             maxFileCount?: number | undefined;
             excludes?: string[] | undefined;
         } | undefined;
+        retryCount?: number | undefined;
+        requestTimeoutSeconds?: number | undefined;
     }> | undefined;
     commands?: z.objectInputType<{
         default: z.ZodOptional<z.ZodObject<{
@@ -1910,7 +1928,7 @@ export declare const default_alias_2: {
 export declare const default_alias_3: {
     readonly name: "generateProjectConfig";
     readonly description: "Analyzes project files to generate polkacodes config sections";
-    readonly prompt: "You are an analyzer agent responsible for examining project files and generating appropriate polkacodes configuration. Your task is to:\n\n1. Read and analyze the provided files using tool_read_file to understand:\n   - Build tools and package manager (e.g., bun, npm)\n   - Testing frameworks and patterns\n   - Code style tools and rules\n   - Project structure and conventions\n   - Common development workflows\n\n2. Generate a YAML configuration that captures:\n   - scripts section based on package.json scripts and CI workflows. If applicable, generate following scripts:\n     - format: Format the code using a code formatter tool\n     - check: Check the code for errors using a linter tool\n     - test: Run tests using a test runner tool\n   - include other relevant scripts based on project conventions, tools, and patterns\n   - rules section based on project conventions, tools, and patterns\n   - excludeFiles section for sensitive files\n\n3. Use tool_attempt_completion to return the final configuration in this format:\n\n<tool_attempt_completion>\n<tool_parameter_result>\nscripts:\n  test:\n    command: \"bun test\"\n    description: \"Run tests with bun:test\"\n  lint:\n    command: \"biome check .\"\n    description: \"Check code style with Biome\"\n\nrules:\n  - \"Use `bun` as package manager\"\n  - \"Write tests using bun:test with snapshots\"\n  - \"Follow Biome code style\"\n\nexcludeFiles:\n  # Sensitive files\n  - \".env\"\n  - \".env.*\"\n  - \".npmrc\"\n</tool_parameter_result>\n</tool_attempt_completion>\n\nFocus on:\n- Package manager and dependency management\n- Testing frameworks and patterns\n- Code style and linting rules\n- File organization and naming conventions\n- Build and development workflows\n- Sensitive files that should not be committed:\n  - Environment files (.env*)\n  - Configuration files with potential secrets\n- Generated files to exclude:\n  - Lock files from package managers\n  - Build artifacts and outputs\n  - Generated file that are not ignored by .gitignore\n\nThe configuration should accurately reflect the project's structure, tools, and conventions.\n";
+    readonly prompt: "\nRole: Analyzer agent\nGoal: Produce a valid polkacodes YAML configuration for the project.\n\nWorkflow\n1. Scan project files with tool_read_file and identify:\n   - Package/build tool (npm, bun, pnpm, etc.)\n   - Test framework and patterns (snapshot tests, coverage, etc.)\n   - Formatter / linter and their rules\n   - Folder structure and naming conventions\n   - CI / development workflows\n\n2. Build a YAML config with three root keys:\n\n```yaml\nscripts:          # derive from package.json and CI\n  format:        # code formatter\n    command: \"<formatter cmd>\"\n    description: \"Format code\"\n  check:         # linter / type checker\n    command: \"<linter cmd>\"\n    description: \"Static checks\"\n  test:          # test runner\n    command: \"<test cmd>\"\n    description: \"Run tests\"\n  # add any other meaningful project scripts\n\nrules:            # bullet list of key conventions/tools\n\nexcludeFiles:     # only files likely to hold secrets\n  - \".env\"\n  - \".env.*\"\n  - \".npmrc\"\n  # do NOT list build artifacts, lockfiles, or paths already in .gitignore\n```\n\n3. Return the YAML exactly once, wrapped like:\n\n<tool_attempt_completion>\n<tool_parameter_result>\n# YAML (2-space indents, double-quoted commands)\n</tool_parameter_result>\n</tool_attempt_completion>\n";
     readonly formatInput: () => string;
     readonly parseOutput: (output: string) => string;
     readonly agent: "analyzer";
@@ -2655,6 +2673,8 @@ declare type SharedAgentOptions = {
     agents?: Readonly<AgentInfo[]>;
     callback?: TaskEventCallback;
     policies: AgentPolicy[];
+    retryCount?: number;
+    requestTimeoutSeconds?: number;
 };
 export { SharedAgentOptions }
 export { SharedAgentOptions as SharedAgentOptions_alias_1 }

package/dist/index.js

@@ -2988,12 +2988,27 @@ ${instance.prompt}`;
       }
     }
     let currentAssistantMessage = "";
-    const retryCount = 5;
+    const retryCount = this.config.retryCount ?? 5;
+    const requestTimeoutSeconds = this.config.requestTimeoutSeconds ?? 10;
     for (let i = 0; i < retryCount; i++) {
       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 * 1e3);
+        }
+      };
       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 });
@@ -3012,6 +3027,10 @@ ${instance.prompt}`;
           break;
         }
         console.error("Error in stream:", error);
+      } finally {
+        if (timeout) {
+          clearTimeout(timeout);
+        }
       }
       if (currentAssistantMessage) {
         break;
@@ -3797,7 +3816,9 @@ var agentSchema = providerModelSchema.extend({
   initialContext: z.object({
     maxFileCount: z.number().int().positive().optional(),
     excludes: z.array(z.string()).optional()
-  }).optional()
+  }).optional(),
+  retryCount: z.number().int().min(0).optional(),
+  requestTimeoutSeconds: z.number().int().positive().optional()
 });
 var configSchema = z.object({
   agent: z.string().optional(),
@@ -3828,6 +3849,8 @@ var configSchema = z.object({
   defaultParameters: z.record(z.string(), z.any()).optional(),
   maxMessageCount: z.number().int().positive().optional(),
   budget: z.number().positive().optional(),
+  retryCount: z.number().int().min(0).optional(),
+  requestTimeoutSeconds: z.number().int().positive().optional(),
   scripts: z.record(
     z.string(),
     z.string().or(
@@ -4323,64 +4346,49 @@ ${output}`);
 };
 
 // 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",

package/package.json

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