This commit is contained in:
aaron 2025-03-10 23:21:54 +08:00
parent bcae21deb7
commit 436d09a5f5

View File

@ -35,14 +35,17 @@ class QwenClient:
messages = [
{
"role": "system",
"content": "是一个专门识别快递的取件码,运单号的助手。请准确提取图片中的所有取件码和运单号信息。"
"content": "现在是一个专门提取驿站名字、快递取件码和运单号等信息的助手"
},
{
"role": "user",
"content": [
{
"type": "text",
"text": "请识别图中驿站的所有取件码或者运单号, 取件码运单号格式有:数字加-分割的 ,字母开头的等,以[{\"station\":\"驿站名字\",\"pickup_codes\":[\"2232\",\"2323\"]}]的格式返回。只返回JSON格式数据不要其他解释。"
"text": "请识别图中信息,提取驿站的名字、取件码、以及运单号(如:顺丰、圆通等快递单号),\
取件码和运单号格式一般有数字加-分割字母+数字字母+数字+字母等\
[{\"station\":\"驿站名字\",\"pickup_codes\":[\"\",\"\"]}]的格式返回给我\
JSON 格式返回"
},
{
"type": "image",
@ -88,6 +91,9 @@ class QwenClient:
if isinstance(item, dict) and 'text' in item:
text_content = item.get('text', '')
print(f"提取的文本内容: {text_content}")
# 剔除 ```json 和 ```
text_content = text_content.replace('```json', '').replace('```', '')
# 尝试直接解析
pickup_data = json.loads(text_content)
print(f"pickup_data: {pickup_data}")