@polka-codes/cli 0.9.59 → 0.9.60

package/dist/index.js

@@ -35579,7 +35579,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.9.59";
+var version = "0.9.60";
 
 // src/commands/code.ts
 import { readFile as readFile3 } from "node:fs/promises";
@@ -48181,7 +48181,8 @@ config(en_default());
 var providerModelSchema = exports_external.object({
   provider: exports_external.string().optional(),
   model: exports_external.string().optional(),
-  parameters: exports_external.record(exports_external.string(), exports_external.any()).optional()
+  parameters: exports_external.record(exports_external.string(), exports_external.any()).optional(),
+  budget: exports_external.number().positive().optional()
 });
 var configSchema = exports_external.object({
   prices: exports_external.record(exports_external.string(), exports_external.record(exports_external.string(), exports_external.object({
@@ -48210,9 +48211,7 @@ var configSchema = exports_external.object({
     command: exports_external.string(),
     description: exports_external.string()
   }))).optional(),
-  commands: exports_external.object({
-    default: providerModelSchema.optional()
-  }).catchall(providerModelSchema).optional(),
+  commands: exports_external.record(exports_external.string(), providerModelSchema).optional(),
   rules: exports_external.array(exports_external.string()).optional().or(exports_external.string()).optional(),
   excludeFiles: exports_external.array(exports_external.string()).optional()
 }).strict();
@@ -77708,7 +77707,9 @@ class ApiProviderConfig {
     this.commands = config4.commands;
   }
   getConfigForCommand(command) {
-    const { provider: provider3, model, parameters } = this.commands?.[command] ?? this.commands?.default ?? {};
+    const commandConfig = this.commands?.[command];
+    const defaultConfig = this.commands?.default;
+    const { provider: provider3, model, parameters, budget } = { ...defaultConfig, ...commandConfig };
     const finalProvider = provider3 ?? this.defaultProvider;
     if (!finalProvider) {
       return;
@@ -77728,7 +77729,8 @@ class ApiProviderConfig {
       project,
       keyFile,
       baseUrl,
-      parameters: finalParameters
+      parameters: finalParameters,
+      budget
     };
   }
 }
@@ -77737,7 +77739,7 @@ class ApiProviderConfig {
 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.", Number.parseInt).option("--budget <budget>", "Budget for the AI service.", Number.parseFloat).option("-v --verbose", "Enable verbose output. Use -v for level 1, -vv for level 2", (_value, prev) => (prev ?? 0) + 1).option("-d --base-dir <path>", "Base directory to run commands in").option("--file <path...>", "File to include in the task").option("--silent", "Enable silent mode").option("-y, --yes", "Skip interactive prompts", false);
 }
-function parseOptions(options, { cwdArg } = {}, home = os2.homedir(), env2 = getEnv()) {
+function parseOptions(options, { cwdArg, commandName } = {}, home = os2.homedir(), env2 = getEnv()) {
   let cwd = cwdArg;
   if (options.baseDir) {
     process.chdir(options.baseDir);
@@ -77773,13 +77775,18 @@ function parseOptions(options, { cwdArg } = {}, home = os2.homedir(), env2 = get
     defaultProvider,
     ...config4
   });
+  const commandConfig = commandName ? providerConfig.getConfigForCommand(commandName) : undefined;
   const verbose = options.silent ? -1 : options.verbose ?? 0;
+  const commandSpecificConfig = commandName ? config4.commands?.[commandName] : undefined;
+  const defaultCommandConfig = config4.commands?.default;
+  const mergedCommandConfig = { ...defaultCommandConfig, ...commandSpecificConfig };
   return {
     maxMessageCount: options.maxMessageCount ?? config4.maxMessageCount ?? 100,
-    budget: options.budget ?? (env2.POLKA_BUDGET ? Number.parseFloat(env2.POLKA_BUDGET) : undefined) ?? config4.budget ?? 10,
+    budget: options.budget ?? mergedCommandConfig.budget ?? (env2.POLKA_BUDGET ? Number.parseFloat(env2.POLKA_BUDGET) : undefined) ?? config4.budget ?? 10,
     verbose,
     config: config4,
     providerConfig,
+    commandConfig,
     file: options.file
   };
 }
@@ -80254,7 +80261,7 @@ async function runCode(task, _options, command) {
   if (stdin) {
     taskInput = stdin;
   }
-  const { file: files } = parseOptions(command.opts());
+  const { file: files } = parseOptions(command.opts(), { commandName: "code" });
   const fileContents = [];
   if (files) {
     for (const file2 of files) {
@@ -80393,7 +80400,7 @@ ${input2.context}
 Commit message:
 ${commitMessage}`);
       await tools2.createCommit({ message: commitMessage });
-      return;
+      return commitMessage;
     }
   }
   context.logger.warn("Failed to generate commit message.", result);
@@ -80842,13 +80849,14 @@ async function runPreflightChecks(epicContext, context) {
 }
 async function performReviewAndFixCycle(iterationCount, taskItem, highLevelPlan, context) {
   const { logger, step, tools: tools2 } = context;
+  const commitMessages = [];
   for (let i = 0;i < MAX_REVIEW_RETRIES; i++) {
     const diffResult = await tools2.executeCommand({ command: "git", args: ["diff", "--name-status", "HEAD~1", "HEAD"] });
     const changedFiles = parseGitDiffNameStatus(diffResult.stdout).filter(({ path }) => path !== ".epic.yml");
     if (changedFiles.length === 0) {
       logger.info(`No files were changed. Skipping review.
 `);
-      return { passed: true };
+      return { passed: true, commitMessages: [] };
     }
     logger.info(`
 Review iteration ${i + 1}/${MAX_REVIEW_RETRIES}`);
@@ -80879,7 +80887,7 @@ ${formatReviewToolInput(changeInfo)}`;
     if (!reviewResult || !reviewResult.specificReviews || reviewResult.specificReviews.length === 0) {
       logger.info(`Review passed. No issues found.
 `);
-      return { passed: true };
+      return { passed: true, commitMessages };
     }
     logger.warn(`Warning: Review found ${reviewResult.specificReviews.length} issue(s). Attempting to fix...
 `);
@@ -80910,17 +80918,22 @@ ${reviewSummary}`;
       }, context);
     });
     await step(`commit-fix-${iterationCount}-${i}`, async () => {
-      await tools2.executeCommand({ command: "git", args: ["add", "."] });
-      await tools2.executeCommand({ command: "git", args: ["commit", "--amend", "--no-edit"] });
+      const commitMessage = await commitWorkflow({
+        all: true,
+        context: reviewSummary
+      }, context);
+      if (commitMessage) {
+        commitMessages.push(commitMessage);
+      }
     });
     if (i === MAX_REVIEW_RETRIES - 1) {
       logger.error(`
 Max retries (${MAX_REVIEW_RETRIES}) reached. Moving to the next task, but issues might remain.
 `);
-      return { passed: false };
+      return { passed: false, commitMessages };
     }
   }
-  return { passed: false };
+  return { passed: false, commitMessages };
 }
 async function findNextTask(tools2) {
   const openRootTasks = await tools2.listTodoItems({ status: "open" });
@@ -80979,7 +80992,8 @@ Focus only on this item, but use the plan for context.`;
       await tools2.executeCommand({ command: "git", args: ["commit", "-m", commitMessage] });
     });
     commitMessages.push(commitMessage);
-    const { passed: reviewPassed } = await performReviewAndFixCycle(iterationCount, nextTask, highLevelPlan, context);
+    const { passed: reviewPassed, commitMessages: fixCommitMessages } = await performReviewAndFixCycle(iterationCount, nextTask, highLevelPlan, context);
+    commitMessages.push(...fixCommitMessages);
     const taskElapsed = Date.now() - taskStartTime;
     const taskElapsedTime = formatElapsedTime(taskElapsed);
     if (reviewPassed) {
@@ -81633,7 +81647,10 @@ async function runEpic(task2, _options, command) {
   const workflowInput = {
     ...epicContext,
     async saveEpicContext(context) {
-      await saveEpicContext(context);
+      await saveEpicContext({
+        ...epicContext,
+        ...context
+      });
     },
     saveUsageSnapshot
   };

package/package.json

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