chore: use uv for running tests

Author: fanzeyi

.github/workflows/main.yml

@@ -12,43 +12,45 @@ on:
       - ready_for_review
   workflow_dispatch:
 
+env:
+  UV_VERSION: "0.9.11"
+
 jobs:
   lint:
     runs-on: ubuntu-latest
     timeout-minutes: 10
     steps:
       - uses: actions/checkout@v4
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
         with:
-          python-version: "3.10"
+          version: ${{ env.UV_VERSION }}
       - name: Install dependencies
         run: |
-          pip install . --group lint
+          uv sync --group lint
       - name: Lint Python code with ruff
         run: |
-          ruff check .
-          ruff format --check .
+          uv run ruff check .
+          uv run ruff format --check .
 
   core_test:
     runs-on: ubuntu-latest
+    name: core_test python-${{ matrix.python-version }}
     strategy:
       matrix:
         python-version: ["3.10", "3.11", "3.12"]
     timeout-minutes: 20
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
         with:
+          version: ${{ env.UV_VERSION }}
           python-version: ${{ matrix.python-version }}
-      - name: Install dependencies
-        run: |
-          pip install ".[memory]" --group dev
       - name: Run tests
         run: |
-          pytest tests/
+          uv run --exact --group integration --extra memory pytest tests/
 
   langchain_test:
     runs-on: ubuntu-latest
@@ -59,17 +61,15 @@ jobs:
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
         with:
+          version: ${{ env.UV_VERSION }}
           python-version: ${{ matrix.python-version }}
-      - name: Install dependencies
-        run: |
-          pip install .
-          pip install integrations/langchain --group dev
       - name: Run tests
+        working-directory: integrations/langchain
         run: |
-          pytest integrations/langchain/tests/unit_tests
+          uv run --exact --group integration pytest tests/unit_tests
 
   lakebase_memory_test:
     runs-on: ubuntu-latest
@@ -80,18 +80,18 @@ jobs:
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
         with:
+          version: ${{ env.UV_VERSION }}
           python-version: ${{ matrix.python-version }}
-      - name: Install dependencies
+      - name: Run databricks-ai-bridge Tests
         run: |
-          pip install .[memory]
-          pip install "integrations/langchain[memory]" --group dev
-      - name: Run tests
+          uv run --exact --group integration --extra memory pytest tests/databricks_ai_bridge/test_lakebase.py
+      - name: Run databricks-langchain Tests
+        working-directory: integrations/langchain
         run: |
-          pytest tests/databricks_ai_bridge/test_lakebase.py
-          pytest integrations/langchain/tests/unit_tests/test_checkpoint.py
+          uv run --exact --group integration --extra memory pytest integrations/langchain/tests/unit_tests/test_checkpoint.py
 
   langchain_cross_version_test:
     runs-on: ubuntu-latest
@@ -149,17 +149,15 @@ jobs:
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
         with:
+          version: ${{ env.UV_VERSION }}
           python-version: ${{ matrix.python-version }}
-      - name: Install dependencies
-        run: |
-          pip install .
-          pip install integrations/openai --group dev
       - name: Run tests
+        working-directory: integrations/openai
         run: |
-          pytest integrations/openai/tests/unit_tests
+          uv run --exact --group integration pytest tests/unit_tests
 
   openai_cross_version_test:
     runs-on: ubuntu-latest
@@ -214,17 +212,15 @@ jobs:
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
         with:
+          version: ${{ env.UV_VERSION }}
           python-version: ${{ matrix.python-version }}
-      - name: Install dependencies
-        run: |
-          pip install .
-          pip install integrations/llamaindex --group dev
       - name: Run tests
+        working-directory: integrations/llamaindex
         run: |
-          pytest integrations/llamaindex/tests/unit_tests
+          uv run --exact --group integration pytest tests/unit_tests
 
   mcp_test:
     runs-on: ubuntu-latest
@@ -235,17 +231,15 @@ jobs:
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
         with:
+          version: ${{ env.UV_VERSION }}
           python-version: ${{ matrix.python-version }}
-      - name: Install dependencies
-        run: |
-          pip install .
-          pip install -e databricks_mcp --group dev
       - name: Run tests
+        working-directory: databricks_mcp
         run: |
-          pytest databricks_mcp/tests/unit_tests
+          uv run --exact --group integration pytest tests/unit_tests
 
   dspy_test:
     runs-on: ubuntu-latest
@@ -256,14 +250,12 @@ jobs:
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
         with:
+          version: ${{ env.UV_VERSION }}
           python-version: ${{ matrix.python-version }}
-      - name: Install dependencies
-        run: |
-          pip install .
-          pip install integrations/dspy --group dev
       - name: Run tests
+        working-directory: integrations/dspy
         run: |
