fix: rename edge_props to router_edge_props to fix type checker error

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2024-12-30 21:18:29 +00:00
parent b0d1d86c26
commit 4604ac618d

View File

@@ -353,14 +353,14 @@ def add_edges(net: Network, flow: Flow[Any],
else: else:
edge_config = {"type": "straight"} edge_config = {"type": "straight"}
edge_props: Dict[str, Any] = { router_edge_props: Dict[str, Any] = {
"color": colors["router_edge"], "color": colors["router_edge"],
"width": 2, "width": 2,
"arrows": "to", "arrows": "to",
"dashes": True, "dashes": True,
"smooth": edge_config, "smooth": edge_config,
} }
net.add_edge(router_method_name, listener_name, **edge_props) net.add_edge(router_method_name, listener_name, **router_edge_props)
else: else:
# Same check here: known router edge and known method? # Same check here: known router edge and known method?
method_known = listener_name in flow._methods method_known = listener_name in flow._methods