Run tests with extra dependencies separately

.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 dependencies
+        run: uv sync -p .venv --extra dev --extra test_extras
+      - name: Run extra tests
+        run: uv run -p .venv pytest tests/ -m extra --extra
 
   build_package:
     name: Build Package

pyproject.toml

@@ -51,6 +51,7 @@ dependencies = [
 anthropic = ["anthropic>=0.18.0,<1.0.0"]
 weaviate = ["weaviate-client~=4.5.4"]
 aws = ["boto3~=1.34.78"]
+mcp = ["mcp; python_version >= '3.10'"]
 dev = [
     "pytest>=6.2.5",
     "pytest-mock>=3.12.0",
@@ -62,9 +63,10 @@ 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'"]
+test_extras = [
+    "mcp; python_version >= '3.10'",
+]
 
 [tool.setuptools.packages.find]
 where = ["."]

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"],