From 6fe2d008b52fc7ad155849751de7ffcea27f71ff Mon Sep 17 00:00:00 2001 From: rafael Date: Fri, 3 Nov 2023 15:21:41 +0000 Subject: [PATCH] guard: Allow calling parse with preconfigured num_reasks (#423) --- guardrails/guard.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/guardrails/guard.py b/guardrails/guard.py index c3260438a..99f6a72f5 100644 --- 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,