better tool and llm tracking

This commit is contained in:
Braelyn Boynton
2024-03-29 17:45:58 -07:00
parent c44c2b6808
commit 5e46514398
3 changed files with 5 additions and 5 deletions

View File

@@ -293,8 +293,10 @@ class Crew(BaseModel):
def _finish_execution(self, output) -> None:
if self.max_rpm:
self._rpm_controller.stop_rpm_counter()
self._telemetry.end_crew(self, output)
print('ending')
agentops.end_session(end_state="Success", end_state_reason="Finished Execution")
self._telemetry.end_crew(self, output)
def __repr__(self):
return f"Crew(id={self.id}, process={self.process}, number_of_agents={len(self.agents)}, number_of_tasks={len(self.tasks)})"