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

📝 (model.py): import Message class from langflow.schema.message to fix the reference error

Browse source 
♻️ (utils.py): remove unnecessary whitespace to improve code readability
This commit is contained in:
ogabrielluiz 2024-06-17 16:34:29 -03:00
commit 44abbabb13
2 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -7,6 +7,7 @@ from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
from langflow.custom import Component from langflow.custom import Component
from langflow.field_typing.prompt import Prompt from langflow.field_typing.prompt import Prompt
from langflow.schema.message import Message
class LCModelComponent(Component): class LCModelComponent(Component):
@ -92,7 +93,7 @@ class LCModelComponent(Component):
if system_message: if system_message:
messages.append(SystemMessage(content=system_message)) messages.append(SystemMessage(content=system_message))
if input_value: if input_value:
if isinstance(input_value, Prompt): if isinstance(input_value, Message):
with warnings.catch_warnings(): with warnings.catch_warnings():
warnings.simplefilter("ignore") warnings.simplefilter("ignore")
if "prompt" in input_value: if "prompt" in input_value:

1
src/backend/base/langflow/graph/utils.py
View file

@ -54,6 +54,7 @@ def flatten_list(list_of_lists: list[Union[list, Any]]) -> list:
def serialize_field(value): def serialize_field(value):
"""Unified serialization function for handling both BaseModel and Document types, """Unified serialization function for handling both BaseModel and Document types,
including handling lists of these types.""" including handling lists of these types."""
if isinstance(value, (list, tuple)): if isinstance(value, (list, tuple)):
return [serialize_field(v) for v in value] return [serialize_field(v) for v in value]
elif isinstance(value, Document): elif isinstance(value, Document):