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

fix: Resolve issue with cognify router graph model default value

Browse source 
Resolve issue with default value for graph model in cognify endpoint

Fix
This commit is contained in:
Igor Ilic 2024-12-17 14:07:50 +01:00
parent dfcec5f41d
commit 63c3dceec6
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -5,11 +5,11 @@ from cognee.modules.users.models import User
from fastapi.responses import JSONResponse 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]
graph_model: Optional[BaseModel] = None graph_model: Optional[BaseModel] = KnowledgeGraph
def get_cognify_router() -> APIRouter: def get_cognify_router() -> APIRouter:
router = APIRouter() router = APIRouter()