@outputai/core 0.11.1-next.a5ded18.0 → 0.11.1-next.b4488bf.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@outputai/core",
-  "version": "0.11.1-next.a5ded18.0",
+  "version": "0.11.1-next.b4488bf.0",
   "description": "The core module of the output framework",
   "type": "module",
   "engines": {
@@ -50,7 +50,6 @@
     "@temporalio/common": "1.20.3",
     "@temporalio/worker": "1.20.3",
     "@temporalio/workflow": "1.20.3",
-    "decimal.js": "10.6.0",
     "folder-hash": "4.1.3",
     "json-stream-stringify": "3.1.6",
     "redis": "5.12.1",

package/src/sdk/runtime/tracing.d.ts

@@ -1,5 +1,7 @@
 import { Attribute } from '#trace_attribute';
 
+export type BaseAttribute = InstanceType<typeof Attribute.BaseAttribute>;
+
 /**
  * Tools to interact with Tracing
  */

package/src/tracing/trace_attribute.d.ts

@@ -1,12 +1,10 @@
 export declare namespace Attribute {
-  export interface Usage {
+  export class BaseAttribute {
     type: string;
-    ppm: number;
-    amount: number;
-    total: number;
+    constructor( type: string );
   }
 
-  export class HTTPRequestCount {
+  export class HTTPRequestCount extends BaseAttribute {
     static TYPE: 'http:request:count';
     type: typeof HTTPRequestCount.TYPE;
     url: string;
@@ -14,7 +12,7 @@ export declare namespace Attribute {
     constructor( url: string, requestId: string );
   }
 
-  export class HTTPRequestCost {
+  export class HTTPRequestCost extends BaseAttribute {
     static TYPE: 'http:request:cost';
     type: typeof HTTPRequestCost.TYPE;
     url: string;
@@ -23,16 +21,5 @@ export declare namespace Attribute {
     constructor( url: string, requestId: string, total: number );
   }
 
-  export class LLMUsage {
-    static TYPE: 'llm:usage';
-    type: typeof LLMUsage.TYPE;
-    modelId: string;
-    usage: Usage[];
-    constructor( modelId: string );
-    addUsage( usage: { type: string; ppm: number; amount: number } ): void;
-    readonly total: number;
-    readonly tokensUsed: number;
-  }
-
-  export type Instance = HTTPRequestCount | HTTPRequestCost | LLMUsage;
+  export type Instance = BaseAttribute;
 }

package/src/tracing/trace_attribute.js

@@ -1,5 +1,3 @@
-import Decimal from 'decimal.js';
-
 /**
  * All attributes inherit from this
  */
@@ -37,36 +35,11 @@ class HTTPRequestCost extends BaseAttribute {
   }
 }
 
-class LLMUsage extends BaseAttribute {
-  static TYPE = 'llm:usage';
-  modelId;
-  usage = [];
-  total = 0;
-  tokensUsed = 0;
-
-  constructor( modelId ) {
-    super( LLMUsage.TYPE );
-    this.modelId = modelId;
-  }
-
-  addUsage( { type, ppm, amount } ) {
-    const total = Decimal( amount ).div( 1_000_000 ).mul( ppm ).toNumber();
-    this.usage.push( {
-      type,
-      ppm,
-      amount,
-      total
-    } );
-    this.total = Decimal( this.total ).add( total ).toNumber();
-    this.tokensUsed = Decimal( this.tokensUsed ).add( amount ).toNumber();
-  }
-}
-
 /**
  * Types of ADD_ATTR attributes
  */
 export const Attribute = {
-  LLMUsage,
+  BaseAttribute,
   HTTPRequestCost,
   HTTPRequestCount
 };