o Æ°iÞã@s\ddlmZmZmZmZmZddlmZddlZddl m Z Gdd„dƒZ Gdd„dƒZ dS) é)ÚDictÚListÚMappingÚOptionalÚUnion)Úparse_qsN)ÚPASS_THROUGH_HEADER_PREFIXc@sŠeZdZe ddejdeeeee ffde e eeee ffde eeee efffdd„ƒZ e dd ed ed e efd d „ƒZdS)ÚBasePassthroughUtilsNÚ existing_urlÚrequest_query_paramsÚdefault_query_paramsÚreturncCsP|j d¡}t|ƒ}dd„| ¡Dƒ}i}|r| |¡| |¡| |¡|S)Nzutf-8cSs*i|]\}}|t|ƒdkr|dn|“qS)ér)Úlen)Ú.0ÚkÚv©rúP/home/app/Keep/.python/lib/python3.10/site-packages/litellm/passthrough/utils.pyÚ sÿzDBasePassthroughUtils.get_merged_query_parameters..)ÚqueryÚdecoderÚitemsÚupdate)r r r Zexisting_query_stringZexisting_query_paramsZupdated_existing_query_paramsZ merged_paramsrrrÚget_merged_query_parameters s ÿ   z0BasePassthroughUtils.get_merged_query_parametersFÚrequest_headersÚheadersÚforward_headerscCsh|dur| dd¡| dd¡i|¥|¥}| ¡D]\}}| ¡ t¡r1|ttƒd…}|||<q|S)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' Tzcontent-lengthNÚhost)ÚpoprÚlowerÚ startswithrr)rrrÚ header_nameÚ header_valueZactual_header_namerrrÚforward_headers_from_request&s   €z1BasePassthroughUtils.forward_headers_from_request)N)F)Ú__name__Ú __module__Ú __qualname__Ú staticmethodÚhttpxÚURLrÚstrrÚlistrrrrÚdictÚboolr$rrrrr s*ýÿþý üýÿþýr c@s"eZdZededefdd„ƒZdS)Ú CommonUtilsÚendpointr cCsNddl}d|vr |Sgd¢}|D]\}}| ||¡r$| |||¡}|Sq|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 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)ÚreÚsearchÚsub)r0r2ÚpatternsÚpatternÚ replacementrrrÚ"encode_bedrock_runtime_modelid_arnEs  üz.CommonUtils.encode_bedrock_runtime_modelid_arnN)r%r&r'r(r+r8rrrrr/Dsr/) ÚtypingrrrrrÚ urllib.parserr)Zlitellm.constantsrr r/rrrrÚs   ;