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",