From 4b6a8595df8e519a24fe1c064baf85c04ebe345b Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Tue, 30 May 2023 21:34:23 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(nodes.py):=20change=20type?= =?UTF-8?q?=20hint=20of=20tools=20list=20to=20include=20ToolkitNode=20The?= =?UTF-8?q?=20type=20hint=20of=20the=20tools=20list=20in=20the=20AgentNode?= =?UTF-8?q?=20class=20has=20been=20updated=20to=20include=20the=20ToolkitN?= =?UTF-8?q?ode=20class.=20This=20is=20because=20the=20tools=20list=20can?= =?UTF-8?q?=20now=20contain=20instances=20of=20the=20ToolkitNode=20class?= =?UTF-8?q?=20in=20addition=20to=20the=20ToolNode=20class.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/graph/nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py index 5f136b3c7..9db6260e9 100644 --- a/src/backend/langflow/graph/nodes.py +++ b/src/backend/langflow/graph/nodes.py @@ -8,7 +8,7 @@ class AgentNode(Node): def __init__(self, data: Dict): super().__init__(data, base_type="agents") - self.tools: List[ToolNode] = [] + self.tools: List[Union[ToolNode, ToolkitNode]] = [] self.chains: List[ChainNode] = [] def _set_tools_and_chains(self) -> None: