From aa30bc97d76b52cf64e7b4596e70b11ddf6516e8 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 26 Jan 2024 17:19:49 -0300 Subject: [PATCH] Remove unused imports and variables in ChatVertex class --- src/backend/langflow/graph/vertex/types.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py index f829e5db9..1e4b2195f 100644 --- a/src/backend/langflow/graph/vertex/types.py +++ b/src/backend/langflow/graph/vertex/types.py @@ -2,7 +2,6 @@ import ast from typing import Callable, Dict, List, Optional, Union from langchain_core.messages import AIMessage - from langflow.graph.utils import UnbuiltObject, flatten_list from langflow.graph.vertex.base import StatefulVertex, StatelessVertex from langflow.interface.utils import extract_input_variables_from_prompt @@ -326,7 +325,6 @@ class ChatVertex(StatelessVertex): artifacts = None sender = self.params.get("sender", None) sender_name = self.params.get("sender_name", None) - message = "" if isinstance(self._built_object, AIMessage): artifacts = ChatOutputResponse.from_message( self._built_object, @@ -334,8 +332,10 @@ class ChatVertex(StatelessVertex): sender_name=sender_name, ) elif not isinstance(self._built_object, UnbuiltObject): + if not isinstance(self._built_object, str): + self._built_object = str(self._built_object) artifacts = ChatOutputResponse( - message=message, + message=self._built_object, sender=sender, sender_name=sender_name, )