fix(client): honor page_size in get_all when page is not passed (#6900)
This commit is contained in:
+4
-10
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user