Merge branch 'main' into rads-1996/rate-limited-sampler-exporter

Author: rads-1996

.github/copilot-instructions.md

@@ -73,9 +73,9 @@ curl -s "https://api.github.com/repos/Azure/azure-rest-api-specs/commits?path=<p
 ## EXECUTION SEQUENCE - 7 MANDATORY STEPS
 
 **ESTIMATED TOTAL TIME: 10-15 minutes**
-- SDK Generation: 5-6 minutes
-- Static Validation: 3-5 minutes  
-- Documentation & Commit: 2-4 minutes
+- SDK Generation: ~2 minutes
+- Static Validation: ~3-5 minutes
+- Documentation & Commit: ~2-4 minutes
 
 **ALWAYS inform users of time expectations before starting any long-running operations.**
 
@@ -89,18 +89,14 @@ IF missing dependencies:
 
 ### STEP 2: SDK GENERATION
 ```
-ACTION: Use azure-sdk-python-mcp sdk generation server tools (init, init_local)
-TIMING: ALWAYS inform user before starting: "This SDK generation step will take approximately 5-6 minutes to complete."
-IF local path provided:
-    USE local mcp tools with tspconfig.yaml path
+ACTION: Use azure-sdk-python-mcp sdk generation server tools (init for new packages, update for existing packages)
 IF commands fail:
     ANALYZE error messages
     DIRECT user to fix TypeSpec errors in source repo
 ```
 
 ### STEP 3: STATIC VALIDATION (SEQUENTIAL)
 ```
-TIMING: Inform user: "Static validation will take approximately 3-5 minutes for each step."
 FOR EACH validation step:
     RUN validation (tox mcp tool)
     IF errors/warnings found:

eng/tools/mcp/azure-sdk-python-mcp/main.py

@@ -196,22 +196,28 @@ def tox_tool(package_path: str, environment: str, repo_path: str, tox_ini_path:
     return run_command(command, cwd=repo_path)
 
 @mcp.tool("init")
-def init_tool(tsp_config_url: str, repo_path: str) -> Dict[str, Any]:
-    """Initializes and generates a typespec client library directory given the url.
+def init_tool(tsp_config_path: str, repo_path: str, is_local: bool = False) -> Dict[str, Any]:
+    """Initializes a new typespec client library directory.
     
     Args:
-        tsp_config_url: The URL to the tspconfig.yaml file.
-        repo_path: The path to the repository root (i.e. ./azure-sdk-for-python/).
+        :param str tsp_config_path: Either the URL to the tspconfig.yaml file or the path to a local tspconfig.yaml file.
+        :param str repo_path: The path to the repository root (i.e. ./azure-sdk-for-python/).
+        :param bool is_local: Whether the tsp_config_path is a local file path (True) or a URL (False).
+
     Returns:
         A dictionary containing the result of the command.
     """
     try:
-        # Get updated URL with latest commit hash
-        updated_url = get_latest_commit(tsp_config_url)
+        config_path = tsp_config_path
+        
+        # If not a local path, get updated URL with latest commit hash
+        if not is_local and tsp_config_path.startswith("http"):
+            config_path = get_latest_commit(tsp_config_path)
+            logger.info(f"Using updated TypeSpec config URL with latest commit: {config_path}")
 
         # Run the init command using the combined function
         return run_command(["init"], cwd=repo_path, is_typespec=True,
-                          typespec_args={"tsp-config": updated_url})
+                          typespec_args={"tsp-config": config_path})
 
     except RuntimeError as e:
         return {
@@ -221,25 +227,42 @@ def init_tool(tsp_config_url: str, repo_path: str) -> Dict[str, Any]:
             "stderr": "",
             "code": 1
         }
+    
+@mcp.tool("update")
+def update_tool(package_path: str, commit_hash: Optional[str] = None, repo: Optional[str] = None, 
+                tsp_config: Optional[str] = None, local_spec: Optional[str] = None) -> Dict[str, Any]:
+    """Updates an existing client library with local or remote TypeSpec changes.
 
-@mcp.tool("init_local")
-def init_local_tool(tsp_config_path: str, repo_path:str) -> Dict[str, Any]:
-    """Initializes and subsequently generates a typespec client library directory from a local azure-rest-api-specs repo.
-
-    This command is used to generate a client library from a local azure-rest-api-specs repository. No additional
-    commands are needed to generate the client library.
+    This command is used to update an existing client library.
 
     Args:
-        tsp_config_path: The path to the local tspconfig.yaml file.
-        repo_path: The path to the repository root (i.e. ./azure-sdk-for-python/).
+        :param str package_path: The path to the directory of the tsp-location.yaml (i.e. ./azure-sdk-for-python/sdk/eventgrid/azure-eventgrid).
+        :param Optional[str] commit_hash: Optional. The commit hash to update to.
+        :param Optional[str] repo: Optional. Repository where the project is defined.
+        :param Optional[str] tsp_config: Optional. Path to tspconfig.yaml.
+        :param Optional[str] local_spec: Optional. Path to the local TypeSpec project (../azure-rest-api-specs/specification/eventgrid/Azure.Messaging.EventGrid/).
 
     Returns:
-        A dictionary containing the result of the command.    """
+        A dictionary containing the result of the command.
+    """
     try:
+        # Build TypeSpec arguments
+        typespec_args = {
+            "output-dir": package_path
+        }
 
-        # Run the init command with local path using the combined function
-        return run_command(["init"], cwd=repo_path, is_typespec=True,
-                          typespec_args={"tsp-config": tsp_config_path})
+        if commit_hash:
+            typespec_args["commit"] = commit_hash
+        if repo:
+            typespec_args["repo"] = repo
+        if tsp_config:
+            typespec_args["tsp-config"] = tsp_config
+        if local_spec:
+            typespec_args["local-spec-repo"] = local_spec
+        
+        # Run the update command
+        return run_command(["update"], cwd=package_path, is_typespec=True,
+                          typespec_args=typespec_args)
 
     except RuntimeError as e:
         return {
@@ -250,7 +273,6 @@ def init_local_tool(tsp_config_path: str, repo_path:str) -> Dict[str, Any]:
             "code": 1
         }
 
-
 @mcp.tool("check_library_health")
 def check_library_health(library_name: str) -> Dict[str, Any]:
     """Checks the health status of a client library.

eng/tools/mcp/azure-sdk-python-mcp/pyproject.toml

@@ -1,9 +1,9 @@
 [project]
 name = "azure-sdk-python-mcp"
-version = "0.1.2"
+version = "0.1.3"
 description = "MCP server for Azure SDK for Python"
 readme = "README.md"
-requires-python = "==3.10.*"
+requires-python = ">=3.10"
 dependencies = [
     "azure-core>=1.34.0",
     "mcp[cli]>=1.9.2",