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

refactor: Update Message class to use jsonable_encoder for prompt serialization

Browse source
This commit is contained in:
ogabrielluiz 2024-06-17 15:47:15 -03:00
commit 6211908d98
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/schema/message.py
View file

@ -1,6 +1,7 @@
from datetime import datetime, timezone from datetime import datetime, timezone
from typing import Annotated, Any, AsyncIterator, Iterator, Optional from typing import Annotated, Any, AsyncIterator, Iterator, Optional
from fastapi.encoders import jsonable_encoder
from langchain_core.load import load from langchain_core.load import load
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
from langchain_core.prompt_values import ImagePromptValue from langchain_core.prompt_values import ImagePromptValue
@ -149,6 +150,6 @@ class Message(Data):
content_dicts = await value.get_file_content_dicts() content_dicts = await value.get_file_content_dicts()
contents.extend(content_dicts) contents.extend(content_dicts)
prompt_template = ChatPromptTemplate.from_messages([HumanMessage(content=contents)]) prompt_template = ChatPromptTemplate.from_messages([HumanMessage(content=contents)])
instance.messages = prompt_template.messages instance.prompt = jsonable_encoder(prompt_template.to_json())
instance.prompt = prompt_template.to_json() instance.messages = instance.prompt.get("kwargs", {}).get("messages", [])
return instance return instance