fix package name

tracker configs

Author: Daniel OBrien

packages/sdk/ai/package.json

@@ -1,5 +1,5 @@
 {
-  "name": "@launchdarkly/node-server-sdk-ai",
+  "name": "@launchdarkly/ai",
   "version": "0.1.0",
   "description": "AI and LLM integration for the LaunchDarkly Server-Side SDK for Node.js",
   "main": "dist/index.js",

packages/sdk/ai/src/LDAIConfigTracker.ts

@@ -5,60 +5,73 @@ import { BedrockTokenUsage, FeedbackKind, TokenUsage, UnderscoreTokenUsage } fro
 
 export class LDAIConfigTracker {
   private ldClient: LDClient;
-  private variationId: Record<string, string>;
+  private configKey: string;
+  private variationId: string;
   private context: LDContext;
 
-  constructor(ldClient: LDClient, variationId: string, context: LDContext) {
+  constructor(ldClient: LDClient, configKey: string, variationId: string, context: LDContext) {
     this.ldClient = ldClient;
-    this.variationId = { variationId };
+    this.configKey = configKey;
+    this.variationId = variationId;
     this.context = context;
   }
 
+  getTrackData() {
+    return {
+      configKey: this.configKey,
+      variationId: this.variationId,
+    };
+  }
+
   trackDuration(duration: number): void {
     this.ldClient.track('$ld:ai:duration:total', this.context, this.variationId, duration);
   }
 
   trackTokens(tokens: TokenUsage | UnderscoreTokenUsage | BedrockTokenUsage) {
+    console.log('tracking LLM tokens', tokens);
     const tokenMetrics = usageToTokenMetrics(tokens);
+    console.log('token metrics', tokenMetrics);
     if (tokenMetrics.total > 0) {
       this.ldClient.track(
         '$ld:ai:tokens:total',
         this.context,
-        this.variationId,
+        this.getTrackData(),
         tokenMetrics.total,
       );
     }
     if (tokenMetrics.input > 0) {
+      console.log('tracking input tokens', tokenMetrics.input);
       this.ldClient.track(
         '$ld:ai:tokens:input',
         this.context,
-        this.variationId,
+        this.getTrackData(),
         tokenMetrics.input,
       );
     }
     if (tokenMetrics.output > 0) {
+      console.log('tracking output tokens', tokenMetrics.output);
       this.ldClient.track(
         '$ld:ai:tokens:output',
         this.context,
-        this.variationId,
+        this.getTrackData(),
         tokenMetrics.output,
       );
     }
   }
 
   trackError(error: number) {
-    this.ldClient.track('$ld:ai:error', this.context, this.variationId, error);
+    this.ldClient.track('$ld:ai:error', this.context, this.getTrackData(), error);
   }
 
   trackGeneration(generation: number) {
-    this.ldClient.track('$ld:ai:generation', this.context, this.variationId, generation);
+    this.ldClient.track('$ld:ai:generation', this.context, this.getTrackData(), generation);
   }
 
   trackFeedback(feedback: { kind: FeedbackKind }) {
     if (feedback.kind === FeedbackKind.Positive) {
-      this.ldClient.track('$ld:ai:feedback:user:positive', this.context, this.variationId, 1);
+      this.ldClient.track('$ld:ai:feedback:user:positive', this.context, this.getTrackData(), 1);
     } else if (feedback.kind === FeedbackKind.Negative) {
-      this.ldClient.track('$ld:ai:feedback:user:negative', this.context, this.variationId, 1);
+      this.ldClient.track('$ld:ai:feedback:user:negative', this.context, this.getTrackData(), 1);
     }
   }
 }

packages/sdk/ai/src/index.ts

@@ -76,11 +76,18 @@ export class AIClient {
 
     return {
       config: detail.value,
-      tracker: new LDAIConfigTracker(this.ldClient, detail.value.variationId, context),
+      tracker: new LDAIConfigTracker(
+        this.ldClient,
+        key,
+        detail.value.variationId || '1234-1234-1234-1234',
+        context,
+      ),
     };
   }
 }
 
 export function init(ldClient: LDClient): AIClient {
   return new AIClient(ldClient);
 }
+
+export { LDAIConfigTracker } from './LDAIConfigTracker';