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

refactor: Allow using dataset name for search endpoint

Browse source
This commit is contained in:
Igor Ilic 2025-06-02 15:07:18 +02:00
parent f0022e2f07
commit e14609bc4f
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/api/v1/search/routers/get_search_router.py
View file

@ -1,5 +1,5 @@
from uuid import UUID from uuid import UUID
from typing import Optional from typing import Optional, Union
from datetime import datetime from datetime import datetime
from fastapi import Depends, APIRouter from fastapi import Depends, APIRouter
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
@ -12,7 +12,7 @@ from cognee.modules.users.methods import get_authenticated_user
class SearchPayloadDTO(InDTO): class SearchPayloadDTO(InDTO):
search_type: SearchType search_type: SearchType
dataset_ids: Optional[list[UUID]] = None datasets: Optional[Union[list[UUID], list[str]]] = None
query: str query: str
@ -44,7 +44,7 @@ def get_search_router() -> APIRouter:
query_text=payload.query, query_text=payload.query,
query_type=payload.search_type, query_type=payload.search_type,
user=user, user=user,
datasets=payload.dataset_ids, datasets=payload.datasets,
) )
return results return results