telemetry-dev-litellm 0.2.0__tar.gz → 0.2.2__tar.gz

{telemetry_dev_litellm-0.2.0 → telemetry_dev_litellm-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: telemetry-dev-litellm
-Version: 0.2.0
+Version: 0.2.2
 Summary: LiteLLM integration for telemetry.dev Python SDK
 Keywords: telemetry,opentelemetry,llm,genai,litellm,tracing
 Author: telemetry.dev
@@ -14,7 +14,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Typing :: Typed
-Requires-Dist: telemetry-dev>=0.2.9
+Requires-Dist: telemetry-dev>=0.3.0
 Requires-Dist: litellm>=1.90.2,<2.0
 Requires-Dist: opentelemetry-api>=1.39.0,<2
 Requires-Python: >=3.10, <3.14

{telemetry_dev_litellm-0.2.0 → telemetry_dev_litellm-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "telemetry-dev-litellm"
-version = "0.2.0"
+version = "0.2.2"
 description = "LiteLLM integration for telemetry.dev Python SDK"
 readme = "README.md"
 license = "MIT"
@@ -25,7 +25,7 @@ classifiers = [
     "Typing :: Typed",
 ]
 dependencies = [
-    "telemetry-dev>=0.2.9",
+    "telemetry-dev>=0.3.0",
     "litellm>=1.90.2,<2.0",
     "opentelemetry-api>=1.39.0,<2",
 ]

{telemetry_dev_litellm-0.2.0 → telemetry_dev_litellm-0.2.2}/pyproject.toml.orig

@@ -1,6 +1,6 @@
 [project]
 name = "telemetry-dev-litellm"
-version = "0.2.0"
+version = "0.2.2"
 description = "LiteLLM integration for telemetry.dev Python SDK"
 readme = "README.md"
 license = "MIT"
@@ -19,7 +19,7 @@ classifiers = [
   "Typing :: Typed",
 ]
 dependencies = [
-  "telemetry-dev>=0.2.9",
+  "telemetry-dev>=0.3.0",
   "litellm>=1.90.2,<2.0",
   "opentelemetry-api>=1.39.0,<2",
 ]

{telemetry_dev_litellm-0.2.0 → telemetry_dev_litellm-0.2.2}/src/telemetry_dev_litellm/__init__.py

@@ -15,7 +15,7 @@ import litellm
 import telemetry_dev
 from opentelemetry import trace
 
-__version__ = "0.2.0"
+__version__ = "0.2.2"
 
 RequestMapper = Callable[[tuple[Any, ...], Mapping[str, Any]], tuple[str, dict[str, Any]]]
 ResponseMapper = Callable[[Any], dict[str, Any]]
@@ -455,6 +455,12 @@ def _output_type(response_format: Any) -> str | None:
     return kind
 
 
+def _prompt_attributes(messages: object) -> dict[str, Any]:
+    # Older cores have no prompt_attributes_for and record a prompt only inside with_prompt.
+    read = getattr(telemetry_dev, "prompt_attributes_for", None)
+    return cast("dict[str, Any]", read(messages)) if callable(read) else {}
+
+
 def _completion_request(
     args: tuple[Any, ...], kwargs: Mapping[str, Any]
 ) -> tuple[str, dict[str, Any]]:
@@ -478,6 +484,7 @@ def _completion_request(
             "presence_penalty": _number(kwargs.get("presence_penalty")),
             "output_type": _output_type(kwargs.get("response_format")),
             "metadata": _request_metadata(kwargs),
+            "attributes": _prompt_attributes(_arg(args, kwargs, "messages", 1)) or None,
         },
     )
 
@@ -545,7 +552,10 @@ def _responses_request(
     input_budget: telemetry_dev.CaptureBudget | None = None
     if capture_input:
         input_value, input_budget = _bounded_responses_native(_arg(args, kwargs, "input", 0))
-    attributes: dict[str, Any] = {"gen_ai.operation.name": "chat"}
+    attributes: dict[str, Any] = {
+        "gen_ai.operation.name": "chat",
+        **_prompt_attributes(_arg(args, kwargs, "input", 0)),
+    }
     if input_budget is not None and input_budget.truncated:
         attributes["telemetry.dev.capture.truncated"] = True
     return (