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

refactor: Allow none through swagger

Browse source
This commit is contained in:
Igor Ilic 2025-09-04 21:05:24 +02:00
parent 9e20103549
commit 0c7ba7c236
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/api/v1/add/routers/get_add_router.py
View file

@ -21,7 +21,8 @@ def get_add_router() -> APIRouter:
async def add( async def add(
data: List[UploadFile] = File(default=None), data: List[UploadFile] = File(default=None),
datasetName: Optional[str] = Form(default=None), datasetName: Optional[str] = Form(default=None),
datasetId: Union[UUID, None] = Form(default=None, examples=[""]), # Note: Literal is needed for Swagger use
datasetId: Union[UUID, Literal[""], None] = Form(default=None, examples=[""]),
node_set: Optional[List[str]] = Form(default=[""], example=[""]), node_set: Optional[List[str]] = Form(default=[""], example=[""]),
user: User = Depends(get_authenticated_user), user: User = Depends(get_authenticated_user),
): ):

5
cognee/api/v1/memify/routers/get_memify_router.py
View file

@ -4,7 +4,7 @@ from fastapi import APIRouter
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
from fastapi import Depends from fastapi import Depends
from pydantic import Field from pydantic import Field
from typing import List, Optional from typing import List, Optional, Union, Literal
from cognee.api.DTO import InDTO from cognee.api.DTO import InDTO
from cognee.modules.users.models import User from cognee.modules.users.models import User
@ -24,7 +24,8 @@ class MemifyPayloadDTO(InDTO):
enrichment_tasks: Optional[List[str]] = Field(default=None, examples=[[]]) enrichment_tasks: Optional[List[str]] = Field(default=None, examples=[[]])
data: Optional[str] = Field(default="") data: Optional[str] = Field(default="")
dataset_name: Optional[str] = Field(default=None) dataset_name: Optional[str] = Field(default=None)
dataset_id: Optional[UUID] = Field(default=None, examples=[[""]]) # Note: Literal is needed for Swagger use
dataset_id: Union[UUID, Literal[""], None] = Field(default=None, examples=[""])
node_name: Optional[List[str]] = Field(default=None, examples=[[]]) node_name: Optional[List[str]] = Field(default=None, examples=[[]])
run_in_background: Optional[bool] = Field(default=False) run_in_background: Optional[bool] = Field(default=False)