Bug importig H2OMLAgent

When I import
from ai_data_science_team.ml_agents import H2OMLAgent

I get the following error:
ModuleNotFoundError: No module named 'langchain.prompts'

I tried using older versions, but it also causes incompatibility with Langchain.

Can you add an exact description of which Python versions and their packages are used so I don't end up with this error?