def test_list_model_auth_bad_key(self, resource_group, location, form_recognizer_account, form_recognizer_account_key): client = FormTrainingClient(form_recognizer_account, AzureKeyCredential("xxxx")) with self.assertRaises(ClientAuthenticationError): result = client.list_model_infos() for res in result: test = res
def manage_custom_models(self): # [START get_account_properties] from azure.core.credentials import AzureKeyCredential from azure.core.exceptions import ResourceNotFoundError from azure.ai.formrecognizer import FormTrainingClient form_training_client = FormTrainingClient( endpoint=self.endpoint, credential=AzureKeyCredential(self.key)) # First, we see how many custom models we have, and what our limit is account_properties = form_training_client.get_account_properties() print( "Our account has {} custom models, and we can have at most {} custom models" .format(account_properties.custom_model_count, account_properties.custom_model_limit)) # [END get_account_properties] # Next, we get a paged list of all of our custom models # [START list_model_infos] custom_models = form_training_client.list_model_infos() print("We have models with the following ids:") # Let's pull out the first model first_model = next(custom_models) print(first_model.model_id) for model in custom_models: print(model.model_id) # [END list_model_infos] # Now we'll get the first custom model in the paged list # [START get_custom_model] custom_model = form_training_client.get_custom_model( model_id=first_model.model_id) print("Model ID: {}".format(custom_model.model_id)) print("Status: {}".format(custom_model.status)) print("Created on: {}".format(custom_model.created_on)) print("Last modified: {}".format(custom_model.last_modified)) # [END get_custom_model] # Finally, we will delete this model by ID # [START delete_model] form_training_client.delete_model(model_id=custom_model.model_id) try: form_training_client.get_custom_model( model_id=custom_model.model_id) except ResourceNotFoundError: print("Successfully deleted model with id {}".format( custom_model.model_id))