MilanM commited on
Commit
83c9f1b
·
verified ·
1 Parent(s): f64cbdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -134,6 +134,19 @@ def _(mo):
134
  def _(client_instantiation_form, mo):
135
  from ibm_watsonx_ai import APIClient, Credentials
136
 
 
 
 
 
 
 
 
 
 
 
 
 
 
137
  if client_instantiation_form.value:
138
  ### Instantiate the watsonx.ai client
139
  wx_credentials = Credentials(
@@ -144,7 +157,7 @@ def _(client_instantiation_form, mo):
144
  # project_client = APIClient(credentials=wx_credentials, project_id=client_instantiation_form.value["project_id"])
145
  deployment_client = APIClient(credentials=wx_credentials, space_id=client_instantiation_form.value["space_id"])
146
 
147
- task_credentials_details = deployment_client.task_credentials.store()
148
  else:
149
  # project_client = None
150
  deployment_client = None
 
134
  def _(client_instantiation_form, mo):
135
  from ibm_watsonx_ai import APIClient, Credentials
136
 
137
+ def setup_task_credentials(deployment_client):
138
+ # Get existing task credentials
139
+ existing_credentials = deployment_client.task_credentials.get_details()
140
+
141
+ # Delete existing credentials if any
142
+ if "resources" in existing_credentials and existing_credentials["resources"]:
143
+ for cred in existing_credentials["resources"]:
144
+ cred_id = deployment_client.task_credentials.get_id(cred)
145
+ deployment_client.task_credentials.delete(cred_id)
146
+
147
+ # Store new credentials
148
+ return deployment_client.task_credentials.store()
149
+
150
  if client_instantiation_form.value:
151
  ### Instantiate the watsonx.ai client
152
  wx_credentials = Credentials(
 
157
  # project_client = APIClient(credentials=wx_credentials, project_id=client_instantiation_form.value["project_id"])
158
  deployment_client = APIClient(credentials=wx_credentials, space_id=client_instantiation_form.value["space_id"])
159
 
160
+ task_credentials_details = setup_task_credentials(deployment_client)
161
  else:
162
  # project_client = None
163
  deployment_client = None