@polka-codes/cli 0.7.12 → 0.7.14

package/dist/index.js

@@ -24629,7 +24629,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.7.12";
+var version = "0.7.14";
 
 // ../core/src/AiService/AiServiceBase.ts
 class AiServiceBase {
@@ -24638,6 +24638,7 @@ class AiServiceBase {
     this.usageMeter = usageMeter;
   }
   async* send(systemPrompt, messages) {
+    this.usageMeter.checkLimit();
     this.usageMeter.incrementMessageCount();
     const stream = this.sendImpl(systemPrompt, messages);
     for await (const chunk of stream) {
@@ -24650,6 +24651,7 @@ class AiServiceBase {
     }
   }
   async request(systemPrompt, messages) {
+    this.usageMeter.checkLimit();
     this.usageMeter.incrementMessageCount();
     const stream = this.sendImpl(systemPrompt, messages);
     const usage = {
@@ -33352,10 +33354,18 @@ class UsageMeter {
     const cost = this.#usage.totalCost >= this.maxCost;
     return {
       messageCount,
+      maxMessageCount: this.maxMessageCount,
       cost,
+      maxCost: this.maxCost,
       result: messageCount || cost
     };
   }
+  checkLimit() {
+    const result = this.isLimitExceeded();
+    if (result.result) {
+      throw new Error(`Usage limit exceeded. Message count: ${result.messageCount}/${result.maxMessageCount}, cost: ${result.cost}/${result.maxCost}`);
+    }
+  }
   get usage() {
     return { ...this.#usage };
   }
@@ -34674,9 +34684,10 @@ class AgentBase {
   ai;
   config;
   handlers;
-  messages = [];
-  constructor(name2, ai, config) {
+  #messages;
+  constructor(name2, ai, config, messages = []) {
     this.ai = ai;
+    this.#messages = messages;
     if (config.agents && config.agents.length > 0) {
       const agents = agentsPrompt(config.agents, name2);
       config.systemPrompt += `
@@ -34689,6 +34700,9 @@ ${agents}`;
     }
     this.handlers = handlers;
   }
+  get messages() {
+    return this.#messages;
+  }
   async#callback(event) {
     await this.config.callback?.(event);
   }
@@ -34696,6 +34710,21 @@ ${agents}`;
     this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
     return await this.#processLoop(prompt);
   }
+  async step(promp, messages) {
+    if (messages) {
+      this.#messages = messages;
+    }
+    if (this.#messages.length === 0) {
+      this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
+    }
+    return await this.#request(promp);
+  }
+  async handleStepResponse(response, messages) {
+    if (messages) {
+      this.#messages = messages;
+    }
+    return this.#handleResponse(response);
+  }
   async#processLoop(userMessage) {
     let nextRequest = userMessage;
     while (true) {
@@ -34720,7 +34749,7 @@ ${agents}`;
       throw new Error("userMessage is missing");
     }
     await this.#callback({ kind: "StartRequest" /* StartRequest */, agent: this, userMessage });
-    this.messages.push({
+    this.#messages.push({
       role: "user",
       content: userMessage
     });
@@ -34728,7 +34757,7 @@ ${agents}`;
     const retryCount = 5;
     for (let i3 = 0;i3 < retryCount; i3++) {
       currentAssistantMessage = "";
-      const stream = this.ai.send(this.config.systemPrompt, this.messages);
+      const stream = this.ai.send(this.config.systemPrompt, this.#messages);
       try {
         for await (const chunk of stream) {
           switch (chunk.type) {
@@ -34755,7 +34784,7 @@ ${agents}`;
     if (!currentAssistantMessage) {
       throw new Error("No assistant message received");
     }
-    this.messages.push({
+    this.#messages.push({
       role: "assistant",
       content: currentAssistantMessage
     });
@@ -47427,12 +47456,12 @@ ${newConfig.provider.toUpperCase()}_API_KEY=${newConfig.apiKey}`;
 // src/commands/pr.ts
 import { execSync as execSync2, spawnSync as spawnSync2 } from "node:child_process";
 var prCommand = new Command("pr").description("Create a GitHub pull request").argument("[message]", "Optional context for the commit message generation").action(async (message, _options, command) => {
-  const spinner = ora("Gathering information...").start();
   const options = command.parent?.opts() ?? {};
   const { providerConfig, config } = parseOptions(options);
   const { provider: provider2, model, apiKey, parameters } = providerConfig.getConfigForCommand("pr") ?? {};
   console.log("Provider:", provider2);
   console.log("Model:", model);
+  const spinner = ora("Gathering information...").start();
   if (!provider2) {
     console.error('Error: No provider specified. Please run "pokla config" to configure your AI provider.');
     process.exit(1);
@@ -47482,6 +47511,8 @@ var prCommand = new Command("pr").description("Create a GitHub pull request").ar
       commitDiff: diff
     });
     spinner.succeed("Pull request details generated");
+    console.log("Title:", prDetails.response.title);
+    console.log(prDetails.response.description);
     await new Promise((resolve3) => setTimeout(resolve3, 10));
     spawnSync2("gh", ["pr", "create", "--title", prDetails.response.title.trim(), "--body", prDetails.response.description.trim()], {
       stdio: "inherit"

package/package.json

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