netra-sdk 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

netra/__init__.py

@@ -2,7 +2,7 @@ import logging
 import threading
 from typing import Any, Dict, Optional, Set
 
-from netra.instrumentation.instruments import NetraInstruments
+from netra.instrumentation.instruments import InstrumentSet, NetraInstruments
 
 from .config import Config
 
@@ -70,6 +70,7 @@ class Netra:
 
             # Instrument all supported modules
             #    Pass trace_content flag to instrumentors that can capture prompts/completions
+
             init_instrumentations(
                 should_enrich_metrics=True,
                 base64_image_uploader=None,
@@ -88,7 +89,10 @@ class Netra:
         Args:
             session_id: Session identifier
         """
-        SessionManager.set_session_context("session_id", session_id)
+        if session_id:
+            SessionManager.set_session_context("session_id", session_id)
+        else:
+            logger.warning("Session ID must be provided for setting session_id.")
 
     @classmethod
     def set_user_id(cls, user_id: str) -> None:
@@ -98,7 +102,10 @@ class Netra:
         Args:
             user_id: User identifier
         """
-        SessionManager.set_session_context("user_id", user_id)
+        if user_id:
+            SessionManager.set_session_context("user_id", user_id)
+        else:
+            logger.warning("User ID must be provided for setting user_id.")
 
     @classmethod
     def set_tenant_id(cls, tenant_id: str) -> None:
@@ -108,7 +115,10 @@ class Netra:
         Args:
             user_account_id: User account identifier
         """
-        SessionManager.set_session_context("tenant_id", tenant_id)
+        if tenant_id:
+            SessionManager.set_session_context("tenant_id", tenant_id)
+        else:
+            logger.warning("Tenant ID must be provided for setting tenant_id.")
 
     @classmethod
     def set_custom_attributes(cls, key: str, value: Any) -> None:
@@ -119,7 +129,10 @@ class Netra:
             key: Custom attribute key
             value: Custom attribute value
         """
-        SessionManager.set_session_context("custom_attributes", {key: value})
+        if key and value:
+            SessionManager.set_session_context("custom_attributes", {key: value})
+        else:
+            logger.warning("Both key and value must be provided for custom attributes.")
 
     @classmethod
     def set_custom_event(cls, event_name: str, attributes: Any) -> None:
@@ -130,7 +143,10 @@ class Netra:
             event_name: Name of the custom event
             attributes: Attributes of the custom event
         """
-        SessionManager.set_custom_event(event_name, attributes)
+        if event_name and attributes:
+            SessionManager.set_custom_event(event_name, attributes)
+        else:
+            logger.warning("Both event_name and attributes must be provided for custom events.")
 
     @classmethod
     def start_span(

netra/instrumentation/__init__.py

@@ -31,6 +31,20 @@ def init_instrumentations(
                 netra_custom_block_instruments.add(getattr(CustomInstruments, instrument.name))
             else:
                 traceloop_block_instruments.add(getattr(Instruments, instrument.name))
+
+    # If no instruments are provided for instrumentation
+    if instruments is None:
+        traceloop_block_instruments = set(Instruments)
+        netra_custom_block_instruments = set(CustomInstruments)
+
+    # If only custom instruments from netra are provided for instrumentation
+    if instruments is not None and not traceloop_instruments and not traceloop_block_instruments:
+        traceloop_block_instruments = set(Instruments)
+
+    # If only traceloop instruments are provided for instrumentation
+    if instruments is not None and not netra_custom_instruments and not netra_custom_block_instruments:
+        netra_custom_block_instruments = set(CustomInstruments)
+
     traceloop_block_instruments.update(
         {
             Instruments.WEAVIATE,
@@ -40,8 +54,6 @@ def init_instrumentations(
             Instruments.OPENAI,
         }
     )
-    if instruments is not None and traceloop_instruments is None and traceloop_block_instruments is None:
-        traceloop_block_instruments = set(Instruments)
 
     init_instrumentations(
         should_enrich_metrics=should_enrich_metrics,

netra/version.py

@@ -1,2 +1 @@
-__version__ = "0.1.15"
-
+__version__ = "0.1.17"

{netra_sdk-0.1.15.dist-info → netra_sdk-0.1.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: netra-sdk
-Version: 0.1.15
+Version: 0.1.17
 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
@@ -18,7 +18,8 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Typing :: Typed
 Provides-Extra: llm-guard
-Requires-Dist: llm-guard (>=0.3.16,<0.4.0) ; extra == "llm-guard"
+Provides-Extra: presidio
+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)
 Requires-Dist: opentelemetry-instrumentation-aiohttp-client (>=0.55b1,<1.0.0)
@@ -66,7 +67,8 @@ Requires-Dist: opentelemetry-instrumentation-tortoiseorm (>=0.55b1,<1.0.0)
 Requires-Dist: opentelemetry-instrumentation-urllib (>=0.55b1,<1.0.0)
 Requires-Dist: opentelemetry-instrumentation-urllib3 (>=0.55b1,<1.0.0)
 Requires-Dist: opentelemetry-sdk (>=1.34.0,<2.0.0)
-Requires-Dist: presidio-analyzer (>=2.2.358,<3.0.0)
+Requires-Dist: presidio-analyzer (==2.2.358) ; extra == "presidio"
+Requires-Dist: presidio-anonymizer (==2.2.358) ; extra == "presidio"
 Requires-Dist: traceloop-sdk (>=0.40.7,<0.43.0)
 Project-URL: Bug Tracker, https://github.com/KeyValueSoftwareSystems/netra-sdk-py/issues
 Project-URL: Documentation, https://github.com/KeyValueSoftwareSystems/netra-sdk-py/blob/main/README.md
@@ -108,6 +110,21 @@ poetry add netra-sdk
 
 Netra SDK supports optional dependencies for enhanced functionality:
 
+#### Presidio for PII Detection
+To use the PII detection features provided by Netra SDK:
+
+```bash
+pip install 'netra-sdk[presidio]'
+```
+
+Or, using Poetry:
+
+```bash
+poetry add netra-sdk --extras "presidio"
+```
+
+
+
 #### LLM-Guard for Prompt Injection Protection
 
 To use the full functionality of prompt injection scanning provided by llm-guard:
@@ -132,6 +149,7 @@ Initialize the Netra SDK at the start of your application:
 
 ```python
 from netra import Netra
+from netra.instrumentation.instruments import InstrumentSet
 
 # Initialize with default settings
 Netra.init(app_name="Your application name")
@@ -143,7 +161,8 @@ Netra.init(
     app_name="Your application name",
     headers=headers,
     trace_content=True,
-    environment="Your Application environment"
+    environment="Your Application environment",
+    instruments={InstrumentSet.OPENAI, InstrumentSet.ANTHROPIC},
 )
 ```
 
@@ -573,7 +592,7 @@ Control which instrumentations are enabled:
 from netra import Netra
 from netra.instrumentation.instruments import InstrumentSet
 
-# Enable specific instruments only
+# Enable specific instruments
 Netra.init(
     app_name="Selective App",
     instruments={
@@ -621,9 +640,10 @@ export NETRA_HEADERS="authorization=Bearer your-token"
 
 ```python
 from netra import Netra
+from netra.instrumentation.instruments import InstrumentSet
 
 # Simple initialization - SDK automatically picks up environment variables
-Netra.init(app_name="Your App")
+Netra.init(app_name="Your App", instruments={InstrumentSet})
 # No endpoint configuration needed in code!
 ```
 

{netra_sdk-0.1.15.dist-info → netra_sdk-0.1.17.dist-info}/RECORD

@@ -1,4 +1,4 @@
-netra/__init__.py,sha256=0w42KiANH31q0sXFL2hLxRjueepfo_jbx2oMDc5miFo,4733
+netra/__init__.py,sha256=49R2bZrVu30HhpE3gfpu0b5C5cNovC6WyNyUwHCOfQ8,5392
 netra/anonymizer/__init__.py,sha256=KeGPPZqKVZbtkbirEKYTYhj6aZHlakjdQhD7QHqBRio,133
 netra/anonymizer/anonymizer.py,sha256=1VeYAsFpF_tYDlqJF-Q82-ZXGOR4YWBqrKUsRw3qOrA,3539
 netra/anonymizer/base.py,sha256=ytPxHCUD2OXlEY6fNTuMmwImNdIjgj294I41FIgoXpU,5946
@@ -9,7 +9,7 @@ netra/exceptions/__init__.py,sha256=uDgcBxmC4WhdS7HRYQk_TtJyxH1s1o6wZmcsnSHLAcM,
 netra/exceptions/injection.py,sha256=ke4eUXRYUFJkMZgdSyPPkPt5PdxToTI6xLEBI0hTWUQ,1332
 netra/exceptions/pii.py,sha256=MT4p_x-zH3VtYudTSxw1Z9qQZADJDspq64WrYqSWlZc,2438
 netra/input_scanner.py,sha256=bzP3s7YudGHQrIbUgQGrcIBEJ6CmOewzuYNSu75cVXM,4988
-netra/instrumentation/__init__.py,sha256=Mz7BlRp4zgKAD3oLBGfBzh5naVVgAP0sLw4ISDydhLw,39567
+netra/instrumentation/__init__.py,sha256=ZORhcso04LPD6-Y7oPsh74rY-jwz1unsfYZBcP--SzM,40081
 netra/instrumentation/aiohttp/__init__.py,sha256=M1kuF0R3gKY5rlbhEC1AR13UWHelmfokluL2yFysKWc,14398
 netra/instrumentation/aiohttp/version.py,sha256=Zy-0Aukx-HS_Mo3NKPWg-hlUoWKDzS0w58gLoVtJec8,24
 netra/instrumentation/cohere/__init__.py,sha256=3XwmCAZwZiMkHdNN3YvcBOLsNCx80ymbU31TyMzv1IY,17685
@@ -39,8 +39,8 @@ netra/scanner.py,sha256=wqjMZnEbVvrGMiUSI352grUyHpkk94oBfHfMiXPhpGU,3866
 netra/session_manager.py,sha256=EVcnWcSj4NdkH--HmqHx0mmzivQiM4GCyFLu6lwi33M,6252
 netra/span_wrapper.py,sha256=DA5jjXkHBUJ8_mdlYP06rcZzFoSih4gdP71Wwr3btcQ,8104
 netra/tracer.py,sha256=In5QPVLz_6BxrolWpav9EuR9_hirD2UUIlyY75QUaKk,3450
-netra/version.py,sha256=s5mBMy--qmLbYWG9g95dqUqvmR31MlOMaNfvabrDFlQ,24
-netra_sdk-0.1.15.dist-info/LICENCE,sha256=8B_UoZ-BAl0AqiHAHUETCgd3I2B9yYJ1WEQtVb_qFMA,11359
-netra_sdk-0.1.15.dist-info/METADATA,sha256=QkYln6aEvwGYpfto8eAE2rCV-7Lk-kgqWGWcWGFxfok,25352
-netra_sdk-0.1.15.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-netra_sdk-0.1.15.dist-info/RECORD,,
+netra/version.py,sha256=BzIjnki8Bz3evNWo6bjGxxpLhy_tN9MRYhtM0MnDiWs,23
+netra_sdk-0.1.17.dist-info/LICENCE,sha256=8B_UoZ-BAl0AqiHAHUETCgd3I2B9yYJ1WEQtVb_qFMA,11359
+netra_sdk-0.1.17.dist-info/METADATA,sha256=BR_VdbI6-h6x9-z23F0uX8Z9HtVcxRduPXC6UBhsrw4,25876
+netra_sdk-0.1.17.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+netra_sdk-0.1.17.dist-info/RECORD,,