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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: netra-sdk
-Version: 0.1.6
+Version: 0.1.9
 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
@@ -402,46 +402,96 @@ Netra.set_custom_event(event_name="conversion", attributes={
     "value": 99.99
 })
 ```
-## 🔄 Custom Session Tracking
+## 🔄 Custom Span Tracking
 
-Use the custom session tracking utility to track external API calls with detailed observability:
+Use the custom span tracking utility to track external API calls with detailed observability:
 
 ```python
-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_model("dall-e-3")
-    session.set_llm_system("openai")
+from netra import Netra, UsageModel
+
+# Start a new span
+with Netra.start_span("image_generation") as span:
+    # Set span attributes
+    span.set_prompt("A beautiful sunset over mountains")
+    span.set_negative_prompt("blurry, low quality")
+    span.set_model("dall-e-3")
+    span.set_llm_system("openai")
 
     # Set usage data with UsageModel
     usage_data = [
         UsageModel(
             model="dall-e-3",
-            type="image_generation",
-            unit_used=1,
+            usage_type="image_generation",
+            units_used=1,
             cost_in_usd=0.02
         )
     ]
-    session.set_usage(usage_data)
+    span.set_usage(usage_data)
 
     # Your API calls here
     # ...
 
     # Set custom attributes
-    session.set_attribute("custom_key", "custom_value")
+    span.set_attribute("custom_key", "custom_value")
 
     # Add events
-    session.add_event("generation_started", {"step": 1, "status": "processing"})
-    session.add_event("processing_completed", {"step": "rendering"})
+    span.add_event("generation_started", {"step": "1", "status": "processing"})
+    span.add_event("processing_completed", {"step": "rendering"})
+
+    # Get the current active open telemetry span
+    current_span = span.get_current_span()
+
+    # Track database operations and other actions
+    action = ActionModel(
+        action="DB",
+        action_type="INSERT",
+        affected_records=[
+            {"record_id": "user_123", "record_type": "user"},
+            {"record_id": "profile_456", "record_type": "profile"}
+        ],
+        metadata={
+            "table": "users",
+            "operation_id": "tx_789",
+            "duration_ms": "45"
+        },
+        success=True
+    )
+    span.set_action([action])
+
+    # Record API calls
+    api_action = ActionModel(
+        action="API",
+        action_type="CALL",
+        metadata={
+            "endpoint": "/api/v1/process",
+            "method": "POST",
+            "status_code": 200,
+            "duration_ms": "120"
+        },
+        success=True
+    )
+    span.set_action([api_action])
+```
+
+### Action Tracking Schema
+
+Action tracking follows this schema:
 
-# Session automatically captures duration, status, and any errors
+```python
+[
+    {
+        "action": str,                # Type of action (e.g., "DB", "API", "CACHE")
+        "action_type": str,           # Action subtype (e.g., "INSERT", "SELECT", "CALL")
+        "affected_records": [         # Optional: List of records affected
+            {
+                "record_id": str,     # ID of the affected record
+                "record_type": str    # Type of the record
+            }
+        ],
+        "metadata": Dict[str, str],   # Additional metadata as key-value pairs
+        "success": bool              # Whether the action succeeded
+    }
+]
 ```
 
 ## 🔧 Advanced Configuration

{netra_sdk-0.1.6 → netra_sdk-0.1.9}/README.md

@@ -326,46 +326,96 @@ Netra.set_custom_event(event_name="conversion", attributes={
     "value": 99.99
 })
 ```
-## 🔄 Custom Session Tracking
+## 🔄 Custom Span Tracking
 
-Use the custom session tracking utility to track external API calls with detailed observability:
+Use the custom span tracking utility to track external API calls with detailed observability:
 
 ```python
-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_model("dall-e-3")
-    session.set_llm_system("openai")
+from netra import Netra, UsageModel
+
+# Start a new span
+with Netra.start_span("image_generation") as span:
+    # Set span attributes
+    span.set_prompt("A beautiful sunset over mountains")
+    span.set_negative_prompt("blurry, low quality")
+    span.set_model("dall-e-3")
+    span.set_llm_system("openai")
 
     # Set usage data with UsageModel
     usage_data = [
         UsageModel(
             model="dall-e-3",
-            type="image_generation",
-            unit_used=1,
+            usage_type="image_generation",
+            units_used=1,
             cost_in_usd=0.02
         )
     ]
-    session.set_usage(usage_data)
+    span.set_usage(usage_data)
 
     # Your API calls here
     # ...
 
     # Set custom attributes
-    session.set_attribute("custom_key", "custom_value")
+    span.set_attribute("custom_key", "custom_value")
 
     # Add events
-    session.add_event("generation_started", {"step": 1, "status": "processing"})
-    session.add_event("processing_completed", {"step": "rendering"})
+    span.add_event("generation_started", {"step": "1", "status": "processing"})
+    span.add_event("processing_completed", {"step": "rendering"})
+
+    # Get the current active open telemetry span
+    current_span = span.get_current_span()
+
+    # Track database operations and other actions
+    action = ActionModel(
+        action="DB",
+        action_type="INSERT",
+        affected_records=[
+            {"record_id": "user_123", "record_type": "user"},
+            {"record_id": "profile_456", "record_type": "profile"}
+        ],
+        metadata={
+            "table": "users",
+            "operation_id": "tx_789",
+            "duration_ms": "45"
+        },
+        success=True
+    )
+    span.set_action([action])
+
+    # Record API calls
+    api_action = ActionModel(
+        action="API",
+        action_type="CALL",
+        metadata={
+            "endpoint": "/api/v1/process",
+            "method": "POST",
+            "status_code": 200,
+            "duration_ms": "120"
+        },
+        success=True
+    )
+    span.set_action([api_action])
+```
+
+### Action Tracking Schema
+
+Action tracking follows this schema:
 
