From 9c53374955f01e115ccce0e4deb55c630b27e724 Mon Sep 17 00:00:00 2001 From: yihong0618 Date: Tue, 30 Jan 2024 22:05:28 +0800 Subject: [PATCH] fix: #379 Signed-off-by: yihong0618 --- book_maker/cli.py | 1 - book_maker/translator/chatgptapi_translator.py | 3 +-- setup.py | 2 +- 3 files changed, 2 insertions(+), 4 deletions(-) diff --git a/book_maker/cli.py b/book_maker/cli.py index 50cbc9c..866baa6 100644 --- a/book_maker/cli.py +++ b/book_maker/cli.py @@ -404,7 +404,6 @@ So you are close to reaching the limit. You have to choose your own value, there e.translate_model.set_deployment_id(options.deployment_id) # TODO refactor, quick fix for gpt4 model if options.model == "chatgptapi": - print(21232) e.translate_model.set_gpt35_models() if options.model == "gpt4": e.translate_model.set_gpt4_models() diff --git a/book_maker/translator/chatgptapi_translator.py b/book_maker/translator/chatgptapi_translator.py index 849c920..8096b10 100644 --- a/book_maker/translator/chatgptapi_translator.py +++ b/book_maker/translator/chatgptapi_translator.py @@ -47,7 +47,7 @@ class ChatGPTAPI(Base): ) -> None: super().__init__(key, language) self.key_len = len(key.split(",")) - self.openai_client = OpenAI(api_key=key, base_url=api_base) + self.openai_client = OpenAI(api_key=next(self.keys), base_url=api_base) self.api_base = api_base self.prompt_template = ( @@ -72,7 +72,6 @@ class ChatGPTAPI(Base): self.openai_client.api_key = next(self.keys) def rotate_model(self): - # TODO self.model = next(self.model_list) def create_chat_completion(self, text): diff --git a/setup.py b/setup.py index 6770444..197bdfd 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ packages = [ setup( name="bbook_maker", description="The bilingual_book_maker is an AI translation tool that uses ChatGPT to assist users in creating multi-language versions of epub/txt files and books.", - version="0.7.3", + version="0.7.5", license="MIT", author="yihong0618", author_email="zouzou0208@gmail.com",