nas/Auto_Bangumi
1
0
Fork 0
mirror of https://github.com/EstrellaXD/Auto_Bangumi.git synced 2026-07-26 16:41:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: flatten openai configuration options

Browse Source
This commit is contained in:
100gle
2023-10-01 13:00:05 +08:00
parent 74bdcb915f
commit 2751bcabb2
8 changed files with 51 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/src/module/parser/title_parser.py
View File

@@ -55,12 +55,9 @@ class TitleParser:
language = settings.rss_parser.language
try:
# use OpenAI ChatGPT to parse raw title and get structured data
if settings.experimental.openai_enable:
gpt = OpenAIParser(
api_key=settings.experimental.openai_api_key,
api_base=settings.experimental.openai_api_base,
model=settings.experimental.openai_model,
)
if settings.experimental_openai.enable:
kwargs = settings.experimental_openai.dict(exclude={"enable"})
gpt = OpenAIParser(**kwargs)
episode_dict = gpt.parse(raw, asdict=True)
episode = Episode(**episode_dict)
else: