diff --git a/guardrails/cli/create.py b/guardrails/cli/create.py index 0048f44f2..4668236ba 100644 --- a/guardrails/cli/create.py +++ b/guardrails/cli/create.py @@ -118,7 +118,7 @@ def generate_template_config( guard_instantiations = [] for i, guard in enumerate(template["guards"]): - guard_instantiations.append(f"guard{i} = AsyncGuard.from_dict(guards[{i}])") + guard_instantiations.append(f"guard{i} = Guard.from_dict(guards[{i}])") guard_instantiations = "\n".join(guard_instantiations) # Interpolate variables output_content = template_content.format( diff --git a/guardrails/cli/hub/template_config.py.template b/guardrails/cli/hub/template_config.py.template index b61ffb2f2..5d5683bdf 100644 --- a/guardrails/cli/hub/template_config.py.template +++ b/guardrails/cli/hub/template_config.py.template @@ -1,6 +1,6 @@ import json import os -from guardrails import AsyncGuard, Guard +from guardrails import Guard from guardrails.hub import {VALIDATOR_IMPORTS} try: diff --git a/server_ci/config.py b/server_ci/config.py index d5a093ca7..b84d637ea 100644 --- a/server_ci/config.py +++ b/server_ci/config.py @@ -1,6 +1,6 @@ import json import os -from guardrails import AsyncGuard +from guardrails import Guard try: file_path = os.path.join(os.getcwd(), "guard-template.json") @@ -11,4 +11,4 @@ SystemExit(1) # instantiate guards -guard0 = AsyncGuard.from_dict(guards[0]) +guard0 = Guard.from_dict(guards[0])