shiyue/stock-scanner
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #4 from yhw17/main

Browse Source 
bugfix: 修复流式输出索引越界错误
This commit is contained in:
兰志宏
2025-03-07 09:14:47 +08:00
committed by GitHub
parent 5dacc9f528 a3c0a57c96
commit 37613b3247
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
stock_analyzer.py
View File

@@ -502,7 +502,8 @@ class StockAnalyzer:
try: try:
json_data = json.loads(data_content) json_data = json.loads(data_content)
if 'choices' in json_data: # 检查 choices 列表是否为空
if 'choices' in json_data and json_data['choices']:
delta = json_data['choices'][0].get('delta', {}) delta = json_data['choices'][0].get('delta', {})
content = delta.get('content', '') content = delta.get('content', '')
@@ -516,6 +517,8 @@ class StockAnalyzer:
"ai_analysis_chunk": content "ai_analysis_chunk": content
}) })
yield chunk_json yield chunk_json
else:
logger.warning(f"收到空的 choices 列表: {data_content}")
except json.JSONDecodeError as e: except json.JSONDecodeError as e:
logger.error(f"[JSON解析错误] {str(e)}, 行内容: {data_content}") logger.error(f"[JSON解析错误] {str(e)}, 行内容: {data_content}")
# 忽略无法解析的JSON # 忽略无法解析的JSON