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

[DO NOT MERGE] Provide inputs on crew creation (#898)

Browse Source 
* Provide inputs on crew creation

* Better naming

* Add crew id and task index to tasks

* Fix type again
This commit is contained in:
Gui Vieira
2024-07-15 09:00:02 -03:00
committed by GitHub
parent bc8cb4984a
commit fbae4909a0
2 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/crewai/crew.py
View File

@@ -170,7 +170,6 @@ class Crew(BaseModel):
self._rpm_controller = RPMController(max_rpm=self.max_rpm, logger=self._logger) self._rpm_controller = RPMController(max_rpm=self.max_rpm, logger=self._logger)
self._telemetry = Telemetry() self._telemetry = Telemetry()
self._telemetry.set_tracer() self._telemetry.set_tracer()
self._telemetry.crew_creation(self)
return self return self
@model_validator(mode="after") @model_validator(mode="after")

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

@@ -80,7 +80,7 @@ class Telemetry:
self.ready = False self.ready = False
self.trace_set = False self.trace_set = False
def crew_creation(self, crew): def crew_creation(self, crew: Crew, inputs: dict[str, Any] | None):
"""Records the creation of a crew.""" """Records the creation of a crew."""
if self.ready: if self.ready:
try: try:
@@ -93,6 +93,12 @@ class Telemetry:
) )
self._add_attribute(span, "python_version", platform.python_version()) self._add_attribute(span, "python_version", platform.python_version())
self._add_attribute(span, "crew_id", str(crew.id)) self._add_attribute(span, "crew_id", str(crew.id))
if crew.share_crew:
self._add_attribute(
span, "crew_inputs", json.dumps(inputs) if inputs else None
)
self._add_attribute(span, "crew_process", crew.process) self._add_attribute(span, "crew_process", crew.process)
self._add_attribute(span, "crew_memory", crew.memory) self._add_attribute(span, "crew_memory", crew.memory)
self._add_attribute(span, "crew_number_of_tasks", len(crew.tasks)) self._add_attribute(span, "crew_number_of_tasks", len(crew.tasks))
@@ -114,7 +120,7 @@ class Telemetry:
"llm": json.dumps(self._safe_llm_attributes(agent.llm)), "llm": json.dumps(self._safe_llm_attributes(agent.llm)),
"delegation_enabled?": agent.allow_delegation, "delegation_enabled?": agent.allow_delegation,
"tools_names": [ "tools_names": [
tool.name.casefold() for tool in agent.tools tool.name.casefold() for tool in agent.tools or []
], ],
} }
for agent in crew.agents for agent in crew.agents
@@ -139,7 +145,7 @@ class Telemetry:
else None else None
), ),
"tools_names": [ "tools_names": [
tool.name.casefold() for tool in task.tools tool.name.casefold() for tool in task.tools or []
], ],
} }
for task in crew.tasks for task in crew.tasks
@@ -161,10 +167,11 @@ class Telemetry:
if self.ready: if self.ready:
try: try:
tracer = trace.get_tracer("crewai.telemetry") tracer = trace.get_tracer("crewai.telemetry")
span = tracer.start_span("Task Execution")
created_span = tracer.start_span("Task Created") created_span = tracer.start_span("Task Created")
self._add_attribute(created_span, "crew_id", str(crew.id))
self._add_attribute(created_span, "task_index", crew.tasks.index(task))
self._add_attribute(created_span, "task_id", str(task.id)) self._add_attribute(created_span, "task_id", str(task.id))
if crew.share_crew: if crew.share_crew:
@@ -178,6 +185,10 @@ class Telemetry:
created_span.set_status(Status(StatusCode.OK)) created_span.set_status(Status(StatusCode.OK))
created_span.end() created_span.end()
span = tracer.start_span("Task Execution")
self._add_attribute(span, "crew_id", str(crew.id))
self._add_attribute(span, "task_index", crew.tasks.index(task))
self._add_attribute(span, "task_id", str(task.id)) self._add_attribute(span, "task_id", str(task.id))
if crew.share_crew: if crew.share_crew:
@@ -275,6 +286,8 @@ class Telemetry:
""" """
if (self.ready) and (crew.share_crew): if (self.ready) and (crew.share_crew):
try: try:
self.crew_creation(crew, inputs)
tracer = trace.get_tracer("crewai.telemetry") tracer = trace.get_tracer("crewai.telemetry")
span = tracer.start_span("Crew Execution") span = tracer.start_span("Crew Execution")
self._add_attribute( self._add_attribute(
@@ -283,7 +296,9 @@ class Telemetry:
pkg_resources.get_distribution("crewai").version, pkg_resources.get_distribution("crewai").version,
) )
self._add_attribute(span, "crew_id", str(crew.id)) self._add_attribute(span, "crew_id", str(crew.id))
self._add_attribute(span, "inputs", json.dumps(inputs)) self._add_attribute(
span, "crew_inputs", json.dumps(inputs) if inputs else None
)
self._add_attribute( self._add_attribute(
span, span,
"crew_agents", "crew_agents",