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

Refactor ChatVertex to handle artifacts and unescape message

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-08 12:17:44 -03:00
commit 7b53178d6f
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/langflow/graph/vertex/types.py
View file

@ -13,6 +13,7 @@ from langflow.interface.utils import extract_input_variables_from_prompt
from langflow.schema import Record from langflow.schema import Record
from langflow.services.monitor.utils import log_vertex_build from langflow.services.monitor.utils import log_vertex_build
from langflow.utils.schemas import ChatOutputResponse from langflow.utils.schemas import ChatOutputResponse
from langflow.utils.util import unescape_string
class AgentVertex(Vertex): class AgentVertex(Vertex):
@ -353,8 +354,13 @@ class ChatVertex(Vertex):
return f"Task {self.task_id} is not running" return f"Task {self.task_id} is not running"
if self.artifacts: if self.artifacts:
# dump as a yaml string # dump as a yaml string
artifacts = {
k.title().replace("_", " "): v
for k, v in self.artifacts.items()
if v is not None
}
yaml_str = yaml.dump( yaml_str = yaml.dump(
self.artifacts, default_flow_style=False, allow_unicode=True artifacts, default_flow_style=False, allow_unicode=True
) )
return yaml_str return yaml_str
return super()._built_object_repr() return super()._built_object_repr()
@ -365,7 +371,7 @@ class ChatVertex(Vertex):
artifacts = None artifacts = None
sender = self.params.get("sender", None) sender = self.params.get("sender", None)
sender_name = self.params.get("sender_name", None) sender_name = self.params.get("sender_name", None)
message = self.params.get(INPUT_FIELD_NAME, None) message = unescape_string(self.params.get(INPUT_FIELD_NAME, None))
stream_url = None stream_url = None
if isinstance(self._built_object, AIMessage): if isinstance(self._built_object, AIMessage):
artifacts = ChatOutputResponse.from_message( artifacts = ChatOutputResponse.from_message(