Add offset parameter

Author: oeken

needle/v1/collections/__init__.py

@@ -135,8 +135,8 @@ def search(
         self,
         collection_id: str,
         text: str,
-        max_distance: Optional[float] = None,
         top_k: Optional[int] = None,
+        offset: Optional[int] = None,
     ):
         """
         Searches within a collection based on the provided parameters.
@@ -153,8 +153,8 @@ def search(
         endpoint = f"{self.search_endpoint}/{collection_id}/search"
         req_body = {
             "text": text,
-            "max_distance": max_distance,
             "top_k": top_k,
+            "offset": offset,
         }
         resp = self.session.post(endpoint, headers=self.headers, json=req_body)
         body = resp.json()

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "needle-python"
-version = "0.4.0"
+version = "0.5.0"
 description = "Needle client library for Python"
 authors = [
     "Onur Eken <m.onureken@gmail.com>",