netra-sdk 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

netra/processors/error_detection_processor.py

@@ -1,5 +1,5 @@
 import logging
-from typing import Any, Optional
+from typing import Any, Optional, Union
 
 import httpx
 from opentelemetry.sdk.trace import SpanProcessor
@@ -66,9 +66,13 @@ class ErrorDetectionProcessor(SpanProcessor):  # type: ignore[misc]
         # Wrap set_status
         original_set_status = span.set_status
 
-        def wrapped_set_status(status: Status) -> Any:
+        def wrapped_set_status(status: Union[Status, StatusCode]) -> Any:
             # Check if status code is ERROR
-            if status.status_code == StatusCode.ERROR:
+            if isinstance(status, Status):
+                status_code = status.status_code
+            elif isinstance(status, StatusCode):
+                status_code = status
+            if status_code == StatusCode.ERROR:
                 event_attributes = {
                     "has_error": True,
                 }

netra/version.py

@@ -1 +1 @@
-__version__ = "0.1.5"
+__version__ = "0.1.6"

{netra_sdk-0.1.5.dist-info → netra_sdk-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: netra-sdk
-Version: 0.1.5
+Version: 0.1.6
 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
 Keywords: netra,tracing,observability,sdk,ai,llm,vector,database
@@ -407,29 +407,38 @@ Netra.set_custom_event(event_name="conversion", attributes={
 Use the custom session tracking utility to track external API calls with detailed observability:
 
 ```python
-from netra import Netra
-
-# Initialize SDK
-Netra.init(app_name="My App")
-
-# Use session context manager for tracking API calls
-with Netra.start_session("video_generation_task") as session:
-    # Set attributes before the API call
-    session.set_prompt("A cat playing piano")
+from netra import Netra, Session
+from netra.session import UsageModel
+
+# Start a new session
+with Netra.start_session("image_generation") as session:
+    # Set session attributes
+    session.set_prompt("A beautiful sunset over mountains")
+    session.set_negative_prompt("blurry, low quality")
     session.set_height("1024")
     session.set_width("1024")
-    session.set_output_type("image")
-    session.set_model("stable-diffusion-xl")
-
-    # Make your external API call
-    result = external_api.generate_video(...)
-
-    # Set post-call attributes
-    session.set_completion_tokens("1250")
-    session.set_credits("30")
-    session.set_completion_tokens_cost("0.15")
-
-    # Add events during session
+    session.set_model("dall-e-3")
+    session.set_llm_system("openai")
+
+    # Set usage data with UsageModel
+    usage_data = [
+        UsageModel(
+            model="dall-e-3",
+            type="image_generation",
+            unit_used=1,
+            cost_in_usd=0.02
+        )
+    ]
+    session.set_usage(usage_data)
+
+    # Your API calls here
+    # ...
+
+    # Set custom attributes
+    session.set_attribute("custom_key", "custom_value")
+
+    # Add events
+    session.add_event("generation_started", {"step": 1, "status": "processing"})
     session.add_event("processing_completed", {"step": "rendering"})
 
 # Session automatically captures duration, status, and any errors

{netra_sdk-0.1.5.dist-info → netra_sdk-0.1.6.dist-info}/RECORD

@@ -29,14 +29,14 @@ netra/instrumentation/weaviate/__init__.py,sha256=EOlpWxobOLHYKqo_kMct_7nu26x1hr
 netra/instrumentation/weaviate/version.py,sha256=PiCZHjonujPbnIn0KmD3Yl68hrjPRG_oKe5vJF3mmG8,24
 netra/pii.py,sha256=S7GnVzoNJEzKiUWnqN9bOCKPeNLsriztgB2E6Rx-yJU,27023
 netra/processors/__init__.py,sha256=G16VumYTpgV4jsWrKNFSgm6xMQAsZ2Rrux25UVeo5YQ,215
-netra/processors/error_detection_processor.py,sha256=vlFQ8-Zgoj83htk_ciVhlT41RfU2GiGmDHGBzjKgtYQ,2701
+netra/processors/error_detection_processor.py,sha256=TtSZoJ7BCMHlVaXWYfqLHSZ6uIx43tdqYb7AFXpt0BA,2898
 netra/processors/session_span_processor.py,sha256=qcsBl-LnILWefsftI8NQhXDGb94OWPc8LvzhVA0JS_c,2432
 netra/scanner.py,sha256=wqjMZnEbVvrGMiUSI352grUyHpkk94oBfHfMiXPhpGU,3866
 netra/session.py,sha256=o1wXrPzMauqj_3P-iNBHVlcIR7zcKcbsmkrcHjQMKuY,7263
 netra/session_manager.py,sha256=EVcnWcSj4NdkH--HmqHx0mmzivQiM4GCyFLu6lwi33M,6252
 netra/tracer.py,sha256=9jAKdIHXbaZ6WV_p8I1syQiMdqXVCXMhpEhCBsbbci8,3538
-netra/version.py,sha256=rPSfWgIeq2YWVPyESOAwCBt8vftsTpIkuLAGDEzyRQc,22
-netra_sdk-0.1.5.dist-info/LICENCE,sha256=8B_UoZ-BAl0AqiHAHUETCgd3I2B9yYJ1WEQtVb_qFMA,11359
-netra_sdk-0.1.5.dist-info/METADATA,sha256=-82nUeU3ogSqDhIpMzYAUkjgS5emC5yy_jA_WTkpsPU,23518
-netra_sdk-0.1.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-netra_sdk-0.1.5.dist-info/RECORD,,
+netra/version.py,sha256=n3oM6B_EMz93NsTI18NNZd-jKFcUPzUkbIKj5VFK5ok,22
+netra_sdk-0.1.6.dist-info/LICENCE,sha256=8B_UoZ-BAl0AqiHAHUETCgd3I2B9yYJ1WEQtVb_qFMA,11359
+netra_sdk-0.1.6.dist-info/METADATA,sha256=5iMVB2PTJpnGBVB-JSsBF9AS5OzGUT2heIdfRoOXR1s,23730
+netra_sdk-0.1.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+netra_sdk-0.1.6.dist-info/RECORD,,