topic Integrating LangGraph with AI Cookbook Agent Template on Databricks using MLflow in Generative AI

Integrating LangGraph with AI Cookbook Agent Template on Databricks using MLflow

aliciachimeno — Sat, 01 Feb 2025 12:07:28 GMT

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:

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

Any advice or resources would be greatly appreciated!

Re: Integrating LangGraph with AI Cookbook Agent Template on Databricks using MLflow

Alberto_Umana — Sat, 01 Feb 2025 15:54:59 GMT

Hi @aliciachimeno,

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

Re: Integrating LangGraph with AI Cookbook Agent Template on Databricks using MLflow

aliciachimeno — Sat, 01 Feb 2025 16:19:01 GMT

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