netra-sdk 0.1.16__tar.gz → 0.1.17__tar.gz

{netra_sdk-0.1.16 → netra_sdk-0.1.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: netra-sdk
-Version: 0.1.16
+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.16 → netra_sdk-0.1.17}/README.md

@@ -32,6 +32,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:
@@ -56,6 +71,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")
@@ -67,7 +83,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},
 )
 ```
 
@@ -497,7 +514,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={
@@ -545,9 +562,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 → netra_sdk-0.1.17}/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_sdk-0.1.16 → netra_sdk-0.1.17}/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_sdk-0.1.17/netra/version.py

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

{netra_sdk-0.1.16 → netra_sdk-0.1.17}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "netra-sdk"
-version = "0.1.16"
+version = "0.1.17"
 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"}
@@ -33,7 +33,6 @@ dependencies = [
     "opentelemetry-sdk>=1.34.0,<2.0.0",
     "opentelemetry-instrumentation-fastapi>=0.55b1,<1.0.0",
     "traceloop-sdk>=0.40.7,<0.43.0",
-    "presidio-analyzer>=2.2.358,<3.0.0",
     "opentelemetry-instrumentation-httpx>=0.55b1,<1.0.0",
     "opentelemetry-instrumentation-aiohttp-client>=0.55b1,<1.0.0",
     "opentelemetry-instrumentation-aio-pika>=0.55b1,<1.0.0",
@@ -91,7 +90,11 @@ include = "netra"
 
 [project.optional-dependencies]
 llm_guard = [
-    "llm-guard>=0.3.16,<0.4.0",
+    "llm-guard==0.3.16",
+]
+presidio = [
+    "presidio-analyzer==2.2.358",
+    "presidio-anonymizer==2.2.358",
 ]
 
 [tool.poetry.group.dev.dependencies]

netra_sdk-0.1.16/netra/version.py

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