Skip to content

Commit 2eddb2a

Browse files
irgolicCalebCourier
authored andcommitted
guard: Allow calling parse with preconfigured num_reasks (guardrails-ai#423)
1 parent e2fbb37 commit 2eddb2a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

guardrails/guard.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -459,7 +459,7 @@ def parse(
459459
llm_output: str,
460460
metadata: Optional[Dict] = None,
461461
llm_api: None = None,
462-
num_reasks: int = 1,
462+
num_reasks: Optional[int] = None,
463463
prompt_params: Optional[Dict] = None,
464464
full_schema_reask: Optional[bool] = None,
465465
*args,
@@ -473,7 +473,7 @@ def parse(
473473
llm_output: str,
474474
metadata: Optional[Dict] = None,
475475
llm_api: Callable[[Any], Awaitable[Any]] = ...,
476-
num_reasks: int = 1,
476+
num_reasks: Optional[int] = None,
477477
prompt_params: Optional[Dict] = None,
478478
full_schema_reask: Optional[bool] = None,
479479
*args,
@@ -487,7 +487,7 @@ def parse(
487487
llm_output: str,
488488
metadata: Optional[Dict] = None,
489489
llm_api: Optional[Callable] = None,
490-
num_reasks: int = 1,
490+
num_reasks: Optional[int] = None,
491491
prompt_params: Optional[Dict] = None,
492492
full_schema_reask: Optional[bool] = None,
493493
*args,
@@ -500,7 +500,7 @@ def parse(
500500
llm_output: str,
501501
metadata: Optional[Dict] = None,
502502
llm_api: Optional[Callable] = None,
503-
num_reasks: int = 1,
503+
num_reasks: Optional[int] = None,
504504
prompt_params: Optional[Dict] = None,
505505
full_schema_reask: Optional[bool] = None,
506506
*args,

0 commit comments

Comments
 (0)