🐛 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:
parent
410ba005cd
commit
4b6a8595df
1 changed files with 1 additions and 1 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue