@polka-codes/core 0.9.90 → 0.9.92

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

@@ -1724,6 +1724,13 @@ export { providerConfigSchema }
 export { providerConfigSchema as providerConfigSchema_alias_1 }
 export { providerConfigSchema as providerConfigSchema_alias_2 }
 
+declare type ProviderMetadataEntry = {
+    provider: string;
+    model: string;
+    metadata: any;
+    timestamp: number;
+};
+
 declare const providerModelSchema: z.ZodObject<{
     provider: z.ZodOptional<z.ZodString>;
     model: z.ZodOptional<z.ZodString>;
@@ -2641,7 +2648,9 @@ declare class UsageMeter {
         modelInfo?: ModelInfo;
     }): void;
     /** Override the running totals (e.g., restore from saved state). */
-    setUsage(newUsage: Partial<Totals>): void;
+    setUsage(newUsage: Partial<Totals>, options?: {
+        clearMetadata?: boolean;
+    }): void;
     /** Manually bump the message count (useful if you record some messages without token info). */
     incrementMessageCount(n?: number): void;
     /** Reset the running totals. */
@@ -2664,6 +2673,18 @@ declare class UsageMeter {
         cost: number;
         messageCount: number;
     };
+    /** Getter for provider metadata entries (immutable copy). */
+    get providerMetadata(): ProviderMetadataEntry[];
+    /** Calculate cache statistics from stored metadata entries. */
+    get cacheStats(): {
+        totalCachedTokens: number;
+        totalRequests: number;
+        requestsWithCache: number;
+        cacheHitRate: number;
+        entries: ProviderMetadataEntry[];
+    };
+    /** Clear stored provider metadata entries. */
+    clearProviderMetadata(): void;
     /** Merge another UsageMeter's totals into this one. */
     merge(other: UsageMeter): void;
     getUsageText(): string;

package/dist/index.js

@@ -166,12 +166,12 @@ import { z as z2 } from "zod";
 var memoryConfigSchema = z2.object({
   enabled: z2.boolean().optional().default(true),
   type: z2.enum(["sqlite", "memory"]).optional().default("sqlite"),
-  path: z2.string().optional().default("~/.config/polka-codes/memory.sqlite")
+  path: z2.string().optional().default("~/.config/polkacodes/memory/memory.sqlite")
 }).strict().optional();
 var DEFAULT_MEMORY_CONFIG = {
   enabled: true,
   type: "sqlite",
-  path: "~/.config/polka-codes/memory.sqlite"
+  path: "~/.config/polkacodes/memory/memory.sqlite"
 };
 function resolveHomePath(path) {
   if (path.startsWith("~")) {
@@ -2269,6 +2269,7 @@ var writeToFile_default = {
 // src/UsageMeter.ts
 var UsageMeter = class {
   #totals = { input: 0, output: 0, cachedRead: 0, cost: 0, messageCount: 0 };
+  #providerMetadataEntries = [];
   #modelInfos;
   #maxMessages;
   #maxCost;
@@ -2352,14 +2353,26 @@ var UsageMeter = class {
     this.#totals.cachedRead += result.cachedRead || 0;
     this.#totals.cost += result.cost || 0;
     this.#totals.messageCount += 1;
+    if (resp.providerMetadata && Object.keys(resp.providerMetadata).length > 0) {
+      const providerKey = Object.keys(resp.providerMetadata)[0];
+      this.#providerMetadataEntries.push({
+        provider: providerKey || llm.provider,
+        model: llm.modelId,
+        metadata: resp.providerMetadata[providerKey] || resp.providerMetadata,
+        timestamp: Date.now()
+      });
+    }
   }
   /** Override the running totals (e.g., restore from saved state). */
-  setUsage(newUsage) {
+  setUsage(newUsage, options = {}) {
     if (newUsage.input != null) this.#totals.input = newUsage.input;
     if (newUsage.output != null) this.#totals.output = newUsage.output;
     if (newUsage.cachedRead != null) this.#totals.cachedRead = newUsage.cachedRead;
     if (newUsage.cost != null) this.#totals.cost = newUsage.cost;
     if (newUsage.messageCount != null) this.#totals.messageCount = newUsage.messageCount;
+    if (options.clearMetadata) {
+      this.#providerMetadataEntries = [];
+    }
   }
   /** Manually bump the message count (useful if you record some messages without token info). */
   incrementMessageCount(n = 1) {
@@ -2368,6 +2381,7 @@ var UsageMeter = class {
   /** Reset the running totals. */
   resetUsage() {
     this.#totals = { input: 0, output: 0, cachedRead: 0, cost: 0, messageCount: 0 };
+    this.#providerMetadataEntries = [];
   }
   /** Return true once either messages or cost exceed the configured caps. */
   isLimitExceeded() {
@@ -2394,6 +2408,37 @@ var UsageMeter = class {
   get usage() {
     return { ...this.#totals };
   }
+  /** Getter for provider metadata entries (immutable copy). */
+  get providerMetadata() {
+    return [...this.#providerMetadataEntries];
+  }
+  /** Calculate cache statistics from stored metadata entries. */
+  get cacheStats() {
+    const entries = this.#providerMetadataEntries;
+    const totalRequests = entries.length;
+    let totalCachedTokens = 0;
+    let requestsWithCache = 0;
+    for (const entry of entries) {
+      const metadata = entry.metadata;
+      const cachedTokens = metadata.cachedPromptTokens ?? metadata.cacheReadTokens ?? metadata.prompt_cache_hit_tokens ?? 0;
+      if (cachedTokens > 0) {
+        totalCachedTokens += cachedTokens;
+        requestsWithCache++;
+      }
+    }
+    const cacheHitRate = totalRequests > 0 ? requestsWithCache / totalRequests : 0;
+    return {
+      totalCachedTokens,
+      totalRequests,
+      requestsWithCache,
+      cacheHitRate,
+      entries: [...this.#providerMetadataEntries]
+    };
+  }
+  /** Clear stored provider metadata entries. */
+  clearProviderMetadata() {
+    this.#providerMetadataEntries = [];
+  }
   /** Merge another UsageMeter's totals into this one. */
   merge(other) {
     const otherUsage = other.usage;
@@ -2402,6 +2447,7 @@ var UsageMeter = class {
     this.#totals.cachedRead += otherUsage.cachedRead;
     this.#totals.cost += otherUsage.cost;
     this.#totals.messageCount += otherUsage.messageCount;
+    this.#providerMetadataEntries.push(...other.providerMetadata);
   }
   getUsageText() {
     const u = this.usage;

package/package.json

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