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.
This commit is contained in:
Thiago Moretto
2026-05-21 11:58:13 -03:00
parent d3e20900e8
commit cd44734df2

View File

@@ -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,