diff --git a/agent/auxiliary_client.py b/agent/auxiliary_client.py
index 5a33d25cc7..aacc2c2a39 100644
--- a/agent/auxiliary_client.py
+++ b/agent/auxiliary_client.py
@@ -8546,6 +8546,7 @@ def call_llm(
             timeout=timeout,
             extra_body=extra_body,
             reasoning_config=reasoning_config,
+            extra_headers=extra_headers,
             api_mode=api_mode,
             stream=stream,
             stream_options=stream_options,
@@ -8590,6 +8591,7 @@ def _call_llm_impl(
     timeout: float = None,
     extra_body: dict = None,
     reasoning_config: Optional[dict] = None,
+    extra_headers: Optional[Dict[str, str]] = None,
     api_mode: str = None,
     stream: bool = False,
     stream_options: dict = None,