From 8e54c87cf31df539ad1fc5728c9bd89e6df9ff4d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 19 Jun 2024 18:11:27 -0300 Subject: [PATCH] refactor: Update model_config in BaseInputMixin to forbid extra fields --- src/backend/base/langflow/inputs/input_mixin.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/inputs/input_mixin.py b/src/backend/base/langflow/inputs/input_mixin.py index d3adf69da..d9107d272 100644 --- a/src/backend/base/langflow/inputs/input_mixin.py +++ b/src/backend/base/langflow/inputs/input_mixin.py @@ -1,10 +1,9 @@ from enum import Enum from typing import Annotated, Any, Optional -from pydantic import BaseModel, ConfigDict, Field, PlainSerializer, field_validator, model_serializer - from langflow.field_typing.range_spec import RangeSpec from langflow.inputs.validators import CoalesceBool +from pydantic import BaseModel, ConfigDict, Field, PlainSerializer, field_validator, model_serializer class FieldTypes(str, Enum): @@ -25,7 +24,7 @@ SerializableFieldTypes = Annotated[FieldTypes, PlainSerializer(lambda v: v.value # Base mixin for common input field attributes and methods class BaseInputMixin(BaseModel, validate_assignment=True): - model_config = ConfigDict(arbitrary_types_allowed=True) + model_config = ConfigDict(arbitrary_types_allowed=True, extra="forbid") field_type: Optional[SerializableFieldTypes] = Field(default=FieldTypes.TEXT)