Crew Integration Changes (#133)

* pydantic support * allow client naming outside decorator * expose record at top level * merge * remove main logs * remove main logs * track agent in event recording
AgentOps-AI · Apr 1, 2024 · c71f8af · c71f8af
1 parent 7bd9163
commit c71f8af
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/agentops/__init__.py b/agentops/__init__.py
@@ -7,7 +7,6 @@
 from .event import Event, ActionEvent, LLMEvent, ToolEvent, ErrorEvent
 from .enums import Models
 from .decorators import record_function
-from os import environ
 
 
 def init(api_key: Optional[str] = None,

diff --git a/agentops/client.py b/agentops/client.py
@@ -110,6 +110,9 @@ def record(self, event: Event | ErrorEvent):
         """
 
         if self._session is not None and not self._session.has_ended:
+            agent_id = check_call_stack_for_agent_id()
+            if agent_id:
+                event.agent_id = agent_id
             self._worker.add_event(event.__dict__)
         else:
             logging.warning(

diff --git a/agentops/http_client.py b/agentops/http_client.py
@@ -97,7 +97,7 @@ def post(url: str, payload: bytes, api_key: Optional[str] = None, parent_key: Op
 
         if result.code == 401:
             logging.warning(
-                'AgentOps: Could not post data - API server rejected your API key')
+                f'AgentOps: Could not post data - API server rejected your API key: {api_key}')
         if result.code == 400:
             logging.warning(f'AgentOps: Could not post data - {result.body}')
         if result.code == 500: