linter

Signed-off-by: Prithvi Kannan <[email protected]>

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.

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.

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.