init on start session

AgentOps-AI · Jun 13, 2024 · ec4a695 · ec4a695
1 parent 35ca376
commit ec4a695
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/agentops/__init__.py b/agentops/__init__.py
@@ -126,7 +126,6 @@ def end_session(
     )
 
 
-@check_init
 def start_session(
     tags: Optional[List[str]] = None,
     config: Optional[Configuration] = None,
@@ -140,7 +139,16 @@ def start_session(
             e.g. ["test_run"].
         config: (Configuration, optional): Client configuration object
     """
-    return Client().start_session(tags, config, inherited_session_id)
+
+    try:
+        sess_result = Client().start_session(tags, config, inherited_session_id)
+
+        global is_initialized
+        is_initialized = True
+
+        return sess_result
+    except Exception:
+        pass
 
 
 @check_init

diff --git a/tests/test_session.py b/tests/test_session.py
@@ -26,7 +26,6 @@ def setup_method(self):
         self.api_key = "random_api_key"
         self.event_type = "test_event_type"
         self.config = agentops.Configuration(api_key=self.api_key, max_wait_time=50)
-        agentops.init(api_key=self.api_key)
 
     def test_session(self, mock_req):
         agentops.start_session(config=self.config)