Multimedika/Bot_Development
0
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 