Databricks SQL 쿼리 도구

Databricks SQL 쿼리 도구 (DatabricksQueryTool)

CrewAI의 DatabricksQueryTool은 Databricks 워크스페이스 테이블에 SQL을 실행하는 도구예요. CLI 프로필을 사용하거나 host/token을 직접 전달하는 두 가지 인증 방식을 지원해서, 에이전트가 Databricks 데이터를 바로 조회할 수 있게 해줍니다.

출처: 문서

본문

Databricks 워크스페이스 테이블에 CLI 프로필 또는 직접 host/token 인증 방식으로 SQL을 실행하는 도구입니다.

설치 (Installation)

uv add crewai-tools[databricks-sdk]

환경변수 (Environment Variables)

DATABRICKS_CONFIG_PROFILE 또는 (DATABRICKS_HOST + DATABRICKS_TOKEN)

개인 액세스 토큰(PAT)을 만들고 host 세부 정보는 Databricks 워크스페이스의 User Settings → Developer에서 찾을 수 있어요. 문서: https://docs.databricks.com/en/dev-tools/auth/pat.html

예시 (Example)

from crewai import Agent, Task, Crew
from crewai_tools import DatabricksQueryTool

tool = DatabricksQueryTool(
    default_catalog="main",
    default_schema="default",
)

agent = Agent(
    role="Data Analyst",
    goal="Query Databricks",
    tools=[tool],
    verbose=True,
)

task = Task(
    description="SELECT * FROM my_table LIMIT 10",
    expected_output="10 rows",
    agent=agent,
)

crew = Crew(
    agents=[agent],
    tasks=[task],
    verbose=True,
)
result = crew.kickoff()

print(result)

파라미터 (Parameters)

query (required): 실행할 SQL 쿼리
catalog (optional): 기본 catalog 재정의
db_schema (optional): 기본 schema 재정의
warehouse_id (optional): 기본 SQL 웨어하우스 재정의
row_limit (optional): 반환할 최대 행 수 (기본값: 1000)

초기화 시 기본값 (Defaults on initialization)

default_catalog
default_schema
default_warehouse_id

에러 처리 & 팁 (Error handling & tips)

인증 오류: DATABRICKS_HOST가 https://로 시작하는지, token이 유효한지 확인하세요.
권한: token이 SQL 웨어하우스와 schema에 접근 가능한지 확인하세요.
한계: 에이전트 루프에서는 긴 실행 쿼리를 피하고 필터/한계를 추가하세요.