fix: serialization of response schema in invocation parameters for google-adk (Arize-ai#2033)

RogerHYang · web-flow · commit 586689f4ec6a · 2025-08-04T13:49:15.000-07:00
diff --git a/python/instrumentation/openinference-instrumentation-google-adk/src/openinference/instrumentation/google_adk/_wrappers.py b/python/instrumentation/openinference-instrumentation-google-adk/src/openinference/instrumentation/google_adk/_wrappers.py
@@ -357,7 +357,10 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> Iterator[tuple[str, AttributeV
 def _get_attributes_from_generate_content_config(
     obj: types.GenerateContentConfig,
 ) -> Iterator[tuple[str, AttributeValue]]:
-    yield SpanAttributes.LLM_INVOCATION_PARAMETERS, obj.model_dump_json(exclude_none=True)
+    yield (
+        SpanAttributes.LLM_INVOCATION_PARAMETERS,
+        obj.model_dump_json(exclude_none=True, fallback=_default),
+    )
 
 
 @stop_on_exception
@@ -536,6 +539,8 @@ def _default(obj: Any) -> Any:
 
     if isinstance(obj, BaseModel):
         return obj.model_dump(exclude_none=True)
+    if inspect.isclass(obj) and issubclass(obj, BaseModel):
+        return obj.model_json_schema()
     if isinstance(obj, bytes):
         return base64.b64encode(obj).decode()
     return str(obj)
