api/events.py

from fastapi import FastAPI
from api.router.topic import db_conn
from llama_index.core import set_global_handler
import os
from dotenv import load_dotenv


load_dotenv()


async def startup() -> None:
    await db_conn.connect()
    os.environ["LANGFUSE_SECRET_KEY"] = os.getenv("LANGFUSE_SECRET_KEY")
    os.environ["LANGFUSE_PUBLIC_KEY"] = os.getenv("LANGFUSE_PUBLIC_KEY")
    os.environ["LANGFUSE_HOST"] = os.getenv("LANGFUSE_HOST")
    set_global_handler("langfuse")


async def shutdown() -> None:
    await db_conn.disconnect()


def register_events(app: FastAPI) -> FastAPI:
    app.add_event_handler("startup", startup)
    app.add_event_handler("shutdown", shutdown)
    return app