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

chapter3/structured-knowledge-extraction/config.py

@@ -0,0 +1,93 @@
+"""
+全局配置：加载环境变量、提供 OpenAI 客户端与默认模型名。
+
+只依赖官方 OpenAI SDK，读取 OPENAI_API_KEY。
+默认模型 gpt-5.6-luna（便宜、够用于因子发现、结构化抽取与文案生成）。
+"""
+import os
+
+from openai import OpenAI
+
+try:
+    # 可选：如果安装了 python-dotenv，则自动加载同目录 .env
+    from dotenv import load_dotenv
+
+    load_dotenv()
+except Exception:  # pragma: no cover - dotenv 是可选依赖
+    pass
+
+def _openrouter_model_id(model) -> str:
+    """将供应商原生模型名映射为 OpenRouter 模型 id（通用 OpenRouter 回退用）。
+    显式的 OPENROUTER_MODEL 环境变量优先。"""
+    override = os.getenv("OPENROUTER_MODEL")
+    if override:
+        return override
+    m = (model or "").strip()
+    if not m:
+        return "openai/gpt-5.6-luna"
+    if "/" in m:
+        return m
+    ml = m.lower()
+    if ml.startswith(("gpt-", "o1", "o3", "o4", "chatgpt")):
+        return "openai/" + m
+    if ml.startswith("claude-"):
+        return "anthropic/claude-opus-4.8"
+    if ml.startswith("kimi"):
+        # kimi-k3 is not on OpenRouter; moonshotai/kimi-k2.6 is the closest hosted id.
+        return "moonshotai/kimi-k2.6"
+    return "openai/gpt-5.6-luna"
+
+
+# Provider: OpenAI by default; DashScope/Qwen/Bailian are OpenAI-compatible.
+PROVIDER = os.getenv("LLM_PROVIDER", "openai").lower()
+PROVIDER = {"qwen": "dashscope", "bailian": "dashscope"}.get(PROVIDER, PROVIDER)
+# 默认模型，可用环境变量覆盖
+MODEL = os.getenv(
+    "DASHSCOPE_MODEL" if PROVIDER == "dashscope" else "OPENAI_MODEL",
+    "qwen3.7-plus" if PROVIDER == "dashscope" else "gpt-5.6-luna",
+)
+
+# 通用 OpenRouter 回退：若没有 OPENAI_API_KEY 但设置了 OPENROUTER_API_KEY，
+# 则把聊天模型路由到 OpenRouter，并把模型名映射为 OpenRouter 的 id。
+_OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+_DASHSCOPE_API_KEY = os.getenv("DASHSCOPE_API_KEY")
+_OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
+# gpt-5.x（含 gpt-5.6*）在 OpenAI 直连 API 上需要组织实名认证；只要设置了
+# OPENROUTER_API_KEY，就优先把这类 id 走 OpenRouter。
+_PREFER_OPENROUTER = bool(_OPENROUTER_API_KEY) and MODEL.lower().startswith("gpt-5")
+_PRIMARY_API_KEY = _DASHSCOPE_API_KEY if PROVIDER == "dashscope" else _OPENAI_API_KEY
+_USE_OPENROUTER = _PREFER_OPENROUTER or ((not _PRIMARY_API_KEY) and bool(_OPENROUTER_API_KEY))
+if _USE_OPENROUTER:
+    MODEL = _openrouter_model_id(MODEL)
+
+
+def get_client() -> OpenAI:
+    """返回一个配置好的 OpenAI 客户端。
+
+    优先使用官方端点（读取 OPENAI_API_KEY）；若缺失则在存在 OPENROUTER_API_KEY 时
+    回退到 OpenRouter（OpenAI 兼容端点）。"""
+    # timeout + 自动重试：发现/抽取阶段要连续发几十次请求，单次瞬时错误
+    # （网络抖动 / 限流 / 5xx）不应中断整条流水线。
+    if _PRIMARY_API_KEY and not _PREFER_OPENROUTER:
+        if PROVIDER == "dashscope":
+            return OpenAI(
+                api_key=_DASHSCOPE_API_KEY,
+                base_url=os.getenv(
+                    "DASHSCOPE_BASE_URL",
+                    "https://dashscope.aliyuncs.com/compatible-mode/v1",
+                ),
+                timeout=60.0,
+                max_retries=5,
+            )
+        return OpenAI(api_key=_OPENAI_API_KEY, timeout=60.0, max_retries=5)
+    if _OPENROUTER_API_KEY:
+        return OpenAI(
+            api_key=_OPENROUTER_API_KEY,
+            base_url="https://openrouter.ai/api/v1",
+            timeout=60.0,
+            max_retries=5,
+        )
+    raise RuntimeError(
+        "未找到所选 provider 的 API Key，请设置 OPENAI_API_KEY、DASHSCOPE_API_KEY "
+        "或 OPENROUTER_API_KEY。"
+    )