update

app/core/qwen_client.py

@@ -43,9 +43,8 @@ class QwenClient:
                         {
                             "type": "text", 
                             "text": "请识别图中信息，提取驿站的名字、取件码、以及运单号（如：顺丰、圆通等快递单号），\
-                                取件码和运单号格式一般有：数字加-分割，字母+数字，字母+数字+字母等，\
-                                以[{\"station\":\"驿站名字\",\"pickup_codes\":[\"\",\"\"]}]的格式返回给我\
-                                已 JSON 格式返回"
+                                取件码和运单号格式一般有：数字加-分割，字母+数字，字母+数字+字母等。\
+                                以JSON 格式：[{\"station\":\"驿站名字\",\"pickup_codes\":[\"\",\"\"]}] 返回"
                         },
                         {
                             "type": "image",
@@ -91,8 +90,11 @@ class QwenClient:
                                 if isinstance(item, dict) and 'text' in item:
                                     text_content = item.get('text', '')
                                     print(f"提取的文本内容: {text_content}")
+                                    # 只获取 ```json 和 ``` 之间的内容
+                                    text_content = re.search(r'```json(.*)```', text_content, re.DOTALL).group(1)
+
                                     # 剔除 ```json 和 ```
-                                    text_content = text_content.replace('```json', '').replace('```', '')
+                                    # text_content = text_content.replace('```json', '').replace('```', '')
 
                                     # 尝试直接解析
                                     pickup_data = json.loads(text_content)