From 46cb7fb85da2442a070782acb1692c370e69e50a Mon Sep 17 00:00:00 2001 From: UtkarshMishra-Microsoft Date: Fri, 28 Mar 2025 17:46:44 +0530 Subject: [PATCH] fixing-AI-Script --- .../Deployment/scripts/aihub_scripts/create_ai_hub.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/ResearchAssistant/Deployment/scripts/aihub_scripts/create_ai_hub.py b/ResearchAssistant/Deployment/scripts/aihub_scripts/create_ai_hub.py index cf0b8c3a6..0ae6db694 100644 --- a/ResearchAssistant/Deployment/scripts/aihub_scripts/create_ai_hub.py +++ b/ResearchAssistant/Deployment/scripts/aihub_scripts/create_ai_hub.py @@ -1,6 +1,7 @@ # Get Azure Key Vault Client key_vault_name = 'kv_to-be-replaced' +import time from azure.ai.ml import MLClient from azure.ai.ml.entities import ( Hub, @@ -73,7 +74,10 @@ def get_secrets_from_kv(kv_name, secret_name): # construct a hub my_hub = Hub(name=aihub_name, location=solutionLocation, display_name=aihub_name) -created_hub = ml_client.workspaces.begin_create(my_hub).result() +created_hub = ml_client.workspaces.begin_create(my_hub) + +time.sleep(60) # wait for hub to be created +created_hub = ml_client.workspaces.get(name=aihub_name) # construct the project my_project = Project( @@ -83,7 +87,9 @@ def get_secrets_from_kv(kv_name, secret_name): hub_id=created_hub.id, ) -created_project = ml_client.workspaces.begin_create(workspace=my_project).result() +created_project = ml_client.workspaces.begin_create(workspace=my_project) + +time.sleep(30) # wait for hub to be created open_ai_connection = AzureOpenAIConnection( name="Azure_OpenAI",