@polka-codes/cli-shared 0.9.90 → 0.9.92

package/dist/index.js

@@ -35106,7 +35106,7 @@ var require_mimeScore = __commonJS((exports, module) => {
 
 // ../../node_modules/sql.js/dist/sql-wasm.js
 var require_sql_wasm = __commonJS((exports, module) => {
-  var __dirname = "/Users/xiliangchen/projects/polka-codes/node_modules/sql.js/dist";
+  var __dirname = "/home/ubuntu/polka-codes/node_modules/sql.js/dist";
   var initSqlJsPromise = undefined;
   var initSqlJs = function(moduleConfig) {
     if (initSqlJsPromise) {
@@ -50959,12 +50959,12 @@ var toolConfigSchema = exports_external.union([
 var memoryConfigSchema = exports_external.object({
   enabled: exports_external.boolean().optional().default(true),
   type: exports_external.enum(["sqlite", "memory"]).optional().default("sqlite"),
-  path: exports_external.string().optional().default("~/.config/polka-codes/memory.sqlite")
+  path: exports_external.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("~")) {
@@ -52333,6 +52333,7 @@ var writeToFile_default = {
 // ../core/src/UsageMeter.ts
 class UsageMeter {
   #totals = { input: 0, output: 0, cachedRead: 0, cost: 0, messageCount: 0 };
+  #providerMetadataEntries = [];
   #modelInfos;
   #maxMessages;
   #maxCost;
@@ -52414,8 +52415,17 @@ class UsageMeter {
     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()
+      });
+    }
   }
-  setUsage(newUsage) {
+  setUsage(newUsage, options = {}) {
     if (newUsage.input != null)
       this.#totals.input = newUsage.input;
     if (newUsage.output != null)
@@ -52426,12 +52436,16 @@ class UsageMeter {
       this.#totals.cost = newUsage.cost;
     if (newUsage.messageCount != null)
       this.#totals.messageCount = newUsage.messageCount;
+    if (options.clearMetadata) {
+      this.#providerMetadataEntries = [];
+    }
   }
   incrementMessageCount(n = 1) {
     this.#totals.messageCount += n;
   }
   resetUsage() {
     this.#totals = { input: 0, output: 0, cachedRead: 0, cost: 0, messageCount: 0 };
+    this.#providerMetadataEntries = [];
   }
   isLimitExceeded() {
     const messageCount = this.#maxMessages > 0 && this.#totals.messageCount >= this.#maxMessages;
@@ -52453,6 +52467,34 @@ class UsageMeter {
   get usage() {
     return { ...this.#totals };
   }
+  get providerMetadata() {
+    return [...this.#providerMetadataEntries];
+  }
+  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]
+    };
+  }
+  clearProviderMetadata() {
+    this.#providerMetadataEntries = [];
+  }
   merge(other) {
     const otherUsage = other.usage;
     this.#totals.input += otherUsage.input;
@@ -52460,6 +52502,7 @@ class UsageMeter {
     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;
@@ -67433,6 +67476,7 @@ import { randomUUID } from "node:crypto";
 import { existsSync as existsSync3 } from "node:fs";
 import { mkdir as mkdir2, readFile as readFile3, rename as rename2, writeFile as writeFile2 } from "node:fs/promises";
 import { dirname as dirname3, resolve as resolve5 } from "node:path";
+import { fileURLToPath } from "node:url";
 var import_sql = __toESM(require_sql_wasm(), 1);
 
 class FileLock {
@@ -67542,7 +67586,24 @@ async function getSqlJs() {
   if (SqlJsInitPromise) {
     return SqlJsInitPromise;
   }
-  SqlJsInitPromise = import_sql.default({});
+  const moduleDir = dirname3(fileURLToPath(import.meta.url));
+  const candidates = [
+    resolve5(moduleDir, "sql-wasm.wasm"),
+    resolve5(moduleDir, "..", "dist", "sql-wasm.wasm"),
+    resolve5(moduleDir, "..", "node_modules", "sql.js", "dist", "sql-wasm.wasm"),
+    resolve5(moduleDir, "..", "..", "..", "node_modules", "sql.js", "dist", "sql-wasm.wasm"),
+    resolve5(process.cwd(), "node_modules", "sql.js", "dist", "sql-wasm.wasm")
+  ];
+  SqlJsInitPromise = import_sql.default({
+    locateFile: (file2) => {
+      for (const candidate of candidates) {
+        if (existsSync3(candidate)) {
+          return candidate;
+        }
+      }
+      return file2;
+    }
+  });
   SqlJs = await SqlJsInitPromise;
   return SqlJs;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/cli-shared",
-  "version": "0.9.90",
+  "version": "0.9.92",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",
@@ -21,7 +21,7 @@
     "@ai-sdk/provider": "^2.0.1",
     "@ai-sdk/provider-utils": "^3.0.20",
     "@inquirer/prompts": "^8.1.0",
-    "@polka-codes/core": "0.9.89",
+    "@polka-codes/core": "0.9.91",
     "ai": "^5.0.117",
     "chalk": "^5.6.2",
     "ignore": "^7.0.5",
@@ -33,6 +33,7 @@
   },
   "devDependencies": {
     "@types/lodash-es": "^4.17.12",
-    "@types/mime-types": "^3.0.1"
+    "@types/mime-types": "^3.0.1",
+    "@types/sql.js": "^1.4.9"
   }
 }