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

Refactor typing import in types.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-27 17:57:29 -03:00
commit 4007b9728f
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -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