diff --git a/ai-hub/app/api/routes/api.py b/ai-hub/app/api/routes/api.py index c6628b3..a68b2b7 100644 --- a/ai-hub/app/api/routes/api.py +++ b/ai-hub/app/api/routes/api.py @@ -2,10 +2,10 @@ from app.api.dependencies import ServiceContainer # Import routers -from .general import router as general_router from .sessions import create_sessions_router from .documents import create_documents_router from .tts import create_tts_router +from .general import create_general_router from .stt import create_stt_router from .workspace import create_workspace_router @@ -16,7 +16,7 @@ router = APIRouter() # Include routers for different functionalities - router.include_router(general_router) + router.include_router(create_general_router(services)) router.include_router(create_sessions_router(services)) router.include_router(create_documents_router(services)) router.include_router(create_tts_router(services)) diff --git a/ai-hub/app/api/routes/general.py b/ai-hub/app/api/routes/general.py index c744fa3..dd6902d 100644 --- a/ai-hub/app/api/routes/general.py +++ b/ai-hub/app/api/routes/general.py @@ -1,7 +1,11 @@ from fastapi import APIRouter +from app.api.dependencies import ServiceContainer -router = APIRouter(tags=["General"]) +def create_general_router(services: ServiceContainer) -> APIRouter: + router = APIRouter(tags=["General"]) -@router.get("/", summary="Check Service Status") -def read_root(): - return {"status": "AI Model Hub is running!"} \ No newline at end of file + @router.get("/", summary="Check Service Status") + def read_root(): + return {"status": "AI Model Hub is running!"} + + return router \ No newline at end of file