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

fix: Fix type-checker issues with _finish_execution method

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-12 06:46:27 +00:00
parent 6c3f4a6446
commit db34516b18
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/crewai/crew.py
View File

@@ -602,7 +602,8 @@ class Crew(BaseModel):
inputs = before_callback(inputs) inputs = before_callback(inputs)
"""Starts the crew to work on its assigned tasks.""" """Starts the crew to work on its assigned tasks."""
self._execution_span = self._telemetry.crew_execution_span(self, inputs) if self._telemetry:
self._execution_span = self._telemetry.crew_execution_span(self, inputs)
self._task_output_handler.reset() self._task_output_handler.reset()
self._logging_color = "bold_purple" self._logging_color = "bold_purple"
@@ -1180,16 +1181,22 @@ class Crew(BaseModel):
for agent in self.agents: for agent in self.agents:
agent.interpolate_inputs(inputs) agent.interpolate_inputs(inputs)
def _finish_execution(self, final_string_output: str) -> None: def _finish_execution(self, final_output: Union[str, CrewOutput]) -> None:
"""Finish execution and cleanup.
Args:
final_output: The final output from crew execution, either as string or CrewOutput
"""
if self.max_rpm: if self.max_rpm:
self._rpm_controller.stop_rpm_counter() self._rpm_controller.stop_rpm_counter()
if agentops: if self._telemetry:
agentops.end_session( self._telemetry.end_crew(self, final_output)
if self._agentops:
self._agentops.end_session(
end_state="Success", end_state="Success",
end_state_reason="Finished Execution", end_state_reason="Finished Execution",
is_auto_end=True, is_auto_end=True,
) )
self._telemetry.end_crew(self, final_string_output)
def calculate_usage_metrics(self) -> UsageMetrics: def calculate_usage_metrics(self) -> UsageMetrics:
"""Calculates and returns the usage metrics.""" """Calculates and returns the usage metrics."""