Skip to content

Commit 50eba6a

Browse files
committed
change empty next_page_token value to None
1 parent d638093 commit 50eba6a

File tree

4 files changed

+20
-10
lines changed

4 files changed

+20
-10
lines changed

src/a2a/server/tasks/database_task_store.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ async def list(
187187
next_page_token = (
188188
str(page_number + 1)
189189
if total_count > (page_number + 1) * page_size
190-
else ''
190+
else None
191191
)
192192

193193
return TasksPage(

src/a2a/server/tasks/inmemory_task_store.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ async def list(
7272
next_page_token = (
7373
str(page_token + 1)
7474
if (page_token + 1) * page_size < total_size
75-
else ''
75+
else None
7676
)
7777

7878
return TasksPage(

tests/server/tasks/test_database_task_store.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ async def test_get_task(db_store_parameterized: DatabaseTaskStore) -> None:
181181
ListTasksParams(),
182182
['task-4', 'task-3', 'task-2', 'task-1', 'task-0'],
183183
5,
184-
'',
184+
None,
185185
),
186186
# Pagination (first page)
187187
(
@@ -198,20 +198,25 @@ async def test_get_task(db_store_parameterized: DatabaseTaskStore) -> None:
198198
'2',
199199
),
200200
# Filtering by context_id
201-
(ListTasksParams(context_id='context-1'), ['task-3', 'task-1'], 2, ''),
201+
(
202+
ListTasksParams(context_id='context-1'),
203+
['task-3', 'task-1'],
204+
2,
205+
None,
206+
),
202207
# Filtering by status
203208
(
204209
ListTasksParams(status=TaskState.working),
205210
['task-3', 'task-1'],
206211
2,
207-
'',
212+
None,
208213
),
209214
# Combined filtering (context_id and status)
210215
(
211216
ListTasksParams(context_id='context-0', status=TaskState.submitted),
212217
['task-2', 'task-0'],
213218
2,
214-
'',
219+
None,
215220
),
216221
# Combined filtering and pagination
217222
(

tests/server/tasks/test_inmemory_task_store.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ async def test_in_memory_task_store_get_nonexistent() -> None:
4141
ListTasksParams(),
4242
['task-0', 'task-1', 'task-2', 'task-3', 'task-4'],
4343
5,
44-
'',
44+
None,
4545
),
4646
# Pagination (first page)
4747
(
@@ -58,20 +58,25 @@ async def test_in_memory_task_store_get_nonexistent() -> None:
5858
'2',
5959
),
6060
# Filtering by context_id
61-
(ListTasksParams(context_id='context-1'), ['task-1', 'task-3'], 2, ''),
61+
(
62+
ListTasksParams(context_id='context-1'),
63+
['task-1', 'task-3'],
64+
2,
65+
None,
66+
),
6267
# Filtering by status
6368
(
6469
ListTasksParams(status=TaskState.working),
6570
['task-1', 'task-3'],
6671
2,
67-
'',
72+
None,
6873
),
6974
# Combined filtering (context_id and status)
7075
(
7176
ListTasksParams(context_id='context-0', status=TaskState.submitted),
7277
['task-0', 'task-2'],
7378
2,
74-
'',
79+
None,
7580
),
7681
# Combined filtering and pagination
7782
(

0 commit comments

Comments
 (0)