fix(agent): Fix application of extra_request_headers in OpenAIProvider

This commit is contained in:
Reinier van der Leer
2024-02-12 12:21:30 +01:00
parent fe0923ba6c
commit 39fddb1214

View File

@@ -490,9 +490,8 @@ class OpenAIProvider(
_functions_compat_fix_kwargs(functions, kwargs)
if extra_headers := self._configuration.extra_request_headers:
kwargs["extra_headers"] = kwargs.get("extra_headers", {}).update(
extra_headers.copy()
)
kwargs["extra_headers"] = kwargs.get("extra_headers", {})
kwargs["extra_headers"].update(extra_headers.copy())
return kwargs
@@ -514,9 +513,8 @@ class OpenAIProvider(
kwargs.update(self._credentials.get_model_access_kwargs(model_name))
if extra_headers := self._configuration.extra_request_headers:
kwargs["extra_headers"] = kwargs.get("extra_headers", {}).update(
extra_headers.copy()
)
kwargs["extra_headers"] = kwargs.get("extra_headers", {})
kwargs["extra_headers"].update(extra_headers.copy())
return kwargs