a2d16c99a1
<!-- .github/pull_request_template.md --> ## Description <!-- Provide a clear description of the changes in this PR --> ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin. --------- Co-authored-by: Igor Ilic <igorilic03@gmail.com>
106 lines
4 KiB
Python
106 lines
4 KiB
Python
from uuid import UUID
|
|
from typing import Optional
|
|
from datetime import datetime
|
|
from fastapi import Depends, APIRouter
|
|
from fastapi.responses import JSONResponse
|
|
from cognee.modules.search.types import SearchType
|
|
from cognee.api.DTO import InDTO, OutDTO
|
|
from cognee.modules.users.exceptions.exceptions import PermissionDeniedError
|
|
from cognee.modules.users.models import User
|
|
from cognee.modules.search.operations import get_history
|
|
from cognee.modules.users.methods import get_authenticated_user
|
|
|
|
|
|
# Note: Datasets sent by name will only map to datasets owned by the request sender
|
|
# To search for datasets not owned by the request sender dataset UUID is needed
|
|
class SearchPayloadDTO(InDTO):
|
|
search_type: SearchType
|
|
datasets: Optional[list[str]] = None
|
|
dataset_ids: Optional[list[UUID]] = None
|
|
query: str
|
|
top_k: Optional[int] = 10
|
|
|
|
|
|
def get_search_router() -> APIRouter:
|
|
router = APIRouter()
|
|
|
|
class SearchHistoryItem(OutDTO):
|
|
id: UUID
|
|
text: str
|
|
user: str
|
|
created_at: datetime
|
|
|
|
@router.get("", response_model=list[SearchHistoryItem])
|
|
async def get_search_history(user: User = Depends(get_authenticated_user)):
|
|
"""
|
|
Get search history for the authenticated user.
|
|
|
|
This endpoint retrieves the search history for the authenticated user,
|
|
returning a list of previously executed searches with their timestamps.
|
|
|
|
Returns:
|
|
List[SearchHistoryItem]: A list of search history items containing:
|
|
- id: Unique identifier for the search
|
|
- text: The search query text
|
|
- user: User who performed the search
|
|
- created_at: When the search was performed
|
|
|
|
Raises:
|
|
HTTPException: If there's an error retrieving the search history
|
|
"""
|
|
try:
|
|
history = await get_history(user.id, limit=0)
|
|
|
|
return history
|
|
except Exception as error:
|
|
return JSONResponse(status_code=500, content={"error": str(error)})
|
|
|
|
@router.post("", response_model=list)
|
|
async def search(payload: SearchPayloadDTO, user: User = Depends(get_authenticated_user)):
|
|
"""
|
|
Search for nodes in the graph database.
|
|
|
|
This endpoint performs semantic search across the knowledge graph to find
|
|
relevant nodes based on the provided query. It supports different search
|
|
types and can be scoped to specific datasets.
|
|
|
|
Args:
|
|
payload (SearchPayloadDTO): Search parameters containing:
|
|
- search_type: Type of search to perform (SearchType)
|
|
- datasets: Optional list of dataset names to search within
|
|
- dataset_ids: Optional list of dataset UUIDs to search within
|
|
- query: The search query string
|
|
- top_k: Maximum number of results to return (default: 10)
|
|
user: The authenticated user performing the search
|
|
|
|
Returns:
|
|
List: A list of search results containing relevant nodes from the graph
|
|
|
|
Raises:
|
|
HTTPException: If there's an error during the search operation
|
|
PermissionDeniedError: If user doesn't have permission to search datasets
|
|
|
|
Note:
|
|
- Datasets sent by name will only map to datasets owned by the request sender
|
|
- To search datasets not owned by the request sender, dataset UUID is needed
|
|
- If permission is denied, returns empty list instead of error
|
|
"""
|
|
from cognee.api.v1.search import search as cognee_search
|
|
|
|
try:
|
|
results = await cognee_search(
|
|
query_text=payload.query,
|
|
query_type=payload.search_type,
|
|
user=user,
|
|
datasets=payload.datasets,
|
|
dataset_ids=payload.dataset_ids,
|
|
top_k=payload.top_k,
|
|
)
|
|
|
|
return results
|
|
except PermissionDeniedError:
|
|
return []
|
|
except Exception as error:
|
|
return JSONResponse(status_code=409, content={"error": str(error)})
|
|
|
|
return router
|