Add langchain-databricks into databricks-langchain (#12)

* Add langchain-databricks into databricks-langchain Signed-off-by: Prithvi Kannan <[email protected]> * fix Signed-off-by: Prithvi Kannan <[email protected]> * min python 3.9 Signed-off-by: Prithvi Kannan <[email protected]> * fix Signed-off-by: Prithvi Kannan <[email protected]> --------- Signed-off-by: Prithvi Kannan <[email protected]>
databricks · Oct 30, 2024 · 9b29c88 · 9b29c88
1 parent 2e7f619
commit 9b29c88
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 4 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -20,7 +20,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
-          python-version: '3.8'
+          python-version: '3.9'
       - name: Install dependencies
         run: |
           pip install -r requirements/lint-requirements.txt
@@ -33,7 +33,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.8', '3.9', '3.10']
+        python-version: ['3.9', '3.10']
     timeout-minutes: 20
     steps:
       - name: Checkout code
@@ -53,7 +53,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.8', '3.9', '3.10']
+        python-version: ['3.9', '3.10']
     timeout-minutes: 20
     steps:
       - name: Checkout code

diff --git a/integrations/langchain/pyproject.toml b/integrations/langchain/pyproject.toml
@@ -7,10 +7,11 @@ authors = [
 ]
 readme = "README.md"
 license = { text="Apache-2.0" }
-requires-python = ">=3.8"
+requires-python = ">=3.9"
 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,16 @@
+# Import modules from langchain-databricks
+from langchain_databricks import (
+    ChatDatabricks,
+    DatabricksEmbeddings,
+    DatabricksVectorSearch,
+)
+
+from .genie import GenieAgent
+
+# Expose all integrations to users under databricks-langchain
+__all__ = [
+    "ChatDatabricks",
+    "DatabricksEmbeddings",
+    "DatabricksVectorSearch",
+    "GenieAgent",
+]