fix: Added UsageInfo

Author: asafgardin

src/models/chat/ChatCompletionsResponse.ts

@@ -23,11 +23,7 @@ export interface ChatCompletionResponseChoice {
 export interface ChatCompletionChunk {
     id: string;
     choices: ChoicesChunk[];
-    usage?: {
-        prompt_tokens: number;
-        completion_tokens: number;
-        total_tokens: number;
-    };
+    usage?: UsageInfo;
 }
 
 export interface ChoiceDelta {
@@ -42,12 +38,19 @@ export interface ChoicesChunk {
     finish_reason?: string;
 }
 
+export interface UsageInfo {
+    prompt_tokens: number;
+    completion_tokens: number;
+    total_tokens: number;
+}
+
 export declare namespace ChatCompletions {
     export {
         type ChatCompletionResponseChoice,
         type ChatCompletionResponse,
         type ChoiceDelta,
         type ChoicesChunk,
         type ChatCompletionChunk,
+        type UsageInfo,
     }
 }

src/models/chat/index.ts

@@ -10,6 +10,7 @@ export {
     type ChoiceDelta,
     type ChoicesChunk,
     type ChatCompletionChunk,
+    type UsageInfo,
 } from "./ChatCompletionsResponse";
 
 export {

src/models/index.ts

@@ -24,4 +24,5 @@ export {
     type ChoicesChunk,
     type ChatCompletionChunk,
     type ChatModel,
+    type UsageInfo,
 } from "./chat";