Move serialization logic to SessionManager

Author: Nithish-KV

CHANGELOG.md

@@ -5,9 +5,10 @@ All notable changes to this project will be documented in this file.
 The format is based on Keep a Changelog and this project adheres to Semantic Versioning.
 
 
-## [0.1.79] - 2026-03-31
+## [0.1.79] - 2026-04-01
 - Added input/output attributes across LLM, traceloop, and custom spans
 - Added utility function to explicitly set input/output attributes on the active span
+- Move serialization logic to SessionManager
 
 
 ## [0.1.78] - 2026-03-31

netra/__init__.py

@@ -1,5 +1,4 @@
 import atexit
-import json
 import logging
 import threading
 from typing import Any, Dict, List, Optional, Set
@@ -389,43 +388,23 @@ def add_conversation(cls, conversation_type: ConversationType, role: str, conten
 
     @classmethod
     def set_input(cls, value: Any) -> None:
-        """Set the ``input`` attribute on the current active span.
-
-        Accepts any value. Dicts and lists are JSON-serialised; primitives are
-        converted with ``str()``. The result is truncated to
-        ``Config.ATTRIBUTE_MAX_LEN`` characters.
+        """
+        Set the input attribute on the current active span.
 
         Args:
-            value: The input value to record.
+            value: The input value to record
         """
-        try:
-            if isinstance(value, (dict, list)):
-                serialized = json.dumps(value, default=str)[: Config.ATTRIBUTE_MAX_LEN]
-            else:
-                serialized = str(value)[: Config.ATTRIBUTE_MAX_LEN]
-            SessionManager.set_attribute_on_active_span("input", serialized)
-        except Exception:
-            logger.exception("Netra.set_input: failed to set input attribute")
+        SessionManager.set_input(value)
 
     @classmethod
     def set_output(cls, value: Any) -> None:
-        """Set the ``output`` attribute on the current active span.
-
-        Accepts any value. Dicts and lists are JSON-serialised; primitives are
-        converted with ``str()``. The result is truncated to
-        ``Config.ATTRIBUTE_MAX_LEN`` characters.
+        """
+        Set the output attribute on the current active span.
 
         Args:
-            value: The output value to record.
+            value: The output value to record
         """
-        try:
-            if isinstance(value, (dict, list)):
-                serialized = json.dumps(value, default=str)[: Config.ATTRIBUTE_MAX_LEN]
-            else:
-                serialized = str(value)[: Config.ATTRIBUTE_MAX_LEN]
-            SessionManager.set_attribute_on_active_span("output", serialized)
-        except Exception:
-            logger.exception("Netra.set_output: failed to set output attribute")
+        SessionManager.set_output(value)
 
     @classmethod
     def start_span(

netra/session_manager.py

@@ -1,3 +1,4 @@
+import json
 import logging
 from datetime import datetime
 from enum import Enum
@@ -372,6 +373,46 @@ def add_conversation(cls, conversation_type: ConversationType, role: str, conten
         except Exception as e:
             logger.exception("Failed to add conversation attribute: %s", e)
 
+    @classmethod
+    def set_input(cls, value: Any) -> None:
+        """Set the ``input`` attribute on the current active span.
+
+        Accepts any value. Dicts and lists are JSON-serialised; primitives are
+        converted with ``str()``. The result is truncated to
+        ``Config.ATTRIBUTE_MAX_LEN`` characters.
+
+        Args:
+            value: The input value to record.
+        """
+        try:
+            if isinstance(value, (dict, list)):
+                serialized = json.dumps(value, default=str)[: Config.ATTRIBUTE_MAX_LEN]
+            else:
+                serialized = str(value)[: Config.ATTRIBUTE_MAX_LEN]
+            cls.set_attribute_on_active_span("input", serialized)
+        except Exception:
+            logger.exception("SessionManager.set_input: failed to set input attribute")
+
+    @classmethod
+    def set_output(cls, value: Any) -> None:
+        """Set the ``output`` attribute on the current active span.
+
+        Accepts any value. Dicts and lists are JSON-serialised; primitives are
+        converted with ``str()``. The result is truncated to
+        ``Config.ATTRIBUTE_MAX_LEN`` characters.
+
+        Args:
+            value: The output value to record.
+        """
+        try:
+            if isinstance(value, (dict, list)):
+                serialized = json.dumps(value, default=str)[: Config.ATTRIBUTE_MAX_LEN]
+            else:
+                serialized = str(value)[: Config.ATTRIBUTE_MAX_LEN]
+            cls.set_attribute_on_active_span("output", serialized)
+        except Exception:
+            logger.exception("SessionManager.set_output: failed to set output attribute")
+
     @staticmethod
     def set_attribute_on_active_span(attr_key: str, attr_value: Any) -> None:
         """