feat: make domains configurable and discoverable

Author: pengqun

src/modelscope_mcp_server/constants.py

@@ -1,13 +1,8 @@
 """Global constants for ModelScope MCP Server."""
 
 # ModelScope domains
-MODELSCOPE_DOMAIN = "https://modelscope.cn"
-MODELSCOPE_API_INFERENCE_DOMAIN = "https://api-inference.modelscope.cn"
-
-# ModelScope API endpoints
-MODELSCOPE_API_ENDPOINT = f"{MODELSCOPE_DOMAIN}/api/v1"
-MODELSCOPE_OPENAPI_ENDPOINT = f"{MODELSCOPE_DOMAIN}/openapi/v1"
-MODELSCOPE_API_INFERENCE_ENDPOINT = f"{MODELSCOPE_API_INFERENCE_DOMAIN}/v1"
+DEFAULT_MODELSCOPE_DOMAIN = "https://modelscope.cn"
+DEFAULT_MODELSCOPE_API_INFERENCE_DOMAIN = "https://api-inference.modelscope.cn"
 
 # Default model IDs for content generation
 DEFAULT_TEXT_TO_IMAGE_MODEL = "MusePublic/489_ckpt_FLUX_1"

src/modelscope_mcp_server/settings.py

@@ -7,10 +7,9 @@
     DEFAULT_API_TIMEOUT_SECONDS,
     DEFAULT_IMAGE_GENERATION_TIMEOUT_SECONDS,
     DEFAULT_IMAGE_TO_IMAGE_MODEL,
+    DEFAULT_MODELSCOPE_API_INFERENCE_DOMAIN,
+    DEFAULT_MODELSCOPE_DOMAIN,
     DEFAULT_TEXT_TO_IMAGE_MODEL,
-    MODELSCOPE_API_ENDPOINT,
-    MODELSCOPE_API_INFERENCE_ENDPOINT,
-    MODELSCOPE_OPENAPI_ENDPOINT,
 )
 
 
@@ -25,19 +24,17 @@ class Settings(BaseSettings):
         extra="ignore",
     )
 
-    # ModelScope API settings
+    # Authentication settings
     api_token: str | None = Field(default=None, description="ModelScope API token for authentication")
