Improve docstrings for AwsLambdaInvokeFunctionOperator (#28233)

apache · Dec 8, 2022 · 1eaedc8 · 1eaedc8
1 parent 3f06efa
commit 1eaedc8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/airflow/providers/amazon/aws/hooks/lambda_function.py b/airflow/providers/amazon/aws/hooks/lambda_function.py
@@ -34,9 +34,13 @@ class LambdaHook(AwsBaseHook):
         :class:`~airflow.providers.amazon.aws.hooks.base_aws.AwsBaseHook`
 
     :param function_name: AWS Lambda Function Name
+    :param invocation_type: AWS Lambda Invocation Type (RequestResponse, Event etc)
     :param log_type: Tail Invocation Request
+    :param client_context: Up to 3,583 bytes of base64-encoded data about the invoking client
+        to pass to the function in the context object.
+    :param payload: The JSON that you want to provide to your Lambda function as input.
     :param qualifier: AWS Lambda Function Version or Alias Name
-    :param invocation_type: AWS Lambda Invocation Type (RequestResponse, Event etc)
+
     """
 
     def __init__(

diff --git a/airflow/providers/amazon/aws/operators/lambda_function.py b/airflow/providers/amazon/aws/operators/lambda_function.py
@@ -37,9 +37,12 @@ class AwsLambdaInvokeFunctionOperator(BaseOperator):
     review the boto3 Lambda invoke docs.
 
     :param function_name: The name of the AWS Lambda function, version, or alias.
-    :param payload: The JSON string that you want to provide to your Lambda function as input.
     :param log_type: Set to Tail to include the execution log in the response. Otherwise, set to "None".
     :param qualifier: Specify a version or alias to invoke a published version of the function.
+    :param invocation_type: One of RequestResponse / Event / DryRun
+    :param client_context: Up to 3,583 bytes of base64-encoded data about the invoking client
+        to pass to the function in the context object.
+    :param payload: The JSON string that you want to provide to your Lambda function as input.
     :param aws_conn_id: The AWS connection ID to use
 
     .. seealso::