🐛 fix(nodes.py): change type hint of tools list to include ToolkitNode

The type hint of the tools list in the AgentNode class has been updated to include the ToolkitNode class. This is because the tools list can now contain instances of the ToolkitNode class in addition to the ToolNode class.
This commit is contained in:
Gabriel Almeida 2023-05-30 21:34:23 -03:00
commit 4b6a8595df

View file

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