@polka-codes/core 0.9.33 → 0.9.34

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

@@ -71,8 +71,6 @@ export { AgentBaseConfig as AgentBaseConfig_alias_1 }
 export { AgentBaseConfig as AgentBaseConfig_alias_2 }
 
 declare type AgentContextParameters = {
-    budget?: number;
-    maxMessages?: number;
     toolFormat?: ToolFormat;
     os?: string;
     policies?: AgentPolicy[];
@@ -81,6 +79,9 @@ declare type AgentContextParameters = {
         command: string;
         description: string;
     }>;
+    retryCount?: number;
+    requestTimeoutSeconds?: number;
+    usageMeter?: UsageMeter;
 };
 export { AgentContextParameters }
 export { AgentContextParameters as AgentContextParameters_alias_1 }
@@ -124,11 +125,9 @@ declare type AgentStepSpec<TInput extends Record<string, Json> = Record<string,
     type: 'agent';
     messages: (UserContent_2 | TemplatedString<TInput>)[];
     outputSchema?: z.ZodType<TOutput>;
-    parseOutput?: (raw: string) => TOutput;
+    parseOutput?: (raw: string) => ParseOutputResult<TOutput>;
     provider?: string;
     model?: string;
-    budget?: number;
-    maxMessages?: number;
     modelParameters?: Record<string, any>;
     toolFormat?: ToolFormat;
     agent?: AgentNameType;
@@ -483,7 +482,7 @@ declare const _default: {
 };
 export { _default as askFollowupQuestion }
 export { _default as askFollowupQuestion_alias_1 }
-export { _default as default_alias_4 }
+export { _default as default_alias_2 }
 
 declare const _default_10: {
     handler: ToolHandler<{
@@ -500,7 +499,7 @@ declare const _default_10: {
         path: z.ZodString;
     }, z.core.$strip>;
 };
-export { _default_10 as default_alias_13 }
+export { _default_10 as default_alias_11 }
 export { _default_10 as removeFile }
 export { _default_10 as removeFile_alias_1 }
 
@@ -521,7 +520,7 @@ declare const _default_11: {
         target_path: z.ZodString;
     }, z.core.$strip>;
 };
-export { _default_11 as default_alias_14 }
+export { _default_11 as default_alias_12 }
 export { _default_11 as renameFile }
 export { _default_11 as renameFile_alias_1 }
 
@@ -542,7 +541,7 @@ declare const _default_12: {
         diff: z.ZodString;
     }, z.core.$strip>;
 };
-export { _default_12 as default_alias_15 }
+export { _default_12 as default_alias_13 }
 export { _default_12 as replaceInFile }
 export { _default_12 as replaceInFile_alias_1 }
 
@@ -565,7 +564,7 @@ declare const _default_13: {
         filePattern: z.ZodOptional<z.ZodString>;
     }, z.core.$strip>;
 };
-export { _default_13 as default_alias_16 }
+export { _default_13 as default_alias_14 }
 export { _default_13 as searchFiles }
 export { _default_13 as searchFiles_alias_1 }
 
@@ -586,7 +585,7 @@ declare const _default_14: {
         content: z.ZodString;
     }, z.core.$strip>;
 };
-export { _default_14 as default_alias_17 }
+export { _default_14 as default_alias_15 }
 export { _default_14 as writeToFile }
 export { _default_14 as writeToFile_alias_1 }
 
@@ -607,7 +606,7 @@ declare const _default_2: {
 };
 export { _default_2 as attemptCompletion }
 export { _default_2 as attemptCompletion_alias_1 }
-export { _default_2 as default_alias_5 }
+export { _default_2 as default_alias_3 }
 
 declare const _default_3: {
     handler: ToolHandler<{
@@ -630,7 +629,7 @@ declare const _default_3: {
         files: z.ZodOptional<z.ZodPipe<z.ZodTransform<string[], unknown>, z.ZodArray<z.ZodString>>>;
     }, z.core.$strip>;
 };
-export { _default_3 as default_alias_6 }
+export { _default_3 as default_alias_4 }
 export { _default_3 as delegate }
 export { _default_3 as delegate_alias_1 }
 
@@ -651,7 +650,7 @@ declare const _default_4: {
         requiresApproval: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodDefault<z.ZodOptional<z.ZodBoolean>>>;
     }, z.core.$strip>;
 };
