Databricks AI Dev Kit

# Python 3.10 이상 필요
python --version

# Databricks CLI 설치 (인증 설정용)
brew install databricks/tap/databricks

# Databricks 인증 프로파일 설정
databricks auth login --host https://<workspace-url>

# pip로 설치
pip install databricks-ai-dev-kit

# 또는 uvx로 직접 실행 (설치 없이)
uvx databricks-ai-dev-kit

# Claude Code에 Databricks MCP 서버 추가
claude mcp add databricks \
  -- uvx databricks-ai-dev-kit@latest

{
  "mcpServers": {
    "databricks": {
      "command": "uvx",
      "args": ["databricks-ai-dev-kit@latest"],
      "env": {
        "DATABRICKS_HOST": "https://<workspace-url>",
        "DATABRICKS_PROFILE": "DEFAULT"
      }
    }
  }
}

{
  "servers": {
    "databricks": {
      "command": "uvx",
      "args": ["databricks-ai-dev-kit@latest"],
      "env": {
        "DATABRICKS_HOST": "https://<workspace-url>"
      }
    }
  }
}

{
  "mcpServers": {
    "databricks": {
      "command": "uvx",
      "args": ["databricks-ai-dev-kit@latest"],
      "env": {
        "DATABRICKS_HOST": "https://<workspace-url>"
      }
    }
  }
}

# OAuth 로그인 (권장)
databricks auth login --host https://<workspace-url>

# 또는 환경 변수로 PAT 설정
export DATABRICKS_HOST="https://<workspace-url>"
export DATABRICKS_TOKEN="dapi..."

사용자: "지난 30일간 일별 주문 건수와 매출 합계를 조회해줘"

AI Dev Kit → execute_sql 도구 호출:
SELECT
  date_trunc('day', order_date) AS order_day,
  COUNT(*) AS order_count,
  SUM(total_amount) AS total_revenue
FROM catalog.schema.orders
WHERE order_date >= current_date() - INTERVAL 30 DAYS
GROUP BY 1
ORDER BY 1

사용자: "현재 실행 중인 클러스터 목록을 보여줘"
→ list_clusters 도구 호출

사용자: "개발용 클러스터를 시작해줘"
→ start_cluster 도구 호출

사용자: "S3의 CSV 파일을 읽어서 bronze 테이블로 적재하는 파이프라인을 만들어줘"
→ create_or_update_pipeline 도구 호출

사용자: "매출 트렌드와 카테고리별 비중을 보여주는 대시보드를 만들어줘"
→ create_or_update_dashboard 도구 호출

사용자: "production 카탈로그에 있는 모든 테이블을 보여줘"
→ manage_uc_objects 도구 호출

사용자: "sales 테이블의 컬럼 정보를 알려줘"
→ get_table_details 도구 호출

사용자: "S3 버킷 s3://raw-data/events/ 에서 JSON 파일을 읽어
bronze_events 테이블로 적재하고,
PII 컬럼을 마스킹한 silver_events 테이블을 만들어줘"

사용자: "customers 테이블의 구조를 보여주고,
가입월별 고객 수 추이를 차트로 시각화해줘"

사용자: "지금 실행 중인 Job 중 1시간 넘게 걸리는 것이 있어?
있으면 상세 정보를 보여줘"

# databricks.yml 예시
bundle:
  name: my-data-project

workspace:
  host: https://<workspace-url>

resources:
  pipelines:
    my_pipeline:
      name: "ETL Pipeline"
      target: "my_catalog.my_schema"