Add limit to AplOptions (#5)

axiomhq · Dec 12, 2024 · dd9ff01 · dd9ff01
1 parent 6282e0d
commit dd9ff01
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/axiom_py/client.py b/src/axiom_py/client.py
@@ -110,6 +110,8 @@ class AplOptions:
     # IncludeCursor will return the Cursor as part of the query result, if set
     # to true.
     includeCursor: bool = field(default=False)
+    # The query limit.
+    limit: Optional[int] = field(default=None)
 
 
 class AxiomError(Exception):
@@ -368,11 +370,13 @@ def _prepare_apl_options(
         self, opts: Optional[AplOptions]
     ) -> Dict[str, object]:
         """Prepare the apl query options for the request."""
-        params = {"format": AplResultFormat.Legacy.value}
+        params: Dict[str, object] = {"format": AplResultFormat.Legacy.value}
 
         if opts is not None:
             if opts.format:
                 params["format"] = opts.format.value
+            if opts.limit is not None:
+                params["request"] = {"limit": opts.limit}
 
         return params