第 7 章：规划与自我恢复

User: 帮我设计一个电商订单处理 Agent

Assistant Plan:
1. 定义订单状态机
2. 列出需要的工具（查库存、扣款、通知物流）
3. 设计异常分支（库存不足、支付失败）
4. 给出核心伪代码

import json

SCHEMA = {
    "type": "object",
    "properties": {
        "service_name": {"type": "string"},
        "port": {"type": "integer", "minimum": 1024, "maximum": 65535},
        "replicas": {"type": "integer", "minimum": 1, "maximum": 10},
        "env": {"type": "array", "items": {"type": "string"}},
    },
    "required": ["service_name", "port", "replicas"],
}


def validate_config(config: dict) -> str | None:
    if not isinstance(config, dict):
        return "Config must be a JSON object."
    for key in ["service_name", "port", "replicas"]:
        if key not in config:
            return f"Missing required field: {key}"
    if not (1024 <= config["port"] <= 65535):
        return "port must be between 1024 and 65535."
    if not (1 <= config["replicas"] <= 10):
        return "replicas must be between 1 and 10."
    return None


def generate_with_recovery(client, request: str, max_retries: int = 3) -> dict:
    messages = [
        {
            "role": "system",
            "content": (
                "You are a configuration generator. "
                f"Return only valid JSON matching this schema: {json.dumps(SCHEMA)}. "
                "No markdown, no explanation."
            ),
        },
        {"role": "user", "content": request},
    ]

    for attempt in range(1, max_retries + 1):
        print(f"\n--- Attempt {attempt} ---")
        response = client.chat_completion(messages=messages, temperature=0.2, max_tokens=300)
        raw = response["choices"][0]["message"]["content"]
        print("Raw output:", raw)

        try:
            parsed = json.loads(raw)
        except json.JSONDecodeError as exc:
            error = f"Invalid JSON: {exc}"
            print(error)
            messages.append({"role": "assistant", "content": raw})
            messages.append({"role": "user", "content": f"That was not valid JSON. {error} Please retry."})
            continue

        error = validate_config(parsed)
        if error is None:
            return parsed

        print("Validation error:", error)
        messages.append({"role": "assistant", "content": raw})
        messages.append({"role": "user", "content": f"Validation failed: {error}. Fix the JSON and retry."})

    raise RuntimeError("Failed to generate valid config after max retries.")

Assistant: {"port": 80, "replicas": 3}
User: Validation failed: port must be between 1024 and 65535. Fix the JSON and retry.

User: The previous output was invalid because port 80 is reserved.
Please reflect on the mistake and produce a corrected config.

for (let attempt = 1; attempt <= maxRetries; attempt++) {
  const response = await client.chatCompletion({ messages, temperature: 0.2, max_tokens: 300 });
  const raw = response.choices[0].message.content ?? "";
  try {
    const parsed = JSON.parse(raw);
    const error = validateConfig(parsed);
    if (!error) return parsed;
    messages.push({ role: "assistant", content: raw });
    messages.push({ role: "user", content: `Validation failed: ${error}. Fix and retry.` });
  } catch (e) {
    messages.push({ role: "assistant", content: raw });
    messages.push({ role: "user", content: `Invalid JSON: ${e}. Please retry.` });
  }
}
throw new Error("Max retries exceeded");

for attempt := 1; attempt <= maxRetries; attempt++ {
    resp, _ := client.ChatCompletion(openaiclient.ChatCompletionRequest{
        Messages: messages,
        Temperature: floatPtr(0.2),
        MaxTokens: intPtr(300),
    })
    raw := resp.Choices[0].Message.Content
    var parsed map[string]interface{}
    if err := json.Unmarshal([]byte(raw), &parsed); err != nil {
        messages = append(messages,
            openaiclient.Message{Role: "assistant", Content: raw},
            openaiclient.Message{Role: "user", Content: fmt.Sprintf("Invalid JSON: %v", err)},
        )
        continue
    }
    if err := validateConfig(parsed); err != nil {
        messages = append(messages,
            openaiclient.Message{Role: "assistant", Content: raw},
            openaiclient.Message{Role: "user", Content: fmt.Sprintf("Validation failed: %v", err)},
        )
        continue
    }
    return parsed
}