Function generate_plan

runner/mobiagent/multi_task/planner.py:267–324

生成多阶段任务计划 Args: task_description: 任务描述 planner_client: Planner客户端 planner_model: Planner模型名称 experience_results: 经验检索结果（可选） temperature: 采样温度 Returns: Plan对象，生成失败返回None

(
    task_description: str,
    planner_client: OpenAI,
    planner_model: str,
    experience_results: str = "",
    temperature: float = 0.0
)

def generate_plan(
    task_description: str,
    planner_client: OpenAI,
    planner_model: str,
    experience_results: str = "",
    temperature: float = 0.0
) -> Optional[Plan]:
    """
    生成多阶段任务计划
    
    Args:
        task_description: 任务描述
        planner_client: Planner客户端
        planner_model: Planner模型名称
        experience_results: 经验检索结果（可选）
        temperature: 采样温度
    
    Returns:
        Plan对象，生成失败返回None
    """
    # 检索相关经验 - 多阶段任务使用多阶段模板
    if PromptTemplateSearch and default_multistage_template_path.exists():
        try:
            search_engine = PromptTemplateSearch(template_path=default_multistage_template_path, storage_dir=DEFAULT_MULTISTORAGE_STORAGE_DIR)
            experience_results = search_engine.get_experience(
                task_description, 
                1  # 获取1个最相关经验用于多阶段任务规划
            )
            logging.info(f"计划生成时检索到的相关经验:\n{experience_results}")
        except Exception as e:
            logging.warning(f"经验检索失败: {e}")
            experience_results = ""
    
    prompt = PLANNER_PLAN_GENERATION_PROMPT.format(
        task_description=task_description,
        experience_content=experience_results or "无相关经验",
        app_list = APP_LIST
    )
    
    logging.info(f"计划生成prompt: \n{prompt}")
    
    response = planner_client.chat.completions.create(
        model=planner_model,
        messages=[{"role": "user", "content": prompt}],
        temperature=temperature
    )
    
    response_str = response.choices[0].message.content
    logging.info(f"计划生成响应: \n{response_str}")
    
    try:
        plan_data = parse_planner_response(response_str)
        plan = Plan.model_validate(plan_data)
        logging.info(f"生成的计划: {plan}")
        return plan
    except Exception as e:
        logging.error(f"生成计划失败: {e}")
        return None