diff --git a/autoblocks/_impl/prompts/autogenerate.py b/autoblocks/_impl/prompts/autogenerate.py index 7347cf48..8ef063aa 100644 --- a/autoblocks/_impl/prompts/autogenerate.py +++ b/autoblocks/_impl/prompts/autogenerate.py @@ -18,11 +18,6 @@ from autoblocks._impl.util import AutoblocksEnvVar from autoblocks._impl.util import encode_uri_component -try: - from pydantic import AliasChoices -except ImportError: # pragma: no cover - older pydantic - AliasChoices = None # type: ignore - class Template(FrozenModel): id: str diff --git a/autoblocks/_impl/prompts/v2/discovery/prompts.py b/autoblocks/_impl/prompts/v2/discovery/prompts.py index 290a8670..cb70b12e 100644 --- a/autoblocks/_impl/prompts/v2/discovery/prompts.py +++ b/autoblocks/_impl/prompts/v2/discovery/prompts.py @@ -9,12 +9,6 @@ from autoblocks._impl.prompts.utils import infer_type from autoblocks._impl.prompts.utils import to_snake_case from autoblocks._impl.prompts.utils import to_title_case - -try: - from pydantic import AliasChoices -except ImportError: # pragma: no cover - older pydantic - AliasChoices = None # type: ignore - from autoblocks._impl.prompts.v2.client import PromptsAPIClient from autoblocks._impl.prompts.v2.discovery.managers import generate_execution_context_class_code from autoblocks._impl.prompts.v2.discovery.managers import generate_factory_class_code diff --git a/tests/autoblocks/discovery/test_prompts.py b/tests/autoblocks/discovery/test_prompts.py index a18007aa..7937b974 100644 --- a/tests/autoblocks/discovery/test_prompts.py +++ b/tests/autoblocks/discovery/test_prompts.py @@ -6,11 +6,6 @@ from autoblocks._impl.prompts.v2.discovery.prompts import generate_prompt_implementations from autoblocks._impl.prompts.v2.discovery.prompts import generate_version_implementations -try: - from pydantic import AliasChoices -except ImportError: # pragma: no cover - older pydantic - AliasChoices = None # type: ignore - class TestPrompts: def test_generate_params_class_code_empty(self): diff --git a/tests/e2e/prompts.py b/tests/e2e/prompts.py index 2e0c62c8..767c65ef 100644 --- a/tests/e2e/prompts.py +++ b/tests/e2e/prompts.py @@ -14,11 +14,6 @@ from autoblocks.prompts.renderer import TemplateRenderer from autoblocks.prompts.renderer import ToolRenderer -try: - from pydantic import AliasChoices -except ImportError: # pragma: no cover - older pydantic - AliasChoices = None # type: ignore - class QuestionAnswererParams(FrozenModel): temperature: Union[float, int] = pydantic.Field(..., alias="temperature")