Handle hooks in instrument_client

Author: xrmx

instrumentation/opentelemetry-instrumentation-httpx/src/opentelemetry/instrumentation/httpx/__init__.py

@@ -192,6 +192,7 @@ async def async_response_hook(span, request, response):
 """
 import logging
 import typing
+from asyncio import iscoroutinefunction
 from types import TracebackType
 
 import httpx
@@ -926,6 +927,20 @@ def instrument_client(
             )
             return
 
+        if iscoroutinefunction(request_hook):
+            self._async_request_hook = request_hook
+            self._request_hook = None
+        else:
+            self._request_hook = request_hook
+            self._async_request_hook = None
+
+        if iscoroutinefunction(response_hook):
+            self._async_response_hook = response_hook
+            self._response_hook = None
+        else:
+            self._response_hook = response_hook
+            self._async_response_hook = None
+
         if hasattr(client._transport, "handle_request"):
             wrap_function_wrapper(
                 client._transport,