openinference-instrumentation-google-adk 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

openinference/instrumentation/google_adk/_wrappers.py

@@ -25,6 +25,7 @@ from google.adk.models.llm_request import LlmRequest
 from google.adk.models.llm_response import LlmResponse
 from google.adk.tools.base_tool import BaseTool
 from google.genai import types
+from google.genai.types import Blob
 from opentelemetry import context as context_api
 from opentelemetry import trace as trace_api
 from opentelemetry.context import _SUPPRESS_INSTRUMENTATION_KEY
@@ -39,6 +40,7 @@ from openinference.instrumentation import (
     using_user,
 )
 from openinference.semconv.trace import (
+    ImageAttributes,
     MessageAttributes,
     MessageContentAttributes,
     OpenInferenceLLMProviderValues,
@@ -159,6 +161,7 @@ class _BaseAgentRunAsync(_WithTracer):
         name = f"agent_run [{instance.name}]"
         attributes = dict(get_attributes_from_context())
         attributes[SpanAttributes.OPENINFERENCE_SPAN_KIND] = OpenInferenceSpanKindValues.AGENT.value
+        attributes[SpanAttributes.AGENT_NAME] = instance.name
 
         class _AsyncGenerator(wrapt.ObjectProxy):  # type: ignore[misc]
             __wrapped__: AsyncGenerator[Event, None]
@@ -439,6 +442,27 @@ def _get_attributes_from_content(
         )
 
 
+def _get_attributes_from_inline_data(
+    inline_data: Blob, prefix: str = ""
+) -> Iterator[tuple[str, AttributeValue]]:
+    # inline_data is typically a Blob-like object with `.data` (bytes)
+    # and `.mime_type` (str). Encode the bytes as base64 and record
+    # it as an image content attribute, along with the mime type.
+    try:
+        mime_type = inline_data.mime_type
+        data = inline_data.data
+        if data and mime_type and "image" in mime_type:
+            image_url = f"data:{inline_data.mime_type};base64,{base64.b64encode(data).decode()}"
+            yield (
+                f"{prefix}{MessageContentAttributes.MESSAGE_CONTENT_IMAGE}.{ImageAttributes.IMAGE_URL}",
+                image_url,
+            )
+            yield f"{prefix}{MessageContentAttributes.MESSAGE_CONTENT_TYPE}", "image"
+
+    except Exception:
+        logger.debug("Failed to extract file data attributes.")
+
+
 @stop_on_exception
 def _get_attributes_from_parts(
     obj: Iterable[types.Part],
@@ -474,6 +498,9 @@ def _get_attributes_from_parts(
                     safe_json_dumps(function_response.response),
                 )
             message_index += 1
+        elif inline_data := part.inline_data:
+            prefix = f"{span_attribute}.{message_index}.{MESSAGE_CONTENTS}.{i}."
+            yield from _get_attributes_from_inline_data(inline_data, prefix)
 
 
 @stop_on_exception
@@ -554,3 +581,6 @@ def _default(obj: Any) -> Any:
     if isinstance(obj, bytes):
         return base64.b64encode(obj).decode()
     return str(obj)
+
+
+MESSAGE_CONTENTS = MessageAttributes.MESSAGE_CONTENTS

openinference/instrumentation/google_adk/version.py

@@ -1 +1 @@
-__version__ = "0.1.7"
+__version__ = "0.1.9"

{openinference_instrumentation_google_adk-0.1.7.dist-info → openinference_instrumentation_google_adk-0.1.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openinference-instrumentation-google-adk
-Version: 0.1.7
+Version: 0.1.9
 Summary: OpenInference Google ADK Instrumentation
 Project-URL: Homepage, https://github.com/Arize-ai/openinference/tree/main/python/instrumentation/openinference-instrumentation-google-adk
 Author-email: OpenInference Authors <oss@arize.com>

openinference_instrumentation_google_adk-0.1.9.dist-info/RECORD

@@ -0,0 +1,9 @@
+openinference/instrumentation/google_adk/__init__.py,sha256=sCrItt4ilGC3u5hHbjw8papCk_yhUbdur0pzAEg0SIY,7012
+openinference/instrumentation/google_adk/_wrappers.py,sha256=CWRgi4BqfmpkW04xd75xNRij_-8mdxiJz7eMrR5SOYM,22081
+openinference/instrumentation/google_adk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+openinference/instrumentation/google_adk/version.py,sha256=XIaxbMbyiP-L3kguR1GhxirFblTXiHR1lMfDVITvHUI,22
+openinference_instrumentation_google_adk-0.1.9.dist-info/METADATA,sha256=fwoPJQaJGApSy3HlP29NMVLWLdQ9_XK0Cd2r8dhPH68,5245
+openinference_instrumentation_google_adk-0.1.9.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+openinference_instrumentation_google_adk-0.1.9.dist-info/entry_points.txt,sha256=SW0GBmqra9efyXhsKjeP6OX4vS7BFpzGeHZgrg_LUzw,211
+openinference_instrumentation_google_adk-0.1.9.dist-info/licenses/LICENSE,sha256=x_Y7OX-lWxSUhmKoivlrD6WN3jWlIHxdug8BJv2SI34,10870
+openinference_instrumentation_google_adk-0.1.9.dist-info/RECORD,,

openinference_instrumentation_google_adk-0.1.7.dist-info/RECORD

@@ -1,9 +0,0 @@
-openinference/instrumentation/google_adk/__init__.py,sha256=sCrItt4ilGC3u5hHbjw8papCk_yhUbdur0pzAEg0SIY,7012
-openinference/instrumentation/google_adk/_wrappers.py,sha256=-HOKXlJJZ_xeIloaH22bVeuKd0OQBp51SWJ2glmlV80,20799
-openinference/instrumentation/google_adk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-openinference/instrumentation/google_adk/version.py,sha256=YpKDcdV7CqL8n45u267wKtyloM13FSVbOdrqgNZnSLM,22
-openinference_instrumentation_google_adk-0.1.7.dist-info/METADATA,sha256=Uz6lLP874V1VtNsYa-om4ezi5Jtpf6-jh5DdRc8B8Bc,5245
-openinference_instrumentation_google_adk-0.1.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-openinference_instrumentation_google_adk-0.1.7.dist-info/entry_points.txt,sha256=SW0GBmqra9efyXhsKjeP6OX4vS7BFpzGeHZgrg_LUzw,211
-openinference_instrumentation_google_adk-0.1.7.dist-info/licenses/LICENSE,sha256=x_Y7OX-lWxSUhmKoivlrD6WN3jWlIHxdug8BJv2SI34,10870
-openinference_instrumentation_google_adk-0.1.7.dist-info/RECORD,,