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 if the input data cannot be parsed to form a valid model.

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

The ID of a Databricks SQL Warehouse to execute functions.

param workspace_client: WorkspaceClient [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