feat: Update AzureCliCredential to include process timeout for improved stability

Author: Pavan-Microsoft

infra/scripts/index_scripts/01_create_search_index.py

@@ -41,7 +41,7 @@ def create_search_index():
     - Semantic search using prioritized fields
     """
     # Shared credential
-    credential = AzureCliCredential()
+    credential = AzureCliCredential(process_timeout=30)
 
     index_client = SearchIndexClient(endpoint=SEARCH_ENDPOINT, credential=credential)
 

infra/scripts/index_scripts/02_create_cu_template_audio.py

@@ -22,7 +22,7 @@
 # Add parent directory to path for imports
 sys.path.append(str(Path.cwd().parent))
 
-credential = AzureCliCredential()
+credential = AzureCliCredential(process_timeout=30)
 # Initialize Content Understanding Client
 token_provider = get_bearer_token_provider(credential, "https://cognitiveservices.azure.com/.default")
 client = AzureContentUnderstandingClient(

infra/scripts/index_scripts/02_create_cu_template_text.py

@@ -17,7 +17,7 @@
 
 ANALYZER_TEMPLATE_FILE = 'infra/data/ckm-analyzer_config_text.json'
 
-credential = AzureCliCredential()
+credential = AzureCliCredential(process_timeout=30)
 # Initialize Content Understanding Client
 token_provider = get_bearer_token_provider(credential, "https://cognitiveservices.azure.com/.default")
 client = AzureContentUnderstandingClient(

infra/scripts/index_scripts/03_cu_process_data_text.py

@@ -53,15 +53,15 @@
 
 # Azure DataLake setup
 account_url = f"https://{STORAGE_ACCOUNT_NAME}.dfs.core.windows.net"
-credential = AzureCliCredential()
+credential = AzureCliCredential(process_timeout=30)
 service_client = DataLakeServiceClient(account_url, credential=credential, api_version='2023-01-03')
 file_system_client = service_client.get_file_system_client(FILE_SYSTEM_CLIENT_NAME)
 directory_name = DIRECTORY
 paths = list(file_system_client.get_paths(path=directory_name))
 print("Azure DataLake setup complete.")
 
 # Azure Search setup
-search_credential = AzureCliCredential()
+search_credential = AzureCliCredential(process_timeout=30)
 search_client = SearchClient(SEARCH_ENDPOINT, INDEX_NAME, search_credential)
 index_client = SearchIndexClient(endpoint=SEARCH_ENDPOINT, credential=search_credential)
 print("Azure Search setup complete.")
@@ -162,7 +162,7 @@ def generate_sql_insert_script(df, table_name, columns, sql_file_name):
 
 
 # Content Understanding client
-cu_credential = AzureCliCredential()
+cu_credential = AzureCliCredential(process_timeout=30)
 cu_token_provider = get_bearer_token_provider(cu_credential, "https://cognitiveservices.azure.com/.default")
 cu_client = AzureContentUnderstandingClient(
     endpoint=CU_ENDPOINT,

infra/scripts/index_scripts/04_cu_process_custom_data.py

@@ -69,15 +69,15 @@
 
 # Azure DataLake setup
 account_url = f"https://{STORAGE_ACCOUNT_NAME}.dfs.core.windows.net"
-credential = AzureCliCredential()
+credential = AzureCliCredential(process_timeout=30)
 service_client = DataLakeServiceClient(account_url, credential=credential, api_version='2023-01-03')
 file_system_client = service_client.get_file_system_client(FILE_SYSTEM_CLIENT_NAME)
 directory_name = DIRECTORY
 paths = list(file_system_client.get_paths(path=directory_name))
 print("Azure DataLake setup complete.")
 
 # Azure Search setup
-search_credential = AzureCliCredential()
+search_credential = AzureCliCredential(process_timeout=30)
 search_client = SearchClient(SEARCH_ENDPOINT, INDEX_NAME, search_credential)
 index_client = SearchIndexClient(endpoint=SEARCH_ENDPOINT, credential=search_credential)
 print("Azure Search setup complete.")
@@ -185,7 +185,7 @@ def create_search_index():
 print("SQL Server connection established.")
 
 # Content Understanding client
-cu_credential = AzureCliCredential()
+cu_credential = AzureCliCredential(process_timeout=30)
 cu_token_provider = get_bearer_token_provider(cu_credential, "https://cognitiveservices.azure.com/.default")
 cu_client = AzureContentUnderstandingClient(
     endpoint=CU_ENDPOINT,