40 lines
1.7 KiB
Python
40 lines
1.7 KiB
Python
import os
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from deerflow.config.app_config import get_app_config
|
|
|
|
|
|
class GatewayConfig(BaseModel):
|
|
"""Configuration for the API Gateway."""
|
|
|
|
host: str = Field(default="0.0.0.0", description="Host to bind the gateway server")
|
|
port: int = Field(default=8001, description="Port to bind the gateway server")
|
|
cors_origins: list[str] = Field(default_factory=lambda: ["http://localhost:3000"], description="Allowed CORS origins")
|
|
enable_docs: bool = Field(default=True, description="Enable Swagger/ReDoc/OpenAPI endpoints")
|
|
|
|
|
|
_gateway_config: GatewayConfig | None = None
|
|
|
|
|
|
def get_gateway_config() -> GatewayConfig:
|
|
"""Get gateway config, preferring the shared ``config.yaml`` settings."""
|
|
global _gateway_config
|
|
if _gateway_config is None:
|
|
cors_origins_str = os.getenv("CORS_ORIGINS", "http://localhost:3000")
|
|
try:
|
|
# Documentation routes are part of the service surface, so the
|
|
# versioned deployment config is authoritative. In particular,
|
|
# a stale shell-level GATEWAY_ENABLE_DOCS=false must not silently
|
|
# suppress the intranet docs configured in config.yaml.
|
|
enable_docs = get_app_config().gateway.enable_docs
|
|
except Exception: # noqa: BLE001 - retain the legacy env fallback if config cannot load
|
|
enable_docs = os.getenv("GATEWAY_ENABLE_DOCS", "true").lower() == "true"
|
|
_gateway_config = GatewayConfig(
|
|
host=os.getenv("GATEWAY_HOST", "0.0.0.0"),
|
|
port=int(os.getenv("GATEWAY_PORT", "8001")),
|
|
cors_origins=cors_origins_str.split(","),
|
|
enable_docs=enable_docs,
|
|
)
|
|
return _gateway_config
|