diff --git a/language/cloud-client/v1/snippets.py b/language/cloud-client/v1/snippets.py index d485752b09c5..704c6347eac0 100644 --- a/language/cloud-client/v1/snippets.py +++ b/language/cloud-client/v1/snippets.py @@ -94,10 +94,14 @@ def entities_text(text): # document.type == enums.Document.Type.HTML entities = client.analyze_entities(document).entities + # entity types from enums.Entity.Type + entity_type = ('UNKNOWN', 'PERSON', 'LOCATION', 'ORGANIZATION', + 'EVENT', 'WORK_OF_ART', 'CONSUMER_GOOD', 'OTHER') + for entity in entities: print('=' * 20) print(u'{:<16}: {}'.format('name', entity.name)) - print(u'{:<16}: {}'.format('type', entity.type)) + print(u'{:<16}: {}'.format('type', entity_type[entity.type])) print(u'{:<16}: {}'.format('metadata', entity.metadata)) print(u'{:<16}: {}'.format('salience', entity.salience)) print(u'{:<16}: {}'.format('wikipedia_url', @@ -120,10 +124,14 @@ def entities_file(gcs_uri): # document.type == enums.Document.Type.HTML entities = client.analyze_entities(document).entities + # entity types from enums.Entity.Type + entity_type = ('UNKNOWN', 'PERSON', 'LOCATION', 'ORGANIZATION', + 'EVENT', 'WORK_OF_ART', 'CONSUMER_GOOD', 'OTHER') + for entity in entities: print('=' * 20) print(u'{:<16}: {}'.format('name', entity.name)) - print(u'{:<16}: {}'.format('type', entity.type)) + print(u'{:<16}: {}'.format('type', entity_type[entity.type])) print(u'{:<16}: {}'.format('metadata', entity.metadata)) print(u'{:<16}: {}'.format('salience', entity.salience)) print(u'{:<16}: {}'.format('wikipedia_url', diff --git a/language/cloud-client/v1beta2/snippets.py b/language/cloud-client/v1beta2/snippets.py index 0ea7352aeb95..0ea15f7fc42d 100644 --- a/language/cloud-client/v1beta2/snippets.py +++ b/language/cloud-client/v1beta2/snippets.py @@ -85,10 +85,14 @@ def entities_text(text): # document.type == enums.Document.Type.HTML entities = client.analyze_entities(document).entities + # entity types from enums.Entity.Type + entity_type = ('UNKNOWN', 'PERSON', 'LOCATION', 'ORGANIZATION', + 'EVENT', 'WORK_OF_ART', 'CONSUMER_GOOD', 'OTHER') + for entity in entities: print('=' * 20) print(u'{:<16}: {}'.format('name', entity.name)) - print(u'{:<16}: {}'.format('type', entity.type)) + print(u'{:<16}: {}'.format('type', entity_type[entity.type])) print(u'{:<16}: {}'.format('metadata', entity.metadata)) print(u'{:<16}: {}'.format('salience', entity.salience)) print(u'{:<16}: {}'.format('wikipedia_url', @@ -108,10 +112,14 @@ def entities_file(gcs_uri): # document.type == enums.Document.Type.HTML entities = client.analyze_entities(document).entities + # entity types from enums.Entity.Type + entity_type = ('UNKNOWN', 'PERSON', 'LOCATION', 'ORGANIZATION', + 'EVENT', 'WORK_OF_ART', 'CONSUMER_GOOD', 'OTHER') + for entity in entities: print('=' * 20) print(u'{:<16}: {}'.format('name', entity.name)) - print(u'{:<16}: {}'.format('type', entity.type)) + print(u'{:<16}: {}'.format('type', entity_type[entity.type])) print(u'{:<16}: {}'.format('metadata', entity.metadata)) print(u'{:<16}: {}'.format('salience', entity.salience)) print(u'{:<16}: {}'.format('wikipedia_url',