netra-sdk 0.1.46__tar.gz → 0.1.47__tar.gz

{netra_sdk-0.1.46 → netra_sdk-0.1.47}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: netra-sdk
-Version: 0.1.46
+Version: 0.1.47
 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-Expression: Apache-2.0
 License-File: LICENCE

{netra_sdk-0.1.46 → netra_sdk-0.1.47}/netra/__init__.py

@@ -183,11 +183,12 @@ class Netra:
             session_id: Session identifier
         """
         if not isinstance(session_id, str):
-            raise TypeError(f"session_id must be a string, got {type(session_id)}")
+            logger.error(f"set_session_id: session_id must be a string, got {type(session_id)}")
+            return
         if session_id:
             SessionManager.set_session_context("session_id", session_id)
         else:
-            logger.warning("Session ID must be provided for setting session_id.")
+            logger.warning("set_session_id: Session ID must be provided for setting session_id.")
 
     @classmethod
     def set_user_id(cls, user_id: str) -> None:
@@ -198,11 +199,12 @@ class Netra:
             user_id: User identifier
         """
         if not isinstance(user_id, str):
-            raise TypeError(f"user_id must be a string, got {type(user_id)}")
+            logger.error(f"set_user_id: user_id must be a string, got {type(user_id)}")
+            return
         if user_id:
             SessionManager.set_session_context("user_id", user_id)
         else:
-            logger.warning("User ID must be provided for setting user_id.")
+            logger.warning("set_user_id: User ID must be provided for setting user_id.")
 
     @classmethod
     def set_tenant_id(cls, tenant_id: str) -> None:
