diff --git a/backend/app/__init__.py b/backend/app/__init__.py index ddd4b1f9..5484ac28 100644 --- a/backend/app/__init__.py +++ b/backend/app/__init__.py @@ -1,8 +1,14 @@ from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware +from traceroot.integrations.fastapi import connect_fastapi + + api = FastAPI() + api.add_middleware( CORSMiddleware, allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"] ) + +connect_fastapi(api) diff --git a/backend/app/controller/chat_controller.py b/backend/app/controller/chat_controller.py index f73610bf..0afa743f 100644 --- a/backend/app/controller/chat_controller.py +++ b/backend/app/controller/chat_controller.py @@ -19,10 +19,14 @@ from app.service.task import ( get_task_lock, ) +import traceroot + +logger = traceroot.get_logger() + router = APIRouter(tags=["chat"]) - +@traceroot.trace() @router.post("/chat", name="start chat") def post(data: Chat, request: Request): load_dotenv(dotenv_path=data.env_path)