Integrating LangGraph with AI Cookbook Agent Template on Databricks using MLflow

[aliciachimeno]

I am working on building an agent using the AI Cookbook Agent Template and would like to integrate LangGraph into the agent template. Specifically, I want to build an agent that uses an adaptive RAG  chain and an SQL agent. My goal is to implement this in Databricks, leveraging MLflow for model management and tracking.

Could you please guide me on how to:

1. Integrate LangGraph with the AI Cookbook agent template in a Databricks environment?
2. Implement an adaptive RAG chain and SQL agent within Databricks using MLflow for model tracking and management?
3. Ensure seamless interaction between these components, including any necessary configuration or code examples?

Any advice or resources would be greatly appreciated!

[Alberto_Umana]

Hi @aliciachimeno,

Please take a look at: https://github.com/databricks/genai-cookbook

[aliciachimeno]

Hi, this is the template i am using, but i want to integrate LangGraph with this template.