From cd44734df2d9a0aa32cb2c704f1f8ffdb336223e Mon Sep 17 00:00:00 2001 From: Thiago Moretto Date: Thu, 21 May 2026 11:58:13 -0300 Subject: [PATCH] feat(tools): declare env_vars on DatabricksQueryTool Add EnvVar import and env_vars field to DatabricksQueryTool so the host UI knows which environment variables the tool requires. Both auth paths (DATABRICKS_HOST+TOKEN or DATABRICKS_CONFIG_PROFILE) are marked required=False with descriptions explaining the alternative. --- .../databricks_query_tool.py | 21 ++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) diff --git a/lib/crewai-tools/src/crewai_tools/tools/databricks_query_tool/databricks_query_tool.py b/lib/crewai-tools/src/crewai_tools/tools/databricks_query_tool/databricks_query_tool.py index ba9a375bd..2ce9338ab 100644 --- a/lib/crewai-tools/src/crewai_tools/tools/databricks_query_tool/databricks_query_tool.py +++ b/lib/crewai-tools/src/crewai_tools/tools/databricks_query_tool/databricks_query_tool.py @@ -4,7 +4,7 @@ import os import time from typing import TYPE_CHECKING, Any, TypeGuard, TypedDict -from crewai.tools import BaseTool +from crewai.tools import BaseTool, EnvVar from pydantic import BaseModel, Field, model_validator @@ -99,6 +99,25 @@ class DatabricksQueryTool(BaseTool): _workspace_client: WorkspaceClient | None = None package_dependencies: list[str] = Field(default_factory=lambda: ["databricks-sdk"]) + env_vars: list[EnvVar] = Field( + default_factory=lambda: [ + EnvVar( + name="DATABRICKS_HOST", + description="Databricks workspace URL (use with DATABRICKS_TOKEN, or set DATABRICKS_CONFIG_PROFILE instead).", + required=False, + ), + EnvVar( + name="DATABRICKS_TOKEN", + description="Databricks personal access token (pair with DATABRICKS_HOST).", + required=False, + ), + EnvVar( + name="DATABRICKS_CONFIG_PROFILE", + description="Databricks CLI profile name (alternative to DATABRICKS_HOST/TOKEN).", + required=False, + ), + ] + ) def __init__( self,