Skip pagination if nothing to paginate (#78)

Skyvern-AI · Oct 3, 2023 · 0db764d · 0db764d
1 parent 647deaf
commit 0db764d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wyvern-ai"
-version = "0.0.21"
+version = "0.0.22"
 description = ""
 authors = ["Wyvern AI <info@wyvern.ai>"]
 readme = "README.md"

diff --git a/wyvern/components/pagination/pagination_component.py b/wyvern/components/pagination/pagination_component.py
@@ -51,6 +51,10 @@ async def execute(self, input: PaginationRequest[T], **kwargs) -> List[T]:
  Returns:
  The paginated entities.
  """
+ if len(input.entities) == 0:
+ logger.info("Found no entities to paginate, skipping pagination")
+ return []
+
  user_page = input.pagination_fields.user_page
  candidate_page = input.pagination_fields.candidate_page
  candidate_page_size = input.pagination_fields.candidate_page_size