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

Remove unnecessary steps in AgentVertex and ChainVertex

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-07 14:59:48 -03:00
commit b79e4a1198
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -14,7 +14,7 @@ class AgentVertex(StatelessVertex):
self.tools: List[Union[ToolkitVertex, ToolVertex]] = [] self.tools: List[Union[ToolkitVertex, ToolVertex]] = []
self.chains: List[ChainVertex] = [] self.chains: List[ChainVertex] = []
self.steps: List[Callable] = [self._custom_build, self._run] self.steps: List[Callable] = [self._custom_build]
def __getstate__(self): def __getstate__(self):
state = super().__getstate__() state = super().__getstate__()
@ -201,7 +201,7 @@ class TextSplitterVertex(StatefulVertex):
class ChainVertex(StatelessVertex): class ChainVertex(StatelessVertex):
def __init__(self, data: Dict, graph): def __init__(self, data: Dict, graph):
super().__init__(data, graph=graph, base_type="chains") super().__init__(data, graph=graph, base_type="chains")
self.steps = [self._custom_build, self._run] self.steps = [self._custom_build]
async def _custom_build(self, *args, **kwargs): async def _custom_build(self, *args, **kwargs):
force = kwargs.get("force", False) force = kwargs.get("force", False)