update args

AgentOps-AI · Apr 1, 2024 · 5e87270 · 5e87270
1 parent 77ef2d5
commit 5e87270
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/jaiqu/cli.py b/jaiqu/cli.py
@@ -24,6 +24,12 @@ def jaiqu(
         "--key-hints",
         help="Extra prompt for the ai to help it complete the task",
     ),
+    max_retries: int = Option(
+        10,
+        "-r",
+        "--max-retries",
+        help="Max number of retries for the ai to complete the task",
+    ),
 ):
     """
     Validate and translate a json schema to jq filter
@@ -46,7 +52,8 @@ def jaiqu(
         output_schema=output_schema,
         input_json=input_json,
         key_hints=key_hints,
-        quiet=quiet,
+        max_retries=max_retries,
+        quiet=quiet
     )
     print(query)