refactor: Update ChatLiteLLMModelComponent to use Message from langflow.schema.message
This commit is contained in:
parent
c04cdb3412
commit
ecac10e416
1 changed files with 3 additions and 2 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue