@omnicross/core 0.1.2 → 0.1.3

package/dist/{chunk-XBSYYZIY.cjs → chunk-3QOSDDNI.cjs}

@@ -1,15 +1,27 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }// src/usage/usage-recorder.ts
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } }// src/pipeline/auditUsageStash.ts
+var stash = /* @__PURE__ */ new WeakMap();
+function stashAuditUsage(key, detail) {
+  const existing = stash.get(key);
+  stash.set(key, existing ? { ...existing, ...detail } : { ...detail });
+}
+function readAuditUsage(key) {
+  return stash.get(key);
+}
+
+// src/usage/usage-recorder.ts
 var UsageRecorder = class {
   constructor(store, pricing, logger, options = {}) {
     this.store = store;
     this.pricing = pricing;
     this.logger = logger;
     this.defer = _nullishCoalesce(options.defer, () => ( ((fn) => setTimeout(fn, 0))));
+    this.onRecord = options.onRecord;
   }
   
   
   
   
+  
   /**
    * Record one LLM request. Returns immediately — the actual store insert is
    * deferred so the caller's response path is unblocked.
@@ -17,6 +29,14 @@ var UsageRecorder = class {
    * Errors during the insert are logged but never thrown.
    */
   record(input) {
+    if (input.auditResponse) {
+      stashAuditUsage(input.auditResponse, {
+        inputTokens: input.usage.inputTokens,
+        outputTokens: input.usage.outputTokens,
+        model: input.model,
+        provider: input.providerId
+      });
+    }
     this.defer(() => {
       void this.recordAsync(input).catch((err) => {
         this.logger.warn("[UsageRecorder] failed to persist usage event", {
@@ -35,6 +55,18 @@ var UsageRecorder = class {
       input.model,
       input.usage
     );
+    if (input.auditResponse) {
+      try {
+        stashAuditUsage(input.auditResponse, { costUsd });
+      } catch (e) {
+      }
+    }
+    if (this.onRecord && input.apiKeyId) {
+      try {
+        this.onRecord(input.apiKeyId, costUsd, Date.now());
+      } catch (e2) {
+      }
+    }
     const row = {
       messageId: _nullishCoalesce(input.messageId, () => ( null)),
       parentMessageId: _nullishCoalesce(input.parentMessageId, () => ( null)),
@@ -73,6 +105,9 @@ var UsageRecorder = class {
   getByApiKey(range) {
     return this.store.getByApiKey(range);
   }
+  getTimeSeries(range, bucket) {
+    return this.store.getTimeSeries(range, bucket);
+  }
   getMessagesForSession(sessionId) {
     return this.store.getMessagesForSession(sessionId);
   }
@@ -84,11 +119,12 @@ var UsageRecorder = class {
 var safeStringify = (value) => {
   try {
     return JSON.stringify(value);
-  } catch (e) {
+  } catch (e3) {
     return "";
   }
 };
 
 
 
-exports.UsageRecorder = UsageRecorder;
+
+exports.readAuditUsage = readAuditUsage; exports.UsageRecorder = UsageRecorder;