fix(google-genai): update logic for deciding whether to use awrap or wrap in the Google Generative AI Instrumentation (#3329)

ashwin153 · web-flow · commit 8fefb8728b7f · 2025-08-22T20:02:22.000+03:00
diff --git a/packages/opentelemetry-instrumentation-google-generativeai/opentelemetry/instrumentation/google_generativeai/__init__.py b/packages/opentelemetry-instrumentation-google-generativeai/opentelemetry/instrumentation/google_generativeai/__init__.py
@@ -281,7 +281,7 @@ def _instrument(self, **kwargs):
                 f"{wrap_object}.{wrap_method}",
                 (
                     _awrap(tracer, event_logger, wrapped_method)
-                    if wrap_method == "generate_content_async"
+                    if wrap_object == "AsyncModels"
                     else _wrap(tracer, event_logger, wrapped_method)
                 ),
             )

Original file line number	Diff line number	Diff line change
`@@ -281,7 +281,7 @@ def _instrument(self, **kwargs):`
`281`	`281`	`f"{wrap_object}.{wrap_method}",`
`282`	`282`	`(`
`283`	`283`	`_awrap(tracer, event_logger, wrapped_method)`
`284`		`- if wrap_method == "generate_content_async"`
	`284`	`+ if wrap_object == "AsyncModels"`
`285`	`285`	`else _wrap(tracer, event_logger, wrapped_method)`
`286`	`286`	`),`
`287`	`287`	`)`