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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: netra-sdk
-Version: 0.1.14
+Version: 0.1.16
 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
@@ -67,7 +67,7 @@ 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: traceloop-sdk (>=0.40.7,<0.41.0)
+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
 Project-URL: Homepage, https://github.com/KeyValueSoftwareSystems/netra-sdk-py

{netra_sdk-0.1.14 → netra_sdk-0.1.16}/netra/__init__.py

@@ -88,7 +88,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 +101,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 +114,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 +128,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 +142,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_sdk-0.1.14 → netra_sdk-0.1.16}/netra/instrumentation/openai/wrappers.py

@@ -12,6 +12,9 @@ from typing import Any, AsyncIterator, Callable, Dict, Iterator, Tuple
 
 from opentelemetry import context as context_api
 from opentelemetry.instrumentation.utils import _SUPPRESS_INSTRUMENTATION_KEY
+from opentelemetry.semconv_ai import (
+    SpanAttributes,
+)
 from opentelemetry.trace import Span, SpanKind, Tracer
 from opentelemetry.trace.status import Status, StatusCode
 from wrapt import ObjectProxy
@@ -55,34 +58,39 @@ def set_request_attributes(span: Span, kwargs: Dict[str, Any], operation_type: s
         return
 
     # Set operation type
-    span.set_attribute("llm.request.type", operation_type)
+    span.set_attribute(f"{SpanAttributes.LLM_REQUEST_TYPE}", operation_type)
 
     # Common attributes
     if kwargs.get("model"):
-        span.set_attribute("llm.request.model", kwargs["model"])
+        span.set_attribute(f"{SpanAttributes.LLM_REQUEST_MODEL}", kwargs["model"])
 
     if kwargs.get("temperature") is not None:
-        span.set_attribute("llm.request.temperature", kwargs["temperature"])
+        span.set_attribute(f"{SpanAttributes.LLM_REQUEST_TEMPERATURE}", kwargs["temperature"])
 
     if kwargs.get("max_tokens") is not None:
-        span.set_attribute("llm.request.max_tokens", kwargs["max_tokens"])
+        span.set_attribute(f"{SpanAttributes.LLM_REQUEST_MAX_TOKENS}", kwargs["max_tokens"])
 
     if kwargs.get("stream") is not None:
-        span.set_attribute("llm.stream", kwargs["stream"])
+        span.set_attribute("gen_ai.stream", kwargs["stream"])
 
     # Chat-specific attributes
     if operation_type == "chat" and kwargs.get("messages"):
         messages = kwargs["messages"]
         if isinstance(messages, list) and len(messages) > 0:
-            span.set_attribute("llm.prompts.0.role", messages[0].get("role", ""))
-            span.set_attribute("llm.prompts.0.content", str(messages[0].get("content", "")))
+            for index, message in enumerate(messages):
+                if hasattr(message, "content"):
+                    span.set_attribute(f"{SpanAttributes.LLM_PROMPTS}.{index}.role", "user")
+                    span.set_attribute(f"{SpanAttributes.LLM_PROMPTS}.{index}.content", message.content)
+                elif isinstance(message, dict):
+                    span.set_attribute(f"{SpanAttributes.LLM_PROMPTS}.{index}.role", message.get("role", "user"))
+                    span.set_attribute(f"{SpanAttributes.LLM_PROMPTS}.{index}.content", str(message.get("content", "")))
 
     # Response-specific attributes
     if operation_type == "response":
         if kwargs.get("instructions"):
-            span.set_attribute("llm.instructions", kwargs["instructions"])
+            span.set_attribute("gen_ai.instructions", kwargs["instructions"])
         if kwargs.get("input"):
-            span.set_attribute("llm.input", kwargs["input"])
+            span.set_attribute("gen_ai.input", kwargs["input"])
 
 
 def set_response_attributes(span: Span, response_dict: Dict[str, Any]) -> None:
@@ -91,33 +99,36 @@ def set_response_attributes(span: Span, response_dict: Dict[str, Any]) -> None:
         return
 
     if response_dict.get("model"):
-        span.set_attribute("llm.response.model", response_dict["model"])
+        span.set_attribute(f"{SpanAttributes.LLM_RESPONSE_MODEL}", response_dict["model"])
 
     if response_dict.get("id"):
-        span.set_attribute("llm.response.id", response_dict["id"])
+        span.set_attribute("gen_ai.response.id", response_dict["id"])
 
     # Usage information
     usage = response_dict.get("usage", {})
     if usage:
         if usage.get("prompt_tokens"):
-            span.set_attribute("llm.usage.prompt_tokens", usage["prompt_tokens"])
+            span.set_attribute(f"{SpanAttributes.LLM_USAGE_PROMPT_TOKENS}", usage["prompt_tokens"])
         if usage.get("completion_tokens"):
-            span.set_attribute("llm.usage.completion_tokens", usage["completion_tokens"])
+            span.set_attribute(f"{SpanAttributes.LLM_USAGE_COMPLETION_TOKENS}", usage["completion_tokens"])
+        if usage.get("cache_read_input_token"):
+            span.set_attribute(f"{SpanAttributes.LLM_USAGE_CACHE_READ_INPUT_TOKENS}", usage["cache_read_input_token"])
         if usage.get("total_tokens"):
-            span.set_attribute("llm.usage.total_tokens", usage["total_tokens"])
+            span.set_attribute(f"{SpanAttributes.LLM_USAGE_TOTAL_TOKENS}", usage["total_tokens"])
 
     # Response content
     choices = response_dict.get("choices", [])
-    if choices and len(choices) > 0:
-        first_choice = choices[0]
-        if first_choice.get("message", {}).get("content"):
-            span.set_attribute("llm.completions.0.content", first_choice["message"]["content"])
-        if first_choice.get("finish_reason"):
-            span.set_attribute("llm.completions.0.finish_reason", first_choice["finish_reason"])
+    for index, choice in enumerate(choices):
+        if choice.get("message", {}).get("role"):
+            span.set_attribute(f"{SpanAttributes.LLM_COMPLETIONS}.{index}.role", choice["message"]["role"])
+        if choice.get("message", {}).get("content"):
+            span.set_attribute(f"{SpanAttributes.LLM_COMPLETIONS}.{index}.content", choice["message"]["content"])
+        if choice.get("finish_reason"):
+            span.set_attribute(f"{SpanAttributes.LLM_COMPLETIONS}.{index}.finish_reason", choice["finish_reason"])
 
     # For responses.create
     if response_dict.get("output_text"):
-        span.set_attribute("llm.response.output_text", response_dict["output_text"])
+        span.set_attribute("gen_ai.response.output_text", response_dict["output_text"])
 
 
 def chat_wrapper(tracer: Tracer) -> Callable[..., Any]:

netra_sdk-0.1.16/netra/version.py

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

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

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "netra-sdk"
-version = "0.1.14"
+version = "0.1.16"
 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"}
@@ -32,7 +32,7 @@ dependencies = [
     "opentelemetry-api>=1.34.0,<2.0.0",
     "opentelemetry-sdk>=1.34.0,<2.0.0",
     "opentelemetry-instrumentation-fastapi>=0.55b1,<1.0.0",
-    "traceloop-sdk>=0.40.7,<0.41.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",

netra_sdk-0.1.14/netra/version.py

@@ -1,2 +0,0 @@
-__version__ = "0.1.14"
-