-          pytest integrations/dspy/tests/unit_tests
+          uv run --exact --group integration pytest tests/unit_tests

databricks_mcp/pyproject.toml

@@ -17,21 +17,25 @@ dependencies = [
 
 [dependency-groups]
 dev = [
-  "pytest",
-  "pytest-asyncio",
   "typing_extensions",
   "databricks-sdk>=0.49.0",
   "ruff==0.6.4",
+  { include-group = "integration" }
 ]
 
 integration = [
+  "pytest",
+  "pytest-asyncio",
   "pytest-timeout>=2.3.1",
 ]
 
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
+[tool.uv.sources]
+databricks-ai-bridge = { path = "../", editable = true }
+
 [tool.hatch.build]
 include = [
   "src/databricks_mcp/*"
@@ -71,3 +75,10 @@ docstring-code-line-length = 88
 
 [tool.ruff.lint.pydocstyle]
 convention = "google"
+
+[tool.pytest.ini_options]
+filterwarnings = [
+    "ignore::Warning",
+    "default::Warning:databricks_mcp",
+    "default::Warning:tests",
+]

integrations/dspy/pyproject.toml

@@ -16,14 +16,20 @@ dependencies = [
 
 [dependency-groups]
 dev = [
-  "pytest",
   "ruff",
+  { include-group = "integration" }
+]
+integration = [
+  "pytest",
 ]
 
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
+[tool.uv.sources]
+databricks-ai-bridge = { path = "../../", editable = true }
+
 [tool.hatch.build]
 include = [
   "src/databricks_dspy/*"
@@ -60,3 +66,10 @@ select = [
 [tool.ruff.format]
 docstring-code-format = true
 docstring-code-line-length = 100
+
+[tool.pytest.ini_options]
+filterwarnings = [
+    "ignore::Warning",
+    "default::Warning:databricks_dspy",
+    "default::Warning:tests",
+]

integrations/langchain/pyproject.toml

@@ -27,12 +27,13 @@ memory = [
 
 [dependency-groups]
 dev = [
-  "pytest",
   "typing_extensions",
   "ruff==0.6.4",
+  { include-group = "integration" }
 ]
 
 integration = [
+  "pytest",
   "langgraph>=0.2.27",
   "pytest-timeout>=2.3.1",
   "pytest-asyncio",
@@ -44,6 +45,9 @@ integration = [
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
+[tool.uv.sources]
+databricks-ai-bridge = { path = "../../", editable = true }
+
 [tool.hatch.build]
 include = [
   "src/databricks_langchain/*"
@@ -83,3 +87,10 @@ docstring-code-line-length = 88
 
 [tool.ruff.lint.pydocstyle]
 convention = "google"
+
+[tool.pytest.ini_options]
+filterwarnings = [
+    "ignore::Warning",
+    "default::Warning:databricks_langchain",
+    "default::Warning:tests",
+]

integrations/llamaindex/pyproject.toml

@@ -17,20 +17,24 @@ dependencies = [
 
 [dependency-groups]
 dev = [
-    "pytest",
     "typing_extensions",
     "databricks-sdk>=0.34.0",
     "ruff==0.6.4",
+    { include-group = "integration" }
 ]
 
 integration = [
+    "pytest",
     "pytest-timeout>=2.3.1",
 ]
 
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
+[tool.uv.sources]
+databricks-ai-bridge = { path = "../../", editable = true }
+
 [tool.hatch.build]
 include = [
     "src/databricks_llamaindex/*"
@@ -70,3 +74,10 @@ docstring-code-line-length = 88
 
 [tool.ruff.lint.pydocstyle]
 convention = "google"
+
+[tool.pytest.ini_options]
+filterwarnings = [
+    "ignore::Warning",
+    "default::Warning:databricks_llamaindex",
+    "default::Warning:tests",
+]

integrations/openai/pyproject.toml

@@ -22,21 +22,25 @@ dependencies = [
 
 [dependency-groups]
 dev = [
-    "pytest",
-    "pytest-asyncio",
     "typing_extensions",
     "databricks-sdk>=0.34.0",
     "ruff==0.6.4",
+    { include-group = "integration" }
 ]
 
 integration = [
+    "pytest",
+    "pytest-asyncio",
     "pytest-timeout>=2.3.1",
 ]
 
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
+[tool.uv.sources]
+databricks-ai-bridge = { path = "../../", editable = true }
+
 [tool.hatch.build]
 include = [
     "src/databricks_openai/*"
@@ -76,3 +80,10 @@ docstring-code-line-length = 88
 
 [tool.ruff.lint.pydocstyle]
 convention = "google"
+
+[tool.pytest.ini_options]
+filterwarnings = [
+    "ignore::Warning",
+    "default::Warning:databricks_openai",
+    "default::Warning:tests",
+]