Merge branch 'issue-254-test-fix' into issue-254-fix

AgentOps-AI · Jun 26, 2024 · bde0804 · bde0804
2 parents 9d9356c + 70d1d3a
commit bde0804
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/agentops/__init__.py b/agentops/__init__.py
@@ -5,7 +5,9 @@
 from typing import Optional, List, Union
 from .client import Client
 from .config import Configuration
-from .event import Event, ErrorEvent
+from .event import Event, ActionEvent, LLMEvent, ToolEvent, ErrorEvent
+from .decorators import record_function
+from .agent import track_agent
 from .log_config import logger
 
 try: