move rerank_count parameter to end

so change is not breaking
marqo-ai · Jan 9, 2025 · 2149916 · 2149916
1 parent 1be6ad0
commit 2149916
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/marqo/index.py b/src/marqo/index.py
@@ -210,7 +210,7 @@ def get_status(self):
             raise UnsupportedOperationError("This operation is only supported for Marqo Cloud")
 
     def search(self, q: Optional[Union[str, dict]] = None, searchable_attributes: Optional[List[str]] = None,
-               limit: int = 10, offset: int = 0, rerank_count: Optional[int] = None,
+               limit: int = 10, offset: int = 0,
                search_method: Union[SearchMethods.TENSOR, str] = SearchMethods.TENSOR,
                highlights=None, device: Optional[str] = None, filter_string: str = None,
                show_highlights=True, reranker=None,
@@ -222,6 +222,7 @@ def search(self, q: Optional[Union[str, dict]] = None, searchable_attributes: Op
                model_auth: Optional[dict] = None,
                ef_search: Optional[int] = None, approximate: Optional[bool] = None,
                text_query_prefix: Optional[str] = None, hybrid_parameters: Optional[dict] = None,
+               rerank_count: Optional[int] = None
                ) -> Dict[str, Any]:
         """Search the index.
 
@@ -238,7 +239,6 @@ def search(self, q: Optional[Union[str, dict]] = None, searchable_attributes: Op
             searchable_attributes:  attributes to search
             limit: The max number of documents to be returned
             offset: The number of search results to skip (for pagination)
-            rerank_count: The number of documents to rerank with score modifiers
             search_method: Indicates TENSOR or LEXICAL (keyword) search
             show_highlights: True if highlights are to be returned
             reranker:
@@ -260,6 +260,7 @@ def search(self, q: Optional[Union[str, dict]] = None, searchable_attributes: Op
             text_query_prefix: a string to prefix the text query with before vectorizing.
                 overrides index setting & model default prefix
             hybrid_parameters: a dictionary of parameters to be used for hybrid search
+            rerank_count: The number of documents to rerank with score modifiers
         Returns:
             Dictionary with hits and other metadata
         """