better tool and llm tracking

src/crewai/agent.py

@@ -25,7 +25,6 @@ from crewai.agents import CacheHandler, CrewAgentExecutor, CrewAgentParser, Tool
 from crewai.utilities import I18N, Logger, Prompts, RPMController
 from crewai.utilities.token_counter_callback import TokenCalcHandler, TokenProcess
 from agentops.agent import track_agent
-from agentops.langchain_callback_handler import LangchainCallbackHandler
 
 
 @track_agent()
@@ -157,8 +156,7 @@ class Agent(BaseModel):
         """set agent executor is set."""
         if hasattr(self.llm, "model_name"):
             self.llm.callbacks = [
-                TokenCalcHandler(self.llm.model_name, self._token_process),
-                LangchainCallbackHandler()
+                TokenCalcHandler(self.llm.model_name, self._token_process)
             ]
         if not self.agent_executor:
             self.set_cache_handler(self.cache_handler)

src/crewai/crew.py

@@ -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)})"

src/crewai/tools/tool_usage.py

@@ -156,7 +156,7 @@ class ToolUsage:
                     self._printer.print(content=f"\n\n{error_message}\n", color="red")
                     return error
                 self.task.increment_tools_errors()
-                agentops.record(agentops.ErrorEvent(details=error, trigger_event=tool_event))
+                agentops.record(agentops.ErrorEvent(details=e, trigger_event=tool_event))
                 return self.use(calling=calling, tool_string=tool_string)
 
             self.tools_handler.on_tool_use(calling=calling, output=result)