Skip to content

Commit b94e8b0

Browse files
algolia-botFluf22
andcommitted
fix(clients): update browse iterator (generated)
algolia/api-clients-automation#4058 Co-authored-by: algolia-bot <[email protected]> Co-authored-by: Thomas Raffray <[email protected]>
1 parent b1645a0 commit b94e8b0

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

algoliasearch/search/client.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -351,6 +351,7 @@ async def browse_objects(
351351
"""
352352
Helper: Iterate on the `browse` method of the client to allow aggregating objects of an index.
353353
"""
354+
browse_params.hits_per_page = browse_params.hits_per_page or 1000
354355

355356
async def _func(_prev: Optional[BrowseResponse]) -> BrowseResponse:
356357
if _prev is not None and _prev.cursor is not None:
@@ -392,7 +393,7 @@ async def _func(_prev: Optional[SearchRulesResponse]) -> SearchRulesResponse:
392393

393394
return await create_iterable(
394395
func=_func,
395-
validate=lambda _resp: _resp.nb_hits < hits_per_page,
396+
validate=lambda _resp: len(_resp.hits) < hits_per_page,
396397
aggregator=aggregator,
397398
)
398399

@@ -427,7 +428,7 @@ async def _func(
427428

428429
return await create_iterable(
429430
func=_func,
430-
validate=lambda _resp: _resp.nb_hits < hits_per_page,
431+
validate=lambda _resp: len(_resp.hits) < hits_per_page,
431432
aggregator=aggregator,
432433
)
433434

@@ -5362,6 +5363,7 @@ def browse_objects(
53625363
"""
53635364
Helper: Iterate on the `browse` method of the client to allow aggregating objects of an index.
53645365
"""
5366+
browse_params.hits_per_page = browse_params.hits_per_page or 1000
53655367

53665368
def _func(_prev: Optional[BrowseResponse]) -> BrowseResponse:
53675369
if _prev is not None and _prev.cursor is not None:
@@ -5403,7 +5405,7 @@ def _func(_prev: Optional[SearchRulesResponse]) -> SearchRulesResponse:
54035405

54045406
return create_iterable_sync(
54055407
func=_func,
5406-
validate=lambda _resp: _resp.nb_hits < hits_per_page,
5408+
validate=lambda _resp: len(_resp.hits) < hits_per_page,
54075409
aggregator=aggregator,
54085410
)
54095411

@@ -5436,7 +5438,7 @@ def _func(_prev: Optional[SearchSynonymsResponse]) -> SearchSynonymsResponse:
54365438

54375439
return create_iterable_sync(
54385440
func=_func,
5439-
validate=lambda _resp: _resp.nb_hits < hits_per_page,
5441+
validate=lambda _resp: len(_resp.hits) < hits_per_page,
54405442
aggregator=aggregator,
54415443
)
54425444

0 commit comments

Comments
 (0)