Fd 45 sift client calc channels and rules

python/lib/sift_client/_internal/low_level_wrappers/__init__.py

@@ -1,4 +1,17 @@
 from sift_client._internal.low_level_wrappers.assets import AssetsLowLevelClient
+from sift_client._internal.low_level_wrappers.calculated_channels import (
+    CalculatedChannelsLowLevelClient,
+)
+from sift_client._internal.low_level_wrappers.channels import ChannelsLowLevelClient
 from sift_client._internal.low_level_wrappers.ping import PingLowLevelClient
+from sift_client._internal.low_level_wrappers.rules import RulesLowLevelClient
+from sift_client._internal.low_level_wrappers.runs import RunsLowLevelClient
 
-__all__ = ["AssetsLowLevelClient", "PingLowLevelClient"]
+__all__ = [
+    "AssetsLowLevelClient",
+    "CalculatedChannelsLowLevelClient",
+    "ChannelsLowLevelClient",
+    "PingLowLevelClient",
+    "RulesLowLevelClient",
+    "RunsLowLevelClient",
+]

python/lib/sift_client/_internal/low_level_wrappers/assets.py

@@ -68,8 +68,8 @@ async def list_all_assets(
         """
         return await self._handle_pagination(
             self.list_assets,
+            kwargs={"query_filter": query_filter},
             page_size=page_size,
-            query_filter=query_filter,
             order_by=order_by,
             max_results=max_results,
         )

python/lib/sift_client/_internal/low_level_wrappers/base.py

@@ -8,12 +8,26 @@ class LowLevelClientBase(ABC):
     @staticmethod
     async def _handle_pagination(
         func: Callable,
+        kwargs: dict[str, Any] = {},
         page_size: int | None = None,
         page_token: str | None = None,
-        query_filter: str | None = None,
         order_by: str | None = None,
         max_results: int | None = None,
     ) -> list[Any]:
+        """
+        Handle pagination for a given function by calling the function until all results are retrieved or the max_results is reached.
+
+        Args:
+            func: The function to call.
+            kwargs: Keyword arguments to pass to the function.
+            page_size: The number of results to return per page.
+            page_token: The token to use for the next page.
+            order_by: How to order the retrieved results.
+            max_results: Maximum number of results to return. Will be in increments of page_size or default page size defined by the call if no page_size is provided.
+
+        Returns:
+            A list of all matching results.
+        """
         results: list[Any] = []
         if page_token is None:
             page_token = ""
@@ -23,8 +37,8 @@ async def _handle_pagination(
             response, page_token = await func(
                 page_size=page_size,
                 page_token=page_token,
-                query_filter=query_filter,
                 order_by=order_by,
+                **kwargs,
             )
             results.extend(response)
             if page_token == "":