"""
main.py — Plan-and-Execute + ReAct Agent

完整流程：
  用户目标 → Planner → 生成任务列表 → Executor (ReAct Agent) → 调用 Tool → 返回结果 → 最终输出
"""

import sys
from pathlib import Path
path = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(path))

from agent_demo.planner import create_plan
from agent_demo.executor import execute_task

# =====================================
# 用户需求
# =====================================
goal = "请分析2026年AI Agent市场发展趋势，并生成一份行业研究报告。"

# =====================================
# 第一阶段：Planner 生成计划
# =====================================
print("\n" + "=" * 60)
print("Planner 开始工作")
print("=" * 60)

plan = create_plan(goal)
print("\n生成任务计划：\n")
for item in plan:
    print(f"Step {item['task_id']}：{item['task']}")

# =====================================
# 第二阶段：Executor 执行任务
# =====================================
print("\n" + "=" * 60)
print("Executor 开始执行")
print("=" * 60)

results = []
for item in plan:
    print(f"\n正在执行 Step {item['task_id']}：{item['task']}")
    task_result = execute_task(item["task"])
    results.append({"task": item["task"], "result": task_result})
    print("\n执行结果：")
    print(task_result)

# =====================================
# 第三阶段：汇总输出
# =====================================
print("\n" + "=" * 60)
print("全部任务完成")
print("=" * 60)

for r in results:
    print("\n任务：")
    print(r["task"])
    print("\n结果：")
    print(r["result"])