netra-sdk 0.1.5__tar.gz → 0.1.6__tar.gz

{netra_sdk-0.1.5 → netra_sdk-0.1.6}/PKG-INFO

@@ -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 → netra_sdk-0.1.6}/README.md

@@ -331,29 +331,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 → netra_sdk-0.1.6}/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_sdk-0.1.6/netra/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.6"

{netra_sdk-0.1.5 → netra_sdk-0.1.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "netra-sdk"
-version = "0.1.5"
+version = "0.1.6"
 description = "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."
 authors = [
     {name = "Sooraj Thomas",email = "sooraj@keyvalue.systems"}

netra_sdk-0.1.5/netra/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.5"