From 63c3dceec6c76956bea3bf5216235ecea06c03a5 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 17 Dec 2024 14:07:50 +0100 Subject: [PATCH] fix: Resolve issue with cognify router graph model default value Resolve issue with default value for graph model in cognify endpoint Fix --- cognee/api/v1/cognify/routers/get_cognify_router.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/api/v1/cognify/routers/get_cognify_router.py b/cognee/api/v1/cognify/routers/get_cognify_router.py index a56dbd7a5..257ac994f 100644 --- a/cognee/api/v1/cognify/routers/get_cognify_router.py +++ b/cognee/api/v1/cognify/routers/get_cognify_router.py @@ -5,11 +5,11 @@ 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 - +from cognee.shared.data_models import KnowledgeGraph class CognifyPayloadDTO(BaseModel): datasets: List[str] - graph_model: Optional[BaseModel] = None + graph_model: Optional[BaseModel] = KnowledgeGraph def get_cognify_router() -> APIRouter: router = APIRouter()