diff --git a/backend/src/module/parser/openai.py b/backend/src/module/parser/openai.py index 1b77ef9a..c172fac0 100644 --- a/backend/src/module/parser/openai.py +++ b/backend/src/module/parser/openai.py @@ -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: diff --git a/backend/src/test/test_openai.py b/backend/src/test/test_openai.py index 7320bcb0..ea40f865 100644 --- a/backend/src/test/test_openai.py +++ b/backend/src/test/test_openai.py @@ -1,3 +1,4 @@ +import json import os from dotenv import load_dotenv @@ -15,4 +16,10 @@ class TestOpenAIParser: def test_parse(self): text = "[梦蓝字幕组]New Doraemon 哆啦A梦新番[747][2023.02.25][AVC][1080P][GB_JP][MP4]" result = self.parser.parse(text=text) - assert result == "XXX" + assert json.loads(result) == { + "group": "梦蓝字幕组", + "title_en": "New Doraemon", + "resolution": "1080P", + "episode": 747, + "season": 1, + }