@polka-codes/cli 0.9.55 → 0.9.57

package/dist/index.js

@@ -35579,7 +35579,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.9.55";
+var version = "0.9.57";
 
 // src/commands/code.ts
 import { readFile as readFile3 } from "node:fs/promises";
@@ -77966,7 +77966,7 @@ var toolInfo18 = {
           return true;
       }
       return val;
-    }, exports_external.boolean().optional().default(false)).describe("Annotate the diff with line numbers for additions and deletions.")
+    }, exports_external.boolean().optional().default(true)).describe("Annotate the diff with line numbers for additions and deletions.")
   })
 };
 var handler18 = async (provider3, args) => {
@@ -78705,6 +78705,7 @@ async function runWorkflow(workflow2, workflowInput, options) {
     maxMessages: config4.maxMessageCount,
     maxCost: config4.budget
   });
+  options.onUsageMeterCreated?.(usage);
   const onEvent = printEvent(verbose, usage, process.stderr);
   const excludeFiles = [".epic.yml", ...config4.excludeFiles ?? []];
   const toolProvider = (options.getProvider ?? getProvider)({ excludeFiles });
@@ -78966,7 +78967,7 @@ When planning an epic:
 Your plan must have two parts: task description and implementation plan.
 
 ### 1. Task Description
-Include the verbatim user request to provide context for the implementation plan.
+Start by providing a detailed description of the task and its expected outcome. This description should be a comprehensive summary based on the user's request, setting the stage for the detailed implementation plan that follows. It should clearly articulate the "what" and "why" of the epic before breaking it down into "how".
 
 ### 2. Implementation Plan
 
@@ -78997,7 +78998,14 @@ For epic-scale work, **use markdown checkboxes** to track progress through multi
 \`\`\`
 ### 1. Task Description
 
-Implement user authentication with JWT tokens.
+This epic is to implement a complete user authentication system using JSON Web Tokens (JWT). The primary goal is to allow users to register for a new account and log in to receive a token that can be used to access protected resources. The system should handle password hashing for security and provide a way to manage user sessions on the frontend.
+
+**Expected Outcome:**
+- A backend with registration and login endpoints.
+- Secure password storage using bcrypt.
+- JWT generation upon successful login.
+- Middleware to protect specific routes.
+- A frontend with a login form, registration form, and context for managing authentication state.
 
 ### 2. Implementation Plan
 
@@ -79251,13 +79259,18 @@ ${createJsonResponseInstruction({
   bailReason: "Unable to identify the root cause of the error. The error message is ambiguous and requires human investigation."
 })}
 `;
