From a42ab3e215f898c83ded7f546a96d7dce1127713 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Wed, 18 Dec 2024 10:58:48 +0100 Subject: [PATCH] fix: Resolve syntax issue with cognify router Resolve syntax issue with cognify router Fix --- cognee/api/v1/cognify/routers/get_cognify_router.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/cognee/api/v1/cognify/routers/get_cognify_router.py b/cognee/api/v1/cognify/routers/get_cognify_router.py index 1a53a7338..739bd1657 100644 --- a/cognee/api/v1/cognify/routers/get_cognify_router.py +++ b/cognee/api/v1/cognify/routers/get_cognify_router.py @@ -6,6 +6,8 @@ from fastapi.responses import JSONResponse from cognee.modules.users.methods import get_authenticated_user from fastapi import Depends +from cognee.shared.data_models import KnowledgeGraph + class CognifyPayloadDTO(BaseModel): datasets: List[str] @@ -32,10 +34,10 @@ def get_cognify_router() -> APIRouter: GraphModelDynamic = json_to_pydantic_model("GraphModelDynamic", graph_model_schema) graph_model_instance = GraphModelDynamic(**graph_model_schema) print(graph_model_instance) - else: - graph_model_instance = None - try: - await cognee_cognify(payload.datasets, user, payload.graph_model) + else: + graph_model_instance = KnowledgeGraph + + await cognee_cognify(payload.datasets, user, graph_model_instance) except Exception as error: return JSONResponse( status_code=409,