Completion for java "expression" does't work with DTO as records #245

@Mapper(unmappedTargetPolicy = ReportingPolicy.IGNORE, componentModel = MappingConstants.ComponentModel.SPRING)
public interface OrderMapper {
    OrderGetAllDto toOrderDto(Order order);

    default <T, R> R mapFromAggregateReference(AggregateReference<T, R> aggregateReference) {
        if (aggregateReference == null) {
            return null;
        }
        return aggregateReference.getId();
    }

    @Mapping(target = "orderPaymentRefPayments", expression = "java(orderPaymentRefsToOrderPaymentRefPayments())")
    OrderDto toOrderGetOneDto(Order order);

    default Set<Long> orderPaymentRefsToOrderPaymentRefPayments(Set<OrderPaymentRef> orderPaymentRefs) {
        return orderPaymentRefs.stream()
            .map(OrderPaymentRef::getPayment)
            .map(AggregateReference::getId)
            .collect(Collectors.toSet());
    }
}

public record OrderDto(Long id, String number, Integer owner, Set<Long> orderPaymentRefPayments,
                       List<OrderItemDto> orderItems, BigDecimal totalCost, LocalDateTime createdDate,
                       LocalDateTime lastModifiedDate) {
    /**
     * DTO for {@link OrderItem}
     */
    public record OrderItemDto(Long id, Long service, Integer quantity, String serviceName, BigDecimal price,
                               BigDecimal cost) {
    }
}

@Mapping(target = "orderPaymentRefPayments", expression = "java()")

PsiElement resolved = references[0].resolve();
                            if ( resolved instanceof PsiMethod resolvedPsiMethod ) {
                                PsiParameter[] psiParameters =
                                        resolvedPsiMethod.getParameterList().getParameters();
                                if ( psiParameters.length > 0) {
                                    targetType = psiParameters[0].getType();
                                }
                            }
                            else if ( resolved instanceof PsiParameter resolvedPsiParameter ) {
                                targetType = resolvedPsiParameter.getType();
                            }
                            else if ( resolved instanceof PsiField resolvedPsiField ) {
                                targetType = resolvedPsiField.getType();
                            }