From 0c7ba7c23610cf966c5660b9ad8d6f5f054dc573 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Thu, 4 Sep 2025 21:05:24 +0200 Subject: [PATCH] refactor: Allow none through swagger --- cognee/api/v1/add/routers/get_add_router.py | 3 ++- cognee/api/v1/memify/routers/get_memify_router.py | 5 +++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/cognee/api/v1/add/routers/get_add_router.py b/cognee/api/v1/add/routers/get_add_router.py index f27d559e1..dfa7d275b 100644 --- a/cognee/api/v1/add/routers/get_add_router.py +++ b/cognee/api/v1/add/routers/get_add_router.py @@ -21,7 +21,8 @@ def get_add_router() -> APIRouter: async def add( data: List[UploadFile] = File(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=[""]), user: User = Depends(get_authenticated_user), ): diff --git a/cognee/api/v1/memify/routers/get_memify_router.py b/cognee/api/v1/memify/routers/get_memify_router.py index cf1df8f71..1976d7414 100644 --- a/cognee/api/v1/memify/routers/get_memify_router.py +++ b/cognee/api/v1/memify/routers/get_memify_router.py @@ -4,7 +4,7 @@ from fastapi import APIRouter from fastapi.responses import JSONResponse from fastapi import Depends from pydantic import Field -from typing import List, Optional +from typing import List, Optional, Union, Literal from cognee.api.DTO import InDTO from cognee.modules.users.models import User @@ -24,7 +24,8 @@ class MemifyPayloadDTO(InDTO): enrichment_tasks: Optional[List[str]] = Field(default=None, examples=[[]]) data: Optional[str] = Field(default="") 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=[[]]) run_in_background: Optional[bool] = Field(default=False)