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

better tool and llm tracking

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai/agent.py
View File

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

4
src/crewai/crew.py
View File

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

2
src/crewai/tools/tool_usage.py
View File

@@ -156,7 +156,7 @@ class ToolUsage:
self._printer.print(content=f"\n\n{error_message}\n", color="red") self._printer.print(content=f"\n\n{error_message}\n", color="red")
return error return error
self.task.increment_tools_errors() 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) return self.use(calling=calling, tool_string=tool_string)
self.tools_handler.on_tool_use(calling=calling, output=result) self.tools_handler.on_tool_use(calling=calling, output=result)