github/bilingual_book_maker
1
1
Fork 0
mirror of https://github.com/yihong0618/bilingual_book_maker.git synced 2025-06-06 11:35:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: #375

Browse Source 
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
yihong0618 2024-01-28 15:24:44 +08:00
parent 66e12f1e54
commit 365d6628bf
2 changed files with 28 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
book_maker/cli.py
View File

@ -403,8 +403,11 @@ So you are close to reaching the limit. You have to choose your own value, there
raise ValueError("`api_base` must be provided when using `deployment_id`")
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("gpt4")
e.translate_model.set_gpt4_models()
if options.block_size > 0:
e.block_size = options.block_size

38
book_maker/translator/chatgptapi_translator.py
View File

@ -66,13 +66,7 @@ class ChatGPTAPI(Base):
self.system_content = environ.get("OPENAI_API_SYS_MSG") or ""
self.deployment_id = None
self.temperature = temperature
# gpt3 all models for save the limit
my_model_list = [
i["id"] for i in self.openai_client.models.list().model_dump()["data"]
]
model_list = list(set(my_model_list) & set(GPT35_MODEL_LIST))
print(f"Using model list {model_list}")
self.model_list = cycle(model_list)
self.model_list = None
def rotate_key(self):
self.openai_client.api_key = next(self.keys)
@ -314,10 +308,26 @@ class ChatGPTAPI(Base):
azure_deployment=self.deployment_id,
)
def set_gpt4_models(self, model="gpt4"):
my_model_list = [
i["id"] for i in self.openai_client.models.list().model_dump()["data"]
]
model_list = list(set(my_model_list) & set(GPT4_MODEL_LIST))
print(f"Using model list {model_list}")
self.model_list = cycle(model_list)
def set_gpt35_models(self):
# gpt3 all models for save the limit
if self.deployment_id:
self.model_list = cycle(["gpt-35-turbo"])
else:
my_model_list = [
i["id"] for i in self.openai_client.models.list().model_dump()["data"]
]
model_list = list(set(my_model_list) & set(GPT35_MODEL_LIST))
print(f"Using model list {model_list}")
self.model_list = cycle(model_list)
def set_gpt4_models(self):
# for issue #375 azure can not use model list
if self.deployment_id:
self.model_list = cycle(["gpt-4"])
else:
my_model_list = [
i["id"] for i in self.openai_client.models.list().model_dump()["data"]
]
model_list = list(set(my_model_list) & set(GPT4_MODEL_LIST))
print(f"Using model list {model_list}")
self.model_list = cycle(model_list)