Fix oss client create function (#1012) #28
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| 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 |