Inline langchain-databricks components (#25)

* Inline langchain-databricks components Signed-off-by: Prithvi Kannan <[email protected]> * add unit tests Signed-off-by: Prithvi Kannan <[email protected]> * format Signed-off-by: Prithvi Kannan <[email protected]> * ruff Signed-off-by: Prithvi Kannan <[email protected]> * fix Signed-off-by: Prithvi Kannan <[email protected]> * deps Signed-off-by: Prithvi Kannan <[email protected]> * ruff Signed-off-by: Prithvi Kannan <[email protected]> * mlflow Signed-off-by: Prithvi Kannan <[email protected]> --------- Signed-off-by: Prithvi Kannan <[email protected]>
databricks · Dec 9, 2024 · 1f43c17 · 1f43c17
1 parent 52333da
commit 1f43c17
Show file tree

Hide file tree

Showing 11 changed files with 3,078 additions and 25 deletions.
diff --git a/integrations/langchain/pyproject.toml b/integrations/langchain/pyproject.toml
@@ -11,7 +11,8 @@ requires-python = ">=3.9"
 dependencies = [
     "langchain>=0.2.0",
     "langchain-community>=0.2.0",
-    "langchain-databricks>=0.1.1",
+    "mlflow",
+    "databricks-vectorsearch>=0.40",
     "databricks-ai-bridge",
 ]
 

diff --git a/integrations/langchain/src/databricks_langchain/__init__.py b/integrations/langchain/src/databricks_langchain/__init__.py
@@ -1,11 +1,7 @@
-# Import modules from langchain-databricks
-from langchain_databricks import (
-    ChatDatabricks,
-    DatabricksEmbeddings,
-    DatabricksVectorSearch,
-)
-
-from .genie import GenieAgent
+from databricks_langchain.chat_models import ChatDatabricks
+from databricks_langchain.embeddings import DatabricksEmbeddings
+from databricks_langchain.genie import GenieAgent
+from databricks_langchain.vectorstores import DatabricksVectorSearch
 
 # Expose all integrations to users under databricks-langchain
 __all__ = [