From 22268409ee19be890cc867d93e0c5ceccf303dc7 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Thu, 13 Jun 2024 00:26:19 -0300 Subject: [PATCH] refactor: Update Message class in message.py to inherit from Data Update the Message class in message.py to inherit from the Data class. This change ensures consistency and improves the organization of the codebase. --- src/backend/base/langflow/schema/message.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/schema/message.py b/src/backend/base/langflow/schema/message.py index 77ba44724..caa7d95bf 100644 --- a/src/backend/base/langflow/schema/message.py +++ b/src/backend/base/langflow/schema/message.py @@ -4,7 +4,7 @@ from typing import Annotated, Any, AsyncIterator, Iterator, Optional from langchain_core.messages import AIMessage, BaseMessage, HumanMessage from langchain_core.prompt_values import ImagePromptValue from langchain_core.prompts.image import ImagePromptTemplate -from pydantic import BaseModel, BeforeValidator, ConfigDict, Field, field_serializer +from pydantic import BeforeValidator, ConfigDict, Field, field_serializer from langflow.schema.data import Data from langflow.schema.image import Image, get_file_paths, is_image_file @@ -14,7 +14,7 @@ def _timestamp_to_str(timestamp: datetime) -> str: return timestamp.strftime("%Y-%m-%d %H:%M:%S") -class Message(BaseModel): +class Message(Data): model_config = ConfigDict(arbitrary_types_allowed=True) # Helper class to deal with image data text: Optional[str | AsyncIterator | Iterator] = Field(default="")