diff --git a/.github/workflows/test_deduplication.yml b/.github/workflows/test_deduplication.yml
index 77a7ddbb78..a1aab3252d 100644
--- a/.github/workflows/test_deduplication.yml
+++ b/.github/workflows/test_deduplication.yml
@@ -17,6 +17,7 @@ jobs:
   run_deduplication_test:
     name: test
     runs-on: ubuntu-latest
+    if:  ${{ github.event.label.name == 'run-checks' }}
     defaults:
       run:
         shell: bash
diff --git a/.github/workflows/test_milvus.yml b/.github/workflows/test_milvus.yml
index 5cad723786..c4214fddf7 100644
--- a/.github/workflows/test_milvus.yml
+++ b/.github/workflows/test_milvus.yml
@@ -18,6 +18,7 @@ jobs:
   run_milvus:
     name: test
     runs-on: ubuntu-latest
+    if:  ${{ github.event.label.name == 'run-checks' }}
     strategy:
       fail-fast: false
     defaults:
diff --git a/.github/workflows/test_neo4j.yml b/.github/workflows/test_neo4j.yml
index 3f3a35e4f5..5f955eefc6 100644
--- a/.github/workflows/test_neo4j.yml
+++ b/.github/workflows/test_neo4j.yml
@@ -15,6 +15,7 @@ env:
 jobs:
   run_neo4j_integration_test:
     name: test
+    if:  ${{ github.event.label.name == 'run-checks' }}
     runs-on: ubuntu-latest
 
     defaults:
diff --git a/.github/workflows/test_pgvector.yml b/.github/workflows/test_pgvector.yml
index a162d2cb41..9f5ffd1da8 100644
--- a/.github/workflows/test_pgvector.yml
+++ b/.github/workflows/test_pgvector.yml
@@ -18,6 +18,7 @@ jobs:
   run_pgvector_integration_test:
     name: test
     runs-on: ubuntu-latest
+    if:  ${{ github.event.label.name == 'run-checks' }}
     defaults:
       run:
         shell: bash
diff --git a/.github/workflows/test_python_3_10.yml b/.github/workflows/test_python_3_10.yml
index 39eb4e57a9..770d2fd636 100644
--- a/.github/workflows/test_python_3_10.yml
+++ b/.github/workflows/test_python_3_10.yml
@@ -18,6 +18,7 @@ jobs:
   run_common:
     name: test
     runs-on: ubuntu-latest
+    if:  ${{ github.event.label.name == 'run-checks' }}
     strategy:
       fail-fast: false
     defaults:
diff --git a/.github/workflows/test_python_3_11.yml b/.github/workflows/test_python_3_11.yml
index 2dd704eb92..69eb875bde 100644
--- a/.github/workflows/test_python_3_11.yml
+++ b/.github/workflows/test_python_3_11.yml
@@ -18,6 +18,7 @@ jobs:
   run_common:
     name: test
     runs-on: ubuntu-latest
+    if:  ${{ github.event.label.name == 'run-checks' }}
     strategy:
       fail-fast: false
     defaults:
diff --git a/.github/workflows/test_python_3_9.yml b/.github/workflows/test_python_3_9.yml
index 99c2b9a7a8..380c894ca4 100644
--- a/.github/workflows/test_python_3_9.yml
+++ b/.github/workflows/test_python_3_9.yml
@@ -18,6 +18,7 @@ jobs:
   run_common:
     name: test
     runs-on: ubuntu-latest
+    if:  ${{ github.event.label.name == 'run-checks' }}
     strategy:
       fail-fast: false
     defaults:
diff --git a/.github/workflows/test_qdrant.yml b/.github/workflows/test_qdrant.yml
index f0a2e3d3fa..17d9ac628f 100644
--- a/.github/workflows/test_qdrant.yml
+++ b/.github/workflows/test_qdrant.yml
@@ -18,6 +18,7 @@ jobs:
   run_qdrant_integration_test:
     name: test
     runs-on: ubuntu-latest
+    if:  ${{ github.event.label.name == 'run-checks' }}
 
     defaults:
       run:
diff --git a/.github/workflows/test_weaviate.yml b/.github/workflows/test_weaviate.yml
index b8eb72383f..9a3651dda0 100644
--- a/.github/workflows/test_weaviate.yml
+++ b/.github/workflows/test_weaviate.yml
@@ -18,6 +18,7 @@ jobs:
   run_weaviate_integration_test:
     name: test
     runs-on: ubuntu-latest
+    if:  ${{ github.event.label.name == 'run-checks' }}
 
     defaults:
       run:
diff --git a/cognee/api/v1/cognify/routers/get_cognify_router.py b/cognee/api/v1/cognify/routers/get_cognify_router.py
index 9616fa71cf..a56dbd7a5e 100644
--- a/cognee/api/v1/cognify/routers/get_cognify_router.py
+++ b/cognee/api/v1/cognify/routers/get_cognify_router.py
@@ -1,13 +1,15 @@
 from fastapi import APIRouter
-from typing import List
+from typing import List, Optional
 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
 
+
 class CognifyPayloadDTO(BaseModel):
     datasets: List[str]
+    graph_model: Optional[BaseModel] = None
 
 def get_cognify_router() -> APIRouter:
     router = APIRouter()
@@ -17,11 +19,11 @@ async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authentic
         """ 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