diff --git a/libs/databricks/langchain_databricks/chat_models.py b/libs/databricks/langchain_databricks/chat_models.py
index 6a4f4d0..ab89396 100644
--- a/libs/databricks/langchain_databricks/chat_models.py
+++ b/libs/databricks/langchain_databricks/chat_models.py
@@ -17,6 +17,7 @@
     Union,
 )
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForLLMRun
 from langchain_core.language_models import BaseChatModel
 from langchain_core.language_models.base import LanguageModelInput
@@ -50,7 +51,6 @@
 from langchain_core.tools import BaseTool
 from langchain_core.utils.function_calling import convert_to_openai_tool
 from langchain_core.utils.pydantic import is_basemodel_subclass
-from langchain_core._api import deprecated
 from mlflow.deployments import BaseDeploymentClient  # type: ignore
 from pydantic import BaseModel, Field
 
@@ -60,7 +60,7 @@
 
 
 @deprecated(
-    since="0.1.2", message="Use databricks_langchain.ChatDatabricks"
+    since="0.1.2", message="Use databricks_langchain.ChatDatabricks", removal="1.0.0"
 )
 class ChatDatabricks(BaseChatModel):
     """Databricks chat model integration.
diff --git a/libs/databricks/langchain_databricks/embeddings.py b/libs/databricks/langchain_databricks/embeddings.py
index b9305c0..1282ea6 100644
--- a/libs/databricks/langchain_databricks/embeddings.py
+++ b/libs/databricks/langchain_databricks/embeddings.py
@@ -1,14 +1,16 @@
 from typing import Any, Dict, Iterator, List
 
-from langchain_core.embeddings import Embeddings
 from langchain_core._api import deprecated
+from langchain_core.embeddings import Embeddings
 from pydantic import BaseModel, PrivateAttr
 
 from langchain_databricks.utils import get_deployment_client
 
 
 @deprecated(
-    since="0.1.2", message="Use databricks_langchain.DatabricksEmbeddings"
+    since="0.1.2",
+    message="Use databricks_langchain.DatabricksEmbeddings",
+    removal="1.0.0",
 )
 class DatabricksEmbeddings(Embeddings, BaseModel):
     """Databricks embedding model integration.
diff --git a/libs/databricks/langchain_databricks/vectorstores.py b/libs/databricks/langchain_databricks/vectorstores.py
index b888845..bb2dec6 100644
--- a/libs/databricks/langchain_databricks/vectorstores.py
+++ b/libs/databricks/langchain_databricks/vectorstores.py
@@ -18,10 +18,10 @@
 )
 
 import numpy as np
+from langchain_core._api import deprecated
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
 from langchain_core.vectorstores import VST, VectorStore
-from langchain_core._api import deprecated
 
 from langchain_databricks.utils import maximal_marginal_relevance
 
@@ -40,7 +40,9 @@ class IndexType(str, Enum):
 
 
 @deprecated(
-    since="0.1.2", message="Use databricks_langchain.DatabricksVectorSearch"
+    since="0.1.2",
+    message="Use databricks_langchain.DatabricksVectorSearch",
+    removal="1.0.0",
 )
 class DatabricksVectorSearch(VectorStore):
     """Databricks vector store integration.