Fix field validation for files in Message class
This commit is contained in:
parent
c31cf6fd1e
commit
ba8754a785
1 changed files with 9 additions and 3 deletions
|
|
@ -7,12 +7,11 @@ from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, System
|
|||
from langchain_core.prompt_values import ImagePromptValue
|
||||
from langchain_core.prompts import BaseChatPromptTemplate, ChatPromptTemplate, PromptTemplate
|
||||
from langchain_core.prompts.image import ImagePromptTemplate
|
||||
from loguru import logger
|
||||
from pydantic import BeforeValidator, ConfigDict, Field, field_serializer
|
||||
|
||||
from langflow.base.prompts.utils import dict_values_to_string
|
||||
from langflow.schema.data import Data
|
||||
from langflow.schema.image import Image, get_file_paths, is_image_file
|
||||
from loguru import logger
|
||||
from pydantic import BeforeValidator, ConfigDict, Field, field_serializer, field_validator
|
||||
|
||||
|
||||
def _timestamp_to_str(timestamp: datetime) -> str:
|
||||
|
|
@ -33,6 +32,13 @@ class Message(Data):
|
|||
)
|
||||
flow_id: Optional[str] = None
|
||||
|
||||
@field_validator("files", mode="before")
|
||||
@classmethod
|
||||
def validate_files(cls, value):
|
||||
if not value:
|
||||
value = []
|
||||
return value
|
||||
|
||||
def model_post_init(self, __context: Any) -> None:
|
||||
new_files = []
|
||||
for file in self.files or []:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue