jYXddlmZmZmZmZddlmZddlZGddZGddZ y))DictListOptionalUnion)parse_qsNc eZdZedej deeeee ffdeeeee efffdZ e d de de de efdZy ) BasePassthroughUtils existing_urlrequest_query_paramsreturnc|jjd}t|}|jDcic]\}}|t |dk(r|dn|}}}i||Scc}}w)Nzutf-8r)querydecoderitemslen)r r existing_query_stringexisting_query_paramskvupdated_existing_query_paramss r/Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/litellm/passthrough/utils.pyget_merged_query_parametersz0BasePassthroughUtils.get_merged_query_parameterss !- 2 2 9 9' B ()> ?7L6Q6Q6S) 6SdaAs1v{qt )6S &) I&H*GHH ) s A#request_headersheadersforward_headerscb|dur*|jdd|jddi||}|S)zA Helper to forward headers from original request Tzcontent-lengthNhost)pop)rrrs rforward_headers_from_requestz1BasePassthroughUtils.forward_headers_from_requestsC d "    0$ 7    -54G4GN)F)__name__ __module__ __qualname__ staticmethodhttpxURLrstrrlistrrdictrboolr r!rr r s Iii I7;CsDyAQ/invoke containing arns with slashes need to be encoded from arn:aws:bedrock:ap-southeast-1:123456789012:application-inference-profile/abdefg12334 => arn:aws:bedrock:ap-southeast-1:123456789012:application-inference-profile%2Fabdefg12334 so that it is treated as one part of the path. Otherwise, the encoded endpoint will return 500 error when passed to Bedrock endpoint. See the apis in https://docs.aws.amazon.com/bedrock/latest/APIReference/API_Operations_Amazon_Bedrock_Runtime.html for more details on the regex patterns of modelId which we use in the regex logic below. Args: endpoint (str): The original endpoint string which may contain ARNs that contain slashes. Returns: str: The endpoint with properly encoded ARN slashes rNzarn:aws:) )z((custom-model)/([a-z0-9.-]+)/([a-z0-9]+)z \1%2F\2%2F\3)z!(:application-inference-profile)/\1%2F)z(:inference-profile)/r1)z(:foundation-model)/r1)z(:imported-model)/r1)z(:provisioned-model)/r1)z (:prompt)/r1)z (:endpoint)/r1)z(:prompt-router)/r1)z(:default-prompt-router)/r1)researchsub)r/r2patternspattern replacements r"encode_bedrock_runtime_modelid_arnz.CommonUtils.encode_bedrock_runtime_modelid_arn*s]*  X %O  %- G[yy(+66';A %- r!N)r"r#r$r%r(r8r,r!rr.r.)s"1S1S11r!r.) typingrrrr urllib.parserr&r r.r,r!rr;s&..!   D33r!