fix: 修复AI API响应处理和token限制问题
- 将max_tokens从100增加到1000,确保AI能生成完整的JSON响应 - 增加AI响应content为None的异常处理,避免NoneType错误 - 解决JSON解析失败导致任务类型全部返回none的问题
This commit is contained in:
parent
22c90728e5
commit
2ad289b374
10
src/parse.py
10
src/parse.py
@ -222,7 +222,7 @@ class PDFParser:
|
||||
{"role": "system", "content": system_prompt},
|
||||
{"role": "user", "content": user_prompt}
|
||||
],
|
||||
"max_tokens": 100, # 需要返回更复杂的JSON格式
|
||||
"max_tokens": 1000, # 增加token数量以确保完整JSON响应
|
||||
"temperature": 0.1 # 降低随机性
|
||||
}
|
||||
|
||||
@ -236,7 +236,13 @@ class PDFParser:
|
||||
|
||||
if response.status_code == 200:
|
||||
result = response.json()
|
||||
ai_response = result['choices'][0]['message']['content'].strip()
|
||||
# 检查AI响应内容是否为None
|
||||
content = result.get('choices', [{}])[0].get('message', {}).get('content')
|
||||
if content is None:
|
||||
logging.error(f"AI API返回的content为None,可能是服务器问题")
|
||||
return {'is_medical': False, 'task_type': 'none', 'medical_confidence': 0.0, 'task_confidence': 0.0}
|
||||
|
||||
ai_response = content.strip()
|
||||
|
||||
try:
|
||||
# 解析JSON响应
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user