Adding extra_headers parameters to ModelSettings

Author: jonnyk20

src/agents/model_settings.py

@@ -3,7 +3,7 @@
 from dataclasses import dataclass, fields, replace
 from typing import Literal
 
-from openai._types import Body, Query
+from openai._types import Body, Query, Headers
 from openai.types.shared import Reasoning
 
 
@@ -67,6 +67,10 @@ class ModelSettings:
     """Additional body fields to provide with the request.
     Defaults to None if not provided."""
 
+    extra_headers: Headers | None = None
+    """Additional headers to provide with the request.
+    Defaults to None if not provided."""
+
     def resolve(self, override: ModelSettings | None) -> ModelSettings:
         """Produce a new ModelSettings by overlaying any non-None values from the
         override on top of this instance."""

src/agents/models/openai_chatcompletions.py

@@ -258,6 +258,7 @@ async def _fetch_response(
             extra_headers=HEADERS,
             extra_query=model_settings.extra_query,
             extra_body=model_settings.extra_body,
+            extra_headers=model_settings.extra_headers,
             metadata=self._non_null_or_not_given(model_settings.metadata),
         )
 

src/agents/models/openai_responses.py

@@ -253,7 +253,7 @@ async def _fetch_response(
             tool_choice=tool_choice,
             parallel_tool_calls=parallel_tool_calls,
             stream=stream,
-            extra_headers=_HEADERS,
+            extra_headers={**_HEADERS, **model_settings.extra_headers},
             extra_query=model_settings.extra_query,
             extra_body=model_settings.extra_body,
             text=response_format,