diff --git a/src/crewai/agent.py b/src/crewai/agent.py index df09ff067..a18f5cc85 100644 --- a/src/crewai/agent.py +++ b/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) diff --git a/src/crewai/crew.py b/src/crewai/crew.py index 44a1cd9e8..f07f1b770 100644 --- a/src/crewai/crew.py +++ b/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)})" diff --git a/src/crewai/tools/tool_usage.py b/src/crewai/tools/tool_usage.py index 28847b9c5..3317b5000 100644 --- a/src/crewai/tools/tool_usage.py +++ b/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)