openinference-instrumentation-google-adk 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

openinference/instrumentation/google_adk/_wrappers.py

@@ -1,3 +1,4 @@
+import base64
 import inspect
 import json
 import logging
@@ -356,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
@@ -530,14 +534,13 @@ def bind_args_kwargs(func: Any, *args: Any, **kwargs: Any) -> OrderedDict[str, A
     return bound.arguments
 
 
-def _default(obj: Any) -> str:
+def _default(obj: Any) -> Any:
     from pydantic import BaseModel
 
     if isinstance(obj, BaseModel):
-        return json.dumps(
-            obj.model_dump(exclude=None),
-            ensure_ascii=False,
-            default=str,
-        )
-    else:
-        return str(obj)
+        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/version.py

@@ -1 +1 @@
-__version__ = "0.1.1"
+__version__ = "0.1.3"

{openinference_instrumentation_google_adk-0.1.1.dist-info → openinference_instrumentation_google_adk-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openinference-instrumentation-google-adk
-Version: 0.1.1
+Version: 0.1.3
 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.3.dist-info/RECORD

@@ -0,0 +1,9 @@
+openinference/instrumentation/google_adk/__init__.py,sha256=HxZ58OnPWAfiCeyEnZom7TmxNKc1kHhuGWkrwFYba2c,6922
+openinference/instrumentation/google_adk/_wrappers.py,sha256=Glf7qYKkVFlwcRFYlfaXne8aroZUrlqF5MEDJvTwR14,20295
+openinference/instrumentation/google_adk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+openinference/instrumentation/google_adk/version.py,sha256=XEqb2aiIn8fzGE68Mph4ck1FtQqsR_am0wRWvrYPffQ,22
+openinference_instrumentation_google_adk-0.1.3.dist-info/METADATA,sha256=jctLEKRSXcT6SJsxSRoq_xOg0xyii55znXejhGHnNF8,5194
+openinference_instrumentation_google_adk-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+openinference_instrumentation_google_adk-0.1.3.dist-info/entry_points.txt,sha256=SW0GBmqra9efyXhsKjeP6OX4vS7BFpzGeHZgrg_LUzw,211
+openinference_instrumentation_google_adk-0.1.3.dist-info/licenses/LICENSE,sha256=x_Y7OX-lWxSUhmKoivlrD6WN3jWlIHxdug8BJv2SI34,10870
+openinference_instrumentation_google_adk-0.1.3.dist-info/RECORD,,

openinference_instrumentation_google_adk-0.1.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-openinference/instrumentation/google_adk/__init__.py,sha256=HxZ58OnPWAfiCeyEnZom7TmxNKc1kHhuGWkrwFYba2c,6922
-openinference/instrumentation/google_adk/_wrappers.py,sha256=B-ORzJ7pkc-4IafoI7xzCZ8Mw_CtFQ6DXMbeZ3GTqzo,20162
-openinference/instrumentation/google_adk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-openinference/instrumentation/google_adk/version.py,sha256=rnObPjuBcEStqSO0S6gsdS_ot8ITOQjVj_-P1LUUYpg,22
-openinference_instrumentation_google_adk-0.1.1.dist-info/METADATA,sha256=sceuxu5UhQLSdL3M6g7F76vKpfZQVaoIFkhgyB0NXzs,5194
-openinference_instrumentation_google_adk-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-openinference_instrumentation_google_adk-0.1.1.dist-info/entry_points.txt,sha256=SW0GBmqra9efyXhsKjeP6OX4vS7BFpzGeHZgrg_LUzw,211
-openinference_instrumentation_google_adk-0.1.1.dist-info/licenses/LICENSE,sha256=x_Y7OX-lWxSUhmKoivlrD6WN3jWlIHxdug8BJv2SI34,10870
-openinference_instrumentation_google_adk-0.1.1.dist-info/RECORD,,