uPjZUddlmZddlmZddlmZddlZddlmZddl m Z ddl m Z e hdZee ed <d Zeeed <Gd d ZGddZy))Mapping)Final)parse_qsN)verbose_logger)PASS_THROUGH_HEADER_PREFIX)contains_aws_arn>api-key x-api-keyx-goog-api-keyhost authorizationcontent-lengthaccept-encoding_PASS_THROUGH_PROTECTED_HEADERS)zx-amz-'_PASS_THROUGH_PROTECTED_HEADER_PREFIXESceZdZe d dej deeeezfde eeezfdzde eeeezffdZ e d de de d e dzfd Z y) BasePassthroughUtilsN existing_urlrequest_query_paramsdefault_query_paramsreturnc8|jjd}t|}|jDcic]\}}|t |dk(r|dn|}}}i}|r|j ||j ||j ||Scc}}w)Nzutf-8r)querydecoderitemslenupdate) rrrexisting_query_stringexisting_query_paramskvupdated_existing_query_params merged_paramss ^/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/passthrough/utils.pyget_merged_query_parametersz0BasePassthroughUtils.get_merged_query_parameterss(4'9'9'@'@'I'/0E'F[pZuZuZw/xRVRSUV3q6Q;1Q4A0M/x%/x "   !5 6 :; 120ys Brequest_headersheadersforward_headersc|dur|jdd|jdd|jdd|Dchc]}|j}}t|jD]'}|j|vs|j|d)i||}|j D]\}}|jj t s*|tt djtvstfdtDrtjd|||<|Scc}w)a$ Helper to forward headers from original request. Also handles 'x-pass-' prefixed headers which are always forwarded with the prefix stripped, regardless of forward_headers setting. e.g., 'x-pass-anthropic-beta: value' becomes 'anthropic-beta: value' TrNr rc3@K|]}j|ywN) startswith).0pactual_header_names r% zDBasePassthroughUtils.forward_headers_from_request..\s$P9:&11!4Psz;x-pass- header %s maps to a protected header name; skipping) poplowerlistkeysrr-rrranyrrdebug)r'r(r) header_namecustom_header_names header_valuer0s @r%forward_headers_from_requestz1BasePassthroughUtils.forward_headers_from_request9sU d "    0$ 7    -    14 8QX)Y++*;*;*=)Y )Y#O$8$8$:; ; $$&*==#'' T: ; 54G4G*9)>)>)@ ; %K  "--.HI%05O1P1R%S%Y%Y%["%)HHCP>ePM#((U#.:*+ ;1*ZsD<r,)F) __name__ __module__ __qualname__ staticmethodhttpxURLrstrr4dictr&boolr;r%rrs>Bii%c3:o6#3d ?3d: c3c?" # 2(-,,,,,rFrc&eZdZededefdZy) CommonUtilsendpointrcddl}t|s|Sgd}|D]-\}}|j||s|j|||}|S|S)aV Encodes any "/" found in the modelId of an AWS Bedrock Runtime Endpoint when arns are passed in. - modelID value can be an ARN which contains slashes that SHOULD NOT be treated as path separators. e.g endpoint: /model//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 rN) )z((custom-model)/([a-z0-9.-]+)/([a-z0-9]+)z \1%2F\2%2F\3)z!(:application-inference-profile)/\1%2F)z(:inference-profile)/rK)z(:foundation-model)/rK)z(:imported-model)/rK)z(:provisioned-model)/rK)z (:prompt)/rK)z (:endpoint)/rK)z(:prompt-router)/rK)z(:default-prompt-router)/rK)rersearchsub)rIrLpatternspattern replacements r%"encode_bedrock_runtime_modelid_arnz.CommonUtils.encode_bedrock_runtime_modelid_arnjsb*  )O  %-  G[yy(+66';A   rFN)r<r=r>r?rBrRrErFr%rHrHis"0S0S00rFrH)collections.abcrtypingr urllib.parserr@litellm._loggingrlitellm.constantsr(litellm.litellm_core_utils.aws_partitionr frozensetr__annotations__rtuplerrHrErFr%r\sa#! +8E 5> 5y!1 9D'uCHHV22rF