klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pass page and page_size in query params (#2052)

Browse Source
This commit is contained in:
Dev Khant
2024-11-26 00:00:49 +05:30
committed by GitHub
parent 7284317cef
commit 86d3e36ace
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mem0/client/main.py
View File

@@ -167,7 +167,11 @@ class MemoryClient:
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"]