Rename search size to limit

openradx · Mar 18, 2024 · 5e346f6 · 5e346f6
1 parent 57b3b8c
commit 5e346f6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/radis/search/site.py b/radis/search/site.py
@@ -43,7 +43,7 @@ class SearchFilters:
 class Search(NamedTuple):
     query: str
     offset: int = 0
-    size: int = 10
+    limit: int = 10
     filters: SearchFilters = SearchFilters()
 
 

diff --git a/radis/vespa/providers.py b/radis/vespa/providers.py
@@ -43,7 +43,7 @@ def search_bm25(search: Search) -> SearchResult:
             "yql": yql,
             "query": search.query,
             "type": "web",
-            "hits": search.size,
+            "hits": search.limit,
             "offset": search.offset,
             "queryProfile": SEARCH_QUERY_PROFILE,
             "ranking": BM25_RANK_PROFILE,
@@ -68,7 +68,7 @@ def search_semantic(search: Search) -> SearchResult:
             "yql": yql,
             "query": search.query,
             "type": "web",
-            "hits": search.size,
+            "hits": search.limit,
             "offset": search.offset,
             "queryProfile": SEARCH_QUERY_PROFILE,
             "ranking": SEMANTIC_RANK_PROFILE,
@@ -95,7 +95,7 @@ def search_hybrid(search: Search) -> SearchResult:
             "yql": yql,
             "query": search.query,
             "type": "web",
-            "hits": search.size,
+            "hits": search.limit,
             "offset": search.offset,
             "queryProfile": SEARCH_QUERY_PROFILE,
             "ranking": FUSION_RANK_PROFILE,
@@ -121,7 +121,7 @@ def retrieve_bm25(search: Search) -> RetrievalResult:
             "yql": yql,
             "query": search.query,
             "type": "web",
-            "hits": search.size,
+            "hits": search.limit,
             "offset": search.offset,
             "queryProfile": RETRIEVAL_QUERY_PROFILE,
             "ranking": "unranked",