api/events.py

import os

from dotenv import load_dotenv
from fastapi import FastAPI

from db.database import engine
from db.models import Base
from llama_index.core import set_global_handler

load_dotenv()


async def startup() -> None:
    Base.metadata.create_all(engine)
    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:
    pass


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