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

refactor: Update langflow schema data module to use BaseMessage instead of HumanMessage or SystemMessage

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 19:05:16 -03:00
commit df3c275db8
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/schema/data.py
View file

@ -3,7 +3,7 @@ import json
from typing import Optional, cast
from langchain_core.documents import Document
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
from langchain_core.prompt_values import ImagePromptValue
from langchain_core.prompts.image import ImagePromptTemplate
from pydantic import BaseModel, model_serializer, model_validator
@ -112,7 +112,7 @@ class Data(BaseModel):
def to_lc_message(
self,
) -> HumanMessage | SystemMessage:
) -> BaseMessage:
"""
Converts the Data to a BaseMessage.