Imagen MCP Server

"""Model catalog resource for MCP.""" from Imagen_MCP.services.model_registry import get_model_registry def get_model_catalog() -> dict: """Get the complete model catalog. Returns: Dictionary containing all available models and their capabilities. """ registry = get_model_registry() return registry.to_catalog_dict() def get_model_info(model_id: str) -> dict | None: """Get information about a specific model. Args: model_id: The model identifier. Returns: Dictionary with model information, or None if not found. """ registry = get_model_registry() model = registry.get_model(model_id) if model: return model.model_dump() return None def list_model_ids() -> list[str]: """Get a list of all available model IDs. Returns: List of model identifiers. """ registry = get_model_registry() return [model.id for model in registry.get_all_models()]