openinference-instrumentation-google-adk 0.1.2__tar.gz → 0.1.4__tar.gz

{openinference_instrumentation_google_adk-0.1.2 → openinference_instrumentation_google_adk-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openinference-instrumentation-google-adk
-Version: 0.1.2
+Version: 0.1.4
 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.2 → openinference_instrumentation_google_adk-0.1.4}/src/openinference/instrumentation/google_adk/_wrappers.py

@@ -357,7 +357,10 @@ def stop_on_exception(
 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)

openinference_instrumentation_google_adk-0.1.4/src/openinference/instrumentation/google_adk/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.4"

openinference_instrumentation_google_adk-0.1.2/src/openinference/instrumentation/google_adk/version.py

@@ -1 +0,0 @@
-__version__ = "0.1.2"