Adding serialization functionality and multion examples

agentops/client.py

@@ -33,6 +33,7 @@
 from .config import Configuration, ConfigurationError
 from .llm_tracker import LlmTracker
 from termcolor import colored
+from typing import Tuple
 
 
 @singleton
@@ -75,7 +76,7 @@ def __init__(
         tags: Optional[List[str]] = None,
         override: Optional[bool] = None,  # Deprecated
         instrument_llm_calls=True,
-        auto_start_session=True,
+        auto_start_session=False,
         inherited_session_id: Optional[str] = None,
         skip_auto_end_session: Optional[bool] = False,
     ):
@@ -133,7 +134,7 @@ def __init__(
 
     def _check_for_partner_frameworks(
         self, instrument_llm_calls, auto_start_session
-    ) -> tuple[bool, bool]:
+    ) -> Tuple[bool, bool]:
         partner_frameworks = get_partner_frameworks()
         for framework in partner_frameworks.keys():
             if framework in sys.modules:
@@ -260,6 +261,10 @@ def _record_event_sync(self, func, event_name, *args, **kwargs):
                 returns = list(returns)
 
             event.returns = returns
+
+            if hasattr(returns, "screenshot"):
+                event.screenshot = returns.screenshot
+
             event.end_timestamp = get_ISO_time()
             self.record(event)
 
@@ -300,6 +305,12 @@ async def _record_event_async(self, func, event_name, *args, **kwargs):
                 returns = list(returns)
 
             event.returns = returns
+
+            # NOTE: Will likely remove in future since this is tightly coupled. Adding it to see how useful we find it for now
+            # TODO: check if screenshot is the url string we expect it to be? And not e.g. "True"
+            if hasattr(returns, "screenshot"):
+                event.screenshot = returns.screenshot
+
             event.end_timestamp = get_ISO_time()
             self.record(event)
 

agentops/decorators.py

@@ -1,5 +1,4 @@
 from .client import Client
-from .event import Event
 import inspect
 import functools
 
@@ -16,7 +15,6 @@ def record_function(event_name: str):
     """
 
     def decorator(func):
-
         if inspect.iscoroutinefunction(func):
 
             @functools.wraps(func)

agentops/helpers.py

@@ -85,19 +85,29 @@ def default(o):
             return o.model_dump_json()
         elif hasattr(o, "to_json"):
             return o.to_json()
+        elif hasattr(o, "json"):
+            return o.json()
+        elif hasattr(o, "to_dict"):
+            return o.to_dict()
+        elif hasattr(o, "dict"):
+            return o.dict()
         else:
             return f"<<non-serializable: {type(o).__qualname__}>>"
 
-    def remove_none_values(value):
-        """Recursively remove keys with None values from dictionaries."""
+    def remove_unwanted_items(value):
+        """Recursively remove self key and None/... values from dictionaries so they aren't serialized"""
         if isinstance(value, dict):
-            return {k: remove_none_values(v) for k, v in value.items() if v is not None}
+            return {
+                k: remove_unwanted_items(v)
+                for k, v in value.items()
+                if v is not None and v is not ... and k != "self"
+            }
         elif isinstance(value, list):
-            return [remove_none_values(item) for item in value]
+            return [remove_unwanted_items(item) for item in value]
         else:
             return value
 
-    cleaned_obj = remove_none_values(obj)
+    cleaned_obj = remove_unwanted_items(obj)
     return json.dumps(cleaned_obj, default=default)