Fix last typecheck issues..

Author: DylanRussell

instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/__init__.py

@@ -43,7 +43,6 @@
 
 from typing import Any, Collection
 
-from typing_extensions import assert_never
 from wrapt import (
     wrap_function_wrapper,  # type: ignore[reportUnknownVariableType]
 )
@@ -152,7 +151,7 @@ def _instrument(self, **kwargs: Any):
                 sem_conv_opt_in_mode,
             )
         else:
-            assert_never(sem_conv_opt_in_mode)
+            raise RuntimeError(f"{sem_conv_opt_in_mode} mode not supported")
         for client_class, method_name, wrapper in _methods_to_wrap(
             method_wrappers
         ):

instrumentation-genai/opentelemetry-instrumentation-vertexai/src/opentelemetry/instrumentation/vertexai/patch.py

@@ -266,9 +266,9 @@ def generate_content(
                 response = None
                 try:
                     response = wrapped(*args, **kwargs)
+                    return response
                 finally:
                     handle_response(response)
-                return response
 
     async def agenerate_content(
         self,
@@ -303,6 +303,6 @@ async def agenerate_content(
                 response = None
                 try:
                     response = await wrapped(*args, **kwargs)
+                    return response
                 finally:
                     handle_response(response)
-                return response