guard: Allow calling parse with preconfigured num_reasks (#423)

guardrails-ai · Nov 3, 2023 · 6fe2d00 · 6fe2d00
1 parent faa8a50
commit 6fe2d00
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/guardrails/guard.py b/guardrails/guard.py
@@ -444,7 +444,7 @@ def parse(
         llm_output: str,
         metadata: Optional[Dict] = None,
         llm_api: None = None,
-        num_reasks: int = 1,
+        num_reasks: Optional[int] = None,
         prompt_params: Optional[Dict] = None,
         full_schema_reask: Optional[bool] = None,
         *args,
@@ -458,7 +458,7 @@ def parse(
         llm_output: str,
         metadata: Optional[Dict] = None,
         llm_api: Callable[[Any], Awaitable[Any]] = ...,
-        num_reasks: int = 1,
+        num_reasks: Optional[int] = None,
         prompt_params: Optional[Dict] = None,
         full_schema_reask: Optional[bool] = None,
         *args,
@@ -472,7 +472,7 @@ def parse(
         llm_output: str,
         metadata: Optional[Dict] = None,
         llm_api: Optional[Callable] = None,
-        num_reasks: int = 1,
+        num_reasks: Optional[int] = None,
         prompt_params: Optional[Dict] = None,
         full_schema_reask: Optional[bool] = None,
         *args,
@@ -485,7 +485,7 @@ def parse(
         llm_output: str,
         metadata: Optional[Dict] = None,
         llm_api: Optional[Callable] = None,
-        num_reasks: int = 1,
+        num_reasks: Optional[int] = None,
         prompt_params: Optional[Dict] = None,
         full_schema_reask: Optional[bool] = None,
         *args,