diff --git a/src/app/search/api/router.py b/src/app/search/api/router.py index 663f4d8..ac20c5c 100644 --- a/src/app/search/api/router.py +++ b/src/app/search/api/router.py @@ -50,6 +50,7 @@ def get_params( query=body.query or "", nb_results=nb_results, corpora=tuple(body.corpora) if body.corpora else None, + lang=body.lang, subject=subject, influence_factor=influence_factor, relevance_factor=relevance_factor, diff --git a/src/app/search/models/search.py b/src/app/search/models/search.py index 41772fb..e8a24c8 100644 --- a/src/app/search/models/search.py +++ b/src/app/search/models/search.py @@ -31,11 +31,13 @@ class SDGFilter(BaseModel): class SearchQuery(SDGFilter): query: str | list[str] | None corpora: list[str] | None = None + lang: list[str] | None = None class EnhancedSearchQuery(SDGFilter): query: str | list[str] corpora: tuple[str, ...] | None = None + lang: list[str] | None = None nb_results: int = 30 subject: str | None = None influence_factor: float = 2 diff --git a/src/app/search/services/search.py b/src/app/search/services/search.py index 7050519..522a2d0 100644 --- a/src/app/search/services/search.py +++ b/src/app/search/services/search.py @@ -267,6 +267,7 @@ async def search_handler( ) filter_content = [ + FilterDefinition(key="document_lang", value=qp.lang), FilterDefinition(key="document_corpus", value=qp.corpora), FilterDefinition(key="document_details.readability", value=qp.readability), FilterDefinition(