Skip to content

Commit 9020db4

Browse files
fixing-AI-Script (#420)
1 parent 7720140 commit 9020db4

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

Diff for: ResearchAssistant/Deployment/scripts/aihub_scripts/create_ai_hub.py

+8-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# Get Azure Key Vault Client
22
key_vault_name = 'kv_to-be-replaced'
33

4+
import time
45
from azure.ai.ml import MLClient
56
from azure.ai.ml.entities import (
67
Hub,
@@ -73,7 +74,10 @@ def get_secrets_from_kv(kv_name, secret_name):
7374
# construct a hub
7475
my_hub = Hub(name=aihub_name, location=solutionLocation, display_name=aihub_name)
7576

76-
created_hub = ml_client.workspaces.begin_create(my_hub).result()
77+
created_hub = ml_client.workspaces.begin_create(my_hub)
78+
79+
time.sleep(60) # wait for hub to be created
80+
created_hub = ml_client.workspaces.get(name=aihub_name)
7781

7882
# construct the project
7983
my_project = Project(
@@ -83,7 +87,9 @@ def get_secrets_from_kv(kv_name, secret_name):
8387
hub_id=created_hub.id,
8488
)
8589

86-
created_project = ml_client.workspaces.begin_create(workspace=my_project).result()
90+
created_project = ml_client.workspaces.begin_create(workspace=my_project)
91+
92+
time.sleep(30) # wait for hub to be created
8793

8894
open_ai_connection = AzureOpenAIConnection(
8995
name="Azure_OpenAI",

0 commit comments

Comments
 (0)