[main]: Adapt async approach from mlflow client (#1013)

[main]: Adapt async approach from mlflow client (#1013) #392