andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 16:18:30 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

children router working. Need to support loops

Browse Source
This commit is contained in:
Brandon Hancock
2024-09-30 11:34:57 -04:00
parent 66e7fc5ce3
commit e07b245c83
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/crewai/flow/flow_visualizer.py
View File

@@ -96,16 +96,19 @@ class PyvisFlowVisualizer(FlowVisualizer):
) )
# Add edges # Add edges
print("self.flow._routers.values()", self.flow._routers)
for method_name in self.flow._listeners: for method_name in self.flow._listeners:
print("Listeners for", method_name)
condition_type, trigger_methods = self.flow._listeners[method_name] condition_type, trigger_methods = self.flow._listeners[method_name]
is_and_condition = condition_type == "AND" is_and_condition = condition_type == "AND"
for trigger in trigger_methods: for trigger in trigger_methods:
print("Trigger", trigger)
if trigger in self.flow._methods: if trigger in self.flow._methods:
is_router_edge = ( is_router_edge = any(
trigger in self.flow._routers.values() trigger in paths for paths in self.flow._router_paths.values()
or method_name in self.flow._routers.values()
) )
print("is_router_edge", is_router_edge)
edge_color = ( edge_color = (
self.colors["router_edge"] self.colors["router_edge"]
if is_router_edge if is_router_edge