fix: #353

Signed-off-by: yihong0618 <zouzou0208@gmail.com>

book_maker/translator/gpt4_translator.py

@@ -2,9 +2,9 @@ import re
 import time
 from copy import copy
 from os import environ, linesep
-from rich import print
 
 import openai
+from rich import print
 
 from .base_translator import Base
 
@@ -81,7 +81,7 @@ class GPT4(Base):
             )
 
         return openai.ChatCompletion.create(
-            model="gpt-4",
+            model="gpt-4-1106-preview",
             messages=messages,
             temperature=self.temperature,
         )

book_maker/translator/litellm_translator.py

@@ -1,16 +1,9 @@
-import re
-import time
-from copy import copy
-from os import environ, linesep
-from rich import print
+from os import linesep
 
-import openai
 from litellm import completion
 
 from book_maker.translator.chatgptapi_translator import ChatGPTAPI
 
-from .base_translator import Base
-
 PROMPT_ENV_MAP = {
     "user": "BBM_CHATGPTAPI_USER_MSG_TEMPLATE",
     "system": "BBM_CHATGPTAPI_SYS_MSG",

setup.py

@@ -3,7 +3,7 @@ from setuptools import find_packages, setup
 
 packages = [
     "bs4",
-    "openai",
+    "openai==0.27.2",
     "litellm",
     "requests",
     "ebooklib",
@@ -17,7 +17,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.5.0",
+    version="0.5.1",
     license="MIT",
     author="yihong0618",
     author_email="zouzou0208@gmail.com",