netra-sdk 0.1.16__py3-none-any.whl → 0.1.18__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,

netra/anonymizer/anonymizer.py

@@ -8,7 +8,13 @@ texts while maintaining privacy.
 
 from typing import Callable, List, Optional
 
-from presidio_analyzer.recognizer_result import RecognizerResult
+try:
+    from presidio_analyzer.recognizer_result import RecognizerResult
+except Exception:
+    raise ImportError(
+        "PII Detetcion requires the 'presidio' packages: Install them explicitly as they are not available with the base SDK. Use pip install 'netra-sdk[presidio]' to install them."
+    )
+
 
 from .base import AnonymizationResult, BaseAnonymizer
 from .fp_anonymizer import FormatPreservingEmailAnonymizer

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 +1 @@
-__version__ = "0.1.16"
+__version__ = "0.1.18"

{netra_sdk-0.1.16.dist-info → netra_sdk-0.1.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: netra-sdk
-Version: 0.1.16
+Version: 0.1.18
 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,9 +149,10 @@ 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")
+Netra.init(app_name="Your application name", instruments={InstrumentSet.OPENAI, InstrumentSet.ANTHROPIC})
 
 # Or with custom configuration
 api_key = "Your API key"
@@ -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},
 )
 ```
 
@@ -375,9 +394,10 @@ Track user sessions and add custom context:
 
 ```python
 from netra import Netra
+from netra.instrumentation.instruments import InstrumentSet
 
 # Initialize SDK
-Netra.init(app_name="My App")
+Netra.init(app_name="My App", instruments={InstrumentSet.OPENAI})
 
 # Set session identification
 Netra.set_session_id("unique-session-id")
@@ -555,13 +575,15 @@ You can also configure the SDK programmatically when initializing:
 
 ```python
 from netra import Netra
+from netra.instrumentation.instruments import InstrumentSet
 
 Netra.init(
     app_name="my-ai-service",
     environment="production",
     resource_attributes={"team": "ai", "version": "1.0.0"},
     trace_content=True,
-    disable_batch=False
+    disable_batch=False,
+    instruments={InstrumentSet.OPENAI}
 )
 ```
 
@@ -573,7 +595,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 +643,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.16.dist-info → netra_sdk-0.1.18.dist-info}/RECORD

@@ -1,6 +1,6 @@
-netra/__init__.py,sha256=8N0U8USfoep2rEnQEPyyFrd1nmuMgpmTSSTyugaW528,5376
+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/anonymizer.py,sha256=IcrYkdwWrFauGWUeAW-0RwrSUM8VSZCFNtoywZhvIqU,3778
 netra/anonymizer/base.py,sha256=ytPxHCUD2OXlEY6fNTuMmwImNdIjgj294I41FIgoXpU,5946
 netra/anonymizer/fp_anonymizer.py,sha256=_6svIYmE0eejdIMkhKBUWCNjGtGimtrGtbLvPSOp8W4,6493
 netra/config.py,sha256=S9GsCvwtakrmryAaV-AhyVB_wAQ6tjwPLLZQemLgXko,5006
@@ -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=yF88-8vL8keLe6gCTumymw0UoMkWkSrJnzLru4zBCLQ,23
-netra_sdk-0.1.16.dist-info/LICENCE,sha256=8B_UoZ-BAl0AqiHAHUETCgd3I2B9yYJ1WEQtVb_qFMA,11359
-netra_sdk-0.1.16.dist-info/METADATA,sha256=5Cq-WcgwKPAsUaikTeP1qjO4MixYZ_VhLXSTvNJQpf0,25352
-netra_sdk-0.1.16.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-netra_sdk-0.1.16.dist-info/RECORD,,
+netra/version.py,sha256=6BiuMUkhwQp6bzUZSF8np8F1NwCltEtK0sPBF__tepU,23
+netra_sdk-0.1.18.dist-info/LICENCE,sha256=8B_UoZ-BAl0AqiHAHUETCgd3I2B9yYJ1WEQtVb_qFMA,11359
+netra_sdk-0.1.18.dist-info/METADATA,sha256=xANrdroAsDTktKVUyaTO6-c6r9Pja6XIMda5-Zhn0Zo,26133
+netra_sdk-0.1.18.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+netra_sdk-0.1.18.dist-info/RECORD,,