From e456b006ea4c88ecbb373f2f688034b57a8a33a3 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 26 Jun 2024 17:38:23 -0300 Subject: [PATCH] refactor: Update from_message method in MessageBase model to accept UUID as flow_id parameter --- .../base/langflow/services/database/models/message/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/base/langflow/services/database/models/message/model.py b/src/backend/base/langflow/services/database/models/message/model.py index 69b461753..7c0b9dc8f 100644 --- a/src/backend/base/langflow/services/database/models/message/model.py +++ b/src/backend/base/langflow/services/database/models/message/model.py @@ -26,7 +26,7 @@ class MessageBase(SQLModel): return value @classmethod - def from_message(cls, message: "Message", flow_id: str | None = None): + def from_message(cls, message: "Message", flow_id: str | UUID | None = None): # first check if the record has all the required fields if message.text is None or not message.sender or not message.sender_name: raise ValueError("The message does not have the required fields (text, sender, sender_name).")