@polka-codes/core 0.7.14 → 0.7.15
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/_tsup-dts-rollup.d.ts +17 -2
- package/dist/index.js +11 -1
- package/package.json +1 -1
|
@@ -22,7 +22,14 @@ declare abstract class AgentBase {
|
|
|
22
22
|
id: string;
|
|
23
23
|
info: ModelInfo;
|
|
24
24
|
};
|
|
25
|
-
get usage():
|
|
25
|
+
get usage(): {
|
|
26
|
+
messageCount: number;
|
|
27
|
+
inputTokens: number;
|
|
28
|
+
outputTokens: number;
|
|
29
|
+
cacheWriteTokens: number;
|
|
30
|
+
cacheReadTokens: number;
|
|
31
|
+
totalCost: number;
|
|
32
|
+
};
|
|
26
33
|
}
|
|
27
34
|
export { AgentBase }
|
|
28
35
|
export { AgentBase as AgentBase_alias_1 }
|
|
@@ -2485,6 +2492,7 @@ declare class UsageMeter {
|
|
|
2485
2492
|
provider: string;
|
|
2486
2493
|
id: string;
|
|
2487
2494
|
}): void;
|
|
2495
|
+
setUsage(usage: Partial<ApiUsage>, messageCount?: number): void;
|
|
2488
2496
|
incrementMessageCount(count?: number): void;
|
|
2489
2497
|
isLimitExceeded(): {
|
|
2490
2498
|
messageCount: boolean;
|
|
@@ -2497,7 +2505,14 @@ declare class UsageMeter {
|
|
|
2497
2505
|
/**
|
|
2498
2506
|
* Get current usage totals
|
|
2499
2507
|
*/
|
|
2500
|
-
get usage():
|
|
2508
|
+
get usage(): {
|
|
2509
|
+
messageCount: number;
|
|
2510
|
+
inputTokens: number;
|
|
2511
|
+
outputTokens: number;
|
|
2512
|
+
cacheWriteTokens: number;
|
|
2513
|
+
cacheReadTokens: number;
|
|
2514
|
+
totalCost: number;
|
|
2515
|
+
};
|
|
2501
2516
|
printUsage(): void;
|
|
2502
2517
|
}
|
|
2503
2518
|
export { UsageMeter }
|
package/dist/index.js
CHANGED
|
@@ -777,6 +777,16 @@ var UsageMeter = class {
|
|
|
777
777
|
}
|
|
778
778
|
this.#usage.totalCost += usage.totalCost ?? 0;
|
|
779
779
|
}
|
|
780
|
+
setUsage(usage, messageCount) {
|
|
781
|
+
this.#usage.inputTokens = usage.inputTokens ?? this.#usage.inputTokens;
|
|
782
|
+
this.#usage.outputTokens = usage.outputTokens ?? this.#usage.outputTokens;
|
|
783
|
+
this.#usage.cacheWriteTokens = usage.cacheWriteTokens ?? this.#usage.cacheWriteTokens;
|
|
784
|
+
this.#usage.cacheReadTokens = usage.cacheReadTokens ?? this.#usage.cacheReadTokens;
|
|
785
|
+
this.#usage.totalCost = usage.totalCost ?? this.#usage.totalCost;
|
|
786
|
+
if (messageCount !== void 0) {
|
|
787
|
+
this.#messageCount = messageCount;
|
|
788
|
+
}
|
|
789
|
+
}
|
|
780
790
|
incrementMessageCount(count = 1) {
|
|
781
791
|
this.#messageCount += count;
|
|
782
792
|
}
|
|
@@ -803,7 +813,7 @@ var UsageMeter = class {
|
|
|
803
813
|
* Get current usage totals
|
|
804
814
|
*/
|
|
805
815
|
get usage() {
|
|
806
|
-
return { ...this.#usage };
|
|
816
|
+
return { ...this.#usage, messageCount: this.#messageCount };
|
|
807
817
|
}
|
|
808
818
|
printUsage() {
|
|
809
819
|
const { inputTokens, outputTokens, cacheReadTokens, cacheWriteTokens } = this.#usage;
|