From ecac10e4160c071e5ba3c5332d2c74794c291be2 Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 18 Jun 2024 15:00:31 -0300 Subject: [PATCH] refactor: Update ChatLiteLLMModelComponent to use Message from langflow.schema.message --- .../base/langflow/components/models/ChatLiteLLMModel.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/models/ChatLiteLLMModel.py b/src/backend/base/langflow/components/models/ChatLiteLLMModel.py index 4528df3bd..f710a7d39 100644 --- a/src/backend/base/langflow/components/models/ChatLiteLLMModel.py +++ b/src/backend/base/langflow/components/models/ChatLiteLLMModel.py @@ -126,8 +126,9 @@ class ChatLiteLLMModelComponent(LCModelComponent): system_message = self.system_message output = self.build_model() result = self.get_chat_result(output, stream, input_value, system_message) - self.status = result - return result + message = Message(text=result) + self.status = message + return message def build_model(self) -> BaseLanguageModel: try: