Plurals

launchdarkly · Nov 5, 2024 · a4ce22e · a4ce22e
1 parent fb05b7e
commit a4ce22e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/packages/sdk/server-ai/src/LDAIConfigTrackerImpl.ts b/packages/sdk/server-ai/src/LDAIConfigTrackerImpl.ts
@@ -53,8 +53,8 @@ export class LDAIConfigTrackerImpl implements LDAIConfigTracker {
     TRes extends {
       usage?: {
         total_tokens?: number;
-        prompt_token?: number;
-        completion_token?: number;
+        prompt_tokens?: number;
+        completion_tokens?: number;
       };
     },
   >(func: () => Promise<TRes>): Promise<TRes> {

diff --git a/packages/sdk/server-ai/src/api/config/LDAIConfigTracker.ts b/packages/sdk/server-ai/src/api/config/LDAIConfigTracker.ts
@@ -49,8 +49,8 @@ export interface LDAIConfigTracker {
     TRes extends {
       usage?: {
         total_tokens?: number;
-        prompt_token?: number;
-        completion_token?: number;
+        prompt_tokens?: number;
+        completion_tokens?: number;
       };
     },
   >(

diff --git a/packages/sdk/server-ai/src/api/metrics/OpenAiUsage.ts b/packages/sdk/server-ai/src/api/metrics/OpenAiUsage.ts
@@ -2,12 +2,12 @@ import { LDTokenUsage } from './LDTokenUsage';
 
 export function createOpenAiUsage(data: {
   total_tokens?: number;
-  prompt_token?: number;
-  completion_token?: number;
+  prompt_tokens?: number;
+  completion_tokens?: number;
 }): LDTokenUsage {
   return {
     total: data.total_tokens ?? 0,
-    input: data.prompt_token ?? 0,
-    output: data.completion_token ?? 0,
+    input: data.prompt_tokens ?? 0,
+    output: data.completion_tokens ?? 0,
   };
 }