ruff

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

integrations/langchain/src/databricks_langchain/chat_models.py

@@ -50,10 +50,11 @@
 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 databricks_langchain.utils import get_deployment_client
 from mlflow.deployments import BaseDeploymentClient  # type: ignore
 from pydantic import BaseModel, Field
 
+from databricks_langchain.utils import get_deployment_client
+
 logger = logging.getLogger(__name__)
 
 

integrations/langchain/src/databricks_langchain/embeddings.py

@@ -1,9 +1,10 @@
 from typing import Any, Dict, Iterator, List
 
 from langchain_core.embeddings import Embeddings
-from databricks_langchain.utils import get_deployment_client
 from pydantic import BaseModel, PrivateAttr
 
+from databricks_langchain.utils import get_deployment_client
+
 
 class DatabricksEmbeddings(Embeddings, BaseModel):
     """Databricks embedding model integration.

integrations/langchain/src/databricks_langchain/vectorstores.py

@@ -22,6 +22,7 @@
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
 from langchain_core.vectorstores import VST, VectorStore
+
 from databricks_langchain.utils import maximal_marginal_relevance
 
 logger = logging.getLogger(__name__)