Fixes

Author: Vasilije1990

cognee/api/v1/cognify/routers/get_cognify_router.py

@@ -1,5 +1,5 @@
 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
@@ -9,13 +9,13 @@
 
 class CognifyPayloadDTO(BaseModel):
     datasets: List[str]
-    graph_model: BaseModel
+    graph_model: Optional[BaseModel] = None
 
 def get_cognify_router() -> APIRouter:
     router = APIRouter()
 
     @router.post("/", response_model=None)
-    async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user), graph_model:BaseModel=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: