From 587df943828dd95596f3588fc6b0f4ff63ecc913 Mon Sep 17 00:00:00 2001
From: qiaoyun680 <549653222@qq.com>
Date: Mon, 19 May 2025 09:24:30 +0800
Subject: [PATCH] =?UTF-8?q?chatgpt:=E5=85=BC=E5=AE=B9=E4=B8=80=E4=BA=9B?=
 =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E8=BF=94=E5=9B=9Ejson=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E4=BF=A1=E6=81=AF=E7=94=A8markdown=E8=AF=AD=E6=B3=95=E5=8C=85?=
 =?UTF-8?q?=E8=A3=B9=E7=9A=84=E6=83=85=E5=86=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 plugins.v2/chatgpt/openai.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/plugins.v2/chatgpt/openai.py b/plugins.v2/chatgpt/openai.py
index 22a1e2c..ff3a3e6 100644
--- a/plugins.v2/chatgpt/openai.py
+++ b/plugins.v2/chatgpt/openai.py
@@ -1,4 +1,5 @@
 import json
+import re
 import time
 from typing import List, Union
 
@@ -127,6 +128,14 @@ class OpenAi:
             _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
+            # 有些模型返回json数据时会使用 ```json ``` 包裹json对象 所以需要进行提取
+            # 定义正则表达式模式，匹配```json开头和```结尾的内容
+            pattern = r'^```json\s*([\s\S]*?)\s*```$'
+            # 使用正则表达式进行匹配
+            match = re.match(pattern, result.strip())
+            if match:
+                # 提取中间的JSON部分
+                result = match.group(1)
             return json.loads(result)
         except Exception as e:
             return {