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

feat: Add field validation for flow_id in MessageTable model

Browse source 
This commit adds field validation for the `flow_id` attribute in the `MessageTable` model. The `validate_flow_id` class method is implemented to ensure that the `flow_id` value is either `None` or a valid UUID. This validation helps maintain data integrity and consistency when working with the `MessageTable` model.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-26 16:13:08 -03:00
commit fab1e327aa
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/base/langflow/services/database/models/message/model.py
View file

@ -54,6 +54,15 @@ class MessageTable(MessageBase, table=True):
flow: "Flow" = Relationship(back_populates="messages") flow: "Flow" = Relationship(back_populates="messages")
files: List[str] = Field(sa_column=Column(JSON)) files: List[str] = Field(sa_column=Column(JSON))
@field_validator("flow_id", mode="before")
@classmethod
def validate_flow_id(cls, value):
if value is None:
return value
if isinstance(value, str):
value = UUID(value)
return value
# Needed for Column(JSON) # Needed for Column(JSON)
class Config: class Config:
arbitrary_types_allowed = True arbitrary_types_allowed = True