import os if __name__ == "__main__": # import needed here when running main.py to debug backend # you will need to run pip install python-dotenv from dotenv import load_dotenv # type: ignore load_dotenv() import sentry_sdk from fastapi import FastAPI, HTTPException from fastapi.responses import JSONResponse from logger import get_logger from middlewares.cors import add_cors_middleware from modules.api_key.controller import api_key_router from modules.chat.controller import chat_router from modules.contact_support.controller import contact_router from modules.knowledge.controller import knowledge_router from modules.misc.controller import misc_router from modules.notification.controller import notification_router from modules.onboarding.controller import onboarding_router from modules.prompt.controller import prompt_router from modules.upload.controller import upload_router from modules.user.controller import user_router from packages.utils import handle_request_validation_error from routes.brain_routes import brain_router from routes.crawl_routes import crawl_router from routes.subscription_routes import subscription_router from sentry_sdk.integrations.fastapi import FastApiIntegration from sentry_sdk.integrations.starlette import StarletteIntegration logger = get_logger(__name__) if os.getenv("DEV_MODE") == "true": import debugpy logger.debug("👨‍💻 Running in dev mode") debugpy.listen(("0.0.0.0", 5678)) sentry_dsn = os.getenv("SENTRY_DSN") if sentry_dsn: sentry_sdk.init( dsn=sentry_dsn, sample_rate=0.1, enable_tracing=True, integrations=[ StarletteIntegration(transaction_style="endpoint"), FastApiIntegration(transaction_style="endpoint"), ], ) # if CREATE_FIRST_USER := os.getenv("CREATE_FIRST_USER", "False").lower() == "true": # try: # from supabase import create_client # supabase_client_auth = create_client( # os.getenv("SUPABASE_URL"), os.getenv("SUPABASE_SERVICE_KEY") # ) # res = supabase_client_auth.from_('users').select('*').eq('email', "admin@quivr.app").execute() # if len(res.data) == 0: # supabase_client_auth.auth.admin.create_user({"email": "admin@quivr.app","email_confirm": True, "password": "admin"}) # logger.info("👨‍💻 Created first user") # else: # logger.info("👨‍💻 First user already exists") # except Exception as e: # logger.error("👨‍💻 Error while creating first user") # logger.error(e) # telemetry_disabled = os.getenv("TELEMETRY_DISABLED", "False").lower() == "true" # if not telemetry_disabled: # try: # logger.info("👨‍💻 You can disable TELEMETRY by addind TELEMETRY_DISABLED=True to your env variables") # logger.info("Telemetry is used to measure the usage of the app. No personal data is collected.") # import os # from supabase import create_client # import uuid # supabase_url = os.environ.get("SUPABASE_URL", "NOT_SET") # supabase_client_telemetry = create_client("https://phcwncasycjransxnmbf.supabase.co","eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InBoY3duY2FzeWNqcmFuc3hubWJmIiwicm9sZSI6ImFub24iLCJpYXQiOjE3MDE0NDM5NDEsImV4cCI6MjAxNzAxOTk0MX0.0MDz2ETHdQve9yVy_YI79iGsrlpLXX1ObrjmnzyVKSo") # ## insert in the usage table id as uuid of supabase_url # uuid_from_string = uuid.uuid5(uuid.NAMESPACE_DNS, supabase_url) # supabase_client_telemetry.table("usage").insert({"id": str(uuid_from_string)}).execute() # except Exception as e: # logger.error("Error while sending telemetry") app = FastAPI() add_cors_middleware(app) # @app.on_event("startup") # async def startup_event(): # if not os.path.exists(pypandoc.get_pandoc_path()): # pypandoc.download_pandoc() app.include_router(brain_router) app.include_router(chat_router) app.include_router(crawl_router) app.include_router(onboarding_router) app.include_router(misc_router) app.include_router(upload_router) app.include_router(user_router) app.include_router(api_key_router) app.include_router(subscription_router) app.include_router(prompt_router) app.include_router(notification_router) app.include_router(knowledge_router) app.include_router(contact_router) @app.exception_handler(HTTPException) async def http_exception_handler(_, exc): return JSONResponse( status_code=exc.status_code, content={"detail": exc.detail}, ) handle_request_validation_error(app) if __name__ == "__main__": # run main.py to debug backend import uvicorn uvicorn.run(app, host="0.0.0.0", port=5050)