fix: use get_router() function to get the router

ai4os · Jun 21, 2024 · 89167f4 · 89167f4
1 parent 9aa1ab1
commit 89167f4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/deepaas/api/v2/debug.py b/deepaas/api/v2/debug.py
@@ -103,3 +103,12 @@ async def get():
         return fastapi.responses.PlainTextResponse(resp)
     else:
         return fastapi.responses.Response(status_code=204)
+
+
+def get_router() -> fastapi.APIRouter:
+    """Auxiliary function to get the router.
+
+    We use this function to be able to include the router in the main
+    application and do things before it gets included.
+    """
+    return router