-function getFixUserPrompt(command, exitCode, stdout, stderr, task) {
+function getFixUserPrompt(command, exitCode, stdout, stderr, task, context) {
   const taskContext = task ? `
 ## Original Task
 
 ${task}
 ` : "";
-  return `## Context${taskContext}
+  const contextSection = context ? `
+## User Context
+
+${context}
+` : "";
+  return `## Context${taskContext}${contextSection}
 
 The following command failed with exit code ${exitCode}:
 \`${command}\`
@@ -79555,82 +79568,93 @@ ${createJsonResponseInstruction({
 `;
 // src/workflows/prompts/review.ts
 var CODE_REVIEW_SYSTEM_PROMPT = `Role: Senior software engineer.
-Goal: Review code changes and provide specific, actionable feedback on any issues found.
+Goal: Review code changes and provide comprehensive, actionable feedback on issues found.
 
 ${TOOL_USAGE_INSTRUCTION}
 
-# Code Review Prompt
-
-You are a senior software engineer reviewing code changes.
-
-## Critical Instructions
-- **ONLY review the actual changes shown in the diff.** Do not comment on existing code that wasn't modified.
-- **ONLY run gitDiff on files that are reviewable source/config files** per the "File Selection for gitDiff" rules below. Do not pass excluded files to gitDiff.
-
-
-## File Selection for gitDiff
-Use <file_status> to decide which files to diff. Include only files likely to contain human-authored source or meaningful configuration.
-
-Include (run gitDiff):
-- Application/source code
-- UI/templates/assets code
-- Infra/config that affects behavior
-
-Exclude (do NOT run gitDiff; do not review):
-- Lockfiles
-- Generated/build artifacts & deps
-- Test artifacts/snapshots
-- Data and fixtures
-- Binary/media/minified/maps
-
-## Viewing Changes
-- For each included file, **use gitDiff** to inspect the actual code changes:
-  - **Pull request:** use the provided commit range for the gitDiff tool with contextLines: 5 and includeLineNumbers: true, but only surface and review the included files.
-  - **Local changes:** diff staged or unstaged included files using gitDiff with contextLines: 5 and includeLineNumbers: true.
-- The diff will include line number annotations: [Line N] for additions and [Line N removed] for deletions.
-- You may receive:
-  - <pr_title>
-  - <pr_description>
-  - <commit_messages>
-- A <review_instructions> tag tells you the focus of the review.
-- Use <file_status> to understand which files were modified, added, deleted, or renamed and to apply the inclusion/exclusion rules above.
-
-
-## Line Number Reporting
-- Use the line numbers from the annotations in the diff output.
-- For additions: use the number from the [Line N] annotation after the + line.
-- For deletions: use the number from the [Line N removed] annotation after the - line.
-- For modifications: report the line number of the new/current code (from [Line N]).
-- Report single lines as "N" and ranges as "N-M".
-
-
-## Review Guidelines
-Focus exclusively on the changed lines (+ additions, - deletions, modified lines):
-- **Specific issues:** Point to exact problems in the changed code with accurate line references from the annotations.
-- **Actionable fixes:** Provide concrete solutions, not vague suggestions.
-- **Clear reasoning:** Explain why each issue matters and how to fix it.
-- **Avoid generic advice** unless directly tied to a specific problem visible in the diff.
-
-## What NOT to review
-- Files excluded by the "File Selection for gitDiff" rules (do not diff or comment on them).
-- Existing unchanged code.
-- Overall project structure/architecture unless directly impacted by the changes.
-- Missing features or functionality not part of this diff.
+## Review Process
+
+1.  **Identify Reviewable Files**: Use the \`<file_status>\` list to determine which files have been modified.
+2.  **Select Files for Diff**: From the modified files, select only the reviewable source and configuration files.
+    -   **Include**: Source code, config files, and template files.
+    -   **Exclude**: Lockfiles, build artifacts, test snapshots, binary/media files, data and fixtures and other generated files.
+3.  **Inspect Changes**: Use the \`gitDiff\` tool on one file at a time to see the exact changes. When reviewing pull requests, use the \`commitRange\` parameter provided in the review instructions.
+4.  **Analyze and Review**: Analyze only the modified lines (additions/deletions) for issues. Provide specific, actionable feedback with accurate line numbers.
+
+## Critical Rules
+
+-   **Focus on Changes**: ONLY review the actual changes shown in the diff. Do not comment on existing, unmodified code.
+-   **Focus Scope**: Do not comment on overall project structure or architecture unless directly impacted by the changes in the diff.
+-   **No Feature Requests**: Do not comment on missing features or functionality that are not part of this diff.
+-   **One File at a Time**: Review files individually using \`gitDiff\` with the specific file path.
+-   **No Empty Diffs**: MUST NOT call \`gitDiff\` with an empty or omitted file parameter.
+-   **Accurate Line Numbers**: Use the line numbers from the diff annotations (\`[Line N]\` for additions, \`[Line N removed]\` for deletions).
+-   **No Praise**: Provide only reviews for actual issues found. Do not include praise or positive feedback.
+-   **Clear Reasoning**: For each issue, provide clear reasoning explaining why it's a problem and what the impact could be.
+-   **Specific Advice**: Avoid generic advice. Provide concrete, actionable suggestions specific to the code being reviewed.
+-   **Assumptions**: Assume all changes have passed linter, type-checking, and unit tests. Do not check for compile errors.
+
+You may receive the following context:
+-   \`<pr_title>\` and \`<pr_description>\`: PR context
+-   \`<commit_messages>\`: Commits in the change
+-   \`<user_context>\`: Specific review focus from the user
+-   \`<file_status>\`: List of modified files with their status
+-   \`<review_instructions>\`: Specific instructions for this review
 
 ## Output Format
-Do not include praise or positive feedback.
-Only include reviews for actual issues found in the changed code.
 
 ${createJsonResponseInstruction({
-  overview: "Summary of specific issues found in the diff changes, 'No issues found', or 'No reviewable changes' if all modified files were excluded.",
+  overview: "Summary of issues found, 'No issues found', or 'No reviewable changes' if all files were excluded.",
   specificReviews: [
     {
-      file: "path/filename.ext",
-      lines: "N or N-M",
-      review: "Specific issue with the changed code and exact actionable fix."
+      file: "path/to/file.ts",
+      lines: "42 or 15-20",
+      review: "Specific issue description and actionable fix."
     }
   ]
 })}
+
+### Examples
+
+**Example 1: Issues found**
+\`\`\`json
+{
+  "overview": "Found 2 security and 1 logic issue in the authentication changes.",
+  "specificReviews": [
+    {
+      "file": "src/auth/login.ts",
+      "lines": "23",
+      "review": "Password is logged in plaintext. Remove the console.log statement or hash the password before logging."
+    },
+    {
+      "file": "src/auth/login.ts",
+      "lines": "45-48",
+      "review": "Missing input validation for email field. Add email format validation before processing the login request."
+    },
+    {
+      "file": "src/utils/token.ts",
+      "lines": "12",
+      "review": "Token expiration is set to 365 days which is too long for security. Reduce to 24 hours or use refresh tokens."
+    }
+  ]
+}
+\`\`\`
+
+**Example 2: No issues**
+\`\`\`json
+{
+  "overview": "No issues found.",
+  "specificReviews": []
+}
+\`\`\`
+
+**Example 3: No reviewable changes**
+\`\`\`json
+{
+  "overview": "No reviewable changes. All modified files are lockfiles or generated artifacts.",
+  "specificReviews": []
+}
+\`\`\`
 `;
 function formatReviewToolInput(params) {
   const parts = [];
@@ -79648,6 +79672,11 @@ ${params.pullRequestDescription}
     parts.push(`<commit_messages>
 ${params.commitMessages}
 </commit_messages>`);
+  }
+  if (params.context) {
+    parts.push(`<user_context>
+${params.context}
+</user_context>`);
   }
   if (params.changedFiles && params.changedFiles.length > 0) {
     const fileList = params.changedFiles.map((file2) => `${file2.status}: ${file2.path}`).join(`
@@ -79679,7 +79708,7 @@ var FixIterationSummarySchema = exports_external.object({
 });
 var fixWorkflow = async (input2, context) => {
   const { tools: tools2, logger, step } = context;
-  const { command: inputCommand, task, interactive = true } = input2;
+  const { command: inputCommand, task, context: userContext, interactive = true } = input2;
   let command = inputCommand;
   const summaries = [];
   let formatCommand;
@@ -79747,7 +79776,7 @@ var fixWorkflow = async (input2, context) => {
     const result = await step(`fix-${i}`, async () => {
       const defaultContext = await getDefaultContext();
       const memoryContext = await tools2.getMemoryContext();
-      const userPrompt = getFixUserPrompt(command, exitCode, stdout, stderr, task);
+      const userPrompt = getFixUserPrompt(command, exitCode, stdout, stderr, task, userContext);
       return await agentWorkflow({
         systemPrompt: FIX_SYSTEM_PROMPT,
         userMessage: [
@@ -80580,6 +80609,7 @@ var MAX_REVIEW_RETRIES = 5;
 var TODO_HANDLING_INSTRUCTIONS = `If you discover that a task is larger than you thought, or that a new task is required, you can add a // TODO comment in the code and create a todo item for it. This will allow you to continue with the current task and address the larger issue later.`;
 async function createPlan2(input2, context) {
   const { task, plan: plan2, files, feedback, messages } = input2;
+  const { tools: tools2 } = context;
   const agentTools = [
     askFollowupQuestion_default,
     fetchUrl_default,
@@ -80600,7 +80630,13 @@ async function createPlan2(input2, context) {
       outputSchema: EpicPlanSchema
     }, context);
   }
-  const content = [{ type: "text", text: getPlanPrompt(task, plan2) }];
+  const defaultContext = await getDefaultContext();
+  const memoryContext = await tools2.getMemoryContext();
+  const prompt = `${memoryContext}
+${getPlanPrompt(task, plan2)}
+
+${defaultContext}`;
+  const content = [{ type: "text", text: prompt }];
   if (files) {
     for (const file2 of files) {
       if (file2.type === "file") {
@@ -81136,6 +81172,37 @@ Summary:`);
     for (const [idx, msg] of commitMessages.entries()) {
       logger.info(`   ${idx + 1}. ${msg}`);
     }
+    if (input2.usages && input2.usages.length > 0) {
+      const sortedUsages = [...input2.usages].sort((a, b) => a.timestamp - b.timestamp);
+      let lastUsage = {
+        input: 0,
+        output: 0,
+        cachedRead: 0,
+        cost: 0,
+        messageCount: 0
+      };
+      logger.info(`
+Usage Breakdown:`);
+      sortedUsages.forEach((usage, index) => {
+        const delta = {
+          input: usage.input - lastUsage.input,
+          output: usage.output - lastUsage.output,
+          cachedRead: usage.cachedRead - lastUsage.cachedRead,
+          cost: usage.cost - lastUsage.cost,
+          messageCount: usage.messageCount - lastUsage.messageCount
+        };
+        const tempMeter = new UsageMeter;
+        tempMeter.setUsage(delta);
+        logger.info(`   Step ${index + 1}: ${tempMeter.getUsageText()}`);
+        lastUsage = usage;
+      });
+      const last = sortedUsages.at(-1);
+      if (last) {
+        const totalMeter = new UsageMeter;
+        totalMeter.setUsage(last);
+        logger.info(`   Total: ${totalMeter.getUsageText()}`);
+      }
+    }
   } catch (error46) {
     logger.error(`
 Epic workflow failed: ${error46 instanceof Error ? error46.message : String(error46)}`);
@@ -81290,7 +81357,7 @@ var prWorkflow = async (input2, context) => {
 // src/workflows/review.workflow.ts
 var reviewWorkflow = async (input2, context) => {
   const { step, tools: tools2, logger } = context;
-  const { pr: pr2 } = input2;
+  const { pr: pr2, context: userContext } = input2;
   let changeInfo;
   if (pr2) {
     const prNumberMatch = pr2.match(/\d+$/);
@@ -81339,7 +81406,8 @@ var reviewWorkflow = async (input2, context) => {
       pullRequestTitle: prDetails.title,
       pullRequestDescription: prDetails.body,
       commitMessages,
-      changedFiles
+      changedFiles,
+      context: userContext
     };
   } else {
     const statusResult = await tools2.executeCommand({ command: "git", args: ["status", "--porcelain=v1"] });
@@ -81353,7 +81421,8 @@ var reviewWorkflow = async (input2, context) => {
       printChangedFiles(logger, changedFiles);
       changeInfo = {
         staged: hasStagedChanges,
-        changedFiles
+        changedFiles,
+        context: userContext
       };
     } else {
       await step("No local changes detected. Falling back to branch diff...", async () => {});
@@ -81390,7 +81459,8 @@ var reviewWorkflow = async (input2, context) => {
       printChangedFiles(logger, branchChangedFiles);
       changeInfo = {
         commitRange: `${defaultBranch}...${currentBranch}`,
-        changedFiles: branchChangedFiles
+        changedFiles: branchChangedFiles,
+        context: userContext
       };
     }
   }
@@ -81437,13 +81507,22 @@ var commitCommand = new Command("commit").description("Create a commit with AI-g
 // src/workflows/epic-context.ts
 import { promises as fs3 } from "node:fs";
 var EPIC_CONTEXT_FILE = ".epic.yml";
+var EpicUsageSchema = exports_external.object({
+  timestamp: exports_external.number(),
+  input: exports_external.number(),
+  output: exports_external.number(),
+  cachedRead: exports_external.number(),
+  cost: exports_external.number(),
+  messageCount: exports_external.number()
+});
 var EpicContextSchema = exports_external.object({
   task: exports_external.string().nullish(),
   plan: exports_external.string().nullish(),
   branchName: exports_external.string().nullish(),
   baseBranch: exports_external.string().nullish(),
   todos: exports_external.array(TodoItemSchema).nullish(),
-  memory: exports_external.record(exports_external.string(), exports_external.string()).nullish()
+  memory: exports_external.record(exports_external.string(), exports_external.string()).nullish(),
+  usages: exports_external.array(EpicUsageSchema).nullish()
 });
 var saveEpicContext = async (context) => {
   const yamlString = $stringify({
@@ -81452,7 +81531,8 @@ var saveEpicContext = async (context) => {
     branchName: context.branchName,
     baseBranch: context.baseBranch,
     todos: context.todos,
-    memory: context.memory
+    memory: context.memory,
+    usages: context.usages
   });
   await fs3.writeFile(EPIC_CONTEXT_FILE, yamlString, "utf-8");
 };
@@ -81526,10 +81606,26 @@ async function runEpic(task2, _options, command) {
     }
     epicContext.task = taskInput;
   }
+  let usageAppended = false;
   const workflowInput = {
     ...epicContext,
-    saveEpicContext
+    async saveEpicContext(context) {
+      if (usageMeter) {
+        const currentUsage = usageMeter.usage;
+        if (!context.usages) {
+          context.usages = [];
+        }
+        if (!usageAppended) {
+          context.usages.push({ ...currentUsage, timestamp: Date.now() });
+          usageAppended = true;
+        } else {
+          context.usages[context.usages.length - 1] = { ...currentUsage, timestamp: Date.now() };
+        }
+      }
+      await saveEpicContext(context);
+    }
   };
+  let usageMeter;
   await runWorkflow(epicWorkflow, workflowInput, {
     commandName: "epic",
     command,
@@ -81539,14 +81635,17 @@ async function runEpic(task2, _options, command) {
       ...opt,
       todoItemStore: new EpicTodoItemStore(workflowInput),
       memoryStore: new EpicMemoryStore(workflowInput)
-    })
+    }),
+    onUsageMeterCreated: (meter) => {
+      usageMeter = meter;
+    }
   });
 }
 var epicCommand = new Command("epic").description("Orchestrates a large feature or epic, breaking it down into smaller tasks.").argument("[task]", "The epic to plan and implement.").action(runEpic);
 
 // src/commands/fix.ts
-var fixCommand = new Command("fix").description("Fix issues by running a command and letting an agent fix it.").argument("[command]", "The command to run").option("-t, --task <task>", "The task to perform, providing specific instructions to the agent.").action(async (command, options, cmd) => {
-  const workflowInput = { command, task: options.task };
+var fixCommand = new Command("fix").description("Fix issues by running a command and letting an agent fix it.").argument("[command]", "The command to run").argument("[context]", "Additional context for the fix").option("-t, --task <task>", "The task to perform, providing specific instructions to the agent.").action(async (command, context, options, cmd) => {
+  const workflowInput = { command, task: options.task, context };
   const globalOpts = (cmd.parent ?? cmd).opts();
   const { verbose } = globalOpts;
   const logger = createLogger({
@@ -81660,7 +81759,7 @@ var prCommand = new Command("pr").description("Create a GitHub pull request").ar
 });
 
 // src/commands/review.ts
-var reviewCommand = new Command("review").description("Review a GitHub pull request or local changes").option("--pr <pr>", "The pull request number or URL to review").option("--json", "Output the review in JSON format", false).option("-y, --yes", "Automatically apply review feedback", false).option("--loop [count]", "Re-review after changes are applied.", (value) => {
+var reviewCommand = new Command("review").description("Review a GitHub pull request or local changes").argument("[context]", "Additional context for the review").option("--pr <pr>", "The pull request number or URL to review").option("--json", "Output the review in JSON format", false).option("-y, --yes", "Automatically apply review feedback", false).option("--loop [count]", "Re-review after changes are applied.", (value) => {
   if (value === undefined || value === null) {
     return 1;
   }
@@ -81669,7 +81768,7 @@ var reviewCommand = new Command("review").description("Review a GitHub pull requ
     throw new InvalidOptionArgumentError("Must be a positive number.");
   }
   return parsedValue;
-}, 1).action(async (options, command) => {
+}, 1).action(async (context, options, command) => {
   const { json: json2, pr: pr3, loop: maxIterations, yes: yesOption } = options;
   const yes = maxIterations > 1 || yesOption;
   let changesAppliedInThisIteration = false;
@@ -81680,7 +81779,7 @@ var reviewCommand = new Command("review").description("Review a GitHub pull requ
   });
   for (let i = 0;i < maxIterations; i++) {
     changesAppliedInThisIteration = false;
-    const input2 = { pr: pr3 };
+    const input2 = { pr: pr3, context };
     if (i > 0) {
       logger.debug(`Re-running review (iteration ${i + 1} of ${maxIterations})...`);
     }

package/package.json

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