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 -4
View File
@@ -404,10 +404,10 @@ export default class MemoryClient {
...(filters && { filters }),
};
let url = `${this.host}/v3/memories/`;
if (page && pageSize) {
url += `?page=${page}&page_size=${pageSize}`;
}
const queryParams: string[] = [];
if (page !== undefined) queryParams.push(`page=${page}`);
if (pageSize !== undefined) queryParams.push(`page_size=${pageSize}`);
const url = `${this.host}/v3/memories/${queryParams.length ? `?${queryParams.join("&")}` : ""}`;
const response = await this._fetchWithErrorHandling(url, {
method: "POST",
@@ -373,3 +373,56 @@ describe("MemoryClient - getAll() entity param rejection", () => {
expect(getFetchBody(call!).show_expired).toBe(true);
});
});
describe("MemoryClient - getAll() page/pageSize query params", () => {
test("sends no query params when neither page nor pageSize is provided", async () => {
const extra = new Map<string, { status: number; body: unknown }>();
extra.set("/v3/memories/", { status: 200, body: { results: [] } });
const mock = setupMockFetch(extra);
const client = new MemoryClient({ apiKey: TEST_API_KEY });
await client.getAll({ filters: { user_id: "u1" } });
const call = findFetchCall(mock, "/v3/memories/", "POST");
expect(call![0]).not.toContain("?");
});
test("sends page alone as a query param", async () => {
const extra = new Map<string, { status: number; body: unknown }>();
extra.set("/v3/memories/", { status: 200, body: { results: [] } });
const mock = setupMockFetch(extra);
const client = new MemoryClient({ apiKey: TEST_API_KEY });
await client.getAll({ filters: { user_id: "u1" }, page: 2 });
const call = findFetchCall(mock, "/v3/memories/", "POST");
expect(call![0]).toContain("?page=2");
expect(call![0]).not.toContain("page_size");
});
test("sends pageSize alone as a query param", async () => {
const extra = new Map<string, { status: number; body: unknown }>();
extra.set("/v3/memories/", { status: 200, body: { results: [] } });
const mock = setupMockFetch(extra);
const client = new MemoryClient({ apiKey: TEST_API_KEY });
await client.getAll({ filters: { user_id: "u1" }, pageSize: 50 });
const call = findFetchCall(mock, "/v3/memories/", "POST");
expect(call![0]).toContain("?page_size=50");
expect(call![0]).not.toContain("?page=50");
});
test("sends both page and pageSize when provided, including falsy 0 values", async () => {
const extra = new Map<string, { status: number; body: unknown }>();
extra.set("/v3/memories/", { status: 200, body: { results: [] } });
const mock = setupMockFetch(extra);
const client = new MemoryClient({ apiKey: TEST_API_KEY });
await client.getAll({ filters: { user_id: "u1" }, page: 0, pageSize: 0 });
const call = findFetchCall(mock, "/v3/memories/", "POST");
expect(call![0]).toContain("page=0");
expect(call![0]).toContain("page_size=0");
});
});
+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)
+145
View File
@@ -8,6 +8,7 @@ import httpx
import pytest
import requests
from mem0.client.main import AsyncMemoryClient
from mem0.client.types import GetAllMemoryOptions, SearchMemoryOptions
@@ -181,6 +182,150 @@ class TestGetAllTypedOptionsParity:
)
class TestGetAllPageSizeQueryParams:
"""MEM-6227: page and page_size must reach the API as query params independently."""
def test_get_all_page_size_alone_lands_in_query_params(self, mock_memory_client):
"""get_all(page_size=...) without page should send page_size as a query param."""
mock_response = MagicMock()
mock_response.json.return_value = {"results": []}
mock_response.raise_for_status.return_value = None
mock_memory_client.client.post.return_value = mock_response
mock_memory_client.get_all(filters={"user_id": "u1"}, page_size=50)
mock_memory_client.client.post.assert_called_once_with(
"/v3/memories/",
json={"filters": {"user_id": "u1"}},
params={"page_size": 50},
)
def test_get_all_page_alone_lands_in_query_params(self, mock_memory_client):
"""get_all(page=...) without page_size should send page as a query param."""
mock_response = MagicMock()
mock_response.json.return_value = {"results": []}
mock_response.raise_for_status.return_value = None
mock_memory_client.client.post.return_value = mock_response
mock_memory_client.get_all(filters={"user_id": "u1"}, page=2)
mock_memory_client.client.post.assert_called_once_with(
"/v3/memories/",
json={"filters": {"user_id": "u1"}},
params={"page": 2},
)
def test_get_all_page_and_page_size_together(self, mock_memory_client):
"""get_all(page=..., page_size=...) should send both as query params."""
mock_response = MagicMock()
mock_response.json.return_value = {"results": []}
mock_response.raise_for_status.return_value = None
mock_memory_client.client.post.return_value = mock_response
mock_memory_client.get_all(filters={"user_id": "u1"}, page=2, page_size=50)
mock_memory_client.client.post.assert_called_once_with(
"/v3/memories/",
json={"filters": {"user_id": "u1"}},
params={"page": 2, "page_size": 50},
)
def test_get_all_neither_page_nor_page_size_sends_no_query_params(self, mock_memory_client):
"""get_all() without page or page_size should not pass a params kwarg."""
mock_response = MagicMock()
mock_response.json.return_value = {"results": []}
mock_response.raise_for_status.return_value = None
mock_memory_client.client.post.return_value = mock_response
mock_memory_client.get_all(filters={"user_id": "u1"})
mock_memory_client.client.post.assert_called_once_with(
"/v3/memories/",
json={"filters": {"user_id": "u1"}},
)
def test_async_get_all_page_size_alone_lands_in_query_params(self):
asyncio.run(self._assert_async_get_all_page_size_alone())
async def _assert_async_get_all_page_size_alone(self):
client = AsyncMemoryClient.__new__(AsyncMemoryClient)
client.async_client = MagicMock()
mock_response = MagicMock()
mock_response.json.return_value = {"results": []}
mock_response.raise_for_status.return_value = None
client.async_client.post = AsyncMock(return_value=mock_response)
with patch("mem0.client.main.capture_client_event"):
await client.get_all(filters={"user_id": "u1"}, page_size=50)
client.async_client.post.assert_called_once_with(
"/v3/memories/",
json={"filters": {"user_id": "u1"}},
params={"page_size": 50},
)
def test_async_get_all_page_alone_lands_in_query_params(self):
asyncio.run(self._assert_async_get_all_page_alone())
async def _assert_async_get_all_page_alone(self):
"""get_all(page=0) without page_size should still send page=0 as a query param."""
client = AsyncMemoryClient.__new__(AsyncMemoryClient)
client.async_client = MagicMock()
mock_response = MagicMock()
mock_response.json.return_value = {"results": []}
mock_response.raise_for_status.return_value = None
client.async_client.post = AsyncMock(return_value=mock_response)
with patch("mem0.client.main.capture_client_event"):
await client.get_all(filters={"user_id": "u1"}, page=0)
client.async_client.post.assert_called_once_with(
"/v3/memories/",
json={"filters": {"user_id": "u1"}},
params={"page": 0},
)
def test_async_get_all_page_and_page_size_together(self):
asyncio.run(self._assert_async_get_all_page_and_page_size_together())
async def _assert_async_get_all_page_and_page_size_together(self):
"""get_all(page=0, page_size=0) should send both as query params despite being falsy."""
client = AsyncMemoryClient.__new__(AsyncMemoryClient)
client.async_client = MagicMock()
mock_response = MagicMock()
mock_response.json.return_value = {"results": []}
mock_response.raise_for_status.return_value = None
client.async_client.post = AsyncMock(return_value=mock_response)
with patch("mem0.client.main.capture_client_event"):
await client.get_all(filters={"user_id": "u1"}, page=0, page_size=0)
client.async_client.post.assert_called_once_with(
"/v3/memories/",
json={"filters": {"user_id": "u1"}},
params={"page": 0, "page_size": 0},
)
def test_async_get_all_neither_page_nor_page_size_sends_no_query_params(self):
asyncio.run(self._assert_async_get_all_neither_page_nor_page_size())
async def _assert_async_get_all_neither_page_nor_page_size(self):
client = AsyncMemoryClient.__new__(AsyncMemoryClient)
client.async_client = MagicMock()
mock_response = MagicMock()
mock_response.json.return_value = {"results": []}
mock_response.raise_for_status.return_value = None
client.async_client.post = AsyncMock(return_value=mock_response)
with patch("mem0.client.main.capture_client_event"):
await client.get_all(filters={"user_id": "u1"})
client.async_client.post.assert_called_once_with(
"/v3/memories/",
json={"filters": {"user_id": "u1"}},
)
class TestUpdateExpirationDate:
"""Tests for update expiration_date payload handling."""