nas/Auto_Bangumi
1
0
Fork 0
mirror of https://github.com/EstrellaXD/Auto_Bangumi.git synced 2026-05-03 01:40:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: adjust default parameters and test assertion

Browse Source
This commit is contained in:
100gle
2023-09-27 17:01:14 +08:00
parent 8477774b62
commit 77201d5de2
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/src/module/parser/openai.py
View File

@@ -44,16 +44,16 @@ class OpenAIParser:
def __init__(
self,
api_key: str,
api_base: str | None = None,
model: str | None = None,
api_base: str = "https://api.openai.com/v1",
model: str = "gpt-3.5-turbo",
**kwargs,
) -> None:
if not api_key:
raise ValueError("API key is required.")
self._api_key = api_key
self.api_base = api_base or "https://api.openai.com/v1"
self.model = model or "gpt-3.5-turbo"
self.api_base = api_base
self.model = model
self.openai_kwargs = kwargs
def parse(self, text: str, prompt: str | None = None) -> str: