From e14609bc4feeb1785c13226af084f410d466cec0 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Mon, 2 Jun 2025 15:07:18 +0200 Subject: [PATCH] refactor: Allow using dataset name for search endpoint --- cognee/api/v1/search/routers/get_search_router.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cognee/api/v1/search/routers/get_search_router.py b/cognee/api/v1/search/routers/get_search_router.py index 3cc935eb6..f626938c5 100644 --- a/cognee/api/v1/search/routers/get_search_router.py +++ b/cognee/api/v1/search/routers/get_search_router.py @@ -1,5 +1,5 @@ from uuid import UUID -from typing import Optional +from typing import Optional, Union from datetime import datetime from fastapi import Depends, APIRouter from fastapi.responses import JSONResponse @@ -12,7 +12,7 @@ from cognee.modules.users.methods import get_authenticated_user class SearchPayloadDTO(InDTO): search_type: SearchType - dataset_ids: Optional[list[UUID]] = None + datasets: Optional[Union[list[UUID], list[str]]] = None query: str @@ -44,7 +44,7 @@ def get_search_router() -> APIRouter: query_text=payload.query, query_type=payload.search_type, user=user, - datasets=payload.dataset_ids, + datasets=payload.datasets, ) return results