Databricks MCP Server
Подключает LLM к Databricks для выполнения SQL-запросов, просмотра джобов и статусов.
Python
⭐
41 stars
Описание
Это MCP-сервер на Python, интегрирующийся с Databricks API. Позволяет LLM запускать SQL-запросы на SQL-складах, перечислять все джобы, получать статус и детали конкретных джобов. Техологии: Python 3.7+, Databricks SDK. Применения: автоматизация анализа данных через естественный язык, мониторинг задач в Databricks для AI-ассистентов и аналитиков.
Возможности
✓
Выполнение SQL-запросов
Запуск SQL-запросов на Databricks SQL warehouses через функцию run_sql_query.
✓
Перечисление джобов
Список всех Databricks jobs в workspace с помощью list_jobs.
✓
Получение статуса джоба
Проверка статуса конкретного джоба по ID с get_job_status.
✓
Детали джоба
Получение подробной информации о джобе по ID с get_job_details.
Установка
- Клонируйте репозиторий.
- Создайте и активируйте виртуальное окружение: python -m venv .venv source .venv/bin/activate (На Windows: .venv\Scripts\activate)
- Установите зависимости: pip install -r requirements.txt
- Создайте файл .env с переменными: DATABRICKS_HOST=your-databricks-instance.cloud.databricks.com DATABRICKS_TOKEN=your-personal-access-token DATABRICKS_HTTP_PATH=/sql/1.0/warehouses/your-warehouse-id
- Запустите сервер: python main.py Опционально: протестируйте соединение с python test_connection.py.