diff --git a/app/main.py b/app/main.py index 6235df6..674756d 100644 --- a/app/main.py +++ b/app/main.py @@ -1,6 +1,7 @@ from fastapi import FastAPI from app.api.routes import router from app.api.tool_routes import router as tool_router +from app.db.database import Base, engine app = FastAPI(title="AI Orquestrador") @@ -9,9 +10,19 @@ app.include_router(tool_router) @app.on_event("startup") -def seed_on_startup(): +def startup_event(): + """ + Inicializa o banco de dados na primeira execução: + - Cria todas as tabelas + - Faz seed das tools iniciais + """ try: + # Cria as tabelas se não existirem + Base.metadata.create_all(bind=engine) + + # Seed das tools from app.db.tool_seed import seed_tools seed_tools() - except Exception: - pass + except Exception as e: + print(f"⚠️ Erro ao inicializar banco de dados: {e}") + # Não falha a aplicação, apenas registra o erro diff --git a/requirements.txt b/requirements.txt index f9da634..fa6d3a0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -21,3 +21,4 @@ websockets==16.0 google-cloud-aiplatform==1.70.0 google-generativeai==0.8.5 httpx==0.27.2 +psycopg2-binary==2.9.10