WIP: More change.

Author: kinyoklion

packages/sdk/ai/src/LDAIConfigTracker.ts

@@ -6,7 +6,6 @@ import {
   OpenAITokenUsage,
   TokenMetrics,
   TokenUsage,
-  UnderScoreTokenUsage,
 } from './api/metrics';
 
 export class LDAIConfigTracker {

packages/sdk/ai/src/api/metrics/BedrockTokenUsage.ts

@@ -1,8 +1,10 @@
+import { TokenUsage } from './TokenUsage';
+
 export function createBedrockTokenUsage(data: {
   totalTokens: number;
   inputTokens: number;
   outputTokens: number;
-}) {
+}): TokenUsage {
   return {
     total: data.totalTokens || 0,
     input: data.inputTokens || 0,

packages/sdk/ai/src/api/metrics/OpenAiUsage.ts

@@ -0,0 +1,9 @@
+import { TokenUsage } from './TokenUsage';
+
+export function createOpenAiUsage(data: any): TokenUsage {
+  return {
+    total: data.total_tokens ?? 0,
+    input: data.prompt_token ?? 0,
+    output: data.completion_token ?? 0,
+  };
+}

packages/sdk/ai/src/api/metrics/TokenUsage.ts

@@ -1,19 +1,5 @@
-export class TokenUsage {
-  totalTokens: number;
-  promptTokens: number;
-  completionTokens: number;
-
-  constructor(data: any) {
-    this.totalTokens = data.total_tokens || 0;
-    this.promptTokens = data.prompt_tokens || 0;
-    this.completionTokens = data.completion_tokens || 0;
-  }
-
-  toMetrics() {
-    return {
-      total: this.totalTokens,
-      input: this.promptTokens,
-      output: this.completionTokens,
-    };
-  }
+export interface TokenUsage {
+  total: number;
+  input: number;
+  output: number;
 }

packages/sdk/ai/src/api/metrics/UnderScoreTokenUsage.ts

@@ -1,19 +1,9 @@
-export class UnderScoreTokenUsage {
-  total_tokens: number;
-  prompt_tokens: number;
-  completion_tokens: number;
+import { TokenUsage } from './TokenUsage';
 
-  constructor(data: any) {
-    this.total_tokens = data.total_tokens || 0;
-    this.prompt_tokens = data.prompt_tokens || 0;
-    this.completion_tokens = data.completion_tokens || 0;
-  }
-
-  toMetrics() {
-    return {
-      total: this.total_tokens,
-      input: this.prompt_tokens,
-      output: this.completion_tokens,
-    };
-  }
+export function createUnderscoreTokenUsage(data: any): TokenUsage {
+  return {
+    total: data.total_tokens || 0,
+    input: data.prompt_tokens || 0,
+    output: data.completion_tokens || 0,
+  };
 }