Usage null check

Cerlancism · Sep 28, 2024 · e94257d · e94257d
1 parent cf3635e
commit e94257d
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/src/translator.mjs b/src/translator.mjs
@@ -211,8 +211,8 @@ export class Translator
                     // process.stdout.write("\n")
                     this.services.onStreamEnd?.()
                 })
-                const prompt_tokens = usage.prompt_tokens
-                const completion_tokens = usage.completion_tokens
+                const prompt_tokens = usage?.prompt_tokens
+                const completion_tokens = usage?.completion_tokens
                 const output = new TranslationOutput(
                     getOutput(streamOutput),
                     prompt_tokens,

diff --git a/src/translatorStructuredArray.js b/src/translatorStructuredArray.js
@@ -89,9 +89,9 @@ export class TranslatorStructuredArray extends TranslatorStructuredBase
 
             const translationOutput = new TranslationOutput(
                 linesOut,
-                output.usage.prompt_tokens,
-                output.usage.completion_tokens,
-                output.usage.total_tokens,
+                output.usage?.prompt_tokens,
+                output.usage?.completion_tokens,
+                output.usage?.total_tokens,
                 output.choices[0].message.refusal
             )
 

diff --git a/src/translatorStructuredObject.mjs b/src/translatorStructuredObject.mjs
@@ -132,9 +132,9 @@ export class TranslatorStructuredObject extends TranslatorStructuredBase
 
             const translationOutput = new TranslationOutput(
                 linesOut,
-                output.usage.prompt_tokens,
-                output.usage.completion_tokens,
-                output.usage.total_tokens,
+                output.usage?.prompt_tokens,
+                output.usage?.completion_tokens,
+                output.usage?.total_tokens,
                 output.choices[0].message.refusal
             )