From a7931de5ab401c705105cbbd463bed35dd928b72 Mon Sep 17 00:00:00 2001 From: Jose Osorio Date: Tue, 29 Jul 2025 16:28:52 -0500 Subject: [PATCH 1/3] Add maxTokens --- .../__snapshots__/test_prompts_autogenerate.ambr | 1 + tests/e2e/prompts_v2/apps/ci_app/prompts.py | 8 ++++++++ 2 files changed, 9 insertions(+) diff --git a/tests/autoblocks/__snapshots__/test_prompts_autogenerate.ambr b/tests/autoblocks/__snapshots__/test_prompts_autogenerate.ambr index a7245e58..9dda3a0c 100644 --- a/tests/autoblocks/__snapshots__/test_prompts_autogenerate.ambr +++ b/tests/autoblocks/__snapshots__/test_prompts_autogenerate.ambr @@ -20,6 +20,7 @@ class PromptAParams(FrozenModel): frequency_penalty: Union[float, int] = pydantic.Field(..., alias="frequencyPenalty") + max_tokens: Union[float, int] = pydantic.Field(..., alias="maxTokens") max_completion_tokens: Union[float, int] = pydantic.Field(..., alias="maxCompletionTokens") model: str = pydantic.Field(..., alias="model") presence_penalty: Union[float, int] = pydantic.Field(..., alias="presencePenalty") diff --git a/tests/e2e/prompts_v2/apps/ci_app/prompts.py b/tests/e2e/prompts_v2/apps/ci_app/prompts.py index 54f471c1..e0c1d306 100644 --- a/tests/e2e/prompts_v2/apps/ci_app/prompts.py +++ b/tests/e2e/prompts_v2/apps/ci_app/prompts.py @@ -13,6 +13,10 @@ class _PromptBasicV4Params(FrozenModel): + max_tokens: Union[float, int] = pydantic.Field( + ..., + alias="maxTokens", + ) max_completion_tokens: Union[float, int] = pydantic.Field( ..., alias="maxCompletionTokens", @@ -56,6 +60,10 @@ class _PromptBasicV4PromptManager(AutoblocksPromptManager[_PromptBasicV4Executio class _PromptBasicV3Params(FrozenModel): + max_tokens: Union[float, int] = pydantic.Field( + ..., + alias="maxTokens", + ) max_completion_tokens: Union[float, int] = pydantic.Field( ..., alias="maxCompletionTokens", From 915d81e55c6889ef14e03f943a8f4726874ada9e Mon Sep 17 00:00:00 2001 From: Jose Osorio Date: Tue, 29 Jul 2025 16:31:50 -0500 Subject: [PATCH 2/3] Update fixture --- tests/autoblocks/__snapshots__/test_prompts_autogenerate.ambr | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/autoblocks/__snapshots__/test_prompts_autogenerate.ambr b/tests/autoblocks/__snapshots__/test_prompts_autogenerate.ambr index 9dda3a0c..a7245e58 100644 --- a/tests/autoblocks/__snapshots__/test_prompts_autogenerate.ambr +++ b/tests/autoblocks/__snapshots__/test_prompts_autogenerate.ambr @@ -20,7 +20,6 @@ class PromptAParams(FrozenModel): frequency_penalty: Union[float, int] = pydantic.Field(..., alias="frequencyPenalty") - max_tokens: Union[float, int] = pydantic.Field(..., alias="maxTokens") max_completion_tokens: Union[float, int] = pydantic.Field(..., alias="maxCompletionTokens") model: str = pydantic.Field(..., alias="model") presence_penalty: Union[float, int] = pydantic.Field(..., alias="presencePenalty") From 5dc3a68fbf61f93dccf6ab599e4464c928dd3331 Mon Sep 17 00:00:00 2001 From: Jose Osorio Date: Tue, 29 Jul 2025 16:42:40 -0500 Subject: [PATCH 3/3] Fix --- autoblocks/_impl/prompts/autogenerate.py | 5 ----- autoblocks/_impl/prompts/v2/discovery/prompts.py | 6 ------ tests/autoblocks/discovery/test_prompts.py | 5 ----- tests/e2e/prompts.py | 5 ----- tests/e2e/prompts_v2/apps/ci_app/prompts.py | 8 -------- 5 files changed, 29 deletions(-) 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") diff --git a/tests/e2e/prompts_v2/apps/ci_app/prompts.py b/tests/e2e/prompts_v2/apps/ci_app/prompts.py index e0c1d306..54f471c1 100644 --- a/tests/e2e/prompts_v2/apps/ci_app/prompts.py +++ b/tests/e2e/prompts_v2/apps/ci_app/prompts.py @@ -13,10 +13,6 @@ class _PromptBasicV4Params(FrozenModel): - max_tokens: Union[float, int] = pydantic.Field( - ..., - alias="maxTokens", - ) max_completion_tokens: Union[float, int] = pydantic.Field( ..., alias="maxCompletionTokens", @@ -60,10 +56,6 @@ class _PromptBasicV4PromptManager(AutoblocksPromptManager[_PromptBasicV4Executio class _PromptBasicV3Params(FrozenModel): - max_tokens: Union[float, int] = pydantic.Field( - ..., - alias="maxTokens", - ) max_completion_tokens: Union[float, int] = pydantic.Field( ..., alias="maxCompletionTokens",