diff --git a/docs/docs/integrations/tools/databricks.ipynb b/docs/docs/integrations/tools/databricks.ipynb
index 9fa37315385..61aa76484f2 100644
--- a/docs/docs/integrations/tools/databricks.ipynb
+++ b/docs/docs/integrations/tools/databricks.ipynb
@@ -66,21 +66,20 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from databricks.sdk import WorkspaceClient\n",
-    "from langchain_community.tools.databricks import UCFunctionToolkit\n",
+    "from databricks_langchain.uc_ai import (\n",
+    "    DatabricksFunctionClient,\n",
+    "    UCFunctionToolkit,\n",
+    "    set_uc_function_client,\n",
+    ")\n",
     "\n",
-    "tools = (\n",
-    "    UCFunctionToolkit(\n",
-    "        # You can find the SQL warehouse ID in its UI after creation.\n",
-    "        warehouse_id=\"xxxx123456789\"\n",
-    "    )\n",
-    "    .include(\n",
-    "        # Include functions as tools using their qualified names.\n",
-    "        # You can use \"{catalog_name}.{schema_name}.*\" to get all functions in a schema.\n",
-    "        \"main.tools.python_exec\",\n",
-    "    )\n",
-    "    .get_tools()\n",
-    ")"
+    "client = DatabricksFunctionClient()\n",
+    "set_uc_function_client(client)\n",
+    "\n",
+    "tools = UCFunctionToolkit(\n",
+    "    # Include functions as tools using their qualified names.\n",
+    "    # You can use \"{catalog_name}.{schema_name}.*\" to get all functions in a schema.\n",
+    "    function_names=[\"main.tools.python_exec\"]\n",
+    ").tools"
    ]
   },
   {
diff --git a/libs/community/langchain_community/tools/databricks/tool.py b/libs/community/langchain_community/tools/databricks/tool.py
index f64616ea5fb..67ef21148e4 100644
--- a/libs/community/langchain_community/tools/databricks/tool.py
+++ b/libs/community/langchain_community/tools/databricks/tool.py
@@ -4,6 +4,7 @@ from decimal import Decimal
 from hashlib import md5
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Type, Union
 
+from langchain_core._api import deprecated
 from langchain_core.tools import BaseTool, StructuredTool
 from langchain_core.tools.base import BaseToolkit
 from pydantic import BaseModel, Field, create_model
@@ -131,6 +132,11 @@ def _get_default_workspace_client() -> Any:
     return WorkspaceClient()
 
 
+@deprecated(
+    since="0.3.18",
+    removal="1.0",
+    alternative_import="databricks_langchain.uc_ai.UCFunctionToolkit",
+)
 class UCFunctionToolkit(BaseToolkit):
     warehouse_id: str = Field(
         description="The ID of a Databricks SQL Warehouse to execute functions."