mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
fix tests, add SearchResult response type for search
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user