Skip to content

Fix oss client create function (#1012) #28

Fix oss client create function (#1012)

Fix oss client create function (#1012) #28

name: Unity Catalog AI Databricks Integration Tests
on:
push:
branches:
- main
paths:
- ai/integrations/**
- .github/workflows/ucai-integration-tests.yml
pull_request:
paths:
- ai/integrations/**
- .github/workflows/ucai-integration-tests.yml
permissions:
contents: read
jobs:
crewai_test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ['3.10']
timeout-minutes: 20
defaults:
run:
working-directory: ai
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: |
pip install core/.
echo "PYTHONPATH=$(pwd)/core/src:\$PYTHONPATH" >> $GITHUB_ENV
pip install integrations/crewai[dev]
pip install mlflow
- name: Run tests
run: |
pytest integrations/crewai/tests --ignore=integrations/crewai/tests/test_crewai_toolkit_oss.py
langchain_test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ['3.9', '3.10']
timeout-minutes: 20
defaults:
run:
working-directory: ai
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
- name: Install Core Library and Langchain Integration
run: |
pip install core/.
echo "PYTHONPATH=$(pwd)/core/src:\$PYTHONPATH" >> $GITHUB_ENV
pip install integrations/langchain[dev]
pip install mlflow
- name: Run Langchain Tests
run: |
pytest integrations/langchain/tests --ignore=integrations/langchain/tests/test_langchain_toolkit_oss.py
anthropic_test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ['3.9', '3.10']
timeout-minutes: 20
defaults:
run:
working-directory: ai
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
- name: Install Core Library and Anthropic Integration
run: |
pip install core/.
echo "PYTHONPATH=$(pwd)/core/src:\$PYTHONPATH" >> $GITHUB_ENV
pip install integrations/anthropic[dev]
pip install mlflow
- name: Run Anthropic Tests
run: |
pytest integrations/anthropic/tests --ignore=integrations/anthropic/tests/test_anthropic_toolkit_oss.py
llamaindex_test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ['3.9', '3.10']
timeout-minutes: 20
defaults:
run:
working-directory: ai
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
- name: Install Core Library and LlamaIndex Integration
run: |
pip install core/.
echo "PYTHONPATH=$(pwd)/core/src:\$PYTHONPATH" >> $GITHUB_ENV
pip install integrations/llama_index[dev]
pip install mlflow
- name: Run LlamaIndex Tests
run: |
pytest integrations/llama_index/tests --ignore=integrations/llama_index/tests/test_llama_index_toolkit_oss.py
openai_test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ['3.9', '3.10']
timeout-minutes: 20
defaults:
run:
working-directory: ai
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
- name: Install Core Library and OpenAI Integration
run: |
pip install core/.
echo "PYTHONPATH=$(pwd)/core/src:\$PYTHONPATH" >> $GITHUB_ENV
pip install integrations/openai[dev]
pip install mlflow
- name: Run OpenAI Tests
run: |
pytest integrations/openai/tests --ignore=integrations/openai/tests/test_openai_toolkit_oss.py
autogen_test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ['3.10']
timeout-minutes: 20
defaults:
run:
working-directory: ai
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: |
pip install core/.
echo "PYTHONPATH=$(pwd)/core/src:\$PYTHONPATH" >> $GITHUB_ENV
pip install integrations/autogen[dev]
pip install mlflow
- name: Run tests
run: |
pytest integrations/autogen/tests --ignore=integrations/autogen/tests/test_autogen_toolkit_oss.py
gemini_test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ['3.9', '3.10']
timeout-minutes: 20
defaults:
run:
working-directory: ai
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: |
pip install core/.
echo "PYTHONPATH=$(pwd)/core/src:\$PYTHONPATH" >> $GITHUB_ENV
pip install integrations/gemini[dev]
pip install mlflow
- name: Run tests
run: |
pytest integrations/gemini/tests --ignore=integrations/gemini/tests/test_gemini_toolkit_oss.py
litellm_test:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: ['3.9', '3.10']
timeout-minutes: 20
defaults:
run:
working-directory: ai
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: |
pip install core/.
echo "PYTHONPATH=$(pwd)/core/src:\$PYTHONPATH" >> $GITHUB_ENV
pip install integrations/litellm[dev]
pip install mlflow
- name: Run tests
run: |
pytest integrations/litellm/tests --ignore=integrations/litellm/tests/test_litellm_toolkit_oss.py