-export { _default_4 as default_alias_7 }
+export { _default_4 as default_alias_5 }
 export { _default_4 as executeCommand }
 export { _default_4 as executeCommand_alias_1 }
 
@@ -670,7 +669,7 @@ declare const _default_5: {
         url: z.ZodPipe<z.ZodTransform<string[], unknown>, z.ZodArray<z.ZodString>>;
     }, z.core.$strip>;
 };
-export { _default_5 as default_alias_8 }
+export { _default_5 as default_alias_6 }
 export { _default_5 as fetchUrl }
 export { _default_5 as fetchUrl_alias_1 }
 
@@ -695,7 +694,7 @@ declare const _default_6: {
         files: z.ZodOptional<z.ZodPipe<z.ZodTransform<string[], unknown>, z.ZodArray<z.ZodString>>>;
     }, z.core.$strip>;
 };
-export { _default_6 as default_alias_9 }
+export { _default_6 as default_alias_7 }
 export { _default_6 as handOver }
 export { _default_6 as handOver_alias_1 }
 
@@ -720,7 +719,7 @@ declare const _default_7: {
         includeIgnored: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodDefault<z.ZodOptional<z.ZodBoolean>>>;
     }, z.core.$strip>;
 };
-export { _default_7 as default_alias_10 }
+export { _default_7 as default_alias_8 }
 export { _default_7 as listFiles }
 export { _default_7 as listFiles_alias_1 }
 
@@ -739,7 +738,7 @@ declare const _default_8: {
         url: z.ZodString;
     }, z.core.$strip>;
 };
-export { _default_8 as default_alias_11 }
+export { _default_8 as default_alias_9 }
 export { _default_8 as readBinaryFile }
 export { _default_8 as readBinaryFile_alias_1 }
 
@@ -760,7 +759,7 @@ declare const _default_9: {
         includeIgnored: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodDefault<z.ZodOptional<z.ZodBoolean>>>;
     }, z.core.$strip>;
 };
-export { _default_9 as default_alias_12 }
+export { _default_9 as default_alias_10 }
 export { _default_9 as readFile }
 export { _default_9 as readFile_alias_1 }
 
@@ -777,30 +776,11 @@ export declare const default_alias: {
     readonly agent: "architect";
 };
 
