andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 08:12:39 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Improve AgentOps listener type hints and formatting

Browse Source 
- 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
This commit is contained in:
Lorenze Jay
2025-02-14 16:00:30 -08:00
parent aa2e7c888e
commit 6d1bcff6d1
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/crewai/utilities/events/third_party/agentops_listener.py vendored
View File

@@ -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):