From 8720bbc58edf234c07d1268b51f7d8f8e0ba723a Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Thu, 11 Sep 2025 19:24:59 +0200 Subject: [PATCH] fix: Combined 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 bd4841d3f..459b5f427 100644 --- a/cognee/api/v1/search/routers/get_search_router.py +++ b/cognee/api/v1/search/routers/get_search_router.py @@ -1,12 +1,12 @@ from uuid import UUID -from typing import Optional +from typing import Optional, Union, List from datetime import datetime from pydantic import Field from fastapi import Depends, APIRouter from fastapi.responses import JSONResponse from fastapi.encoders import jsonable_encoder -from cognee.modules.search.types import SearchType +from cognee.modules.search.types import SearchType, SearchResult, CombinedSearchResult from cognee.api.DTO import InDTO, OutDTO from cognee.modules.users.exceptions.exceptions import PermissionDeniedError from cognee.modules.users.models import User @@ -73,7 +73,7 @@ def get_search_router() -> APIRouter: except Exception as error: return JSONResponse(status_code=500, content={"error": str(error)}) - @router.post("", response_model=list) + @router.post("", response_model=Union[List[SearchResult], CombinedSearchResult]) async def search(payload: SearchPayloadDTO, user: User = Depends(get_authenticated_user)): """ Search for nodes in the graph database.