105 lines
3.5 KiB
Python
105 lines
3.5 KiB
Python
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
from dataclasses import dataclass
|
|||
|
|
from pathlib import Path
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from dotenv import load_dotenv
|
|||
|
|
except Exception: # pragma: no cover
|
|||
|
|
load_dotenv = None
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass(slots=True)
|
|||
|
|
class CoreAgentConfig:
|
|||
|
|
model: str
|
|||
|
|
api_key: Optional[str]
|
|||
|
|
base_url: Optional[str]
|
|||
|
|
timeout: float = 120.0
|
|||
|
|
temperature: float = 0.2
|
|||
|
|
max_iterations: int = 12
|
|||
|
|
user_name: str = "User"
|
|||
|
|
agent_name: str = "MeetingAgent"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def load_core_agent_env(env_dir: str | Path | None = None) -> Optional[Path]:
|
|||
|
|
candidates = []
|
|||
|
|
if env_dir:
|
|||
|
|
candidates.append(Path(env_dir) / ".env")
|
|||
|
|
candidates.append(Path.cwd() / ".env")
|
|||
|
|
for path in candidates:
|
|||
|
|
if not path.exists():
|
|||
|
|
continue
|
|||
|
|
if load_dotenv is not None:
|
|||
|
|
load_dotenv(path, override=False)
|
|||
|
|
else:
|
|||
|
|
_load_env_without_dependency(path)
|
|||
|
|
return path
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def apply_compat_env_aliases() -> None:
|
|||
|
|
if not os.getenv("OPENAI_API_KEY") and os.getenv("API_KEY"):
|
|||
|
|
os.environ["OPENAI_API_KEY"] = os.environ["API_KEY"]
|
|||
|
|
if not os.getenv("OPENAI_BASE_URL") and os.getenv("BASE_URL"):
|
|||
|
|
os.environ["OPENAI_BASE_URL"] = os.environ["BASE_URL"]
|
|||
|
|
if not os.getenv("CORE_AGENT_MODEL") and os.getenv("MODEL_NAME"):
|
|||
|
|
os.environ["CORE_AGENT_MODEL"] = os.environ["MODEL_NAME"]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def build_core_agent_config() -> CoreAgentConfig:
|
|||
|
|
return CoreAgentConfig(
|
|||
|
|
model=os.getenv("CORE_AGENT_MODEL") or os.getenv("MODEL_NAME") or os.getenv("MODEL") or "",
|
|||
|
|
api_key=os.getenv("OPENAI_API_KEY") or os.getenv("API_KEY"),
|
|||
|
|
base_url=os.getenv("OPENAI_BASE_URL") or os.getenv("BASE_URL"),
|
|||
|
|
timeout=float(os.getenv("OPENAI_TIMEOUT", "120")),
|
|||
|
|
temperature=float(os.getenv("OPENAI_TEMPERATURE", "0.2")),
|
|||
|
|
max_iterations=int(os.getenv("CORE_AGENT_MAX_ITERATIONS", "12")),
|
|||
|
|
user_name=os.getenv("USER_NAME", "User"),
|
|||
|
|
agent_name=os.getenv("AGENT_NAME", "MeetingAgent"),
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def require_model_config(config: CoreAgentConfig) -> None:
|
|||
|
|
missing = []
|
|||
|
|
if _is_missing_or_placeholder(config.api_key):
|
|||
|
|
missing.append("OPENAI_API_KEY")
|
|||
|
|
if _is_missing_or_placeholder(config.base_url):
|
|||
|
|
missing.append("OPENAI_BASE_URL")
|
|||
|
|
if _is_missing_or_placeholder(config.model):
|
|||
|
|
missing.append("MODEL_NAME or CORE_AGENT_MODEL")
|
|||
|
|
if missing:
|
|||
|
|
joined = ", ".join(missing)
|
|||
|
|
raise RuntimeError(
|
|||
|
|
"缺少大模型配置：" + joined + "。请复制 .env.example 为 .env，并填写 OpenAI-compatible API 配置；"
|
|||
|
|
"如果只想本地演示工具流程，请加 --offline。"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _load_env_without_dependency(path: Path) -> None:
|
|||
|
|
for raw_line in path.read_text(encoding="utf-8").splitlines():
|
|||
|
|
line = raw_line.strip()
|
|||
|
|
if not line or line.startswith("#") or "=" not in line:
|
|||
|
|
continue
|
|||
|
|
key, value = line.split("=", 1)
|
|||
|
|
key = key.strip()
|
|||
|
|
value = value.strip().strip('"').strip("'")
|
|||
|
|
if key and key not in os.environ:
|
|||
|
|
os.environ[key] = value
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _is_missing_or_placeholder(value: Optional[str]) -> bool:
|
|||
|
|
if value is None or not str(value).strip():
|
|||
|
|
return True
|
|||
|
|
normalized = str(value).strip().lower()
|
|||
|
|
placeholders = {
|
|||
|
|
"your-api-key-here",
|
|||
|
|
"your-base-url-here",
|
|||
|
|
"your-model-name-here",
|
|||
|
|
"sk-xxx",
|
|||
|
|
"xxx",
|
|||
|
|
"changeme",
|
|||
|
|
}
|
|||
|
|
return normalized in placeholders or normalized.startswith("your-")
|