Skip to content

Commit 6fe2d00

Browse files
authored
guard: Allow calling parse with preconfigured num_reasks (#423)
1 parent faa8a50 commit 6fe2d00

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
@@ -444,7 +444,7 @@ def parse(
444444
llm_output: str,
445445
metadata: Optional[Dict] = None,
446446
llm_api: None = None,
447-
num_reasks: int = 1,
447+
num_reasks: Optional[int] = None,
448448
prompt_params: Optional[Dict] = None,
449449
full_schema_reask: Optional[bool] = None,
450450
*args,
@@ -458,7 +458,7 @@ def parse(
458458
llm_output: str,
459459
metadata: Optional[Dict] = None,
460460
llm_api: Callable[[Any], Awaitable[Any]] = ...,
461-
num_reasks: int = 1,
461+
num_reasks: Optional[int] = None,
462462
prompt_params: Optional[Dict] = None,
463463
full_schema_reask: Optional[bool] = None,
464464
*args,
@@ -472,7 +472,7 @@ def parse(
472472
llm_output: str,
473473
metadata: Optional[Dict] = None,
474474
llm_api: Optional[Callable] = None,
475-
num_reasks: int = 1,
475+
num_reasks: Optional[int] = None,
476476
prompt_params: Optional[Dict] = None,
477477
full_schema_reask: Optional[bool] = None,
478478
*args,
@@ -485,7 +485,7 @@ def parse(
485485
llm_output: str,
486486
metadata: Optional[Dict] = None,
487487
llm_api: Optional[Callable] = None,
488-
num_reasks: int = 1,
488+
num_reasks: Optional[int] = None,
489489
prompt_params: Optional[Dict] = None,
490490
full_schema_reask: Optional[bool] = None,
491491
*args,

0 commit comments

Comments
 (0)