From 3bb3ca7f3a17fd38ea19678d7ad67c04353b8813 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 20 Feb 2024 14:32:13 -0300 Subject: [PATCH] Add support for dumping artifacts as YAML string --- src/backend/langflow/graph/vertex/types.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py index 8d23059ce..cb4684d08 100644 --- a/src/backend/langflow/graph/vertex/types.py +++ b/src/backend/langflow/graph/vertex/types.py @@ -2,6 +2,7 @@ import ast import json from typing import Callable, Dict, List, Optional, Union +import yaml from langchain_core.messages import AIMessage from langflow.graph.utils import UnbuiltObject, flatten_list from langflow.graph.vertex.base import StatefulVertex, StatelessVertex @@ -343,8 +344,11 @@ class ChatVertex(StatelessVertex): return str(task.info) else: return f"Task {self.task_id} is not running" - if self.artifacts and "repr" in self.artifacts: - return self.artifacts["repr"] or super()._built_object_repr() + if self.artifacts: + # 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): if self.is_interface_component: