feat: add modelscope_url to user and model info

Author: pengqun

src/modelscope_mcp_server/tools/context.py

@@ -53,12 +53,16 @@ async def get_current_user() -> UserInfo:
 
         user_data = response.get("Data", {})
 
+        username = user_data.get("Name")
+        modelscope_url = f"{settings.main_domain}/profile/{username}"
+
         return UserInfo(
             authenticated=True,
-            username=user_data.get("Name"),
+            username=username,
             email=user_data.get("Email"),
             avatar_url=user_data.get("Avatar"),
             description=user_data.get("Description") or "",
+            modelscope_url=modelscope_url,
         )
 
     @mcp.tool(

src/modelscope_mcp_server/tools/model.py

@@ -106,6 +106,7 @@ async def search_models(
         for model_data in models_data:
             path = model_data.get("Path", "")
             name = model_data.get("Name", "")
+            modelscope_url = f"{settings.main_domain}/models/{path}/{name}"
 
             if not path or not name:
                 logger.warning(f"Skipping model with invalid path or name: {model_data}")
@@ -117,6 +118,7 @@ async def search_models(
                 name=name,
                 chinese_name=model_data.get("ChineseName", ""),
                 created_by=model_data.get("CreatedBy"),
+                modelscope_url=modelscope_url,
                 # Non-empty value means True, else False
                 support_inference=bool(model_data.get("SupportInference", "")),
                 downloads_count=model_data.get("Downloads", 0),

src/modelscope_mcp_server/types.py

@@ -16,13 +16,19 @@ class GenerationType(str, Enum):
 class UserInfo(BaseModel):
     """User information."""
 
+    # Authentication result
     authenticated: Annotated[bool, Field(description="Whether the user is authenticated")]
     reason: Annotated[str | None, Field(description="Reason for failed authentication")] = None
+
+    # Basic information
     username: Annotated[str | None, Field(description="Username")] = None
     email: Annotated[str | None, Field(description="Email")] = None
     avatar_url: Annotated[str | None, Field(description="Avatar URL")] = None
     description: Annotated[str | None, Field(description="Description")] = None
 
+    # Links
+    modelscope_url: Annotated[str | None, Field(description="Profile page URL on ModelScope")] = None
+
 
 class Model(BaseModel):
     """Model information."""
@@ -34,6 +40,9 @@ class Model(BaseModel):
     chinese_name: Annotated[str, Field(description="Chinese name")]
     created_by: Annotated[str, Field(description="User who created the model")]
 
+    # Links
+    modelscope_url: Annotated[str, Field(description="Detail page URL on ModelScope")]
+
     # Capabilities
     support_inference: Annotated[bool, Field(description="Whether the model supports inference API")] = False
 
@@ -58,7 +67,7 @@ class Paper(BaseModel):
     abstract_en: Annotated[str, Field(description="Abstract in English")]
 
     # Links
-    modelscope_url: Annotated[str, Field(description="ModelScope page URL")]
+    modelscope_url: Annotated[str, Field(description="Detail page URL on ModelScope")]
     arxiv_url: Annotated[str, Field(description="Arxiv page URL")]
     pdf_url: Annotated[str, Field(description="PDF URL")]
     code_link: Annotated[str | None, Field(description="Code link")] = None
@@ -81,7 +90,7 @@ class McpServer(BaseModel):
     tags: Annotated[list[str], Field(description="Tags")] = []
 
     # Links
-    modelscope_url: Annotated[str, Field(description="ModelScope page URL")]
+    modelscope_url: Annotated[str, Field(description="Detail page URL on ModelScope")]
 
     # Metrics
     view_count: Annotated[int, Field(description="View count")] = 0

tests/tools/test_get_current_user.py

@@ -22,6 +22,7 @@ async def test_get_current_user_success(mcp_server):
         assert user_info.authenticated is True, "User should be authenticated"
         assert user_info.username is not None, "Username should be present"
         assert user_info.email is not None, "Email should be present"
+        assert user_info.modelscope_url is not None, "ModelScope URL should be present"
 
 
 @pytest.mark.integration

tests/tools/test_search_models.py

@@ -42,6 +42,7 @@ def validate_model_fields(model):
         "path",
         "chinese_name",
         "created_by",
+        "modelscope_url",
         "support_inference",
         "downloads_count",
         "stars_count",