Revert "AgentOps Implementation (#411)"

This reverts commit 3d5257592b.
This commit is contained in:
João Moura
2024-04-21 10:51:29 -05:00
parent b862e464f8
commit caa0e5db8d
4 changed files with 0 additions and 13 deletions

View File

@@ -28,7 +28,6 @@ click = "^8.1.7"
python-dotenv = "^1.0.0"
embedchain = "^0.1.98"
appdirs = "^1.4.4"
agentops = "0.1.6"
[tool.poetry.extras]
tools = ["crewai-tools"]

View File

@@ -24,10 +24,8 @@ from crewai.agents import CacheHandler, CrewAgentExecutor, CrewAgentParser, Tool
from crewai.memory.contextual.contextual_memory import ContextualMemory
from crewai.utilities import I18N, Logger, Prompts, RPMController
from crewai.utilities.token_counter_callback import TokenCalcHandler, TokenProcess
from agentops.agent import track_agent
@track_agent()
class Agent(BaseModel):
"""Represents an agent in a system.
@@ -57,8 +55,6 @@ class Agent(BaseModel):
_rpm_controller: RPMController = PrivateAttr(default=None)
_request_within_rpm_limit: Any = PrivateAttr(default=None)
_token_process: TokenProcess = TokenProcess()
agent_ops_agent_name: str = None
agent_ops_agent_id: str = None
formatting_errors: int = 0
model_config = ConfigDict(arbitrary_types_allowed=True)
@@ -133,7 +129,6 @@ class Agent(BaseModel):
def __init__(__pydantic_self__, **data):
config = data.pop("config", {})
super().__init__(**config, **data)
__pydantic_self__.agent_ops_agent_name = __pydantic_self__.role
@field_validator("id", mode="before")
@classmethod

View File

@@ -26,7 +26,6 @@ from crewai.task import Task
from crewai.telemetry import Telemetry
from crewai.tools.agent_tools import AgentTools
from crewai.utilities import I18N, FileHandler, Logger, RPMController
import agentops
class Crew(BaseModel):
@@ -240,7 +239,6 @@ class Crew(BaseModel):
self._set_tasks_callbacks()
i18n = I18N(language=self.language, language_file=self.language_file)
agentops.set_parent_key("daebe730-f54d-4af5-98df-e6946fb76d13")
for agent in self.agents:
agent.i18n = i18n
@@ -376,7 +374,6 @@ class Crew(BaseModel):
def _finish_execution(self, output) -> None:
if self.max_rpm:
self._rpm_controller.stop_rpm_counter()
agentops.end_session(end_state="Success", end_state_reason="Finished Execution")
self._telemetry.end_crew(self, output)
def __repr__(self):

View File

@@ -9,7 +9,6 @@ from crewai.agents.tools_handler import ToolsHandler
from crewai.telemetry import Telemetry
from crewai.tools.tool_calling import InstructorToolCalling, ToolCalling
from crewai.utilities import I18N, Converter, ConverterError, Printer
import agentops
OPENAI_BIGGER_MODELS = ["gpt-4"]
@@ -97,7 +96,6 @@ class ToolUsage:
tool: BaseTool,
calling: Union[ToolCalling, InstructorToolCalling],
) -> None:
tool_event = agentops.ToolEvent(name=calling.tool_name)
if self._check_tool_repeated_usage(calling=calling):
try:
result = self._i18n.errors("task_repeated_usage").format(
@@ -161,7 +159,6 @@ 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=e, trigger_event=tool_event))
return self.use(calling=calling, tool_string=tool_string)
if self.tools_handler:
@@ -182,7 +179,6 @@ class ToolUsage:
)
self._printer.print(content=f"\n\n{result}\n", color="purple")
agentops.record(tool_event)
self._telemetry.tool_usage(
llm=self.function_calling_llm,
tool_name=tool.name,