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

refactor: Add field_validator for files in MessageBase model

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-25 21:48:50 -03:00
commit 5bdd035dc3
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -2,6 +2,7 @@ from datetime import datetime, timezone
from typing import TYPE_CHECKING, List, Optional from typing import TYPE_CHECKING, List, Optional
from uuid import UUID, uuid4 from uuid import UUID, uuid4
from pydantic import field_validator
from sqlmodel import JSON, Column, Field, Relationship, SQLModel from sqlmodel import JSON, Column, Field, Relationship, SQLModel
if TYPE_CHECKING: if TYPE_CHECKING:
@ -17,6 +18,13 @@ class MessageBase(SQLModel):
text: str text: str
files: list[str] = Field(default_factory=list) files: list[str] = Field(default_factory=list)
@field_validator("files", mode="before")
@classmethod
def validate_files(cls, value):
if not value:
value = []
return value
@classmethod @classmethod
def from_message(cls, message: "Message", flow_id: str | None = None): def from_message(cls, message: "Message", flow_id: str | None = None):
# first check if the record has all the required fields # first check if the record has all the required fields