refactor: Update ChatLiteLLMModelComponent to use Message from langflow.schema.message

This commit is contained in:
ogabrielluiz 2024-06-18 15:00:31 -03:00
commit ecac10e416

View file

@ -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: