From e94257d2ad5ffc3c94f0595a98c2c10ea7866a8d Mon Sep 17 00:00:00 2001 From: Chen Yu Date: Sun, 29 Sep 2024 03:58:37 +0800 Subject: [PATCH] Usage null check --- src/translator.mjs | 4 ++-- src/translatorStructuredArray.js | 6 +++--- src/translatorStructuredObject.mjs | 6 +++--- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/src/translator.mjs b/src/translator.mjs index d7d6035..23e5367 100644 --- 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 index b604c30..dca2763 100644 --- 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 index ec94b3b..0903c7a 100644 --- 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 )