diff --git a/cognee/api/v1/cognify/routers/get_cognify_router.py b/cognee/api/v1/cognify/routers/get_cognify_router.py
index 9616fa71c..5e321bf33 100644
--- a/cognee/api/v1/cognify/routers/get_cognify_router.py
+++ b/cognee/api/v1/cognify/routers/get_cognify_router.py
@@ -6,6 +6,7 @@ from fastapi.responses import JSONResponse
 from cognee.modules.users.methods import get_authenticated_user
 from fastapi import Depends
 
+
 class CognifyPayloadDTO(BaseModel):
     datasets: List[str]
 
@@ -13,15 +14,15 @@ def get_cognify_router() -> APIRouter:
     router = APIRouter()
 
     @router.post("/", response_model=None)
-    async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user)):
+    async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user), graph_model:BaseModel):
         """ 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)
+            await cognee_cognify(payload.datasets, user, payload.graph_model)
         except Exception as error:
             return JSONResponse(
                 status_code=409,
                 content={"error": str(error)}
             )
 
-    return router
\ No newline at end of file
+    return router