Agent 模式

import os
from openai import OpenAI

client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

def run_agent(task: str, tools: list, max_iters: int = 3):
    plan = client.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": f"Plan steps for: {task}"}]
    ).choices[0].message.content
    
    for i in range(max_iters):
        result = execute_steps(plan, tools)
        if validate(result):
            return result
        plan = refine_plan(plan, result)
    return {"status": "failed", "reason": "max iterations reached"}

def execute_steps(plan, tools):
    # 简化工具调用逻辑
    return {"output": "executed", "steps": plan}

def validate(result):
    return result.get("output") == "executed"

def refine_plan(plan, result):
    return f"Revised plan based on: {result}"

Planner-Executor 基础实现

{
  "status": "success",
  "result": {
    "output": "Task completed successfully",
    "steps_executed": 3,
    "tools_used": [
      "search",
      "calculator"
    ]
  },
  "metadata": {
    "iterations": 2,
    "reflection_triggered": true,
    "execution_time_ms": 1250
  }
}