U h @sDdZddlZddlmZmZmZddlmZmZGdddeZ dS)zUtil that calls Lambda.N)AnyDictOptional) BaseModelroot_validatorc@seZdZUdZeed<dZeeed<dZ eeed<dZ eeed<GdddZ e d d e e d d d ZeedddZdS) LambdaWrappera7Wrapper for AWS Lambda SDK. To use, you should have the ``boto3`` package installed and a lambda functions built from the AWS Console or CLI. Set up your AWS credentials with ``aws configure`` Example: .. code-block:: bash pip install boto3 aws configure lambda_clientN function_nameawslambda_tool_nameawslambda_tool_descriptionc@seZdZdZdS)zLambdaWrapper.ConfigZforbidN)__name__ __module__ __qualname__extrarrK/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/awslambda.pyConfig!srT)pre)valuesreturncCs<z ddl}Wntk r(tdYnX|d|d<|S)z3Validate that python package exists in environment.rNzBboto3 is not installed. Please install it with `pip install boto3`lambdar)boto3 ImportErrorclient)clsrrrrrvalidate_environment$s  z"LambdaWrapper.validate_environment)queryrcCs|jj|jdtd|id}z(|d}|d}t|d}Wntk r\YdSX|dksn|dkrrd Sd |SdS) z Invokes the lambda function and returns the result. Args: query: an input to passed to the lambda function as the ``body`` of a JSON object. ZRequestResponsebody)Z FunctionNameZInvocationTypePayloadrzutf-8z$Failed to parse response from LambdaNzRequest failed.zResult: ) rZinvoker jsondumpsreaddecodeloads StopIteration)selfrresZpayload_streamZpayload_stringZanswerrrrrun3s  zLambdaWrapper.run)r r r__doc__r__annotations__r rstrr r rrrrr(rrrrr s r) r)r typingrrrZlangchain_core.pydantic_v1rrrrrrrs