Add langchain-databricks into databricks-langchain

Signed-off-by: Prithvi Kannan <[email protected]>
databricks · Oct 29, 2024 · 519a0a0 · 519a0a0
1 parent 2e7f619
commit 519a0a0
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/integrations/langchain/pyproject.toml b/integrations/langchain/pyproject.toml
@@ -11,6 +11,7 @@ requires-python = ">=3.8"
 dependencies = [
     "langchain>=0.2.0",
     "langchain-community>=0.2.0",
+    "langchain-databricks>=0.1.1",
     "databricks-ai-bridge",
 ]
 

diff --git a/integrations/langchain/src/databricks_langchain/__init__.py b/integrations/langchain/src/databricks_langchain/__init__.py
@@ -0,0 +1,17 @@
+# Import modules from langchain-databricks
+from langchain_databricks import (
+    ChatDatabricks,
+    DatabricksEmbeddings,
+    DatabricksVectorSearch,
+)
+
+# Import any additional modules specific to databricks-langchain
+from .genie import GenieAgent  # Example for GenieAgent functionality
+
+# Expose all integrations to users under databricks-langchain
+__all__ = [
+    "ChatDatabricks",
+    "DatabricksEmbeddings",
+    "DatabricksVectorSearch",
+    "GenieAgent",
+]