fix(client): honor page_size in get_all when page is not passed (#6900)

This commit is contained in:
Kartik
2026-08-14 16:49:58 +05:30
committed by GitHub
parent a133287015
commit bfb51c6b93
4 changed files with 206 additions and 14 deletions
+4 -10
View File
@@ -275,11 +275,8 @@ class MemoryClient:
kwargs = {**(options.model_dump(exclude_unset=True) if options else {}), **kwargs}
params = self._prepare_params(kwargs)
if "page" in params and "page_size" in params:
query_params = {
"page": params.pop("page"),
"page_size": params.pop("page_size"),
}
query_params = {key: params.pop(key) for key in ("page", "page_size") if key in params}
if query_params:
response = self.client.post("/v3/memories/", json=params, params=query_params)
else:
response = self.client.post("/v3/memories/", json=params)
@@ -1198,11 +1195,8 @@ class AsyncMemoryClient:
kwargs = {**(options.model_dump(exclude_unset=True) if options else {}), **kwargs}
params = self._prepare_params(kwargs)
if "page" in params and "page_size" in params:
query_params = {
"page": params.pop("page"),
"page_size": params.pop("page_size"),
}
query_params = {key: params.pop(key) for key in ("page", "page_size") if key in params}
if query_params:
response = await self.async_client.post("/v3/memories/", json=params, params=query_params)
else:
response = await self.async_client.post("/v3/memories/", json=params)