gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Resolve syntax issue with cognify router

Browse source 
Resolve syntax issue with cognify router

Fix
This commit is contained in:
Igor Ilic 2024-12-18 10:58:48 +01:00
parent 57ae93b3cd
commit a42ab3e215
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
cognee/api/v1/cognify/routers/get_cognify_router.py
View file

@ -6,6 +6,8 @@ from fastapi.responses import JSONResponse
from cognee.modules.users.methods import get_authenticated_user from cognee.modules.users.methods import get_authenticated_user
from fastapi import Depends from fastapi import Depends
from cognee.shared.data_models import KnowledgeGraph
class CognifyPayloadDTO(BaseModel): class CognifyPayloadDTO(BaseModel):
datasets: List[str] datasets: List[str]
@ -32,10 +34,10 @@ def get_cognify_router() -> APIRouter:
GraphModelDynamic = json_to_pydantic_model("GraphModelDynamic", graph_model_schema) GraphModelDynamic = json_to_pydantic_model("GraphModelDynamic", graph_model_schema)
graph_model_instance = GraphModelDynamic(**graph_model_schema) graph_model_instance = GraphModelDynamic(**graph_model_schema)
print(graph_model_instance) print(graph_model_instance)
else: else:
graph_model_instance = None graph_model_instance = KnowledgeGraph
try:
await cognee_cognify(payload.datasets, user, payload.graph_model) await cognee_cognify(payload.datasets, user, graph_model_instance)
except Exception as error: except Exception as error:
return JSONResponse( return JSONResponse(
status_code=409, status_code=409,