Commit 8606d373 authored by dmMaze's avatar dmMaze
Browse files

add fancy scrollbars

parent 87f8c4d1

modules/translators/trans_chatgpt.py

+18 −6
Original line number Diff line number Diff line
@@ -266,6 +266,11 @@ class GPTTranslator(BaseTranslator): 
            temperature=self.temperature,

            top_p=self.top_p,

        )



        if OPENAPI_V1_API:

            self.token_count += response.usage.total_tokens

            self.token_count_last = response.usage.total_tokens

        else:

            self.token_count += response.usage['total_tokens']

            self.token_count_last = response.usage['total_tokens']

        return response.choices[0].text
@@ -292,9 +297,16 @@ class GPTTranslator(BaseTranslator): 
            top_p=self.top_p,

        )



        if OPENAPI_V1_API:

            self.token_count += response.usage.total_tokens

            self.token_count_last = response.usage.total_tokens

        else:

            self.token_count += response.usage['total_tokens']

            self.token_count_last = response.usage['total_tokens']

        for choice in response.choices:

            if OPENAPI_V1_API:

                return choice.message.content

            else:

                if 'text' in choice:

                    return choice.text