-    api_base_url: str = Field(
-        default=MODELSCOPE_API_ENDPOINT,
-        description="Base URL for ModelScope API",
-    )
-    openapi_base_url: str = Field(
-        default=MODELSCOPE_OPENAPI_ENDPOINT,
-        description="Base URL for ModelScope OpenAPI",
+
+    # Domain settings
+    main_domain: str = Field(
+        default=DEFAULT_MODELSCOPE_DOMAIN,
+        description="ModelScope website domain",
     )
-    api_inference_base_url: str = Field(
-        default=MODELSCOPE_API_INFERENCE_ENDPOINT,
-        description="Base URL for ModelScope API Inference",
+    api_inference_domain: str = Field(
+        default=DEFAULT_MODELSCOPE_API_INFERENCE_DOMAIN,
+        description="ModelScope API inference domain",
     )
 
     # Default model settings
@@ -99,9 +96,8 @@ def show_settings(self) -> None:
             self.api_token[:-8] + "********" if self.api_token and len(self.api_token) > 4 else "Not configured"
         )
         print(f"  • Token: {masked_token}")
-        print(f"  • Base URL: {self.api_base_url}")
-        print(f"  • OpenAPI URL: {self.openapi_base_url}")
-        print(f"  • Inference URL: {self.api_inference_base_url}")
+        print(f"  • Main Domain: {self.main_domain}")
+        print(f"  • API Inference Domain: {self.api_inference_domain}")
         print()
 
         # Default Models

src/modelscope_mcp_server/tools/aigc.py

@@ -76,7 +76,7 @@ async def generate_image(
         if not settings.is_api_token_configured():
             raise ValueError("API token is not set")
 
-        url = f"{settings.api_inference_base_url}/images/generations"
+        url = f"{settings.api_inference_domain}/v1/images/generations"
 
         payload = {
             "model": model,

src/modelscope_mcp_server/tools/context.py

@@ -39,7 +39,7 @@ async def get_current_user() -> UserInfo:
         if not settings.is_api_token_configured():
             return UserInfo(authenticated=False, reason="API token is not set")
 
-        url = f"{settings.api_base_url}/users/login/info"
+        url = f"{settings.main_domain}/api/v1/users/login/info"
 
         try:
             response = default_client.get(url)
@@ -78,4 +78,8 @@ async def get_environment_info() -> EnvironmentInfo:
             fastmcp_version=get_fastmcp_version(),
             mcp_protocol_version=get_mcp_protocol_version(),
             python_version=get_python_version(),
+            main_domain=settings.main_domain,
+            api_inference_domain=settings.api_inference_domain,
+            default_text_to_image_model=settings.default_text_to_image_model,
+            default_image_to_image_model=settings.default_image_to_image_model,
         )

src/modelscope_mcp_server/tools/mcp.py

@@ -10,7 +10,7 @@
 from pydantic import Field
 
 from ..client import default_client
-from ..constants import MODELSCOPE_DOMAIN, MODELSCOPE_OPENAPI_ENDPOINT
+from ..settings import settings
 from ..types import McpServer
 
 logger = logging.get_logger(__name__)
@@ -63,7 +63,7 @@ async def search_mcp_servers(
         limit: Annotated[int, Field(description="Maximum number of servers to return", ge=1, le=100)] = 10,
     ) -> list[McpServer]:
         """Search for MCP servers on ModelScope."""
-        url = f"{MODELSCOPE_OPENAPI_ENDPOINT}/mcp/servers"
+        url = f"{settings.main_domain}/openapi/v1/mcp/servers"
 
         # Build filter object
         filter_obj = {}
@@ -86,7 +86,7 @@ async def search_mcp_servers(
         servers = []
         for server_data in servers_data:
             id = server_data.get("id", "")
-            modelscope_url = f"{MODELSCOPE_DOMAIN}/mcp/servers/{id}"
+            modelscope_url = f"{settings.main_domain}/mcp/servers/{id}"
 
             server = McpServer(
                 id=id,

src/modelscope_mcp_server/tools/model.py

@@ -53,7 +53,7 @@ async def search_models(
         limit: Annotated[int, Field(description="Maximum number of models to return", ge=1, le=30)] = 10,
     ) -> list[Model]:
         """Search for models on ModelScope."""
-        url = f"{settings.api_base_url}/dolphin/models"
+        url = f"{settings.main_domain}/api/v1/dolphin/models"
 
         # Build criterion for task filter
         criterion = []

src/modelscope_mcp_server/tools/paper.py

@@ -10,7 +10,6 @@
 from pydantic import Field
 
 from ..client import default_client
-from ..constants import MODELSCOPE_DOMAIN
 from ..settings import settings
 from ..types import Paper
 
@@ -39,7 +38,7 @@ async def search_papers(
         limit: Annotated[int, Field(description="Maximum number of papers to return", ge=1, le=100)] = 10,
     ) -> list[Paper]:
         """Search for papers on ModelScope."""
-        url = f"{settings.api_base_url}/dolphin/papers"
+        url = f"{settings.main_domain}/api/v1/dolphin/papers"
 
         request_data = {
             "Query": query,
@@ -56,7 +55,7 @@ async def search_papers(
         papers = []
         for paper_data in papers_data:
             arxiv_id = paper_data.get("ArxivId")
-            modelscope_url = f"{MODELSCOPE_DOMAIN}/papers/{arxiv_id}"
+            modelscope_url = f"{settings.main_domain}/papers/{arxiv_id}"
 
             paper = Paper(
                 arxiv_id=arxiv_id,

src/modelscope_mcp_server/types.py

@@ -98,7 +98,16 @@ class ImageGenerationResult(BaseModel):
 class EnvironmentInfo(BaseModel):
     """Environment information."""
 
+    # Versions
     server_version: Annotated[str, Field(description="ModelScope MCP Server version")]
     fastmcp_version: Annotated[str, Field(description="FastMCP framework version")]
     mcp_protocol_version: Annotated[str, Field(description="MCP protocol version")]
     python_version: Annotated[str, Field(description="Python runtime version")]
+
+    # Domains
+    main_domain: Annotated[str, Field(description="ModelScope website domain")]
+    api_inference_domain: Annotated[str, Field(description="ModelScope API inference domain")]
+
+    # Settings
+    default_text_to_image_model: Annotated[str, Field(description="Default text-to-image model")]
+    default_image_to_image_model: Annotated[str, Field(description="Default image-to-image model")]