guard: Allow calling parse with preconfigured num_reasks (guardrails-…

…ai#423)
guardrails-ai · Nov 3, 2023 · 2eddb2a · 2eddb2a
1 parent e2fbb37
commit 2eddb2a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/guardrails/guard.py b/guardrails/guard.py
@@ -459,7 +459,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,
@@ -473,7 +473,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,
@@ -487,7 +487,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,
@@ -500,7 +500,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,