Ng- VdZddlZddlmZmZmZddlmZmZm Z GddeZ dS)zUtil that calls Lambda.N)AnyDictOptional) BaseModel ConfigDictmodel_validatorceZdZUdZdZeed< dZee ed< dZ ee ed< dZ ee ed< e dZ ed ed ed efd Zde d e fdZdS) 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 N lambda_client function_nameawslambda_tool_nameawslambda_tool_descriptionforbid)extrabefore)modevaluesreturnc| ddl}n#t$rtdwxYw|d|d<|S)z3Validate that python package exists in environment.rNzBboto3 is not installed. Please install it with `pip install boto3`lambdar )boto3 ImportErrorclient)clsrrs c/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/utilities/awslambda.pyvalidate_environmentz"LambdaWrapper.validate_environment%s_   LLLL   T   #(,,x"8"8 s!querycD|j|jdtjd|i} |d}|d}tj|d}n#t$rYdSwxYw||dkrd Sd |S) 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. RequestResponsebody) FunctionNameInvocationTypePayloadr#zutf-8z$Failed to parse response from LambdaNzRequest failed.zResult: ) r invoker jsondumpsreaddecodeloads StopIteration)selfrrespayload_streampayload_stringanswers rrunzLambdaWrapper.run5s ''+,J//(    : ^N+002299'BBNZ//7FF : : :999 : >Vr\\$$&f&& &sA B BB)__name__ __module__ __qualname____doc__r r__annotations__r rstrr rr model_configr classmethodrrr1rr r s  M3%#'M8C='''*)-#---904 444;:L_(### $ 3   [$# ''''''''r;r ) r5r&typingrrrpydanticrrrr r:r;rr>s &&&&&&&&&&;;;;;;;;;;H'H'H'H'H'IH'H'H'H'H'r;