diff --git a/tests/tools/test_managed_media_gateways.py b/tests/tools/test_managed_media_gateways.py index d7d35e4b7b69..6dc76374d5ab 100644 --- a/tests/tools/test_managed_media_gateways.py +++ b/tests/tools/test_managed_media_gateways.py @@ -169,6 +169,7 @@ def _install_fake_openai_module(captured, transcription_response=None): APIError=Exception, APIConnectionError=Exception, APITimeoutError=Exception, + BadRequestError=type("BadRequestError", (Exception,), {}), ) sys.modules["openai"] = fake_module