Commit 3a404293 authored by dmMaze's avatar dmMaze
Browse files

fix usage not available for gpt #640

parent 631427c9
Loading
Loading
Loading
Loading
+6 −4
Original line number Diff line number Diff line
@@ -292,6 +292,7 @@ class GPTTranslator(BaseTranslator):
        )

        if OPENAPI_V1_API:
            if response.usage is not None:
                self.token_count += response.usage.total_tokens
                self.token_count_last = response.usage.total_tokens
        else:
@@ -322,6 +323,7 @@ class GPTTranslator(BaseTranslator):
        )

        if OPENAPI_V1_API:
            if response.usage is not None:
                self.token_count += response.usage.total_tokens
                self.token_count_last = response.usage.total_tokens
        else:
+3 −2
Original line number Diff line number Diff line
@@ -269,6 +269,7 @@ Then stop, without any other explanations or notes.
        self.logger.debug(f'openai response: \n {response}')

        if OPENAPI_V1_API:
            if response.usage is not None:
                self.token_count += response.usage.total_tokens
                self.token_count_last = response.usage.total_tokens
        else: