UCFunctionToolkit#

class langchain_community.tools.databricks.tool.UCFunctionToolkit[source]#

Bases: BaseToolkit

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

param tools: Dict[str, BaseTool] [Optional]#
param warehouse_id: str [Required]#

The ID of a Databricks SQL Warehouse to execute functions.

param workspace_client: Any [Optional]#

Databricks workspace client.

get_tools() → List[BaseTool][source]#

Get the tools in the toolkit.

Return type:

List[BaseTool]

include(*function_names: str, **kwargs: Any) → Self[source]#

Includes UC functions to the toolkit.

Parameters:

  • functions – A list of UC function names in the format “catalog_name.schema_name.function_name” or “catalog_name.schema_name.*”. If the function name ends with “.*”, all functions in the schema will be added.

  • kwargs (Any) – Extra arguments to pass to StructuredTool, e.g., return_direct.

  • function_names (str)

Return type:

Self

Examples using UCFunctionToolkit