mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-10 22:41:41 +00:00
use common get_request_headers for BaseAWS
This commit is contained in:
parent
3cc4afbd6f
commit
be2ba6b1b3
1 changed files with 7 additions and 2 deletions
|
|
@ -87,7 +87,7 @@ class BaseAWSLLM:
|
|||
"""
|
||||
import litellm
|
||||
from litellm.secret_managers.main import str_to_bool
|
||||
|
||||
|
||||
# Check environment variable first (highest priority)
|
||||
ssl_verify = os.getenv("SSL_VERIFY", litellm.ssl_verify)
|
||||
|
||||
|
|
@ -1232,15 +1232,20 @@ class BaseAWSLLM:
|
|||
else:
|
||||
headers = {"Content-Type": "application/json"}
|
||||
|
||||
aws_signature_headers = self._filter_headers_for_aws_signature(headers)
|
||||
request = AWSRequest(
|
||||
method="POST",
|
||||
url=api_base,
|
||||
data=json.dumps(request_data),
|
||||
headers=headers,
|
||||
headers=aws_signature_headers,
|
||||
)
|
||||
sigv4.add_auth(request)
|
||||
|
||||
request_headers_dict = dict(request.headers)
|
||||
# Add back original headers after signing. Only headers in SignedHeaders
|
||||
# are integrity-protected; forwarded headers (x-forwarded-*) must remain unsigned.
|
||||
for header_name, header_value in headers.items():
|
||||
request_headers_dict[header_name] = header_value
|
||||
if (
|
||||
headers is not None and "Authorization" in headers
|
||||
): # prevent sigv4 from overwriting the auth header
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue