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

refactor: Update LCModelComponent to use BaseMessage instead of HumanMessage and SystemMessage

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:09:10 -03:00
commit 43058045eb
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/base/models/model.py
View file

@ -3,7 +3,7 @@ import warnings
from typing import Optional, Union from typing import Optional, Union
from langchain_core.language_models.llms import LLM from langchain_core.language_models.llms import LLM
from langchain_core.messages import AIMessage, HumanMessage, SystemMessage from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage
from langflow.custom import Component from langflow.custom import Component
from langflow.field_typing import LanguageModel from langflow.field_typing import LanguageModel
@ -120,7 +120,7 @@ class LCModelComponent(Component):
def get_chat_result( def get_chat_result(
self, runnable: LanguageModel, stream: bool, input_value: str | Message, system_message: Optional[str] = None self, runnable: LanguageModel, stream: bool, input_value: str | Message, system_message: Optional[str] = None
): ):
messages: list[Union[HumanMessage, SystemMessage]] = [] messages: list[Union[BaseMessage]] = []
if not input_value and not system_message: if not input_value and not system_message:
raise ValueError("The message you want to send to the model is empty.") raise ValueError("The message you want to send to the model is empty.")
if system_message: if system_message: