Plurals

Author: kinyoklion

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> {

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;
       };
     },
   >(

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,
   };
 }