From aabd7173096ef5651be1ec4f5661f6c0eb776809 Mon Sep 17 00:00:00 2001
From: Praveen Kuttappan <praveen.kuttappan@gmail.com>
Date: Fri, 13 Jun 2025 12:12:27 -0400
Subject: [PATCH] Update script to support running on GH action

---
 eng/common/mcp/azure-sdk-mcp.ps1 | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/eng/common/mcp/azure-sdk-mcp.ps1 b/eng/common/mcp/azure-sdk-mcp.ps1
index d78c61a751..caca58ca24 100755
--- a/eng/common/mcp/azure-sdk-mcp.ps1
+++ b/eng/common/mcp/azure-sdk-mcp.ps1
@@ -4,7 +4,7 @@ param(
     [string]$FileName = 'Azure.Sdk.Tools.Cli',
     [string]$Package = 'azsdk',
     [string]$Version, # Default to latest
-    [string]$InstallDirectory = (Join-Path $HOME ".azure-sdk-mcp" "azsdk"),
+    [string]$InstallDirectory = '',
     [string]$Repository = 'Azure/azure-sdk-tools',
     [string]$RunDirectory = (Resolve-Path (Join-Path $PSScriptRoot .. .. ..)),
     [switch]$Run,
@@ -14,6 +14,11 @@ param(
 
 $ErrorActionPreference = "Stop"
 
+if (-not $InstallDirectory)
+{
+    $homeDir = if ($env:HOME) { $env:HOME } else { $env:USERPROFILE }
+    $InstallDirectory = (Join-Path $homeDir ".azure-sdk-mcp" "azsdk")
+}
 . (Join-Path $PSScriptRoot '..' 'scripts' 'Helpers' 'AzSdkTool-Helpers.ps1')
 
 if ($Clean) {