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

Update ai_analyzer.py

Browse Source
This commit is contained in:
Cassianvale
2025-03-06 19:38:36 +08:00
parent 9246e7f02b
commit 35cf295b29
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
services/ai_analyzer.py
View File

@@ -66,7 +66,6 @@ class AIAnalyzer:
'volume_trend': 'increasing' if df.iloc[-1]['Volume_Ratio'] > 1 else 'decreasing', 'volume_trend': 'increasing' if df.iloc[-1]['Volume_Ratio'] > 1 else 'decreasing',
'rsi_level': df.iloc[-1]['RSI'] 'rsi_level': df.iloc[-1]['RSI']
} }
print(recent_data, technical_summary)
# 根据市场类型调整分析提示 # 根据市场类型调整分析提示
if market_type in ['ETF', 'LOF']: if market_type in ['ETF', 'LOF']:
@@ -197,7 +196,7 @@ class AIAnalyzer:
if len(buffer) > 100: if len(buffer) > 100:
yield json.dumps({ yield json.dumps({
"stock_code": stock_code, "stock_code": stock_code,
"partial_content": buffer "ai_analysis_chunk": buffer
}) })
collected_messages.append(buffer) collected_messages.append(buffer)
buffer = "" buffer = ""
@@ -209,13 +208,13 @@ class AIAnalyzer:
if buffer: if buffer:
yield json.dumps({ yield json.dumps({
"stock_code": stock_code, "stock_code": stock_code,
"partial_content": buffer "ai_analysis_chunk": buffer
}) })
collected_messages.append(buffer) collected_messages.append(buffer)
# 尝试从整个内容中提取JSON # 尝试从整个内容中提取JSON
full_content = "".join(collected_messages) full_content = "".join(collected_messages)
print(f"尝试从整个内容中提取JSON: {full_content}")
# 如果没有成功解析JSON，返回原始内容 # 如果没有成功解析JSON，返回原始内容
if not full_content.strip().startswith("{"): if not full_content.strip().startswith("{"):
yield json.dumps({ yield json.dumps({