@@ -213,11 +215,12 @@ class Netra:
             user_account_id: User account identifier
         """
         if not isinstance(tenant_id, str):
-            raise TypeError(f"tenant_id must be a string, got {type(tenant_id)}")
+            logger.error(f"set_tenant_id: tenant_id must be a string, got {type(tenant_id)}")
+            return
         if tenant_id:
             SessionManager.set_session_context("tenant_id", tenant_id)
         else:
-            logger.warning("Tenant ID must be provided for setting tenant_id.")
+            logger.warning("set_tenant_id: Tenant ID must be provided for setting tenant_id.")
 
     @classmethod
     def set_custom_attributes(cls, key: str, value: Any) -> None:

{netra_sdk-0.1.46 → netra_sdk-0.1.47}/netra/decorators.py

@@ -285,7 +285,8 @@ def _create_function_wrapper(
             # Set span type if provided
 
             if not isinstance(as_type, SpanType):
-                raise ValueError(f"Invalid span type: {as_type}")
+                logger.error("Invalid span type: %s", as_type)
+                return
             try:
                 span.set_attribute("netra.span.type", as_type.value)
             except Exception:
@@ -344,7 +345,8 @@ def _create_function_wrapper(
             # Set span type if provided
             if as_type is not None:
                 if not isinstance(as_type, SpanType):
-                    raise ValueError(f"Invalid span type: {as_type}")
+                    logger.error("Invalid span type: %s", as_type)
+                    return
                 try:
                     span.set_attribute("netra.span.type", as_type.value)
                 except Exception:

{netra_sdk-0.1.46 → netra_sdk-0.1.47}/netra/span_wrapper.py

@@ -98,7 +98,8 @@ class SpanWrapper:
         if isinstance(as_type, SpanType):
             self.attributes["netra.span.type"] = as_type.value
         else:
-            raise ValueError(f"Invalid span type: {as_type}")
+            logger.error("Invalid span type: %s", as_type)
+            return
 
     def __enter__(self) -> "SpanWrapper":
         """Start the span wrapper, begin time tracking, and create OpenTelemetry span."""

netra_sdk-0.1.47/netra/tracer.py

@@ -0,0 +1,134 @@
+"""Netra OpenTelemetry tracer configuration module.
+
+This module handles the initialization and configuration of OpenTelemetry tracing,
+including exporter setup and span processor configuration.
+"""
+
+import logging
+import threading
+from typing import Any, Dict
+
+from opentelemetry import trace
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk import trace as sdk_trace
+from opentelemetry.sdk.resources import DEPLOYMENT_ENVIRONMENT, SERVICE_NAME, Resource
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import (
+    BatchSpanProcessor,
+    ConsoleSpanExporter,
+    SimpleSpanProcessor,
+)
+
+from netra.config import Config
+from netra.exporters import FilteringSpanExporter
+
+logger = logging.getLogger(__name__)
+
+_provider_install_lock = threading.Lock()
+
+
+class Tracer:
+    """
+    Configures Netra's OpenTelemetry tracer with OTLP exporter (or Console exporter as fallback)
+    and appropriate span processor.
+    """
+
+    def __init__(self, cfg: Config) -> None:
+        """Initialize the Netra tracer with the provided configuration.
+
+        Args:
+            cfg: Configuration object with tracer settings
+        """
+        self.cfg = cfg
+        self._setup_tracer()
+
+    def _setup_tracer(self) -> None:
+        """Set up the OpenTelemetry tracer with appropriate exporters and processors.
+
+        Creates a resource with service name and custom attributes,
+        configures the appropriate exporter (OTLP or Console fallback),
+        and sets up either a batch or simple span processor based on configuration.
+        """
+        # Create Resource with service.name + custom attributes
+        resource_attrs: Dict[str, Any] = {
+            SERVICE_NAME: self.cfg.app_name,
+            DEPLOYMENT_ENVIRONMENT: self.cfg.environment,
+        }
+        if self.cfg.resource_attributes:
+            resource_attrs.update(self.cfg.resource_attributes)
+        resource = Resource(attributes=resource_attrs)
+
+        # Build TracerProvider
+        current_provider = trace.get_tracer_provider()
+        if isinstance(current_provider, sdk_trace.TracerProvider):
+            provider = current_provider
+            logger.info("Reusing existing TracerProvider. Possible loss of Resource attributes")
+        else:
+            provider = TracerProvider(resource=resource)
+            trace.set_tracer_provider(provider)
+            logger.info("Using Netra TracerProvider")
+
+        with _provider_install_lock:
+            if getattr(provider, "_netra_processors_installed", False):
+                logger.info("Netra processors already installed on provider; skipping setup")
+                return
+
+            if not self.cfg.otlp_endpoint:
+                logger.warning("OTLP endpoint not provided, falling back to console exporter")
+                exporter = ConsoleSpanExporter()
+            else:
+                exporter = OTLPSpanExporter(
+                    endpoint=self._format_endpoint(self.cfg.otlp_endpoint),
+                    headers=self.cfg.headers,
+                )
+            original_exporter = exporter
+            try:
+                patterns = getattr(self.cfg, "blocked_spans", None) or []
+                exporter = FilteringSpanExporter(exporter, patterns)
+                if patterns:
+                    logger.info("Enabled FilteringSpanExporter with %d global pattern(s)", len(patterns))
+                else:
+                    logger.info("Enabled FilteringSpanExporter with local-only rules")
+            except (ValueError, TypeError) as e:
+                logger.warning("Failed to enable FilteringSpanExporter: %s; using unwrapped exporter", e)
+                exporter = original_exporter
+
+            from netra.processors import (
+                InstrumentationSpanProcessor,
+                LocalFilteringSpanProcessor,
+                ScrubbingSpanProcessor,
+                SessionSpanProcessor,
+            )
+
+            provider.add_span_processor(LocalFilteringSpanProcessor())
+            provider.add_span_processor(InstrumentationSpanProcessor())
+            provider.add_span_processor(SessionSpanProcessor())
+
+            if self.cfg.enable_scrubbing:
+                provider.add_span_processor(ScrubbingSpanProcessor())  # type: ignore[no-untyped-call]
+
+            if self.cfg.disable_batch:
+                provider.add_span_processor(SimpleSpanProcessor(exporter))
+            else:
+                provider.add_span_processor(BatchSpanProcessor(exporter))
+
+            setattr(provider, "_netra_processors_installed", True)
+
+            logger.info(
+                "Netra initialized: endpoint=%s, disable_batch=%s",
+                self.cfg.otlp_endpoint,
+                self.cfg.disable_batch,
+            )
+
+    def _format_endpoint(self, endpoint: str) -> str:
+        """Format the OTLP endpoint URL to ensure it ends with '/v1/traces'.
+
+        Args:
+            endpoint: Base OTLP endpoint URL
+
+        Returns:
+            Properly formatted endpoint URL
+        """
+        if not endpoint.endswith("/v1/traces"):
+            return endpoint.rstrip("/") + "/v1/traces"
+        return endpoint

netra_sdk-0.1.47/netra/version.py

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

{netra_sdk-0.1.46 → netra_sdk-0.1.47}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "netra-sdk"
-version = "0.1.46"
+version = "0.1.47"
 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.46/netra/tracer.py

@@ -1,121 +0,0 @@
-"""Netra OpenTelemetry tracer configuration module.
-
-This module handles the initialization and configuration of OpenTelemetry tracing,
-including exporter setup and span processor configuration.
-"""
-
-import logging
-from typing import Any, Dict
-
-from opentelemetry import trace
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
-from opentelemetry.sdk.resources import DEPLOYMENT_ENVIRONMENT, SERVICE_NAME, Resource
-from opentelemetry.sdk.trace import TracerProvider
-from opentelemetry.sdk.trace.export import (
-    BatchSpanProcessor,
-    ConsoleSpanExporter,
-    SimpleSpanProcessor,
-)
-
-from netra.config import Config
-from netra.exporters import FilteringSpanExporter
-
-logger = logging.getLogger(__name__)
-
-
-class Tracer:
-    """
-    Configures Netra's OpenTelemetry tracer with OTLP exporter (or Console exporter as fallback)
-    and appropriate span processor.
-    """
-
-    def __init__(self, cfg: Config) -> None:
-        """Initialize the Netra tracer with the provided configuration.
-
-        Args:
-            cfg: Configuration object with tracer settings
-        """
-        self.cfg = cfg
-        self._setup_tracer()
-
-    def _setup_tracer(self) -> None:
-        """Set up the OpenTelemetry tracer with appropriate exporters and processors.
-
-        Creates a resource with service name and custom attributes,
-        configures the appropriate exporter (OTLP or Console fallback),
-        and sets up either a batch or simple span processor based on configuration.
-        """
-        # Create Resource with service.name + custom attributes
-        resource_attrs: Dict[str, Any] = {
-            SERVICE_NAME: self.cfg.app_name,
-            DEPLOYMENT_ENVIRONMENT: self.cfg.environment,
-        }
-        if self.cfg.resource_attributes:
-            resource_attrs.update(self.cfg.resource_attributes)
-        resource = Resource(attributes=resource_attrs)
-
-        # Build TracerProvider
-        provider = TracerProvider(resource=resource)
-
-        # Configure exporter based on configuration
-        if not self.cfg.otlp_endpoint:
-            logger.warning("OTLP endpoint not provided, falling back to console exporter")
-            exporter = ConsoleSpanExporter()
-        else:
-            exporter = OTLPSpanExporter(
-                endpoint=self._format_endpoint(self.cfg.otlp_endpoint),
-                headers=self.cfg.headers,
-            )
-        # Wrap exporter with filtering to support both global and local (baggage-based) rules
-        try:
-            patterns = getattr(self.cfg, "blocked_spans", None) or []
-            exporter = FilteringSpanExporter(exporter, patterns)
-            if patterns:
-                logger.info("Enabled FilteringSpanExporter with %d global pattern(s)", len(patterns))
-            else:
-                logger.info("Enabled FilteringSpanExporter with local-only rules")
-        except Exception as e:
-            logger.warning("Failed to enable FilteringSpanExporter: %s", e)
-        # Add span processors: first instrumentation wrapper, then session processor
-        from netra.processors import (
-            InstrumentationSpanProcessor,
-            LocalFilteringSpanProcessor,
-            ScrubbingSpanProcessor,
-            SessionSpanProcessor,
-        )
-
-        # Apply local filtering propagation first so later processors and spans see attributes
-        provider.add_span_processor(LocalFilteringSpanProcessor())
-        provider.add_span_processor(InstrumentationSpanProcessor())
-        provider.add_span_processor(SessionSpanProcessor())
-
-        # Add scrubbing processor if enabled
-        if self.cfg.enable_scrubbing:
-            provider.add_span_processor(ScrubbingSpanProcessor())  # type: ignore[no-untyped-call]
-
-        # Install appropriate span processor
-        if self.cfg.disable_batch:
-            provider.add_span_processor(SimpleSpanProcessor(exporter))
-        else:
-            provider.add_span_processor(BatchSpanProcessor(exporter))
-
-        # Set global tracer provider
-        trace.set_tracer_provider(provider)
-        logger.info(
-            "Netra TracerProvider initialized: endpoint=%s, disable_batch=%s",
-            self.cfg.otlp_endpoint,
-            self.cfg.disable_batch,
-        )
-
-    def _format_endpoint(self, endpoint: str) -> str:
-        """Format the OTLP endpoint URL to ensure it ends with '/v1/traces'.
-
-        Args:
-            endpoint: Base OTLP endpoint URL
-
-        Returns:
-            Properly formatted endpoint URL
-        """
-        if not endpoint.endswith("/v1/traces"):
-            return endpoint.rstrip("/") + "/v1/traces"
-        return endpoint

netra_sdk-0.1.46/netra/version.py

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