@polka-codes/core 0.9.35 → 0.9.37

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

@@ -1,6 +1,7 @@
 import type { FilePart } from 'ai';
 import type { ImagePart } from 'ai';
 import type { JSONValue } from '@ai-sdk/provider';
+import { LanguageModelUsage } from 'ai';
 import type { LanguageModelV2 } from '@ai-sdk/provider';
 import type { LanguageModelV2ToolResultOutput } from '@ai-sdk/provider';
 import type { LanguageModelV2Usage } from '@ai-sdk/provider';
@@ -321,6 +322,7 @@ declare type CommandProvider = {
         stdout: string;
         stderr: string;
         exitCode: number;
+        summary?: string;
     }>;
 };
 export { CommandProvider }
@@ -381,6 +383,7 @@ declare const configSchema: z.ZodObject<{
     budget: z.ZodOptional<z.ZodNumber>;
     retryCount: z.ZodOptional<z.ZodNumber>;
     requestTimeoutSeconds: z.ZodOptional<z.ZodNumber>;
+    summaryThreshold: z.ZodOptional<z.ZodNumber>;
     scripts: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnion<[z.ZodString, z.ZodObject<{
         command: z.ZodString;
         description: z.ZodString;
@@ -1036,6 +1039,7 @@ declare class MockProvider implements ToolProvider {
         stdout: string;
         stderr: string;
         exitCode: number;
+        summary?: string;
     }>;
     askFollowupQuestion(_question: string, _options?: string[]): Promise<string>;
     attemptCompletion(_result: string): Promise<string | undefined>;
@@ -1506,6 +1510,7 @@ export { TaskEventToolUse as TaskEventToolUse_alias_2 }
  */
 declare interface TaskEventUsage extends TaskEventBase {
     kind: TaskEventKind.Usage;
+    usage: LanguageModelUsage;
 }
 export { TaskEventUsage }
 export { TaskEventUsage as TaskEventUsage_alias_1 }
@@ -1931,6 +1936,8 @@ declare class UsageMeter {
         cost: number;
         messageCount: number;
     };
+    /** Merge another UsageMeter's totals into this one. */
+    merge(other: UsageMeter): void;
     /** Print a concise usage summary to console. */
     printUsage(customConsole?: Console): void;
     onFinishHandler(llm: LanguageModelV2): (evt: {

package/dist/index.js

@@ -266,14 +266,23 @@ var handler4 = async (provider, args) => {
   try {
     console.log("Executing command:", command2, "Requires approval:", requiresApproval);
     const result = await provider.executeCommand(command2, requiresApproval);
-    const message = `<command>${command2}</command>
+    let message = `<command>${command2}</command>
 <command_exit_code>${result.exitCode}</command_exit_code>
-<command_stdout>
+`;
+    if (result.summary) {
+      message += `<command_output_summary>
+${result.summary}
+</command_output_summary>
+`;
+    } else {
+      message += `<command_stdout>
 ${result.stdout}
 </command_stdout>
 <command_stderr>
 ${result.stderr}
-</command_stderr>`;
+</command_stderr>
+`;
+    }
     if (result.exitCode === 0) {
       return {
         type: "Reply" /* Reply */,
@@ -1369,6 +1378,15 @@ var UsageMeter = class {
   get usage() {
     return { ...this.#totals };
   }
+  /** Merge another UsageMeter's totals into this one. */
+  merge(other) {
+    const otherUsage = other.usage;
+    this.#totals.input += otherUsage.input;
+    this.#totals.output += otherUsage.output;
+    this.#totals.cachedRead += otherUsage.cachedRead;
+    this.#totals.cost += otherUsage.cost;
+    this.#totals.messageCount += otherUsage.messageCount;
+  }
   /** Print a concise usage summary to console. */
   printUsage(customConsole = console) {
     const u = this.usage;
@@ -2047,6 +2065,7 @@ Request timeout after ${requestTimeoutSeconds} seconds. Canceling current reques
       };
       try {
         resetTimeout();
+        const usageMeterOnFinishHandler = this.config.usageMeter.onFinishHandler(this.ai);
         const streamTextOptions = {
           model: this.ai,
           temperature: 0,
@@ -2065,7 +2084,10 @@ Request timeout after ${requestTimeoutSeconds} seconds. Canceling current reques
                 break;
             }
           },
-          onFinish: this.config.usageMeter.onFinishHandler(this.ai),
+          onFinish: (evt) => {
+            usageMeterOnFinishHandler(evt);
+            this.#callback({ kind: "Usage" /* Usage */, agent: this, usage: evt.totalUsage });
+          },
           onError: async (error) => {
             console.error("Error in stream:", error);
           },
@@ -3191,6 +3213,7 @@ var configSchema = z16.object({
   budget: z16.number().positive().optional(),
   retryCount: z16.number().int().min(0).optional(),
   requestTimeoutSeconds: z16.number().int().positive().optional(),
+  summaryThreshold: z16.number().int().positive().optional(),
   scripts: z16.record(
     z16.string(),
     z16.string().or(

package/package.json

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