langchain_community.tools.databricks.tool.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