fix typing in worker.py and set body value correctly in `http_cli…

…ent.py`

agentops/http_client.py

@@ -28,9 +28,7 @@ class Response:
     def __init__(self, status: HttpStatus = HttpStatus.UNKNOWN, body: Optional[dict] = None):
         self.status: HttpStatus = status
         self.code: int = status.value
-        self.body = body
-        if not self.body:
-            self.body = {}
+        self.body = body if body else {}
 
     def parse(self, res: requests.models.Response):
         res_body = res.json()

agentops/worker.py

@@ -2,10 +2,10 @@
 import threading
 import time
 from .http_client import HttpClient
-from .config import Configuration, ConfigurationError
+from .config import Configuration
 from .session import Session
 from .helpers import safe_serialize, filter_unjsonable
-from typing import Dict
+from typing import Dict, Optional
 
 
 class Worker:
@@ -17,7 +17,7 @@ def __init__(self, config: Configuration) -> None:
         self.thread = threading.Thread(target=self.run)
         self.thread.daemon = True
         self.thread.start()
-        self._session: Session | None = None
+        self._session: Optional[Session] = None
 
     def add_event(self, event: dict) -> None:
         with self.lock:
@@ -32,7 +32,7 @@ def flush_queue(self) -> None:
                 self.queue = []
 
                 payload = {
-                    "session_id": self._session.session_id,
+                    "session_id": getattr(self._session, "session_id", None),
                     "events": events
                 }
 
@@ -42,7 +42,7 @@ def flush_queue(self) -> None:
                                 self.config.api_key,
                                 self.config.parent_key)
 
-    def start_session(self, session: Session) -> None:
+    def start_session(self, session: Session) -> bool:
         self._session = session
         with self.lock:
             payload = {
@@ -94,7 +94,7 @@ def create_agent(self, agent_id, name):
         payload = {
             "id": agent_id,
             "name": name,
-            "session_id": self._session.session_id
+            "session_id": getattr(self._session, "session_id", None),
         }
 
         serialized_payload = \