-export declare const default_alias_1: {
-    readonly name: "generateGitCommitMessage";
-    readonly description: "Generates git commit messages from git diff output";
-    readonly prompt: "\nYou are an advanced assistant specialized in creating concise and accurate Git commit messages. When you receive:\n- A Git diff inside the <tool_input> tag.\n- Additional user-supplied context inside the <tool_input_context> tag (if any).\n\nYou will produce a single commit message enclosed within <tool_output> tags. The commit message must accurately reflect the changes shown in the diff and should be clear, descriptive, and devoid of unnecessary or repeated information. If a context is provided, it MUST be incorporated into the commit message.\n\nHere’s an example of the input and the expected output format:\n\n<tool_input>\n--- a/example_file.py\n+++ b/example_file.py\n@@ -10,7 +10,7 @@ def example_function():\n-    print(\"Old behavior\")\n+    print(\"New behavior\")\n</tool_input>\n<tool_input_context>\nChanging print statement to update the user-facing message.\n</tool_input_context>\n\nExample Output:\n\n<tool_output>\nUpdate print statement for revised user-facing message\n</tool_output>\n\nFollow the same structure for any new input. Never repeat questions; focus on generating a concise commit message that captures the essence of the changes.\n";
-    readonly formatInput: (params: {
-        diff: string;
-        context?: string;
-    }) => string;
-    readonly parseOutput: (output: string) => string;
-};
-
-export declare const default_alias_2: {
-    readonly name: "generateGithubPullRequestDetails";
-    readonly description: "Generates a GitHub pull request title and description from git commits";
-    readonly prompt: "\n# Generate Github Pull Request Details\n\nYou are given:\n- A branch name in <tool_input_branch_name>.\n- An optional context message in <tool_input_context> (which may or may not be present).\n- All commit messages combined in <tool_input_commit_messages>.\n- All diffs combined in <tool_input_commit_diff>.\n\nYour task:\n1. Consider the optional context (if provided).\n   - If an issue number is found, add \"Closes #xxx\" at the beginning of the PR description\n   - IMPORTANT: Use ONLY the exact format \"Closes #xxx\" at the beginning of the description\n   - DO NOT use variations like \"Closes issue #xxx\" or other formats\n2. Analyze the combined commit messages and diffs.\n3. Produce a single GitHub Pull Request title.\n4. Produce a Pull Request description that explains the changes.\n\nUse the following template for the Pull Request description:\n\n---\n**Context (if provided)**:\n- Acknowledge any guiding concerns or instructions.\n\n**Summary of Changes**:\n- Provide a concise list or overview of what changed.\n\n**Highlights of Changed Code**:\n- Mention only the specific sections or functionalities updated, without showing full surrounding context.\n\n**Additional Information (if needed)**:\n- Testing steps (if applicable).\n- Any notes or caveats.\n\n---\n\nOutput format:\n<tool_output>\n  <tool_output_pr_title>YOUR PR TITLE HERE</tool_output_pr_title>\n  <tool_output_pr_description>\n    YOUR PR DESCRIPTION HERE\n  </tool_output_pr_description>\n</tool_output>\n\nBelow is an **example** of the input and output:\n\nExample Input:\n<tool_input>\n<tool_input_branch_name>feature/refactor-logging</tool_input_branch_name>\n<tool_input_context>Implementing changes for issue #123 - Focus on clean code and maintainability</tool_input_context>\n<tool_input_commit_messages>\nRemove debug logs\nRefactor order validation logic\n</tool_input_commit_messages>\n<tool_input_commit_diff>\ndiff --git a/user_service.py b/user_service.py\n- print(\"Debug info\")\n+ # Removed debug print statements\n\ndiff --git a/order_service.py b/order_service.py\n- if is_valid_order(order):\n-     process_order(order)\n+ validate_and_process(order)\n</tool_input_commit_diff>\n</tool_input>\n\nExample Output:\n<tool_output>\n<tool_output_pr_title>Refactor Order Validation and Remove Debug Logs</tool_output_pr_title>\n<tool_output_pr_description>\nCloses #123\n\n**Context**:\n- Implementing changes for issue #123 - Focus on clean code and maintainability\n\n**Summary of Changes**:\n- Refactored order validation logic to use a new `validate_and_process` method.\n- Removed debug print statements from `user_service.py`.\n\n**Highlights of Changed Code**:\n- `order_service.py`: Replaced direct call to `process_order` with `validate_and_process`.\n- `user_service.py`: Removed `print(\"Debug info\")`.\n</tool_output_pr_description>\n</tool_output>\n\n---\n\nUse the above format whenever you receive <tool_input> that may include a branch name, an optional context, aggregated commit messages in a single tag, and a combined diff in a single tag. Provide your final output strictly in <tool_output> with <tool_output_pr_title> and <tool_output_pr_description>. Only highlight the changed code and avoid including the context around the changes in the description.\n";
-    readonly formatInput: (params: Input) => string;
-    readonly parseOutput: (output: string) => Output;
-};
-
 /**
  * AI tool for analyzing project files and generating polkacodes config sections.
  * Generated by polka.codes
  */
