update comment

app.py

@@ -57,6 +57,7 @@ def create_interface():
                 exp_create_button = gr.Button("Create Experiment")
                 
                 def create_exp_with_tags(name, tags_str):
+                    """Create a new experiment. Given the name and tags"""
                     try:
                         tags = json.loads(tags_str) if tags_str else None
                     except json.JSONDecodeError:

mcp_mlflow_tools.py

@@ -43,10 +43,9 @@ def get_system_info() -> Dict:
         return {"error": True, "message": f"Failed to fetch system info: {str(e)}"}
 
 def list_experiments(name_contains: Optional[str] = "", max_results: Optional[int] = 100) -> Dict:
+    """List all experiments in the MLflow tracking server, with optional filtering. Includes run count for each experiment."""
+
     """
-    List all experiments in the MLflow tracking server, with optional filtering.
-    Includes run count for each experiment.
-    
     Args:
         name_contains: Optional filter to only include experiments whose names contain this string (case-insensitive).
         max_results: Maximum number of results to return (default: 100). None means no limit after filtering.
@@ -141,7 +140,7 @@ def list_experiments(name_contains: Optional[str] = "", max_results: Optional[in
         return {"error": True, "message": error_msg}
 
 def create_experiment(name: str, tags: Optional[Dict[str, str]] = None) -> Dict:
-    """Create a new experiment."""
+    """Create a new experiment. Given the name and tags"""
     if not name:
         return {"error": True, "message": "Experiment name cannot be empty"}
     
@@ -160,9 +159,9 @@ def search_runs(
     order_string: Optional[str] = None,
     max_results: int = 100
 ) -> Dict:
-    """
-    Search runs in a given experiment, with filtering and ordering.
+    """Search runs in a given experiment, with filtering and ordering."""
 
+    """
     Args:
         experiment_id: The ID of the experiment to search runs in.
         filter_string: A filter query string used to search for runs. 
@@ -320,7 +319,6 @@ def register_model(
 ) -> Dict:
     """
     Register a model from a run, with optional description and tags.
-    Assumes artifact path is 'model'.
     """
     if not all([run_id, model_name]):
         return {"error": True, "message": "Run ID and model name must be non-empty"}