diff --git a/cognee/api/client.py b/cognee/api/client.py index 7aeddd795..29efad813 100644 --- a/cognee/api/client.py +++ b/cognee/api/client.py @@ -64,8 +64,9 @@ app.add_middleware( from cognee.api.v1.users.routers import get_auth_router, get_register_router,\ get_reset_password_router, get_verify_router, get_users_router -from cognee.api.v1.permissions.get_permissions_router import get_permissions_router -from cognee.api.v1.settings.routers.get_settings_router import get_settings_router +from cognee.api.v1.permissions.routers import get_permissions_router +from cognee.api.v1.settings.routers import get_settings_router +from cognee.api.v1.cognify.routers import get_cognify_router from cognee.api.v1.search.routers import get_search_router from fastapi import Request @@ -313,21 +314,11 @@ async def add( content = {"error": str(error)} ) - -class CognifyPayloadDTO(BaseModel): - datasets: List[str] - -@app.post("/api/v1/cognify", response_model = None) -async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user)): - """ This endpoint is responsible for the cognitive processing of the content.""" - from cognee.api.v1.cognify.cognify_v2 import cognify as cognee_cognify - try: - await cognee_cognify(payload.datasets, user) - except Exception as error: - return JSONResponse( - status_code = 409, - content = {"error": str(error)} - ) +app.include_router( + get_cognify_router(), + prefix="/api/v1/cognify", + tags=["cognify"] +) app.include_router( get_search_router(), diff --git a/cognee/api/v1/cognify/routers/__init__.py b/cognee/api/v1/cognify/routers/__init__.py new file mode 100644 index 000000000..c6d52bfa2 --- /dev/null +++ b/cognee/api/v1/cognify/routers/__init__.py @@ -0,0 +1 @@ +from .get_cognify_router import get_cognify_router \ No newline at end of file diff --git a/cognee/api/v1/cognify/routers/get_cognify_router.py b/cognee/api/v1/cognify/routers/get_cognify_router.py new file mode 100644 index 000000000..987c15d8a --- /dev/null +++ b/cognee/api/v1/cognify/routers/get_cognify_router.py @@ -0,0 +1,27 @@ +from fastapi import APIRouter +from typing import List +from pydantic import BaseModel +from cognee.modules.users.models import User +from fastapi.responses import JSONResponse +from cognee.modules.users.methods import get_authenticated_user +from fastapi import Depends + +def get_cognify_router(): + router = APIRouter() + + class CognifyPayloadDTO(BaseModel): + datasets: List[str] + + @router.post("/", response_model=None) + async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user)): + """ This endpoint is responsible for the cognitive processing of the content.""" + from cognee.api.v1.cognify.cognify_v2 import cognify as cognee_cognify + try: + await cognee_cognify(payload.datasets, user) + except Exception as error: + return JSONResponse( + status_code=409, + content={"error": str(error)} + ) + + return router \ No newline at end of file diff --git a/cognee/api/v1/permissions/routers/__init__.py b/cognee/api/v1/permissions/routers/__init__.py new file mode 100644 index 000000000..986b52c3e --- /dev/null +++ b/cognee/api/v1/permissions/routers/__init__.py @@ -0,0 +1 @@ +from .get_permissions_router import get_permissions_router \ No newline at end of file diff --git a/cognee/api/v1/permissions/get_permissions_router.py b/cognee/api/v1/permissions/routers/get_permissions_router.py similarity index 100% rename from cognee/api/v1/permissions/get_permissions_router.py rename to cognee/api/v1/permissions/routers/get_permissions_router.py