Call to undefined method in images of media field

Problem/Motivation

Message Error: Call to undefined method Drupal\media_library\Form\FileUploadForm::getEntity() in _image_field_caption_widget_process() (line 50 of /Applications/XAMPP/xamppfiles/htdocs/eg-blog/web/modules/contrib/image_field_caption/image_field_caption.module)

This error appears in a field that uses the Media field type, when trying to upload an image to it.
Image is not loaded and the error appears in the log messages

Steps to reproduce

Add a new field of type media on a content type, and enable the image media type
Add content in the content type and try to upload an image to that field

Proposed resolution

This modification of the code works and the image is uploaded and the textarea for the caption appears

  // Get the entity.
  $field_definition = NULL;
  $form_object = $form_state->getFormObject();
  if ($form_object instanceof EntityFormInterface) {
    $entity = $form_object->getEntity();
    // Get the fields definitions.
    $field_definitions = $entity->getFieldDefinitions();
    // Get the current field definition.
    if (!empty($field_definitions[$element['#field_name']])) {
      $field_definition = $field_definitions[$element['#field_name']];
    }
    elseif (!empty($field_definitions[$element['#field_parents'][0]])) {
      $field_definition = $field_definitions[$element['#field_parents'][0]];
    }
  }

Remaining tasks

Create the patch with the proposed code.

User interface changes

API changes

Data model changes