diff --git a/app/main.py b/app/main.py index 7cc170b..6235df6 100644 --- a/app/main.py +++ b/app/main.py @@ -5,4 +5,13 @@ from app.api.tool_routes import router as tool_router app = FastAPI(title="AI Orquestrador") app.include_router(router) -app.include_router(tool_router) \ No newline at end of file +app.include_router(tool_router) + + +@app.on_event("startup") +def seed_on_startup(): + try: + from app.db.tool_seed import seed_tools + seed_tools() + except Exception: + pass diff --git a/app/services/orquestrador_service.py b/app/services/orquestrador_service.py index 1e66db2..a315080 100644 --- a/app/services/orquestrador_service.py +++ b/app/services/orquestrador_service.py @@ -1,12 +1,14 @@ +from sqlalchemy.orm import Session + from app.services.llm_service import LLMService from app.services.tool_registry import ToolRegistry class OrquestradorService: - def __init__(self): - self.llm = LLMService() # Instancia o serviço que conversa com o Gemini - self.registry = ToolRegistry() # Instancia o registro de tools disponíveis no sistema + def __init__(self, db: Session): + self.llm = LLMService() + self.registry = ToolRegistry(db) """ diff --git a/requirements.txt b/requirements.txt index 76ab3bd..f9da634 100644 Binary files a/requirements.txt and b/requirements.txt differ