Hi Pawan.

Thanks for your answer. However, the issue is more related to MLflow than LangGraph. My graph is working perfectly in the notebook, but I need to put it into production and create the API, and to do that, I need to register the graph and be able to use it.

The question is, how MLflow or Databricks would know that a threat_id is needed for every chat session? How should I register the graph to be able to use the short-term memory in the API?

Thanks!