fix tests, add SearchResult response type for search

This commit is contained in:
phernandez
2025-01-04 22:22:38 -06:00
parent e35dbab156
commit fa7e986d3b
10 changed files with 235 additions and 472 deletions
@@ -5,18 +5,19 @@ from typing import List
from loguru import logger
from basic_memory.services.search_service import SearchService
from basic_memory.schemas.search import SearchQuery, SearchResult
from basic_memory.schemas.search import SearchQuery, SearchResult, SearchResponse
from basic_memory.deps import get_search_service
router = APIRouter(prefix="/search", tags=["search"])
@router.post("/", response_model=List[SearchResult])
@router.post("/", response_model=SearchResponse)
async def search(
query: SearchQuery,
search_service: SearchService = Depends(get_search_service)
):
"""Search across all knowledge and documents."""
return await search_service.search(query)
results = await search_service.search(query)
return SearchResponse(results=results)
@router.post("/reindex")
async def reindex(