run tests with extra dependencies separately

Author: TomeHirata

.github/workflows/run_tests.yml

@@ -83,6 +83,10 @@ jobs:
           args: check --fix-only
       - name: Run tests with pytest
         run: uv run -p .venv pytest tests/
+      - name: Install optional depencies
+        run: uv sync -p .venv --extra mcp
+      - name: Run extra tests
+        run: uv run -p .venv pytest tests/ -m extra --extra
 
   build_package:
     name: Build Package

pyproject.toml

@@ -62,7 +62,6 @@ dev = [
     "build>=1.0.3",
     "litellm>=1.60.3; sys_platform == 'win32'",
     "litellm[proxy]>=1.60.3; sys_platform != 'win32'",
-    "mcp>=1.5.0; python_version >= '3.10'",
 ]
 mcp = ["mcp; python_version >= '3.10'"]
 

tests/conftest.py

@@ -4,7 +4,7 @@
 
 from tests.test_utils.server import litellm_test_server, read_litellm_test_server_request_logs  # noqa: F401
 
-SKIP_DEFAULT_FLAGS = ["reliability"]
+SKIP_DEFAULT_FLAGS = ["reliability", "extra"]
 
 
 @pytest.fixture(autouse=True)

tests/utils/test_mcp.py

@@ -1,14 +1,19 @@
 import asyncio
+import importlib
 
 import pytest
-from mcp import ClientSession, StdioServerParameters
-from mcp.client.stdio import stdio_client
 
 from dspy.utils.mcp import convert_mcp_tool
 
+if importlib.util.find_spec("mcp") is None:
+    pytest.skip(reason="mcp is not installed", allow_module_level=True)
+
 
 @pytest.mark.asyncio
+@pytest.mark.extra
 async def test_convert_mcp_tool():
+    from mcp import ClientSession, StdioServerParameters
+    from mcp.client.stdio import stdio_client
     server_params = StdioServerParameters(
         command="python",
         args=["tests/utils/resources/mcp_server.py"],