Fix update docstring to include context (#1785)

Author: Sushisource

workflow/workflow.go

@@ -545,19 +545,17 @@ func SetUpdateHandler(ctx Context, updateName string, handler interface{}) error
 	return SetUpdateHandlerWithOptions(ctx, updateName, handler, UpdateHandlerOptions{})
 }
 
-// SetUpdateHandlerWithOptions binds an update handler function to the specified
-// name such that update invocations specifying that name will invoke the
-// handler.  The handler function can take as input any number of parameters so
-// long as they can be serialized/deserialized by the system. The handler can
-// take a [workflow.Context] as its first parameter but this is not required. The
-// update handler must return either a single error or a single serializable
-// object along with a single error. The update handler function is invoked in
-// the context of the workflow and thus is subject to the same restrictions as
-// workflow code, namely, the update handler must be deterministic. As with
-// other workflow code, update code is free to invoke and wait on the results of
-// activities. Update handler code is free to mutate workflow state.
-//
-// This registration can optionally specify (through UpdateHandlerOptions) an
+// SetUpdateHandlerWithOptions binds an update handler function to the specified name such that
+// update invocations specifying that name will invoke the handler. The handler function can take as
+// input any number of parameters so long as they can be serialized/deserialized by the system. The
+// handler must take a [workflow.Context] as its first parameter. The update handler must return
+// either a single error or a single serializable object along with a single error. The update
+// handler function is invoked in the context of the workflow and thus is subject to the same
+// restrictions as workflow code, namely, the update handler must be deterministic. As with other
+// workflow code, update code is free to invoke and wait on the results of activities. Update
+// handler code is free to mutate workflow state.
+//
+
 // update validation function. If provided, this function will be invoked before
 // the update handler itself is invoked and if this function returns an error,
 // the update request will be considered to have been rejected and as such will
@@ -575,7 +573,7 @@ func SetUpdateHandler(ctx Context, updateName string, handler interface{}) error
 //		err := workflow.SetUpdateHandlerWithOptions(
 //			ctx,
 //			"add",
-//			func(val int) (int, error) { // Calls
+//			func(ctx workflow.Context, val int) (int, error) { // Calls
 //				counter += val // note that this mutates workflow state
 //				return counter, nil
 //			},