jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
ogabrielluiz 2024-06-18 15:00:31 -03:00
commit ecac10e416
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/base/langflow/components/models/ChatLiteLLMModel.py
View file

@ -126,8 +126,9 @@ class ChatLiteLLMModelComponent(LCModelComponent):
system_message = self.system_message system_message = self.system_message
output = self.build_model() output = self.build_model()
result = self.get_chat_result(output, stream, input_value, system_message) result = self.get_chat_result(output, stream, input_value, system_message)
self.status = result message = Message(text=result)
return result self.status = message
return message
def build_model(self) -> BaseLanguageModel: def build_model(self) -> BaseLanguageModel:
try: try: