From a530a426e9f9db2e22ee5ffbba73755f734634c0 Mon Sep 17 00:00:00 2001 From: Youth Lin Date: Wed, 5 Mar 2025 10:57:32 +0800 Subject: [PATCH] Update openai.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Prompt增加对文件名中存在谐音字或字母替代情况的辅助识别。 --- plugins.v2/chatgpt/openai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/plugins.v2/chatgpt/openai.py b/plugins.v2/chatgpt/openai.py index 139ecc9..22a1e2c 100644 --- a/plugins.v2/chatgpt/openai.py +++ b/plugins.v2/chatgpt/openai.py @@ -124,7 +124,7 @@ class OpenAi: return None result = "" try: - _filename_prompt = '接下来我会给你一个电影或电视剧的文件名，你需要识别文件名中的名称、版本、分段、年份、分瓣率、季集等信息，并按以下JSON格式返回：{"name":string,"version":string,"part":string,"year":string,"resolution":string,"season":number|null,"episode":number|null}，特别注意返回结果需要严格附合JSON格式，不需要有任何其它的字符。' + _filename_prompt = '接下来我会给你一个电影或电视剧的文件名，你需要识别文件名中的名称、版本、分段、年份、分瓣率、季集等信息，并按以下JSON格式返回：{"name":string,"version":string,"part":string,"year":string,"resolution":string,"season":number|null,"episode":number|null}，特别注意返回结果需要严格附合JSON格式，不需要有任何其它的字符。如果中文电影或电视剧的文件名中存在谐音字或字母替代的情况，请还原最有可能的结果。' completion = self.__get_model(prompt=_filename_prompt, message=filename) result = completion.choices[0].message.content return json.loads(result)