chendandan_homework/03_work/planner.py

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

import json
from llm_client import get_llm

llm = get_llm()


def create_plan(goal: str) -> list:
    """根据用户目标生成任务列表，返回 [{"task_id": 1, "task": "..."}]"""
    prompt = f"""你是一名专业任务规划专家。

用户目标：
{goal}

请将这个复杂目标拆解成多个独立任务。

要求：
1. 每个任务可以单独执行
2. 不要执行任务，只负责规划
3. 返回 JSON，不要 Markdown

格式：
[
    {{
        "task_id": 1,
        "task": "任务描述"
    }}
]
"""
    response = llm.invoke(prompt)
    plan = json.loads(response.content)
    return plan


# =====================================
# 测试 Planner
# =====================================
if __name__ == "__main__":
    goal = "分析2025年AI Agent市场，并生成行业研究报告。"
    plan = create_plan(goal)
    print("=" * 50)
    print("生成计划：")
    print("=" * 50)
    for item in plan:
        print(f"Step {item['task_id']}：{item['task']}")