Fix issue with file validation in Message class

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 16:01:56 -03:00
commit aec4292239

View file

@ -37,6 +37,8 @@ class Message(Data):
def validate_files(cls, value):
if not value:
value = []
elif not isinstance(value, list):
value = [value]
return value
def model_post_init(self, __context: Any) -> None:
@ -174,3 +176,4 @@ class Message(Data):
instance.prompt = jsonable_encoder(prompt_template.to_json())
instance.messages = instance.prompt.get("kwargs", {}).get("messages", [])
return instance
return instance