diff --git a/headroom/proxy/server.py b/headroom/proxy/server.py index 3b5729337..4b74034da 100644 --- a/headroom/proxy/server.py +++ b/headroom/proxy/server.py @@ -2532,7 +2532,8 @@ if __name__ == "__main__": "--openai-api-url", help=f"Custom OpenAI API URL (default: {HeadroomProxy.OPENAI_API_URL})" ) parser.add_argument( - "--anthropic-api-url", help=f"Custom Anthropic API URL (default: {HeadroomProxy.ANTHROPIC_API_URL})" + "--anthropic-api-url", + help=f"Custom Anthropic API URL (default: {HeadroomProxy.ANTHROPIC_API_URL})", ) # Backend (anthropic direct, bedrock, openrouter, anyllm, or litellm-) diff --git a/headroom/release_version.py b/headroom/release_version.py index 00dccadf7..076fb75cb 100644 --- a/headroom/release_version.py +++ b/headroom/release_version.py @@ -9,8 +9,6 @@ from collections.abc import Sequence from dataclasses import dataclass, replace from pathlib import Path -import tomllib - SEMVER_RE = re.compile(r"^(\d+)\.(\d+)\.(\d+)$") RELEASE_TAG_RE = re.compile(r"^v(\d+)\.(\d+)\.(\d+)(?:\.(\d+))?$") @@ -126,6 +124,11 @@ def compute_release_version( def get_canonical_version(root: Path) -> str: """Read the canonical project version from pyproject.toml.""" + try: + import tomllib + except ModuleNotFoundError: # pragma: no cover - Python 3.10 compatibility + import tomli as tomllib + with open(root / "pyproject.toml", "rb") as file: data = tomllib.load(file) return data["project"]["version"]