diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 3e91c38..fa5e819 100644 --- 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 index 40527ed..2b2e31f 100644 --- 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 index e69de29..4b7a9da 100644 --- 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", +]