昺h VdZddlZddlmZmZmZddlmZmZm Z GddeZ dS)zUtil that calls Lambda.N)AnyDictOptional) BaseModelExtraroot_validatorceZdZUdZeed< dZeeed< dZ eeed< dZ eeed< GddZ e d e d e fd Zd ed efd ZdS) 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 eZdZdZejZdS)LambdaWrapper.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__rforbidextrao/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/utilities/awslambda.pyConfigr s55 rrvaluesreturnc ddl}n#t$rtdwxYw|d|d<|d|d<|S)z3Validate that python package exists in environment.rNzBboto3 is not installed. Please install it with `pip install boto3`lambdar r )boto3 ImportErrorclient)clsrrs rvalidate_environmentz"LambdaWrapper.validate_environment%sn  LLLL   T   #(,,x"8"8"("9 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)selfr$respayload_streampayload_stringanswers rrunzLambdaWrapper.run6s ''+,J//(    : ^N+002299'BBNZ//7FF : : :999 : >Vr\\$$&f&& &sA B BB)rrrrr__annotations__r rstrr rrrrr#r8rrrr r s  %#'M8C='''*)-#---904 444; ^$4 ''''''''rr ) rr-typingrrrlangchain_core.pydantic_v1rrrr rrrr=s &&&&&&&&&&GGGGGGGGGGJ'J'J'J'J'IJ'J'J'J'J'r