From 5e872702bde8c4304a6890f923940401471f9233 Mon Sep 17 00:00:00 2001
From: reibs <areibman@gmail.com>
Date: Mon, 1 Apr 2024 12:55:31 -0700
Subject: [PATCH] update args

---
 jaiqu/cli.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/jaiqu/cli.py b/jaiqu/cli.py
index 11cf2ec..44bf3b5 100644
--- 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)