From 4007b9728fde7402cbeecf0ee1b0ee2fcea9c19b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 27 Feb 2024 17:57:29 -0300 Subject: [PATCH] Refactor typing import in types.py --- src/backend/langflow/graph/vertex/types.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py index 45bcb9ccd..92d4a1bd7 100644 --- a/src/backend/langflow/graph/vertex/types.py +++ b/src/backend/langflow/graph/vertex/types.py @@ -1,7 +1,6 @@ import ast import json -from typing import (AsyncIterator, Callable, Dict, Iterator, List, Optional, - Union) +from typing import AsyncIterator, Callable, Dict, Iterator, List, Optional, Union import yaml from langchain_core.messages import AIMessage @@ -396,7 +395,7 @@ class ChatVertex(StatelessVertex): if artifacts: self.artifacts = artifacts.model_dump() if isinstance(self._built_object, (AsyncIterator, Iterator)): - if self.params["as_record"]: + if self.params["return_record"]: self._built_object = Record(text=message, data=self.artifacts) else: self._built_object = message