jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update model_config in BaseInputMixin to forbid extra fields

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 18:11:27 -03:00
commit 8e54c87cf3
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/inputs/input_mixin.py
View file

@ -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)