mirror of
https://github.com/furyhawk/agent_alpha.git
synced 2026-07-21 10:35:34 +00:00
102 lines
3.4 KiB
Python
102 lines
3.4 KiB
Python
"""FastAPI application factory — encapsulated in an AppBuilder class."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from uvicorn.logging import DefaultFormatter
|
|
|
|
from backend.core.config import Settings
|
|
from backend.core.database import close_engine, close_valkey
|
|
from backend.core.dependencies import get_agent_service
|
|
from backend.routes.chat import router as chat_router
|
|
from backend.routes.health import router as health_router
|
|
|
|
logger = logging.getLogger("agent_alpha")
|
|
|
|
|
|
def _configure_logging() -> None:
|
|
"""Configure standard logging to match FastAPI/uvicorn's log format."""
|
|
handler = logging.StreamHandler()
|
|
handler.setFormatter(DefaultFormatter("%(levelprefix)s %(message)s"))
|
|
logging.getLogger("agent_alpha").addHandler(handler)
|
|
logging.getLogger("agent_alpha").setLevel(logging.INFO)
|
|
logging.getLogger("agent_alpha").propagate = False
|
|
|
|
|
|
_configure_logging()
|
|
|
|
|
|
class AppBuilder:
|
|
"""Builds and configures the FastAPI application with DI wiring.
|
|
|
|
Accepts a ``Settings`` instance so tests can supply custom config.
|
|
The ``AgentService`` is lazily initialised inside the lifespan.
|
|
"""
|
|
|
|
def __init__(self, settings: Settings | None = None) -> None:
|
|
from backend.core.config import settings as _default_settings
|
|
|
|
self._settings = settings or _default_settings
|
|
|
|
# ── Factory ────────────────────────────────────────────────────────────
|
|
|
|
def build(self) -> FastAPI:
|
|
"""Construct the fully-configured FastAPI application."""
|
|
app = FastAPI(
|
|
title="Agent Alpha",
|
|
description="Agentic AI backend powered by pydantic-ai",
|
|
version="0.1.0",
|
|
lifespan=self._lifespan,
|
|
)
|
|
|
|
# Allow the frontend dev-server to call the API.
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["http://localhost:5173", "http://127.0.0.1:5173"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
app.include_router(health_router)
|
|
app.include_router(chat_router)
|
|
|
|
return app
|
|
|
|
# ── Lifecycle ──────────────────────────────────────────────────────────
|
|
|
|
@asynccontextmanager
|
|
async def _lifespan(self, _app: FastAPI):
|
|
"""Startup / shutdown lifecycle."""
|
|
logger.info("Agent Alpha backend starting")
|
|
|
|
# Initialise the agent service on startup.
|
|
agent_service = await get_agent_service()
|
|
agent_service.initialize()
|
|
|
|
yield
|
|
|
|
# Graceful teardown on shutdown.
|
|
await agent_service.shutdown()
|
|
await close_valkey()
|
|
await close_engine()
|
|
logger.info("Agent Alpha backend shutting down")
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Module-level factory function (backward-compatible).
|
|
# ---------------------------------------------------------------------------
|
|
|
|
_app_builder: AppBuilder | None = None
|
|
|
|
|
|
def create_app() -> FastAPI:
|
|
"""Return a new FastAPI application instance."""
|
|
global _app_builder
|
|
_app_builder = AppBuilder()
|
|
return _app_builder.build()
|