@polka-codes/core 0.9.36 → 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';
@@ -1509,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 }

package/dist/index.js

@@ -2065,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,
@@ -2083,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);
           },

package/package.json

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