From cfad366c5ca9c1961a2b162a0bdec3a3680ef09c Mon Sep 17 00:00:00 2001 From: vasilije Date: Mon, 16 Dec 2024 11:45:33 +0100 Subject: [PATCH] update --- cognee/api/v1/cognify/routers/get_cognify_router.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cognee/api/v1/cognify/routers/get_cognify_router.py b/cognee/api/v1/cognify/routers/get_cognify_router.py index 5e321bf33..761740bab 100644 --- a/cognee/api/v1/cognify/routers/get_cognify_router.py +++ b/cognee/api/v1/cognify/routers/get_cognify_router.py @@ -9,12 +9,13 @@ from fastapi import Depends class CognifyPayloadDTO(BaseModel): datasets: List[str] + graph_model: BaseModel 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): + async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user), graph_model:BaseModel=None): """ This endpoint is responsible for the cognitive processing of the content.""" from cognee.api.v1.cognify.cognify_v2 import cognify as cognee_cognify try: