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

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/span_wrapper.py

@@ -1,5 +1,6 @@
 import json
 import logging
+import re
 import time
 from datetime import datetime, timezone
 from typing import Any, Dict, List, Literal, Optional
@@ -8,7 +9,7 @@ from opentelemetry import context as context_api
 from opentelemetry import trace
 from opentelemetry.trace import SpanKind, Status, StatusCode
 from opentelemetry.trace.propagation import set_span_in_context
-from pydantic import BaseModel
+from pydantic import BaseModel, field_validator
 
 from netra.config import Config
 
@@ -25,6 +26,21 @@ class ActionModel(BaseModel):  # type: ignore[misc]
     affected_records: Optional[List[Dict[str, str]]] = None
     metadata: Optional[Dict[str, str]] = None
 
+    @field_validator("start_time")  # type: ignore[misc]
+    @classmethod
+    def validate_time_format(cls, value: str) -> str:
+        """Validate that start_time is in ISO 8601 format with microseconds and Z suffix."""
+        # Pattern for ISO 8601 with microseconds: YYYY-MM-DDTHH:MM:SS.ffffffZ
+        pattern = r"^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\.\d{6}Z$"
+
+        if not re.match(pattern, value):
+            raise ValueError(
+                f"start_time must be in ISO 8601 format with microseconds: "
+                f"YYYY-MM-DDTHH:MM:SS.ffffffZ (e.g., 2025-07-18T14:30:45.123456Z). "
+                f"Got: {value}"
+            )
+        return value
+
 
 class UsageModel(BaseModel):  # type: ignore[misc]
     model: str

netra/version.py

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: netra-sdk
-Version: 0.1.13
+Version: 0.1.15
 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
@@ -482,7 +482,7 @@ Action tracking follows this schema:
 ```python
 [
     {
-        "start_time": str,            # Start time of the action
+        "start_time": str,            # Start time of the action in ISO 8601 format with microseconds and Z suffix (e.g., 2025-07-18T14:30:45.123456Z)
         "action": str,                # Type of action (e.g., "DB", "API", "CACHE")
         "action_type": str,           # Action subtype (e.g., "INSERT", "SELECT", "CALL")
         "affected_records": [         # Optional: List of records affected

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

@@ -29,7 +29,7 @@ netra/instrumentation/mistralai/utils.py,sha256=nhdIer5gJFxuGwg8FCT222hggDHeMQDh
 netra/instrumentation/mistralai/version.py,sha256=d6593s-XBNvVxri9lr2qLUDZQ3Zk3-VXHEwdb4pj8qA,22
 netra/instrumentation/openai/__init__.py,sha256=HztqLMw8Tf30-Ydqr4N7FcvAwj-5cnGZNqI-S3wIZ_4,5143
 netra/instrumentation/openai/version.py,sha256=_J-N1qG50GykJDM356BSQf0E8LoLbB8AaC3RKho494A,23
-netra/instrumentation/openai/wrappers.py,sha256=hXUgWKAs2_LCKIBnScoIJt_AkHhdQKnZWk7D94UjPGU,20685
+netra/instrumentation/openai/wrappers.py,sha256=4VQwIBLYaGovO9gE5TSMC-Ot84IaDuDhGqHndgR-Am4,21637
 netra/instrumentation/weaviate/__init__.py,sha256=EOlpWxobOLHYKqo_kMct_7nu26x1hr8qkeG5_h99wtg,4330
 netra/instrumentation/weaviate/version.py,sha256=PiCZHjonujPbnIn0KmD3Yl68hrjPRG_oKe5vJF3mmG8,24
 netra/pii.py,sha256=S7GnVzoNJEzKiUWnqN9bOCKPeNLsriztgB2E6Rx-yJU,27023
@@ -37,10 +37,10 @@ netra/processors/__init__.py,sha256=wfnSskRBtMT90hO7LqFJoEW374LgoH_gnTxhynqtByI,
 netra/processors/session_span_processor.py,sha256=qcsBl-LnILWefsftI8NQhXDGb94OWPc8LvzhVA0JS_c,2432
 netra/scanner.py,sha256=wqjMZnEbVvrGMiUSI352grUyHpkk94oBfHfMiXPhpGU,3866
 netra/session_manager.py,sha256=EVcnWcSj4NdkH--HmqHx0mmzivQiM4GCyFLu6lwi33M,6252
-netra/span_wrapper.py,sha256=BJvWMJWYX95NRMj9Yjs6ZL3BY6c5v1VGVN9i88VfewQ,7409
+netra/span_wrapper.py,sha256=DA5jjXkHBUJ8_mdlYP06rcZzFoSih4gdP71Wwr3btcQ,8104
 netra/tracer.py,sha256=In5QPVLz_6BxrolWpav9EuR9_hirD2UUIlyY75QUaKk,3450
-netra/version.py,sha256=PBhf_BIMfa1fmMCOO5V1SBLORGuuPOjZ5FtpjeZKVI4,24
-netra_sdk-0.1.13.dist-info/LICENCE,sha256=8B_UoZ-BAl0AqiHAHUETCgd3I2B9yYJ1WEQtVb_qFMA,11359
-netra_sdk-0.1.13.dist-info/METADATA,sha256=ZTfWpwu5b22-e_OCrRaTWut7CayCEtGNE6O0hKQFbNM,25266
-netra_sdk-0.1.13.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-netra_sdk-0.1.13.dist-info/RECORD,,
+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,,