Team Ai
Apppublic

Multimedika/Bot_Development

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes
events.py29 linesDownload Raw Back to api
1import os2 3from dotenv import load_dotenv4from fastapi import FastAPI5 6from db.database import engine7from db.models import Base8from llama_index.core import set_global_handler9 10load_dotenv()11 12 13async def startup() -> None:14    Base.metadata.create_all(engine)15    os.environ["LANGFUSE_SECRET_KEY"] = os.getenv("LANGFUSE_SECRET_KEY")16    os.environ["LANGFUSE_PUBLIC_KEY"] = os.getenv("LANGFUSE_PUBLIC_KEY")17    os.environ["LANGFUSE_HOST"] = os.getenv("LANGFUSE_HOST")18    set_global_handler("langfuse")19 20 21async def shutdown() -> None:22    pass23 24 25def register_events(app: FastAPI) -> FastAPI:26    app.add_event_handler("startup", startup)27    app.add_event_handler("shutdown", shutdown)28    return app29