netra-sdk 0.1.41__py3-none-any.whl → 0.1.43__py3-none-any.whl

netra/__init__.py

@@ -249,13 +249,13 @@ class Netra:
             logger.warning("Both event_name and attributes must be provided for custom events.")
 
     @classmethod
-    def add_conversation(cls, conversation_type: ConversationType, field_name: str, value: Any) -> None:
+    def add_conversation(cls, conversation_type: ConversationType, role: str, content: Any) -> None:
         """
         Append a conversation entry and set span attribute 'conversation' as an array.
         If a conversation array already exists for the current active span, this appends
         to it; otherwise, it initializes a new array.
         """
-        SessionManager.add_conversation(conversation_type=conversation_type, field_name=field_name, value=value)
+        SessionManager.add_conversation(conversation_type=conversation_type, role=role, content=content)
 
     @classmethod
     def start_span(

netra/config.py

@@ -28,7 +28,9 @@ class Config:
     LIBRARY_NAME = "netra"
     LIBRARY_VERSION = __version__
     # Maximum length for any attribute value (strings and bytes). Processors should honor this.
-    ATTRIBUTE_MAX_LEN = 1000
+    ATTRIBUTE_MAX_LEN = 2000
+    # Maximum length specifically for conversation entry content (strings or JSON when serialized)
+    CONVERSATION_CONTENT_MAX_LEN = 1000
 
     def __init__(
         self,

netra/session_manager.py

@@ -13,6 +13,7 @@ from opentelemetry import context as otel_context
 from opentelemetry import trace
 
 from netra.config import Config
+from netra.utils import process_content_for_max_len
 
 logger = logging.getLogger(__name__)
 
@@ -20,7 +21,6 @@ logger = logging.getLogger(__name__)
 class ConversationType(str, Enum):
     INPUT = "input"
     OUTPUT = "output"
-    SYSTEM = "system"
 
 
 class SessionManager:
@@ -262,7 +262,7 @@ class SessionManager:
             logger.exception(f"Failed to add custom event: {name} - {e}")
 
     @staticmethod
-    def add_conversation(conversation_type: ConversationType, field_name: str, value: Any) -> None:
+    def add_conversation(conversation_type: ConversationType, role: str, content: Any) -> None:
         """
         Append a conversation entry and set span attribute 'conversation' as an array.
 
@@ -278,20 +278,27 @@ class SessionManager:
             raise TypeError("conversation_type must be a ConversationType enum value (input, output, system)")
         normalized_type = conversation_type.value
 
-        if not isinstance(field_name, str):
-            raise TypeError(f"field_name must be a string, got {type(field_name)}")
-        if not field_name:
-            raise ValueError("field_name must be a non-empty string")
+        if not isinstance(role, str):
+            raise TypeError(f"role must be a string, got {type(role)}")
 
-        if not value:
-            raise ValueError("value must not be empty")
+        if not isinstance(content, (str, dict)):
+            raise TypeError(f"content must be a string or dict, got {type(content)}")
+
+        if not role:
+            raise ValueError("role must be a non-empty string")
+
+        if not content:
+            raise ValueError("content must not be empty")
 
         try:
+
+            # Get active recording span
             span = trace.get_current_span()
             if not (span and getattr(span, "is_recording", lambda: False)()):
                 logger.warning("No active span to add conversation attribute.")
                 return
 
+            # Load existing conversation (JSON string -> list)
             existing: List[Dict[str, Any]] = []
             raw_data = None
 
@@ -301,10 +308,12 @@ class SessionManager:
                     raw_data = attrs.get("conversation")
             except Exception:
                 logger.exception("Failed to retrieve conversation attribute")
+
             if raw_data:
                 try:
                     import json
 
+                    parsed: Any = None
                     if isinstance(raw_data, str):
                         parsed = json.loads(raw_data)
                     if isinstance(parsed, list):
@@ -312,11 +321,30 @@ class SessionManager:
                 except Exception:
                     existing = []
 
-            # Append new entry
-            entry: Dict[str, Any] = {"type": normalized_type, "field_name": field_name, "value": value}
+            # Enforce per-entry content length limit without breaking the entire conversation structure
+            max_len = Config.CONVERSATION_CONTENT_MAX_LEN
+            processed_content = process_content_for_max_len(content, max_len)
+
+            # Create a conversation entry
+            entry: Dict[str, Any] = {"type": normalized_type, "role": role, "content": processed_content}
+
+            # Add format based on processed value type for backend parsing
+            if isinstance(processed_content, str):
+                entry["format"] = "text"
+            elif isinstance(processed_content, dict):
+                entry["format"] = "json"
             existing.append(entry)
 
-            SessionManager.set_attribute_on_active_span("conversation", existing)
+            # Bypass global attribute value truncation by writing directly to the span's
+            # private attribute store. We intentionally avoid span.set_attribute here.
+            try:
+                import json
+
+                payload = json.dumps(existing, default=str)
+                attrs = getattr(span, "_attributes", None)
+                attrs["conversation"] = payload  # type: ignore[index]
+            except Exception:
+                logger.exception("Failed to set conversation attribute directly on span")
         except Exception as e:
             logger.exception("Failed to add conversation attribute: %s", e)
 

netra/utils.py

@@ -0,0 +1,81 @@
+"""General utility helpers for Netra SDK.
+
+This module centralizes common helpers that can be reused across the codebase.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+
+def truncate_string(value: str, max_len: int) -> str:
+    """Truncate a string to max_len characters.
+
+    If value is not a string, it is returned unchanged.
+    """
+    try:
+        if not isinstance(value, str):
+            return value
+        return value if len(value) <= max_len else value[:max_len]
+    except Exception:
+        return value
+
+
+def truncate_and_repair_json(content: Any, max_len: int) -> Any:
+    """Truncate a dict/list by JSON-serializing and hard-cutting, then attempt repair.
+
+    The function will:
+    - json.dumps(content, default=str)
+    - hard-cut the string to max_len
+    - try to repair using `json-repair` (optional dependency)
+    - parse back with json.loads
+
+    On failure, returns a minimal safe container with a preview of the truncated text.
+    """
+    try:
+        import json
+
+        json_str = json.dumps(content, default=str)
+        if len(json_str) <= max_len:
+            return content
+
+        truncated = json_str[:max_len]
+
+        # Try json_repair if available
+        repaired_obj: Any = None
+        try:
+            try:
+                from json_repair import repair_json as _repair_json
+            except Exception:  # pragma: no cover - optional dependency not installed
+                _repair_json = None
+
+            if _repair_json is not None:
+                repaired_str = _repair_json(truncated)
+                repaired_obj = json.loads(repaired_str)
+        except Exception:
+            repaired_obj = None
+
+        if repaired_obj is not None:
+            return repaired_obj
+
+        # Fallback: safe container preserving a preview
+        return {"__truncated__": True, "preview": truncated}
+    except Exception:
+        # If anything goes wrong, return original content as-is
+        return content
+
+
+def process_content_for_max_len(content: Any, max_len: int) -> Any:
+    """Ensure the content fits within max_len when serialized.
+
+    - If content is a string: truncate to max_len.
+    - If content is a dict or list: attempt truncate+repair to keep it valid JSON.
+    """
+    try:
+        if isinstance(content, str):
+            return truncate_string(content, max_len)
+        if isinstance(content, (dict, list)):
+            return truncate_and_repair_json(content, max_len)
+        return content
+    except Exception:
+        return content

netra/version.py

@@ -1 +1 @@
-__version__ = "0.1.41"
+__version__ = "0.1.43"

{netra_sdk-0.1.41.dist-info → netra_sdk-0.1.43.dist-info}/METADATA

@@ -1,8 +1,9 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: netra-sdk
-Version: 0.1.41
+Version: 0.1.43
 Summary: A Python SDK for AI application observability that provides OpenTelemetry-based monitoring, tracing, and PII protection for LLM and vector database applications. Enables easy instrumentation, session tracking, and privacy-focused data collection for AI systems in production environments.
-License: Apache-2.0
+License-Expression: Apache-2.0
+License-File: LICENCE
 Keywords: netra,tracing,observability,sdk,ai,llm,vector,database
 Author: Sooraj Thomas
 Author-email: sooraj@keyvalue.systems
@@ -19,6 +20,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Typing :: Typed
 Provides-Extra: llm-guard
 Provides-Extra: presidio
+Requires-Dist: json-repair (==0.44.1)
 Requires-Dist: llm-guard (==0.3.16) ; extra == "llm-guard"
 Requires-Dist: opentelemetry-api (>=1.34.0,<2.0.0)
 Requires-Dist: opentelemetry-instrumentation-aio-pika (>=0.55b1,<1.0.0)
@@ -73,8 +75,8 @@ Requires-Dist: stanza (>=1.10.1,<2.0.0) ; extra == "presidio"
 Requires-Dist: traceloop-sdk (>=0.40.7,<0.43.0)
 Requires-Dist: transformers (==4.51.3) ; extra == "presidio"
 Project-URL: Changelog, https://github.com/KeyValueSoftwareSystems/netra-sdk-py/blob/main/CHANGELOG.md
-Project-URL: Documentation, https://github.com/KeyValueSoftwareSystems/netra-sdk-py/blob/main/README.md
-Project-URL: Homepage, https://github.com/KeyValueSoftwareSystems/netra-sdk-py
+Project-URL: Documentation, https://docs.getnetra.ai/introduction
+Project-URL: Homepage, https://getnetra.ai/
 Project-URL: Repository, https://github.com/KeyValueSoftwareSystems/netra-sdk-py
 Description-Content-Type: text/markdown
 

{netra_sdk-0.1.41.dist-info → netra_sdk-0.1.43.dist-info}/RECORD

@@ -1,9 +1,9 @@
-netra/__init__.py,sha256=imPVPlcpM85yPJNXMexewJ7p30blBbbY6hn8_lOrjlA,10365
+netra/__init__.py,sha256=-eEBq3ndX3JAWDo4_Ra0L19KrTfrJhNDjlAwChb3_IA,10353
 netra/anonymizer/__init__.py,sha256=KeGPPZqKVZbtkbirEKYTYhj6aZHlakjdQhD7QHqBRio,133
 netra/anonymizer/anonymizer.py,sha256=IcrYkdwWrFauGWUeAW-0RwrSUM8VSZCFNtoywZhvIqU,3778
 netra/anonymizer/base.py,sha256=ytPxHCUD2OXlEY6fNTuMmwImNdIjgj294I41FIgoXpU,5946
 netra/anonymizer/fp_anonymizer.py,sha256=_6svIYmE0eejdIMkhKBUWCNjGtGimtrGtbLvPSOp8W4,6493
-netra/config.py,sha256=51m8R0NoOrw58gMV7arOniEuFdJ7EIu3PNdFtIQ5xfg,6893
+netra/config.py,sha256=MMSAKrX_HCZ7QECjTocs_jsNFgcOnbP8aAVfOdq9YEs,7032
 netra/decorators.py,sha256=qZFHrwdj10FsTFqggo3XjdGB12aMxsrrDMMmslDqZ-0,17424
 netra/exceptions/__init__.py,sha256=uDgcBxmC4WhdS7HRYQk_TtJyxH1s1o6wZmcsnSHLAcM,174
 netra/exceptions/injection.py,sha256=ke4eUXRYUFJkMZgdSyPPkPt5PdxToTI6xLEBI0hTWUQ,1332
@@ -45,11 +45,12 @@ netra/processors/instrumentation_span_processor.py,sha256=VzurzwtGleFltxzKD_gjVk
 netra/processors/scrubbing_span_processor.py,sha256=dJ86Ncmjvmrhm_uAdGTwcGvRpZbVVWqD9AOFwEMWHZY,6701
 netra/processors/session_span_processor.py,sha256=qcsBl-LnILWefsftI8NQhXDGb94OWPc8LvzhVA0JS_c,2432
 netra/scanner.py,sha256=kyDpeZiscCPb6pjuhS-sfsVj-dviBFRepdUWh0sLoEY,11554
-netra/session_manager.py,sha256=wRGaZZQr1-OeG2ZpuiOAT_A53sQaLa-KHECZCXEc6gg,12492
+netra/session_manager.py,sha256=VzmSAiP63ODCuOWv-irsxyU2LvHoqjOBUuXtyxboBU0,13740
 netra/span_wrapper.py,sha256=IygQX78xQRlL_Z1MfKfUbv0okihx92qNClnRlYFtRNc,8004
 netra/tracer.py,sha256=FJO8Cine-WL9K_4wn6RVjQOgX6c1JCp_8QowUbRSVHk,7718
-netra/version.py,sha256=y9uk7vaddYq7x48m1qAG2qgbHu814vUzjs_fjnrQ3Es,23
-netra_sdk-0.1.41.dist-info/LICENCE,sha256=8B_UoZ-BAl0AqiHAHUETCgd3I2B9yYJ1WEQtVb_qFMA,11359
-netra_sdk-0.1.41.dist-info/METADATA,sha256=nhjMP6HyrFA9ZgJ3sF_lOywJKX8InO5Y75MmhDUah0c,28210
-netra_sdk-0.1.41.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-netra_sdk-0.1.41.dist-info/RECORD,,
+netra/utils.py,sha256=FblSzI8qMTfEbusakGBKE9CNELW0GEBHl09mPPxgI-w,2521
+netra/version.py,sha256=PwEipAcIgfEks4UpVgOERoR8Az-FCfCGocVnb0rkncM,23
+netra_sdk-0.1.43.dist-info/METADATA,sha256=3IMsH-b9Wcy9tMvG62MgOa3Zj76kidZbt1biWT3jyj0,28208
+netra_sdk-0.1.43.dist-info/WHEEL,sha256=M5asmiAlL6HEcOq52Yi5mmk9KmTVjY2RDPtO4p9DMrc,88
+netra_sdk-0.1.43.dist-info/licenses/LICENCE,sha256=8B_UoZ-BAl0AqiHAHUETCgd3I2B9yYJ1WEQtVb_qFMA,11359
+netra_sdk-0.1.43.dist-info/RECORD,,

{netra_sdk-0.1.41.dist-info → netra_sdk-0.1.43.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: poetry-core 2.2.0
 Root-Is-Purelib: true
 Tag: py3-none-any