lakebase-setup

Configure Lakebase memory storage for Databricks LangGraph agents.

Updated Mar 15, 2026
One-click install
npx skills add https://github.com/sumitsaxena-git/databricks-app --skill lakebase-setup-sumitsaxena-git
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: lakebase-setup
Source: https://github.com/sumitsaxena-git/databricks-app/tree/main/agent-langgraph-long-term-memory/.claude/skills/lakebase-setup
Command: npx skills add https://github.com/sumitsaxena-git/databricks-app --skill lakebase-setup-sumitsaxena-git

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires databricks-langchain[memory], databricks-ai-bridge, and includes scripts (resource) components.

What problem does it solve?

Configure Lakebase to provide persistent memory storage for Databricks LangGraph agents, enabling long-term memory across sessions and reliable state for agents.

Core Features & Use Cases

  • Provision and connect to Lakebase for short-term memory (conversation history) and long-term memory (user facts)
  • Attach Lakebase as a resource in databricks.yml and configure environment variables for runtime access
  • Initialize Lakebase tables and perform required permissions grants for the app's service principal
  • Guidance for provisioned vs autoscaling Lakebase setups, including memory templates and deployment steps

Quick Start

Follow the steps to provision Lakebase, configure your app, initialize tables, grant permissions, and deploy.

Frequently Asked Questions about lakebase-setup

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I configure persistent memory for Databricks LangGraph agents?

To configure persistent memory for Databricks LangGraph agents, provision Lakebase and attach it as a resource in databricks.yml to enable short-term chat history and long-term user facts across sessions and deployments.

What's the best way to set up short-term and long-term agent memory in Databricks?

Setting up short-term conversation history and long-term user facts involves provisioning Lakebase, initializing the memory tables, and configuring environment variables for runtime access within your Databricks LangGraph app.

Do I need to grant service principal permissions for Lakebase memory storage?

Yes, you need to grant service principal permissions for Lakebase memory storage. The setup process includes performing required permission grants via databricks.yml and related scripts to secure access to the initialized Lakebase tables.

Can I use Lakebase with an autoscaling Databricks setup?

Yes, you can use Lakebase with an autoscaling Databricks setup. The configuration provides specific guidance for both provisioned and autoscaling Lakebase setups, including memory templates and deployment steps.

What dependencies are required to enable agent persistence in Databricks?

To enable agent persistence in Databricks, the required dependencies are databricks-langchain with memory features and databricks-ai-bridge, which ensure proper environment configuration and runtime access to Lakebase storage.