From c00e48a94a9999200c0cca0b3fa22c4c47fcc23d Mon Sep 17 00:00:00 2001 From: Willem Momma Date: Thu, 30 Oct 2025 15:17:50 +0100 Subject: [PATCH] [Bugfix] Fix OpenAI authentication header formatting for external openai-compatible providers --- genai_bench/user/openai_user.py | 3 ++- tests/user/test_openai_user.py | 5 +---- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/genai_bench/user/openai_user.py b/genai_bench/user/openai_user.py index fd74606..715a392 100644 --- a/genai_bench/user/openai_user.py +++ b/genai_bench/user/openai_user.py @@ -40,8 +40,9 @@ class OpenAIUser(BaseUser): def on_start(self): if not self.host or not self.auth_provider: raise ValueError("API key and base must be set for OpenAIUser.") + auth_headers = self.auth_provider.get_headers() self.headers = { - "Authorization": f"Bearer {self.auth_provider.get_credentials()}", + **auth_headers, "Content-Type": "application/json", } super().on_start() diff --git a/tests/user/test_openai_user.py b/tests/user/test_openai_user.py index ccb3761..d7db0bb 100644 --- a/tests/user/test_openai_user.py +++ b/tests/user/test_openai_user.py @@ -19,6 +19,7 @@ def mock_openai_user(): # Set up mock auth provider mock_auth = MagicMock() mock_auth.get_credentials.return_value = "fake_api_key" + mock_auth.get_headers.return_value = {"Authorization": "Bearer fake_api_key"} mock_auth.get_config.return_value = { "api_base": "http://example.com", "api_key": "fake_api_key", @@ -27,10 +28,6 @@ def mock_openai_user(): OpenAIUser.host = "http://example.com" user = OpenAIUser(environment=MagicMock()) - user.headers = { - "Authorization": "Bearer fake_api_key", - "Content-Type": "application/json", - } user.user_requests = [ UserChatRequest( model="gpt-3",