telemetry-dev-openai 0.2.1__tar.gz → 0.2.2__tar.gz

{telemetry_dev_openai-0.2.1 → telemetry_dev_openai-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: telemetry-dev-openai
-Version: 0.2.1
+Version: 0.2.2
 Summary: OpenAI integration for telemetry.dev Python SDK
 Keywords: telemetry,opentelemetry,openai,llm,genai,tracing
 Author: telemetry.dev

{telemetry_dev_openai-0.2.1 → telemetry_dev_openai-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "telemetry-dev-openai"
-version = "0.2.1"
+version = "0.2.2"
 description = "OpenAI integration for telemetry.dev Python SDK"
 readme = "README.md"
 license = "MIT"

{telemetry_dev_openai-0.2.1 → telemetry_dev_openai-0.2.2}/pyproject.toml.orig

@@ -1,6 +1,6 @@
 [project]
 name = "telemetry-dev-openai"
-version = "0.2.1"
+version = "0.2.2"
 description = "OpenAI integration for telemetry.dev Python SDK"
 readme = "README.md"
 license = "MIT"

{telemetry_dev_openai-0.2.1 → telemetry_dev_openai-0.2.2}/src/telemetry_dev_openai/__init__.py

@@ -20,7 +20,7 @@ from openai.resources.embeddings import AsyncEmbeddings, Embeddings
 from openai.resources.images import AsyncImages, Images
 from openai.resources.responses.responses import AsyncResponses, Responses
 
-__version__ = "0.2.1"
+__version__ = "0.2.2"
 
 ProviderResolver = Callable[[object | None], str]
 RequestMapper = Callable[[Mapping[str, Any]], tuple[str, dict[str, Any]]]
@@ -349,9 +349,18 @@ def _capture_text(value: str, budget: telemetry_dev.CaptureBudget) -> str:
     return value
 
 
+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 _chat_request(params: Mapping[str, Any]) -> tuple[str, dict[str, Any]]:
     model = _string(params.get("model"))
+    attributes = _prompt_attributes(params.get("messages"))
     captured, truncated = _bounded_responses_capture(params.get("messages"), "input")
+    if truncated:
+        attributes["telemetry.dev.capture.truncated"] = True
     return (
         f"chat {model or 'unknown'}",
         {
@@ -366,7 +375,7 @@ def _chat_request(params: Mapping[str, Any]) -> tuple[str, dict[str, Any]]:
             "seed": _number(params.get("seed")),
             "frequency_penalty": _number(params.get("frequency_penalty")),
             "presence_penalty": _number(params.get("presence_penalty")),
-            "attributes": {"telemetry.dev.capture.truncated": True} if truncated else None,
+            "attributes": attributes or None,
         },
     )
 
@@ -432,7 +441,10 @@ def _chat_response(response: Any) -> dict[str, Any]:
 
 def _responses_request(params: Mapping[str, Any]) -> tuple[str, dict[str, Any]]:
     model = _string(params.get("model"))
+    attributes = _prompt_attributes(params.get("input"))
     captured, truncated = _bounded_responses_capture(params.get("input"), "input")
+    if truncated:
+        attributes["telemetry.dev.capture.truncated"] = True
     return (
         f"chat {model or 'unknown'}",
         {
@@ -443,7 +455,7 @@ def _responses_request(params: Mapping[str, Any]) -> tuple[str, dict[str, Any]]:
             "temperature": _number(params.get("temperature")),
             "top_p": _number(params.get("top_p")),
             "max_tokens": _number(params.get("max_output_tokens")),
-            "attributes": {"telemetry.dev.capture.truncated": True} if truncated else None,
+            "attributes": attributes or None,
         },
     )