1  """
 2  python examples/mlflow-3/langchain_databricks_example.py
 3  """
 4  
 5  from databricks.sdk import WorkspaceClient
 6  from langchain_core.runnables import RunnableLambda
 7  
 8  import mlflow
 9  
10  mlflow.langchain.autolog(log_models=True)
11  
12  wc = WorkspaceClient()
13  mlflow.set_tracking_uri("databricks")
14  mlflow.set_experiment(f"/Users/{wc.current_user.me().user_name}/langchain-autolog")
15  
16  with mlflow.start_run() as run:
17      r = RunnableLambda(lambda x: x + 1)
18      r.invoke(3)
19  
20  print(mlflow.search_traces(max_results=1))