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