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

store node names

Browse Source
This commit is contained in:
Brandon Hancock
2024-10-03 13:20:26 -04:00
parent 2418069081
commit 1693be3e53
2 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/crewai/flow/flow.py
View File

@@ -193,7 +193,9 @@ class Flow(Generic[T], metaclass=FlowMeta):
if not self._start_methods: if not self._start_methods:
raise ValueError("No start method defined") raise ValueError("No start method defined")
self._telemetry.flow_execution_span(self.__class__.__name__) self._telemetry.flow_execution_span(
self.__class__.__name__, list(self._methods.keys())
)
# Create tasks for all start methods # Create tasks for all start methods
tasks = [ tasks = [
@@ -276,6 +278,8 @@ class Flow(Generic[T], metaclass=FlowMeta):
traceback.print_exc() traceback.print_exc()
def plot(self, filename: str = "crewai_flow"): def plot(self, filename: str = "crewai_flow"):
self._telemetry.flow_plotting_span(self.__class__.__name__) self._telemetry.flow_plotting_span(
self.__class__.__name__, list(self._methods.keys())
)
plot_flow(self, filename) plot_flow(self, filename)

6
src/crewai/telemetry/telemetry.py
View File

@@ -586,23 +586,25 @@ class Telemetry:
except Exception: except Exception:
pass pass
def flow_plotting_span(self, flow_name: str): def flow_plotting_span(self, flow_name: str, node_names: list[str]):
if self.ready: if self.ready:
try: try:
tracer = trace.get_tracer("crewai.telemetry") tracer = trace.get_tracer("crewai.telemetry")
span = tracer.start_span("Flow Plotting") span = tracer.start_span("Flow Plotting")
self._add_attribute(span, "flow_name", flow_name) self._add_attribute(span, "flow_name", flow_name)
self._add_attribute(span, "node_names", json.dumps(node_names))
span.set_status(Status(StatusCode.OK)) span.set_status(Status(StatusCode.OK))
span.end() span.end()
except Exception: except Exception:
pass pass
def flow_execution_span(self, flow_name: str): def flow_execution_span(self, flow_name: str, node_names: list[str]):
if self.ready: if self.ready:
try: try:
tracer = trace.get_tracer("crewai.telemetry") tracer = trace.get_tracer("crewai.telemetry")
span = tracer.start_span("Flow Execution") span = tracer.start_span("Flow Execution")
self._add_attribute(span, "flow_name", flow_name) self._add_attribute(span, "flow_name", flow_name)
self._add_attribute(span, "node_names", json.dumps(node_names))
span.set_status(Status(StatusCode.OK)) span.set_status(Status(StatusCode.OK))
span.end() span.end()
except Exception: except Exception: