fix Bedrock usage

Author: Daniel OBrien

packages/sdk/ai/src/LDAIConfigTracker.ts

@@ -2,6 +2,7 @@ import { LDClient, LDContext } from '@launchdarkly/node-server-sdk';
 
 import {
   BedrockTokenUsage,
+  createBedrockTokenUsage,
   FeedbackKind,
   OpenAITokenUsage,
   TokenUsage,
@@ -66,17 +67,25 @@ export class LDAIConfigTracker {
     return result;
   }
 
-  async trackBedrockConverse(res: any): Promise<any> {
+  async trackBedrockConverse(res: {
+    $metadata?: { httpStatusCode: number };
+    metrics?: { latencyMs: number };
+    usage?: {
+      inputTokens: number;
+      outputTokens: number;
+      totalTokens: number;
+    };
+  }): Promise<any> {
     if (res.$metadata?.httpStatusCode === 200) {
       this.trackGeneration(1);
-    } else if (res.$metadata?.httpStatusCode >= 400) {
+    } else if (res.$metadata?.httpStatusCode && res.$metadata.httpStatusCode >= 400) {
       this.trackError(res.$metadata.httpStatusCode);
     }
     if (res.metrics) {
       this.trackDuration(res.metrics.latencyMs);
     }
     if (res.usage) {
-      this.trackTokens(new BedrockTokenUsage(res.usage));
+      this.trackTokens(createBedrockTokenUsage(res.usage));
     }
     return res;
   }

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

@@ -1,19 +1,11 @@
-export class BedrockTokenUsage {
+export function createBedrockTokenUsage(data: {
   totalTokens: number;
   inputTokens: number;
   outputTokens: number;
-
-  constructor(data: any) {
-    this.totalTokens = data.totalTokens || 0;
-    this.inputTokens = data.inputTokens || 0;
-    this.outputTokens = data.outputTokens || 0;
-  }
-
-  toMetrics() {
-    return {
-      total: this.totalTokens,
-      input: this.inputTokens,
-      output: this.outputTokens,
-    };
-  }
+}) {
+  return {
+    total: data.totalTokens || 0,
+    input: data.inputTokens || 0,
+    output: data.outputTokens || 0,
+  };
 }