From d85be6526650149db645eaba0441b5194c9e5991 Mon Sep 17 00:00:00 2001
From: yihong0618 <zouzou0208@gmail.com>
Date: Sun, 26 Nov 2023 22:24:58 +0800
Subject: [PATCH] fix: #353

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
---
 book_maker/translator/gpt4_translator.py    | 4 ++--
 book_maker/translator/litellm_translator.py | 9 +--------
 setup.py                                    | 4 ++--
 3 files changed, 5 insertions(+), 12 deletions(-)

diff --git a/book_maker/translator/gpt4_translator.py b/book_maker/translator/gpt4_translator.py
index d3daeeb..194728d 100644
--- a/book_maker/translator/gpt4_translator.py
+++ b/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,
         )
diff --git a/book_maker/translator/litellm_translator.py b/book_maker/translator/litellm_translator.py
index 22b9e3e..31101eb 100644
--- a/book_maker/translator/litellm_translator.py
+++ b/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",
diff --git a/setup.py b/setup.py
index 88798e0..e82e4fa 100644
--- a/setup.py
+++ b/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",