-export declare const default_alias_3: {
+export declare const default_alias_1: {
     readonly name: "generateProjectConfig";
     readonly description: "Analyzes project files to generate polkacodes config sections";
     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";
@@ -907,25 +887,6 @@ declare type FullToolInfoV2 = ToolInfoV2 & {
 export { FullToolInfoV2 }
 export { FullToolInfoV2 as FullToolInfoV2_alias_1 }
 
-declare const generateGitCommitMessage: (ai: LanguageModelV2, params: {
-    diff: string;
-    context?: string;
-}, usageMeter: UsageMeter) => Promise<string>;
-export { generateGitCommitMessage }
-export { generateGitCommitMessage as generateGitCommitMessage_alias_1 }
-
-declare const generateGithubPullRequestDetails: (ai: LanguageModelV2, params: {
-    commitMessages: string;
-    commitDiff: string;
-    context?: string;
-    branchName: string;
-}, usageMeter: UsageMeter) => Promise<{
-    title: string;
-    description: string;
-}>;
-export { generateGithubPullRequestDetails }
-export { generateGithubPullRequestDetails as generateGithubPullRequestDetails_alias_1 }
-
 declare const generateProjectConfig: (agent: MultiAgent, params: unknown) => Promise<string>;
 export { generateProjectConfig }
 export { generateProjectConfig as generateProjectConfig_alias_1 }
@@ -993,13 +954,6 @@ export declare const handler_alias_8: ToolHandler<typeof toolInfo_alias_8, Files
 
 export declare const handler_alias_9: ToolHandler<typeof toolInfo_alias_9, FilesystemProvider>;
 
-declare type Input = {
-    commitMessages: string;
-    commitDiff: string;
-    context?: string;
-    branchName: string;
-};
-
 declare type InputType = {
     $: Record<string, Record<string, Json>>;
     [key: string]: Json;
@@ -1126,11 +1080,6 @@ export { MultiAgentConfig as MultiAgentConfig_alias_2 }
 
 export declare const objectives: (toolNamePrefix: string) => string;
 
-declare type Output = {
-    title: string;
-    description: string;
-};
-
 declare interface ParallelStepSpec<TInput extends Record<string, Json> = Record<string, Json>, TOutput extends Record<string, Json> = Record<string, Json>> extends BaseStepSpec<Record<string, Json>, Record<string, Json>> {
     type: 'parallel';
     step: BaseStepSpec<TInput, TOutput>;
@@ -1211,11 +1160,22 @@ export { parseAssistantMessage }
 export { parseAssistantMessage as parseAssistantMessage_alias_1 }
 export { parseAssistantMessage as parseAssistantMessage_alias_2 }
 
-declare const parseJsonFromMarkdown: (markdown: string) => any;
+declare const parseJsonFromMarkdown: (markdown: string) => ParseOutputResult<any>;
 export { parseJsonFromMarkdown }
 export { parseJsonFromMarkdown as parseJsonFromMarkdown_alias_1 }
 export { parseJsonFromMarkdown as parseJsonFromMarkdown_alias_2 }
 
+declare type ParseOutputResult<T> = {
+    success: true;
+    data: T;
+} | {
+    success: false;
+    error?: string;
+};
+export { ParseOutputResult }
+export { ParseOutputResult as ParseOutputResult_alias_1 }
+export { ParseOutputResult as ParseOutputResult_alias_2 }
+
 declare enum Policies {
     TruncateContext = "truncatecontext",
     EnableCache = "enablecache"

package/dist/index.d.ts

@@ -60,8 +60,6 @@ export { executeAgentTool } from './_tsup-dts-rollup.js';
 export { makeTool } from './_tsup-dts-rollup.js';
 export { makeMultiAgentTool } from './_tsup-dts-rollup.js';
 export { makeAgentTool } from './_tsup-dts-rollup.js';
-export { generateGitCommitMessage } from './_tsup-dts-rollup.js';
-export { generateGithubPullRequestDetails } from './_tsup-dts-rollup.js';
 export { generateProjectConfig } from './_tsup-dts-rollup.js';
 export { createNewProject } from './_tsup-dts-rollup.js';
 export { AiToolDefinition } from './_tsup-dts-rollup.js';
@@ -145,5 +143,6 @@ export { WorkflowRunResultSuccess } from './_tsup-dts-rollup.js';
 export { WorkflowRunResultError } from './_tsup-dts-rollup.js';
 export { WorkflowRunResultPaused } from './_tsup-dts-rollup.js';
 export { WorkflowRunResult } from './_tsup-dts-rollup.js';
+export { ParseOutputResult } from './_tsup-dts-rollup.js';
 export { run } from './_tsup-dts-rollup.js';
 export { resume } from './_tsup-dts-rollup.js';

package/dist/index.js

@@ -87,9 +87,9 @@ var handler = async (provider, args) => {
   }
   const answers = [];
   for (const question of questions) {
-    const { prompt: prompt5, options } = question;
-    const answer = await provider.askFollowupQuestion(prompt5, options);
-    answers.push(`<ask_followup_question_answer question="${prompt5}">
+    const { prompt: prompt3, options } = question;
+    const answer = await provider.askFollowupQuestion(prompt3, options);
+    answers.push(`<ask_followup_question_answer question="${prompt3}">
 ${answer}
 </ask_followup_question_answer>`);
   }
@@ -1958,9 +1958,9 @@ ${instance.prompt}`;
   async #callback(event) {
     await this.config.callback?.(event);
   }
-  async start(prompt5) {
+  async start(prompt3) {
     this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
-    return await this.#processLoop(prompt5);
+    return await this.#processLoop(prompt3);
   }
   async step(message) {
     if (this.#messages.length === 0) {
@@ -3033,24 +3033,24 @@ var MultiAgent = class {
     switch (exitReason.type) {
       case "HandOver" /* HandOver */: {
         this.#agents.pop();
-        const prompt5 = await this.#config.getPrompt?.(
+        const prompt3 = await this.#config.getPrompt?.(
           exitReason.agentName,
           exitReason.task,
           exitReason.context,
           exitReason.files,
           this.#originalTask
         ) ?? exitReason.task;
-        return await this.#startTask(exitReason.agentName, prompt5);
+        return await this.#startTask(exitReason.agentName, prompt3);
       }
       case "Delegate" /* Delegate */: {
-        const prompt5 = await this.#config.getPrompt?.(
+        const prompt3 = await this.#config.getPrompt?.(
           exitReason.agentName,
           exitReason.task,
           exitReason.context,
           exitReason.files,
           this.#originalTask
         ) ?? exitReason.task;
-        const delegateResult = await this.#startTask(exitReason.agentName, prompt5);
+        const delegateResult = await this.#startTask(exitReason.agentName, prompt3);
         switch (delegateResult.type) {
           case "HandOver" /* HandOver */:
           case "Delegate" /* Delegate */:
@@ -3113,22 +3113,30 @@ var MultiAgent = class {
 var parseJsonFromMarkdown = (markdown) => {
   const jsonRegex = /```(?:json)?\n([\s\S]*?)\n```/;
   const match = markdown.match(jsonRegex);
-  if (match?.[1]) {
-    const content = match[1].trim();
-    let parsed = JSON.parse(content);
-    if (typeof parsed === "string") {
-      try {
-        parsed = JSON.parse(parsed);
-      } catch {
+  const tryParse = (str) => {
+    try {
+      let parsed = JSON.parse(str);
+      if (typeof parsed === "string") {
+        try {
+          parsed = JSON.parse(parsed);
+        } catch {
+        }
       }
+      return { success: true, data: parsed };
+    } catch (e) {
+      const error = e instanceof Error ? e.message : String(e);
+      return { success: false, error: `Failed to parse JSON: ${error}` };
     }
-    return parsed;
+  };
+  if (match?.[1]) {
+    const content = match[1].trim();
+    return tryParse(content);
   }
-  try {
-    return JSON.parse(markdown);
-  } catch (_error) {
+  const parseResult = tryParse(markdown);
+  if (parseResult.success) {
+    return parseResult;
   }
-  return markdown;
+  return { success: false, error: "No JSON object found in the string." };
 };
 
 // src/config.ts
@@ -3457,189 +3465,8 @@ var createNewProject_default = {
   agent: "architect"
 };
 
-// src/AiTool/generateGitCommitMessage.ts
-var prompt2 = `
-You are an advanced assistant specialized in creating concise and accurate Git commit messages. When you receive:
-- A Git diff inside the <tool_input> tag.
-- Additional user-supplied context inside the <tool_input_context> tag (if any).
-
-You will produce a single commit message enclosed within <tool_output> tags. The commit message must accurately reflect the changes shown in the diff and should be clear, descriptive, and devoid of unnecessary or repeated information. If a context is provided, it MUST be incorporated into the commit message.
-
-Here\u2019s an example of the input and the expected output format:
-
-<tool_input>
---- a/example_file.py
-+++ b/example_file.py
-@@ -10,7 +10,7 @@ def example_function():
--    print("Old behavior")
-+    print("New behavior")
-</tool_input>
-<tool_input_context>
-Changing print statement to update the user-facing message.
-</tool_input_context>
-
-Example Output:
-
-<tool_output>
-Update print statement for revised user-facing message
-</tool_output>
-
-Follow the same structure for any new input. Never repeat questions; focus on generating a concise commit message that captures the essence of the changes.
-`;
-var generateGitCommitMessage_default = {
-  name: "generateGitCommitMessage",
-  description: "Generates git commit messages from git diff output",
-  prompt: prompt2,
-  formatInput: (params) => {
-    let ret = `<tool_input>
-${params.diff}
-</tool_input>`;
-    if (params.context) {
-      ret += `
-<tool_input_context>
-${params.context}
-</tool_input_context>`;
-    }
-    return ret;
-  },
-  parseOutput: (output) => {
-    const regex = /<tool_output>([\s\S]*)<\/tool_output>/gm;
-    const match = regex.exec(output);
-    if (match) {
-      return match[1];
-    }
-    throw new Error(`Could not parse output:
-${output}`);
-  }
-};
-
-// src/AiTool/generateGithubPullRequestDetails.ts
-var prompt3 = `
-# Generate Github Pull Request Details
-
-You are given:
-- A branch name in <tool_input_branch_name>.
-- An optional context message in <tool_input_context> (which may or may not be present).
-- All commit messages combined in <tool_input_commit_messages>.
-- All diffs combined in <tool_input_commit_diff>.
-
-Your task:
-1. Consider the optional context (if provided).
-   - If an issue number is found, add "Closes #xxx" at the beginning of the PR description
-   - IMPORTANT: Use ONLY the exact format "Closes #xxx" at the beginning of the description
-   - DO NOT use variations like "Closes issue #xxx" or other formats
-2. Analyze the combined commit messages and diffs.
-3. Produce a single GitHub Pull Request title.
-4. Produce a Pull Request description that explains the changes.
-
-Use the following template for the Pull Request description:
-
----
-**Context (if provided)**:
-- Acknowledge any guiding concerns or instructions.
-
-**Summary of Changes**:
-- Provide a concise list or overview of what changed.
-
-**Highlights of Changed Code**:
-- Mention only the specific sections or functionalities updated, without showing full surrounding context.
-
-**Additional Information (if needed)**:
-- Testing steps (if applicable).
-- Any notes or caveats.
-
----
-
-Output format:
-<tool_output>
-  <tool_output_pr_title>YOUR PR TITLE HERE</tool_output_pr_title>
-  <tool_output_pr_description>
-    YOUR PR DESCRIPTION HERE
-  </tool_output_pr_description>
-</tool_output>
-
-Below is an **example** of the input and output:
-
-Example Input:
-<tool_input>
-<tool_input_branch_name>feature/refactor-logging</tool_input_branch_name>
-<tool_input_context>Implementing changes for issue #123 - Focus on clean code and maintainability</tool_input_context>
-<tool_input_commit_messages>
-Remove debug logs
-Refactor order validation logic
-</tool_input_commit_messages>
-<tool_input_commit_diff>
-diff --git a/user_service.py b/user_service.py
-- print("Debug info")
-+ # Removed debug print statements
-
-diff --git a/order_service.py b/order_service.py
-- if is_valid_order(order):
--     process_order(order)
-+ validate_and_process(order)
-</tool_input_commit_diff>
-</tool_input>
-
-Example Output:
-<tool_output>
-<tool_output_pr_title>Refactor Order Validation and Remove Debug Logs</tool_output_pr_title>
-<tool_output_pr_description>
-Closes #123
-
-**Context**:
-- Implementing changes for issue #123 - Focus on clean code and maintainability
-
-**Summary of Changes**:
-- Refactored order validation logic to use a new \`validate_and_process\` method.
-- Removed debug print statements from \`user_service.py\`.
-
-**Highlights of Changed Code**:
-- \`order_service.py\`: Replaced direct call to \`process_order\` with \`validate_and_process\`.
-- \`user_service.py\`: Removed \`print("Debug info")\`.
-</tool_output_pr_description>
-</tool_output>
-
----
-
-Use the above format whenever you receive <tool_input> that may include a branch name, an optional context, aggregated commit messages in a single tag, and a combined diff in a single tag. Provide your final output strictly in <tool_output> with <tool_output_pr_title> and <tool_output_pr_description>. Only highlight the changed code and avoid including the context around the changes in the description.
-`;
-var generateGithubPullRequestDetails_default = {
-  name: "generateGithubPullRequestDetails",
-  description: "Generates a GitHub pull request title and description from git commits",
-  prompt: prompt3,
-  formatInput: (params) => {
-    return `<tool_input>
-<tool_input_branch_name>${params.branchName}</tool_input_branch_name>${params.context ? `
-<tool_input_context>${params.context}</tool_input_context>` : ""}
-<tool_input_commit_messages>${params.commitMessages}</tool_input_commit_messages>
-<tool_input_commit_diff>${params.commitDiff}</tool_input_commit_diff>
-</tool_input>`;
-  },
-  parseOutput: (output) => {
-    const regex = /<tool_output>([\s\S]*)<\/tool_output>/gm;
-    const match = regex.exec(output);
-    if (!match) {
-      throw new Error(`Could not parse output:
-${output}`);
-    }
-    const [, outputContent] = match;
-    const titleRegex = /<tool_output_pr_title>([\s\S]*)<\/tool_output_pr_title>/gm;
-    const titleMatch = titleRegex.exec(outputContent);
-    const descriptionRegex = /<tool_output_pr_description>([\s\S]*)<\/tool_output_pr_description>/gm;
-    const descriptionMatch = descriptionRegex.exec(outputContent);
-    if (titleMatch && descriptionMatch) {
-      return {
-        title: titleMatch[1],
-        description: descriptionMatch[1]
-      };
-    }
-    throw new Error(`Could not parse output:
-${output}`);
-  }
-};
-
 // src/AiTool/generateProjectConfig.ts
-var prompt4 = `
+var prompt2 = `
 Role: Analyzer agent
 Goal: Produce a valid polkacodes YAML configuration for the project.
 
@@ -3686,7 +3513,7 @@ excludeFiles:     # only files likely to hold secrets
 var generateProjectConfig_default = {
   name: "generateProjectConfig",
   description: "Analyzes project files to generate polkacodes config sections",
-  prompt: prompt4,
+  prompt: prompt2,
   formatInput: () => {
     return "";
   },
@@ -3748,11 +3575,12 @@ var makeAgentTool = (definition) => {
     return executeAgentTool(definition, options, params);
   };
 };
-var generateGitCommitMessage = makeTool(generateGitCommitMessage_default);
-var generateGithubPullRequestDetails = makeTool(generateGithubPullRequestDetails_default);
 var generateProjectConfig = makeMultiAgentTool(generateProjectConfig_default);
 var createNewProject = makeMultiAgentTool(createNewProject_default);
 
+// src/workflow/agent.ts
+import { toJSONSchema as toJSONSchema2 } from "zod/v4";
+
 // src/workflow/utils.ts
 import { template } from "lodash";
 var resolveTemplatedString = (templatedString, input) => {
@@ -3790,9 +3618,7 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
         try {
           const model = await getModelFn(step2, context);
           const parameters = context.parameters ?? {};
-          const budget = step2.budget ?? parameters?.budget;
-          const maxMessages = step2.maxMessages ?? parameters?.maxMessages;
-          const usageMeter = new UsageMeter({}, { maxMessages, maxCost: budget });
+          const usageMeter = parameters?.usageMeter ?? new UsageMeter();
           const toolFormat = step2.toolFormat ?? parameters.toolFormat ?? "native";
           const policies = parameters.policies ?? [];
           const modelParameters = { ...parameters.modelParameters ?? {}, ...step2.modelParameters ?? {} };
@@ -3820,7 +3646,9 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
                 parameters: modelParameters,
                 scripts: parameters.scripts,
                 callback: context.agentCallback,
-                requireToolUse: false
+                requireToolUse: false,
+                retryCount: parameters.retryCount,
+                requestTimeoutSeconds: parameters.requestTimeoutSeconds
               });
             } else {
               if (!step2.systemPrompt) {
@@ -3842,7 +3670,9 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
                 toolFormat,
                 parameters: modelParameters,
                 usageMeter,
-                requireToolUse: false
+                requireToolUse: false,
+                retryCount: parameters.retryCount,
+                requestTimeoutSeconds: parameters.requestTimeoutSeconds
               });
             }
           };
@@ -3871,7 +3701,7 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
           if (context.verbose && context.verbose >= 1) {
             logger.log(`[agent-step] Agent exited with reason:`, exitReason);
           }
-          const handleExitReason = (reason) => {
+          const handleExitReason = async (reason) => {
             switch (reason.type) {
               case "Pause":
                 return { type: "paused", state: { messages: agent.messages } };
@@ -3880,24 +3710,34 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
               case "Exit" /* Exit */: {
                 const raw = reason.message;
                 if (step2.parseOutput) {
-                  try {
-                    const parsed = step2.parseOutput(raw);
+                  const result = step2.parseOutput(raw);
+                  if (result.success) {
                     if (step2.outputSchema) {
-                      const validationResult = step2.outputSchema.safeParse(parsed);
+                      const validationResult = step2.outputSchema.safeParse(result.data);
                       if (validationResult.success) {
                         return { type: "success", output: validationResult.data };
                       }
-                      return {
-                        type: "error",
-                        error: new Error(`Output validation failed: ${validationResult.error.toString()}`)
-                      };
+                      let errorMessage2 = `Output validation failed: ${validationResult.error.toString()}`;
+                      if (step2.outputSchema) {
+                        errorMessage2 += `
+
+Expected JSON schema:
+${JSON.stringify(toJSONSchema2(step2.outputSchema), null, 2)}`;
+                      }
+                      const newReason2 = await agent.continueTask(errorMessage2);
+                      return handleExitReason(newReason2);
                     }
-                    return { type: "success", output: parsed };
-                  } catch (e) {
-                    const error = e instanceof Error ? e : new Error(String(e));
-                    error.message = `Failed to parse agent output with parseOutput: ${error.message}`;
-                    return { type: "error", error };
+                    return { type: "success", output: result.data };
+                  }
+                  let errorMessage = result.error ?? "Invalid format.";
+                  if (step2.outputSchema) {
+                    errorMessage += `
+
+Expected JSON schema:
+${JSON.stringify(toJSONSchema2(step2.outputSchema), null, 2)}`;
                   }
+                  const newReason = await agent.continueTask(errorMessage);
+                  return handleExitReason(newReason);
                 } else if (step2.outputSchema) {
                   try {
                     const output = JSON.parse(raw);
@@ -3905,14 +3745,22 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
                     if (validationResult.success) {
                       return { type: "success", output: validationResult.data };
                     }
-                    return {
-                      type: "error",
-                      error: new Error(`Output validation failed: ${validationResult.error.toString()}`)
-                    };
+                    let errorMessage = `Output validation failed: ${validationResult.error.toString()}`;
+                    errorMessage += `
+
+Expected JSON schema:
+${JSON.stringify(toJSONSchema2(step2.outputSchema), null, 2)}`;
+                    const newReason = await agent.continueTask(errorMessage);
+                    return handleExitReason(newReason);
                   } catch (e) {
                     const error = e instanceof Error ? e : new Error(String(e));
-                    error.message = `Failed to parse agent output as JSON: ${error.message}`;
-                    return { type: "error", error };
+                    let errorMessage = `Failed to parse agent output as JSON: ${error.message}`;
+                    errorMessage += `
+
+Expected JSON schema:
+${JSON.stringify(toJSONSchema2(step2.outputSchema), null, 2)}`;
+                    const newReason = await agent.continueTask(errorMessage);
+                    return handleExitReason(newReason);
                   }
                 }
                 return { type: "success", output: reason };
@@ -3932,7 +3780,7 @@ var makeAgentStepSpecHandler = (getModelFn) => ({
                 };
             }
           };
-          return handleExitReason(exitReason);
+          return await handleExitReason(exitReason);
         } catch (e) {
           return { type: "error", error: e instanceof Error ? e : new Error(String(e)) };
         }
@@ -4248,8 +4096,6 @@ export {
   executeMultiAgentTool,
   executeTool,
   fetchUrl_default as fetchUrl,
-  generateGitCommitMessage,
-  generateGithubPullRequestDetails,
   generateProjectConfig,
   getAvailableTools,
   handOver_default as handOver,

package/package.json

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