-# Session automatically captures duration, status, and any errors
+```python
+[
+    {
+        "action": str,                # Type of action (e.g., "DB", "API", "CACHE")
+        "action_type": str,           # Action subtype (e.g., "INSERT", "SELECT", "CALL")
+        "affected_records": [         # Optional: List of records affected
+            {
+                "record_id": str,     # ID of the affected record
+                "record_type": str    # Type of the record
+            }
+        ],
+        "metadata": Dict[str, str],   # Additional metadata as key-value pairs
+        "success": bool              # Whether the action succeeded
+    }
+]
 ```
 
 ## 🔧 Advanced Configuration

{netra_sdk-0.1.6 → netra_sdk-0.1.9}/netra/__init__.py

@@ -8,8 +8,8 @@ from .config import Config
 
 # Instrumentor functions
 from .instrumentation import init_instrumentations
-from .session import Session
 from .session_manager import SessionManager
+from .span_wrapper import ActionModel, SpanWrapper, UsageModel
 from .tracer import Tracer
 
 logger = logging.getLogger(__name__)
@@ -133,16 +133,16 @@ class Netra:
         SessionManager.set_custom_event(event_name, attributes)
 
     @classmethod
-    def start_session(
+    def start_span(
         cls,
         name: str,
         attributes: Optional[Dict[str, str]] = None,
         module_name: str = "combat_sdk",
-    ) -> Session:
+    ) -> SpanWrapper:
         """
         Start a new session.
         """
-        return Session(name, attributes, module_name)
+        return SpanWrapper(name, attributes, module_name)
 
 
-__all__ = ["Netra"]
+__all__ = ["Netra", "UsageModel", "ActionModel"]

{netra_sdk-0.1.6 → netra_sdk-0.1.9}/netra/config.py

@@ -51,7 +51,7 @@ class Config:
         if isinstance(headers, str):
             self.headers = parse_env_headers(headers)
 
-        if self.otlp_endpoint == "https://api.dev.getcombat.ai" and not self.api_key:
+        if self.otlp_endpoint in ["https://api.dev.getcombat.ai", "https://api.eu.getnetra.ai"] and not self.api_key:
             print("Error: Missing Netra API key, go to https://app.dev.getcombat.ai/api-key to create one")
             print("Set the NETRA_API_KEY environment variable to the key")
             return
@@ -59,7 +59,7 @@ class Config:
         # Handle API key authentication based on OTLP endpoint
         if self.api_key and self.otlp_endpoint:
             # For Netra endpoints, use x-api-key header
-            if "getcombat" in self.otlp_endpoint.lower():
+            if "getcombat" in self.otlp_endpoint.lower() or "getnetra" in self.otlp_endpoint.lower():
                 if not self.headers:
                     self.headers = {"x-api-key": self.api_key}
                 elif "x-api-key" not in self.headers:
@@ -86,6 +86,11 @@ class Config:
             env_tc = os.getenv("NETRA_TRACE_CONTENT")
             self.trace_content = False if (env_tc is not None and env_tc.lower() in ("0", "false")) else True
 
+        if not self.trace_content:
+            os.environ["TRACELOOP_TRACE_CONTENT"] = "false"
+        else:
+            os.environ["TRACELOOP_TRACE_CONTENT"] = "true"
+
         # 7. Environment: param override, else env
         if environment is not None:
             self.environment = environment

{netra_sdk-0.1.6 → netra_sdk-0.1.9}/netra/instrumentation/__init__.py

@@ -37,6 +37,7 @@ def init_instrumentations(
             Instruments.QDRANT,
             Instruments.GOOGLE_GENERATIVEAI,
             Instruments.MISTRAL,
+            Instruments.OPENAI,
         }
     )
     if instruments is not None and traceloop_instruments is None and traceloop_block_instruments is None:
@@ -82,6 +83,10 @@ def init_instrumentations(
     if CustomInstruments.MISTRALAI in netra_custom_instruments:
         init_mistral_instrumentor()
 
+    # Initialize OpenAI instrumentation.
+    if CustomInstruments.OPENAI in netra_custom_instruments:
+        init_openai_instrumentation()
+
     # Initialize aio_pika instrumentation.
     if CustomInstruments.AIO_PIKA in netra_custom_instruments:
         init_aio_pika_instrumentation()
@@ -282,7 +287,7 @@ def init_fastapi_instrumentation() -> bool:
     """
     try:
         if is_package_installed("fastapi"):
-            from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+            from netra.instrumentation.fastapi import FastAPIInstrumentor
 
             instrumentor = FastAPIInstrumentor()
             if not instrumentor.is_instrumented_by_opentelemetry:
@@ -416,6 +421,26 @@ def init_mistral_instrumentor() -> bool:
         return False
 
 
+def init_openai_instrumentation() -> bool:
+    """Initialize OpenAI instrumentation.
+
+    Returns:
+        bool: True if initialization was successful, False otherwise.
+    """
+    try:
+        if is_package_installed("openai"):
+            from netra.instrumentation.openai import NetraOpenAIInstrumentor
+
+            instrumentor = NetraOpenAIInstrumentor()
+            if not instrumentor.is_instrumented_by_opentelemetry:
+                instrumentor.instrument()
+        return True
+    except Exception as e:
+        logging.error(f"Error initializing OpenAI instrumentor: {e}")
+        Telemetry().log_exception(e)
+        return False
+
+
 def init_aio_pika_instrumentation() -> bool:
     """Initialize aio_pika instrumentation."""
     try: