Make pydantic-monty + conda-forge-metadata optional (#187)

Author: dbast

conda_meta_mcp/tools/import_mapping.py

@@ -8,17 +8,35 @@
 from __future__ import annotations
 
 import asyncio
+from collections.abc import Callable
 from functools import lru_cache
 from typing import Any
 
-from conda_forge_metadata.autotick_bot.import_to_pkg import (
-    get_pkgs_for_import,
-    map_import_to_package,
-)
 from fastmcp.exceptions import ToolError
 
 from .registry import register_tool
 
+DISABLED_MESSAGE = "Disabled, enable via installing the package conda-forge-metadata"
+
+GetPkgsForImport = Callable[[str], tuple[set[str] | None, str]]
+MapImportToPackage = Callable[[str], str]
+get_pkgs_for_import: GetPkgsForImport | None
+map_import_to_package: MapImportToPackage | None
+
+try:
+    from conda_forge_metadata.autotick_bot.import_to_pkg import (
+        get_pkgs_for_import as _get_pkgs_for_import,
+    )
+    from conda_forge_metadata.autotick_bot.import_to_pkg import (
+        map_import_to_package as _map_import_to_package,
+    )
+except ImportError:
+    get_pkgs_for_import = None
+    map_import_to_package = None
+else:
+    get_pkgs_for_import = _get_pkgs_for_import
+    map_import_to_package = _map_import_to_package
+
 
 @lru_cache(maxsize=1024)
 def _map_import(import_name: str, get_keys: str = "") -> dict[str, Any]:
@@ -28,6 +46,8 @@ def _map_import(import_name: str, get_keys: str = "") -> dict[str, Any]:
     """
     if not import_name or not import_name.strip():
         raise ValueError("import_name must be a non-empty string")
+    if get_pkgs_for_import is None or map_import_to_package is None:
+        raise ToolError(DISABLED_MESSAGE)
 
     query = import_name.strip()
 
@@ -105,6 +125,8 @@ async def import_mapping(import_name: str, get_keys: str = "") -> dict[str, Any]
     """
     try:
         return await asyncio.to_thread(_map_import, import_name, get_keys)
+    except ToolError:
+        raise
     except ValueError as ve:
         raise ToolError(f"'import_mapping' invalid input: {ve}") from ve
     except Exception as e:  # pragma: no cover - generic protection

conda_meta_mcp/tools/pypi_to_conda.py

@@ -8,14 +8,28 @@
 from __future__ import annotations
 
 import asyncio
+from collections.abc import Callable
 from functools import lru_cache
 from typing import Any
 
-from conda_forge_metadata.autotick_bot.pypi_to_conda import map_pypi_to_conda
 from fastmcp.exceptions import ToolError
 
 from .registry import register_tool
 
+DISABLED_MESSAGE = "Disabled, enable via installing the package conda-forge-metadata"
+
+MapPypiToConda = Callable[[str], str]
+map_pypi_to_conda: MapPypiToConda | None
+
+try:
+    from conda_forge_metadata.autotick_bot.pypi_to_conda import (
+        map_pypi_to_conda as _map_pypi_to_conda,
+    )
+except ImportError:
+    map_pypi_to_conda = None
+else:
+    map_pypi_to_conda = _map_pypi_to_conda
+
 
 @lru_cache(maxsize=4096)
 def _map_pypi_name(pypi_name: str) -> dict[str, Any]:
@@ -25,6 +39,8 @@ def _map_pypi_name(pypi_name: str) -> dict[str, Any]:
     """
     if not pypi_name or not pypi_name.strip():
         raise ValueError("pypi_name must be a non-empty string")
+    if map_pypi_to_conda is None:
+        raise ToolError(DISABLED_MESSAGE)
 
     original = pypi_name.strip()
     conda_name = map_pypi_to_conda(original)
@@ -51,6 +67,8 @@ async def pypi_to_conda(pypi_name: str) -> dict[str, Any]:
     """
     try:
         return await asyncio.to_thread(_map_pypi_name, pypi_name)
+    except ToolError:
+        raise
     except ValueError as ve:
         raise ToolError(f"'pypi_to_conda' invalid input: {ve}") from ve
     except Exception as e:  # pragma: no cover

pixi.lock

@@ -11,7 +11,14 @@ classifiers = [
   "Programming Language :: Python :: 3.13",
 ]
 dependencies = [
-  "fastmcp",
+  "argparse-manpage>=4.7,<5",
+  "conda>=26.1,<27",
+  "conda-libmamba-solver",
+  "conda-package-streaming>=0.12.0,<0.13",
+  "fastmcp>=3.1.0",
+  "pydantic>=2",
+  "pyyaml>=6.0.3,<7",
+  "requests>=2",
 ]
 description = "A Model Context Protocol server providing conda eco system context"
 dynamic = ["version"]
@@ -20,6 +27,10 @@ name = "conda-meta-mcp"
 readme = "README.md"
 requires-python = ">=3.13"
 
+[project.optional-dependencies]
+code-mode = ["pydantic-monty"]
+conda-forge-metadata = ["conda-forge-metadata>=0.13,<0.14"]
+
 [project.scripts]
 cmm = "conda_meta_mcp.cli:main"
 
@@ -39,11 +50,13 @@ source = "vcs"
 argparse-manpage = ">=4.7,<5"
 conda = ">=26.1,<27"
 conda-forge-metadata = ">=0.13,<0.14"
+conda-libmamba-solver = "*"
 conda-package-streaming = ">=0.12.0,<0.13"
-fastmcp = ">=3.2.4,<3.3.0"
+fastmcp = ">=3.1.0,<3.3.0"
 libmambapy = ">=2.4.0,<3"
 pydantic-monty = "*"
 pyyaml = ">=6.0.3,<7"
+requests = ">=2,<3"
 
 [tool.pixi.environments]
 dev = ["test"]
@@ -67,7 +80,7 @@ prek = 'prek run --all-files --color=always'
 test = 'python -m pytest'
 
 [tool.pixi.pypi-dependencies]
-conda-meta-mcp = {path = ".", editable = true}
+conda-meta-mcp = {path = ".", editable = true, extras = ["code-mode", "conda-forge-metadata"]}
 
 [tool.pixi.workspace]
 channels = ["conda-forge"]

pyproject.toml

@@ -25,11 +25,12 @@ requirements:
   run:
     - python >=3.13
     - argparse-manpage >=4.7
-    - fastmcp >=2.13.1
+    - fastmcp >=3.1.0
     - conda >=26.1
     - conda-forge-metadata >=0.13.0
     - conda-package-streaming >=0.12.0
     - conda-libmamba-solver
+    - pydantic-monty
     - pyyaml >=6.0.3
     - requests >=2.28.0
 

recipe/meta.yaml

@@ -4,6 +4,8 @@
 from fastmcp import Client
 from fastmcp.exceptions import ToolError
 
+from conda_meta_mcp.tools import import_mapping as import_mapping_module
+
 # Heuristic labels the tool may legitimately emit. Keeping a central set makes the
 # success test resilient to minor internal mapping changes upstream.
 VALID_HEURISTICS = {
@@ -65,6 +67,18 @@ async def test_import_mapping__error_on_empty_input(server):
         assert "invalid input" in str(exc.value).lower()
 
 
+@pytest.mark.asyncio
+async def test_import_mapping__disabled_when_dependency_missing(monkeypatch):
+    import_mapping_module._map_import.cache_clear()
+    monkeypatch.setattr(import_mapping_module, "get_pkgs_for_import", None)
+
+    with pytest.raises(ToolError) as exc:
+        await import_mapping_module.import_mapping("numpy")
+
+    assert str(exc.value) == "Disabled, enable via installing the package conda-forge-metadata"
+    import_mapping_module._map_import.cache_clear()
+
+
 @pytest.mark.asyncio
 async def test_import_mapping__get_keys_empty_returns_all(server):
     """Empty get_keys should return all fields (backward compatible)."""

tests/tools/test_import_mapping.py

@@ -4,6 +4,8 @@
 from fastmcp import Client
 from fastmcp.exceptions import ToolError
 
+from conda_meta_mcp.tools import pypi_to_conda as pypi_to_conda_module
+
 
 @pytest.mark.asyncio
 @pytest.mark.parametrize(
@@ -47,3 +49,15 @@ async def test_pypi_to_conda__error_empty_input(server):
     async with Client(server) as client:
         with pytest.raises(ToolError):
             await client.call_tool("pypi_to_conda", {"pypi_name": ""})
+
+
+@pytest.mark.asyncio
+async def test_pypi_to_conda__disabled_when_dependency_missing(monkeypatch):
+    pypi_to_conda_module._map_pypi_name.cache_clear()
+    monkeypatch.setattr(pypi_to_conda_module, "map_pypi_to_conda", None)
+
+    with pytest.raises(ToolError) as exc:
+        await pypi_to_conda_module.pypi_to_conda("authzed")
+
+    assert str(exc.value) == "Disabled, enable via installing the package conda-forge-metadata"
+    pypi_to_conda_module._map_pypi_name.cache_clear()