@polka-codes/cli 0.9.55 → 0.9.56

package/dist/index.js

@@ -35579,7 +35579,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.9.55";
+var version = "0.9.56";
 
 // src/commands/code.ts
 import { readFile as readFile3 } from "node:fs/promises";
@@ -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
 
@@ -80580,6 +80588,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 +80609,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 +81151,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)}`);
@@ -81437,13 +81483,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 +81507,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 +81582,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,7 +81611,10 @@ 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);

package/package.json

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