mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-14 04:09:09 +00:00
fix(bedrock): mark retrieval request metadata handoffs
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
85caa4a4ce
commit
303f95374b
1 changed files with 5 additions and 4 deletions
|
|
@ -1378,10 +1378,11 @@ class BedrockFilesConfig(BaseAWSLLM, BaseFilesConfig):
|
|||
if not isinstance(signed_headers_object, Mapping):
|
||||
raise TypeError("S3 request signing did not produce request headers")
|
||||
signed_headers: Final = _S3_SIGNED_REQUEST_HEADERS_ADAPTER.validate_python(signed_headers_object)
|
||||
litellm_params[S3_SIGNED_REQUEST_HEADERS_PARAM] = MappingProxyType({**signed_headers, "Range": "bytes=0-0"})
|
||||
litellm_params[S3_RETRIEVE_FILE_ID_PARAM] = file_id
|
||||
litellm_params[S3_RETRIEVE_FILE_KEY_PARAM] = object_key
|
||||
litellm_params[S3_RETRIEVE_FILE_RELATIVE_KEY_PARAM] = relative_key
|
||||
range_headers: Final = MappingProxyType({**signed_headers, "Range": "bytes=0-0"})
|
||||
litellm_params[S3_SIGNED_REQUEST_HEADERS_PARAM] = range_headers # rebind-ok: handed to validate_environment
|
||||
litellm_params[S3_RETRIEVE_FILE_ID_PARAM] = file_id # rebind-ok: required by response transform
|
||||
litellm_params[S3_RETRIEVE_FILE_KEY_PARAM] = object_key # rebind-ok: required by response transform
|
||||
litellm_params[S3_RETRIEVE_FILE_RELATIVE_KEY_PARAM] = relative_key # rebind-ok: required by response transform
|
||||
return url, params
|
||||
|
||||
def transform_retrieve_file_response(
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue