def _instrument_requests()

in aidial_sdk/header_propagator.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

    def _instrument_requests(self):
        try:
            import requests
        except ImportError:
            return

        def instrumented_send(wrapped, instance, args, kwargs):
            request: requests.PreparedRequest = args[0]
            self._modify_headers(request.url or "", request.headers)
            return wrapped(*args, **kwargs)

        wrapt.wrap_function_wrapper(requests.Session, "send", instrumented_send)