oficcejo
150 lines
5.3 KiB
Python
150 lines
5.3 KiB
Python
"""
|
||
宏观周期分析 - 综合研判引擎
|
||
协调数据采集与AI分析，生成完整的宏观周期分析报告
|
||
"""
|
||
|
||
from macro_cycle_agents import MacroCycleAgents
|
||
from macro_cycle_data import MacroCycleDataFetcher
|
||
from typing import Dict, Any
|
||
import time
|
||
import json
|
||
import logging
|
||
import config
|
||
|
||
|
||
class MacroCycleEngine:
|
||
"""宏观周期综合研判引擎"""
|
||
|
||
def __init__(self, model=None):
|
||
self.model = model or config.DEFAULT_MODEL_NAME
|
||
self.agents = MacroCycleAgents(model=self.model)
|
||
self.data_fetcher = MacroCycleDataFetcher()
|
||
self.logger = logging.getLogger(__name__)
|
||
if not self.logger.handlers:
|
||
logging.basicConfig(level=logging.INFO, format='[%(asctime)s] %(levelname)s %(name)s: %(message)s')
|
||
print(f"[宏观周期引擎] 初始化完成 (模型: {self.model})")
|
||
|
||
def run_full_analysis(self, progress_callback=None) -> Dict[str, Any]:
|
||
"""
|
||
运行完整的宏观周期分析流程
|
||
|
||
Args:
|
||
progress_callback: 进度回调函数 (progress_pct, status_text)
|
||
|
||
Returns:
|
||
完整的分析结果
|
||
"""
|
||
print("\n" + "=" * 60)
|
||
print("🧭 宏观周期分析系统启动")
|
||
print("=" * 60)
|
||
|
||
results = {
|
||
"success": False,
|
||
"timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
|
||
"raw_data": {},
|
||
"formatted_data": "",
|
||
"agents_analysis": {},
|
||
"data_errors": []
|
||
}
|
||
|
||
try:
|
||
# 阶段1：数据采集
|
||
if progress_callback:
|
||
progress_callback(5, "📊 正在采集宏观经济数据...")
|
||
print("\n[阶段1] 宏观经济数据采集...")
|
||
print("-" * 60)
|
||
|
||
raw_data = self.data_fetcher.get_all_macro_data()
|
||
results["raw_data"] = raw_data
|
||
results["data_errors"] = raw_data.get("errors", [])
|
||
|
||
if not raw_data.get("success"):
|
||
print("⚠ 数据采集未完全成功，尝试继续分析...")
|
||
|
||
# 格式化数据
|
||
formatted_text = self.data_fetcher.format_data_for_ai(raw_data)
|
||
results["formatted_data"] = formatted_text
|
||
|
||
if progress_callback:
|
||
progress_callback(15, "✓ 数据采集完成")
|
||
print("✓ 数据采集和格式化完成")
|
||
print(f" 数据文本长度: {len(formatted_text)} 字符")
|
||
|
||
# 阶段2：AI智能体分析
|
||
print("\n[阶段2] AI智能体分析集群工作中...")
|
||
print("-" * 60)
|
||
|
||
agents_results = {}
|
||
|
||
# 2.1 康波周期分析师
|
||
if progress_callback:
|
||
progress_callback(20, "🌊 康波周期分析师正在分析...")
|
||
print("1/4 康波周期分析师...")
|
||
kondratieff_result = self.agents.kondratieff_wave_agent(formatted_text)
|
||
agents_results["kondratieff"] = kondratieff_result
|
||
|
||
if progress_callback:
|
||
progress_callback(35, "✓ 康波分析完成")
|
||
|
||
# 2.2 美林时钟分析师
|
||
if progress_callback:
|
||
progress_callback(40, "⏰ 美林时钟分析师正在分析...")
|
||
print("2/4 美林时钟分析师...")
|
||
merrill_result = self.agents.merrill_lynch_clock_agent(formatted_text)
|
||
agents_results["merrill"] = merrill_result
|
||
|
||
if progress_callback:
|
||
progress_callback(55, "✓ 美林时钟分析完成")
|
||
|
||
# 2.3 中国政策分析师
|
||
if progress_callback:
|
||
progress_callback(60, "🏛️ 中国政策分析师正在分析...")
|
||
print("3/4 中国政策分析师...")
|
||
policy_result = self.agents.china_policy_agent(formatted_text)
|
||
agents_results["policy"] = policy_result
|
||
|
||
if progress_callback:
|
||
progress_callback(75, "✓ 政策分析完成")
|
||
|
||
# 2.4 首席宏观策略师（综合三位分析师的报告）
|
||
if progress_callback:
|
||
progress_callback(80, "👔 首席宏观策略师正在综合研判...")
|
||
print("4/4 首席宏观策略师综合研判...")
|
||
chief_result = self.agents.chief_macro_strategist_agent(
|
||
kondratieff_report=kondratieff_result.get("analysis", ""),
|
||
merrill_report=merrill_result.get("analysis", ""),
|
||
policy_report=policy_result.get("analysis", ""),
|
||
macro_data_text=formatted_text
|
||
)
|
||
agents_results["chief"] = chief_result
|
||
|
||
if progress_callback:
|
||
progress_callback(95, "✓ 综合研判完成")
|
||
|
||
results["agents_analysis"] = agents_results
|
||
results["success"] = True
|
||
|
||
print("\n" + "=" * 60)
|
||
print("✓ 宏观周期分析完成！")
|
||
print("=" * 60)
|
||
|
||
if progress_callback:
|
||
progress_callback(100, "✅ 分析完成！")
|
||
|
||
except Exception as e:
|
||
print(f"\n✗ 分析过程出错: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
results["error"] = str(e)
|
||
|
||
return results
|
||
|
||
|
||
# 测试
|
||
if __name__ == "__main__":
|
||
print("=" * 60)
|
||
print("测试宏观周期分析引擎")
|
||
print("=" * 60)
|
||
engine = MacroCycleEngine()
|
||
print("引擎初始化完成")
|