Merge pull request #81 from databricks-industry-solutions/fix/serving…

…_secret

Added token creation and secret, minor fix to time.sleep

monailabel_model/ModelServing.py

@@ -195,6 +195,28 @@
 
 # COMMAND ----------
 
+# MAGIC %md
+# MAGIC ## Creating a personal token and assign it to a secret
+# MAGIC
+# MAGIC The next cell demonstrates how to create a personal access token (PAT) and store it securely in Databricks secrets. This token will be used for api calls authentication in the monailabel server to Volumes and SQL Warehouses.
+
+# COMMAND ----------
+
+from databricks.sdk import WorkspaceClient
+
+w = WorkspaceClient()
+
+scope_name = "pixels-scope"
+
+if scope_name not in [scope.name for scope in w.secrets.list_scopes()]:
+  w.secrets.create_scope(scope=scope_name)
+
+token = w.tokens.create(comment=f'pixels_serving_endpoint_token')
+
+w.secrets.put_secret(scope=scope_name, key="pixels_token", string_value=token.token_value)
+
+# COMMAND ----------
+
 # MAGIC %md
 # MAGIC # Model Deployment with MLflow on Databricks
 # MAGIC
@@ -259,7 +281,7 @@
 
 while client.get_endpoint(serving_endpoint_name).state.ready != 'READY':
   print("ENDPOINT NOT READY YET")
-  time.sleet(60)
+  time.sleep(60)
 
 client.predict(
     endpoint=serving_endpoint_name,