From 6d1bcff6d164c268e8ee9614ecebd9c5acc62934 Mon Sep 17 00:00:00 2001 From: Lorenze Jay Date: Fri, 14 Feb 2025 16:00:30 -0800 Subject: [PATCH] Improve AgentOps listener type hints and formatting - Add string type hints for AgentOps classes to resolve potential import issues - Clean up unnecessary whitespace and improve code indentation - Simplify initialization and event handling logic --- .../events/third_party/agentops_listener.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/src/crewai/utilities/events/third_party/agentops_listener.py b/src/crewai/utilities/events/third_party/agentops_listener.py index 77e492965..b4fcf841f 100644 --- a/src/crewai/utilities/events/third_party/agentops_listener.py +++ b/src/crewai/utilities/events/third_party/agentops_listener.py @@ -18,8 +18,8 @@ except ImportError: class AgentOpsListener(BaseEventListener): - tool_event: Optional[agentops.ToolEvent] = None - session: Optional[agentops.Session] = None + tool_event: Optional["agentops.ToolEvent"] = None + session: Optional["agentops.Session"] = None def __init__(self): super().__init__() @@ -28,8 +28,6 @@ class AgentOpsListener(BaseEventListener): if not AGENTOPS_INSTALLED: return - - @event_bus.on(CrewKickoffStartedEvent) def on_crew_kickoff_started(source, event: CrewKickoffStartedEvent): self.session = agentops.init() @@ -50,9 +48,9 @@ class AgentOpsListener(BaseEventListener): @event_bus.on(ToolUsageStartedEvent) def on_tool_usage_started(source, event: ToolUsageStartedEvent): - self.tool_event = agentops.ToolEvent(name=event.tool_name) - if self.session: - self.session.record(self.tool_event) + self.tool_event = agentops.ToolEvent(name=event.tool_name) + if self.session: + self.session.record(self.tool_event) @event_bus.on(ToolUsageErrorEvent) def on_tool_usage_error(source, event: ToolUsageErrorEvent):