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

Add support for dumping artifacts as YAML string

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-20 14:32:13 -03:00
commit 3bb3ca7f3a
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -2,6 +2,7 @@ import ast
import json import json
from typing import Callable, Dict, List, Optional, Union from typing import Callable, Dict, List, Optional, Union
import yaml
from langchain_core.messages import AIMessage from langchain_core.messages import AIMessage
from langflow.graph.utils import UnbuiltObject, flatten_list from langflow.graph.utils import UnbuiltObject, flatten_list
from langflow.graph.vertex.base import StatefulVertex, StatelessVertex from langflow.graph.vertex.base import StatefulVertex, StatelessVertex
@ -343,8 +344,11 @@ class ChatVertex(StatelessVertex):
return str(task.info) return str(task.info)
else: else:
return f"Task {self.task_id} is not running" return f"Task {self.task_id} is not running"
if self.artifacts and "repr" in self.artifacts: if self.artifacts:
return self.artifacts["repr"] or super()._built_object_repr() # dump as a yaml string
yaml_str = yaml.dump(self.artifacts, default_flow_style=False)
return yaml_str
return super()._built_object_repr()
async def _run(self, *args, **kwargs): async def _run(self, *args, **kwargs):
if self.is_interface_component: if self.is_interface_component: