From b286d063591bf17472fd8ed2182513709b7a7574 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 18 Jun 2024 18:42:42 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=93=9D=20(message.py):=20add=20logging=20?= =?UTF-8?q?using=20loguru=20to=20warn=20about=20missing=20required=20keys?= =?UTF-8?q?=20in=20Message=20class=20and=20default=20to=20HumanMessage=20i?= =?UTF-8?q?f=20keys=20are=20missing?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/schema/message.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/schema/message.py b/src/backend/base/langflow/schema/message.py index 84b125c26..bb4eacaaf 100644 --- a/src/backend/base/langflow/schema/message.py +++ b/src/backend/base/langflow/schema/message.py @@ -7,6 +7,7 @@ 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 @@ -61,9 +62,9 @@ class Message(Data): # But first we check if all required keys are present in the data dictionary # they are: "text", "sender" if self.text is None or not self.sender: - raise ValueError("Missing required keys ('text', 'sender') in Message.") + logger.warn("Missing required keys ('text', 'sender') in Message, defaulting to HumanMessage.") - if self.sender == "User": + if self.sender == "User" or not self.sender: if self.files: contents = [{"type": "text", "text": self.text}] contents.extend(self.get_file_content_dicts())