jtddlZddlmZmZmZmZmZddlmZm Z m Z ddl m Z GddZ Gddee Zy) N)AnyDictListOptionalTuple) BaseModel ConfigDictmodel_validator)BaseCrossEncodercNeZdZdZdZdZdeeeefde fdZ de dee fdZ y) CrossEncoderContentHandlerz'Content handler for CrossEncoder class.zapplication/json text_pairsreturncRtjd|i}|jdS)Nrutf-8)jsondumpsencode)selfr input_strs /var/www/html/Fitness-lenito-AI-main/venv/lib/python3.12/site-packages/langchain_community/cross_encoders/sagemaker_endpoint.pytransform_inputz*CrossEncoderContentHandler.transform_inputs&JJ j9: ((outputcttj|jjd}|d}|S)Nrscores)rloadsreaddecode)rr response_jsonrs rtransform_outputz+CrossEncoderContentHandler.transform_outputs0 6;;=#7#7#@A x( rN)__name__ __module__ __qualname____doc__ content_typeacceptsrrstrbytesrrfloatr!rrr r sF1%L G)$uS#X*?)E)stE{rr c eZdZUdZ dZeed<dZeed< dZ eed< dZ e eed< e Z e ed<dZe eed < dZe eed < ed d d ZedededefdZdeeeefdeefdZy)SagemakerEndpointCrossEncoderaSageMaker Inference CrossEncoder endpoint. To use, you must supply the endpoint name from your deployed Sagemaker model & the region where it is deployed. To authenticate, the AWS client uses the following methods to automatically load credentials: https://boto3.amazonaws.com/v1/documentation/api/latest/guide/credentials.html If a specific credential profile should be used, you must pass the name of the profile from the ~/.aws/credentials file that is to be used. Make sure the credentials / roles used have the required policies to access the Sagemaker endpoint. See: https://docs.aws.amazon.com/IAM/latest/UserGuide/access_policies.html Nclient endpoint_name region_namecredentials_profile_namecontent_handler model_kwargsendpoint_kwargsTforbidr+)arbitrary_types_allowedextraprotected_namespacesbefore)modevaluesrc ddl} |jdr|j|d}n|j}|jd|d|d<|S#t$r}t d |d}~wwxYw#t $r t d wxYw) zJValidate that AWS credentials to and python package exists in environment.rNr2) profile_namezsagemaker-runtimer1)r1r.zCould not load credentials to authenticate with AWS client. Please check that credentials in the specified profile name are valid.zRCould not import boto3 python package. Please install it with `pip install boto3`.)boto3getSessionr. Exception ValueError ImportError)clsr<r?sessiones rvalidate_environmentz2SagemakerEndpointCrossEncoder.validate_environment`s   ::89#mm%+,F%G,G $mmoG#*>>'VM5J$2$x    .   >  s)A6AA A3" A..A33A66B rc|jxsi}|jj|}|jj}|jj} |j j d|j|||d|}|jj|dS#t$r}td|d}~wwxYw)z6Call out to SageMaker Inference CrossEncoder endpoint.) EndpointNameBody ContentTypeAcceptz$Error raised by inference endpoint: NrKr+) r5r3rr&r'r.invoke_endpointr0rBrCr!)rr_endpoint_kwargsbodyr&r'responserGs rscorez#SagemakerEndpointCrossEncoder.scores//52##33J?++88 &&.. I2t{{22!//(  # H##44Xf5EFF ICA3GH H Is+B"" B>+B99B>)r"r#r$r%r.r__annotations__r0r(r1r2rr r3r4rr5r model_configr classmethodrHrrr*rRr+rrr-r-s"*FCM3+KP.2hsm23M2NO/N#'L(4.'1&*OXd^*  $H2L(#$3$@GU38_ 5G$u+Grr-)rtypingrrrrrpydanticrr r 'langchain_community.cross_encoders.baser r r-r+rrrYs4 33;;D   }GI/?}Gr