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)