pass page and page_size in query params (#2052)

mem0ai · Nov 25, 2024 · 86d3e36 · 86d3e36
1 parent 7284317
commit 86d3e36
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mem0/client/main.py b/mem0/client/main.py
@@ -167,7 +167,11 @@ def get_all(self, version: str = "v1", **kwargs) -> List[Dict[str, Any]]:
         if version == "v1":
             response = self.client.get(f"/{version}/memories/", params=params)
         elif version == "v2":
-            response = self.client.post(f"/{version}/memories/", json=params)
+            if "page" in params and "page_size" in params:
+                query_params = {"page": params.pop("page"), "page_size": params.pop("page_size")}
+                response = self.client.post(f"/{version}/memories/", json=params, params=query_params)
+            else:
+                response = self.client.post(f"/{version}/memories/", json=params)
         response.raise_for_status()
         if "metadata" in kwargs:
             del kwargs["metadata"]