diff --git a/api/oss/src/core/applications/dtos.py b/api/oss/src/core/applications/dtos.py index 1b186c8ce1..3b740fc72f 100644 --- a/api/oss/src/core/applications/dtos.py +++ b/api/oss/src/core/applications/dtos.py @@ -1,7 +1,7 @@ from typing import Optional, List from uuid import UUID -from pydantic import Field +from pydantic import ConfigDict, Field from oss.src.core.shared.dtos import sync_alias, AliasConfig from oss.src.core.shared.dtos import ( @@ -212,6 +212,7 @@ class ApplicationVariantQuery(WorkflowVariantQuery): class ApplicationRevisionData(WorkflowRevisionData): + model_config = ConfigDict(extra="forbid") pass diff --git a/api/oss/src/core/evaluators/dtos.py b/api/oss/src/core/evaluators/dtos.py index e68f4c53ec..7df5a71831 100644 --- a/api/oss/src/core/evaluators/dtos.py +++ b/api/oss/src/core/evaluators/dtos.py @@ -1,7 +1,7 @@ from typing import Optional, List from uuid import UUID -from pydantic import Field +from pydantic import ConfigDict, Field from oss.src.core.shared.dtos import sync_alias, AliasConfig from oss.src.core.shared.dtos import ( @@ -208,6 +208,7 @@ class EvaluatorVariantQuery(WorkflowVariantQuery): class EvaluatorRevisionData(WorkflowRevisionData): + model_config = ConfigDict(extra="forbid") pass diff --git a/api/oss/src/core/queries/dtos.py b/api/oss/src/core/queries/dtos.py index 3b17bfb9f7..301d6fec7b 100644 --- a/api/oss/src/core/queries/dtos.py +++ b/api/oss/src/core/queries/dtos.py @@ -1,7 +1,7 @@ from typing import List, Optional from uuid import UUID -from pydantic import BaseModel, Field +from pydantic import BaseModel, ConfigDict, Field from oss.src.core.tracing.dtos import Filtering, Formatting @@ -139,6 +139,7 @@ class QueryVariantQuery(VariantQuery): class QueryRevisionData(BaseModel): + model_config = ConfigDict(extra="forbid") formatting: Optional[Formatting] = None filtering: Optional[Filtering] = None windowing: Optional[Windowing] = None diff --git a/api/oss/src/core/testsets/dtos.py b/api/oss/src/core/testsets/dtos.py index 3ab2ffb2f3..6828922dc9 100644 --- a/api/oss/src/core/testsets/dtos.py +++ b/api/oss/src/core/testsets/dtos.py @@ -1,7 +1,7 @@ from typing import Optional, List, Tuple from uuid import UUID -from pydantic import BaseModel, Field +from pydantic import BaseModel, ConfigDict, Field from oss.src.core.shared.dtos import ( sync_alias, @@ -154,6 +154,7 @@ class TestsetVariantQuery(VariantQuery): class TestsetRevisionData(BaseModel): + model_config = ConfigDict(extra="forbid") testcase_ids: Optional[List[UUID]] = None testcases: Optional[List[Testcase]] = None diff --git a/sdks/python/agenta/sdk/models/testsets.py b/sdks/python/agenta/sdk/models/testsets.py index c3e4eb1b64..7a9073c17b 100644 --- a/sdks/python/agenta/sdk/models/testsets.py +++ b/sdks/python/agenta/sdk/models/testsets.py @@ -1,7 +1,7 @@ from typing import List, Optional, Dict, Any from uuid import UUID -from pydantic import BaseModel, Field +from pydantic import BaseModel, ConfigDict, Field from agenta.sdk.models.shared import ( Identifier, @@ -53,6 +53,7 @@ class TestsetFlags(BaseModel): class TestsetRevisionData(BaseModel): + model_config = ConfigDict(extra="forbid") testcase_ids: Optional[List[UUID]] = None testcases: Optional[List[Testcase]] = None diff --git a/sdks/python/agenta/sdk/models/workflows.py b/sdks/python/agenta/sdk/models/workflows.py index 67dfc3ac21..6c019463b6 100644 --- a/sdks/python/agenta/sdk/models/workflows.py +++ b/sdks/python/agenta/sdk/models/workflows.py @@ -119,6 +119,7 @@ class WorkflowQueryFlags(BaseModel): class WorkflowRevisionData(BaseModel): + model_config = ConfigDict(extra="forbid") uri: Optional[str] = None url: Optional[str] = None @@ -562,6 +563,7 @@ class EvaluatorVariantIdAlias(AliasConfig): class EvaluatorRevisionData(WorkflowRevisionData): + model_config = ConfigDict(extra="forbid") pass @@ -712,6 +714,7 @@ class ApplicationVariantEdit(WorkflowVariantEdit): class ApplicationRevisionData(WorkflowRevisionData): + model_config = ConfigDict(extra="forbid") pass