ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

2026-08-20 13:12:50 +00:00
commit b119135836
10275 changed files with 3284984 additions and 0 deletions
@@ -0,0 +1,70 @@
+"""
+配置模块：集中读取环境变量。
+
+阶段化系统提示词附加项目 使用 OpenAI 官方 SDK，所有可调项都通过环境变量注入，
+方便切换到兼容 OpenAI 协议的其他厂商（Kimi / Doubao 等）。
+"""
+
+import os
+
+try:
+    # 允许把配置写在 .env 里（可选依赖）
+    from dotenv import load_dotenv
+
+    load_dotenv()
+except Exception:  # pragma: no cover - dotenv 不是硬性依赖
+    pass
+
+
+def _to_openrouter_model(model: str) -> str:
+    """把模型名映射到 OpenRouter 命名空间（用于无 OPENAI_API_KEY 的回退路径）。"""
+    if "/" in model:
+        return model                      # 已是 OpenRouter 命名空间，原样使用
+    if model.startswith("gpt-"):
+        return "openai/" + model          # gpt-* -> openai/gpt-*
+    if model.startswith("claude-"):
+        return "anthropic/claude-opus-4.8"
+    return "openai/gpt-5.6-luna"          # 兜底：当前便宜旗舰
+
+
+class Config:
+    """运行时配置。"""
+
+    # 必填：OpenAI API Key（本实验默认用 OPENAI_API_KEY）
+    API_KEY: str = os.environ.get("OPENAI_API_KEY", "")
+
+    # 可选：兼容 OpenAI 协议的 base_url，默认官方地址
+    BASE_URL: str = os.environ.get("OPENAI_BASE_URL", "https://api.openai.com/v1")
+
+    # 可选：模型名，默认用当前便宜旗舰 gpt-5.6-luna 控制演示成本
+    MODEL: str = os.environ.get("OPENAI_MODEL", "gpt-5.6-luna")
+
+    # 采样温度，稍低一些让行为更稳定可复现
+    TEMPERATURE: float = float(os.environ.get("OPENAI_TEMPERATURE", "0.3"))
+
+    @classmethod
+    def validate(cls) -> None:
+        """校验并按需应用通用回退。
+
+        0) gpt-5.x 系列直连 OpenAI 需组织验证（且这类推理模型只接受默认温度），
+           因此只要设置了 OPENROUTER_API_KEY，即便同时有 OPENAI_API_KEY，也优先走
+           OpenRouter，省去组织验证的麻烦；
+        1) 否则有 OPENAI_API_KEY -> 直连 OpenAI（尊重 OPENAI_BASE_URL）；
+        2) 否则有 OPENROUTER_API_KEY -> 改走 OpenRouter，并映射模型名；
+        3) 都没有则报清晰错误。
+        """
+        or_key = os.environ.get("OPENROUTER_API_KEY", "")
+        # gpt-5.x（含 gpt-5.6-luna 等推理旗舰）优先走 OpenRouter，规避组织验证。
+        needs_openrouter = cls.MODEL.startswith("gpt-5") and "/" not in cls.MODEL
+        if or_key and (needs_openrouter or not cls.API_KEY):
+            cls.API_KEY = or_key
+            cls.BASE_URL = "https://openrouter.ai/api/v1"
+            cls.MODEL = _to_openrouter_model(cls.MODEL)
+            return
+        if cls.API_KEY:
+            return
+        raise SystemExit(
+            "错误：未检测到 OPENAI_API_KEY 或 OPENROUTER_API_KEY 环境变量。\n"
+            "请先 `export OPENAI_API_KEY=...`（或 OPENROUTER_API_KEY），"
+            "或复制 env.example 为 .env 并填写。"
+        )