73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
import logging
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from gateway.v1.auth import router as auth_router
|
|
from gateway.v1.chat import router as v1_router
|
|
from gateway.jellystat.api import router as jellystat_router
|
|
from src.config import DEEPSEEK_API_KEY, get_config
|
|
from src.llm import create_client
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Logging — tool calls will appear in the uvicorn console
|
|
# ---------------------------------------------------------------------------
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s [%(name)s] %(levelname)s: %(message)s",
|
|
datefmt="%H:%M:%S",
|
|
)
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Load all agents, skills, AND auth services so they self-register at startup
|
|
# ---------------------------------------------------------------------------
|
|
from agents import load_all_agents # noqa: E402
|
|
|
|
load_all_agents()
|
|
|
|
import gateway.auth.jellyfin # noqa: E402 — self-registers JellyfinAuth
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Lifespan
|
|
# ---------------------------------------------------------------------------
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
from gateway.discord.bot import start_in_background # noqa: E402
|
|
from gateway.jellystat.db import init_pool, close_pool # noqa: E402
|
|
|
|
await init_pool(app)
|
|
start_in_background()
|
|
|
|
yield
|
|
|
|
await close_pool(app)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# App
|
|
# ---------------------------------------------------------------------------
|
|
app = FastAPI(lifespan=lifespan)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Singletons (stored on app.state so every module can reach them via Depends)
|
|
# ---------------------------------------------------------------------------
|
|
app.state.llm_client = create_client(DEEPSEEK_API_KEY)
|
|
|
|
# Lazy-compiled LangGraph graphs — populated on first use per agent
|
|
app.state.agent_graphs: dict = {}
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Routers
|
|
# ---------------------------------------------------------------------------
|
|
app.include_router(v1_router, prefix="/v1")
|
|
app.include_router(auth_router)
|
|
app.include_router(jellystat_router) |