Compare commits

...

2 Commits

Author SHA1 Message Date
github-actions[bot]
cded952701 chore: update tool specifications 2026-05-21 14:59:35 +00:00
Thiago Moretto
cd44734df2 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.
2026-05-21 11:58:13 -03:00
2 changed files with 40 additions and 2 deletions

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,

View File

@@ -6821,7 +6821,26 @@
},
{
"description": "Execute SQL queries against Databricks workspace tables and return the results. Provide a 'query' parameter with the SQL query to execute.",
"env_vars": [],
"env_vars": [
{
"default": null,
"description": "Databricks workspace URL (use with DATABRICKS_TOKEN, or set DATABRICKS_CONFIG_PROFILE instead).",
"name": "DATABRICKS_HOST",
"required": false
},
{
"default": null,
"description": "Databricks personal access token (pair with DATABRICKS_HOST).",
"name": "DATABRICKS_TOKEN",
"required": false
},
{
"default": null,
"description": "Databricks CLI profile name (alternative to DATABRICKS_HOST/TOKEN).",
"name": "DATABRICKS_CONFIG_PROFILE",
"required": false
}
],
"humanized_name": "Databricks SQL Query",
"init_params_schema": {
"$defs": {