I am experiencing this same issue. As a temporary work-around I had to go back to using the deprecated package. What I can't tell is if it's an MLFlow bug or a databricks-langchain bug. It could be MLFlow is expecting the langchain-databricks package for ChatDatabricks. Or the databricks-langchain package has a messed up dependency.