use of com.ibm.watson.developer_cloud.discovery.v1.model.NluEnrichmentEntities in project java-sdk by watson-developer-cloud.
the class DiscoveryServiceIT method createConfigurationIsSuccessful.
@Test
public void createConfigurationIsSuccessful() {
Date start = new Date();
String uniqueConfigName = uniqueName + "-config";
String description = "Description of " + uniqueConfigName;
Conversions conversions = new Conversions();
HtmlSettings htmlSettings = new HtmlSettings();
htmlSettings.setExcludeTagsCompletely(Arrays.asList("table", "h6", "header"));
conversions.setHtml(htmlSettings);
NormalizationOperation operation = new NormalizationOperation();
operation.setOperation("foo");
operation.setSourceField("bar");
operation.setDestinationField("baz");
List<NormalizationOperation> normalizations = Arrays.asList(operation);
Enrichment enrichment = new Enrichment();
enrichment.setSourceField("foo");
enrichment.setDestinationField("bar");
enrichment.setEnrichmentName("baz");
enrichment.setDescription("Erich foo to bar with baz");
enrichment.setIgnoreDownstreamErrors(true);
enrichment.setOverwrite(false);
NluEnrichmentSentiment sentiment = new NluEnrichmentSentiment.Builder().document(true).build();
NluEnrichmentEmotion emotion = new NluEnrichmentEmotion.Builder().document(true).build();
NluEnrichmentEntities entities = new NluEnrichmentEntities.Builder().emotion(true).sentiment(true).model("WhatComesAfterQux").build();
NluEnrichmentKeywords keywords = new NluEnrichmentKeywords.Builder().emotion(true).sentiment(true).build();
NluEnrichmentSemanticRoles semanticRoles = new NluEnrichmentSemanticRoles.Builder().entities(true).build();
NluEnrichmentFeatures features = new NluEnrichmentFeatures.Builder().sentiment(sentiment).emotion(emotion).entities(entities).keywords(keywords).semanticRoles(semanticRoles).build();
EnrichmentOptions options = new EnrichmentOptions.Builder().features(features).build();
enrichment.setOptions(options);
List<Enrichment> enrichments = Arrays.asList(enrichment);
CreateConfigurationOptions createOptions = new CreateConfigurationOptions.Builder().environmentId(environmentId).name(uniqueConfigName).description(description).conversions(conversions).normalizations(normalizations).enrichments(enrichments).build();
Configuration createResponse = createConfiguration(createOptions);
assertEquals(uniqueConfigName, createResponse.getName());
assertEquals(description, createResponse.getDescription());
assertEquals(conversions, createResponse.getConversions());
assertEquals(normalizations, createResponse.getNormalizations());
assertEquals(enrichments, createResponse.getEnrichments());
Date now = new Date();
assertTrue(fuzzyBefore(createResponse.getCreated(), now));
assertTrue(fuzzyAfter(createResponse.getCreated(), start));
assertTrue(fuzzyBefore(createResponse.getUpdated(), now));
assertTrue(fuzzyAfter(createResponse.getUpdated(), start));
}
use of com.ibm.watson.developer_cloud.discovery.v1.model.NluEnrichmentEntities in project java-sdk by watson-developer-cloud.
the class DiscoveryServiceIT method updateConfigurationIsSuccessful.
@Test
public void updateConfigurationIsSuccessful() {
Configuration testConfig = createTestConfig();
Date start = new Date();
String updatedName = testConfig.getName() + UUID.randomUUID().toString();
String updatedDescription = "Description of " + updatedName;
Conversions updatedConversions = new Conversions();
HtmlSettings newHtmlSettings = new HtmlSettings();
newHtmlSettings.setExcludeTagsCompletely(Arrays.asList("table", "h6", "header"));
updatedConversions.setHtml(newHtmlSettings);
NormalizationOperation operation = new NormalizationOperation();
operation.setOperation("foo");
operation.setSourceField("bar");
operation.setDestinationField("baz");
List<NormalizationOperation> updatedNormalizations = Arrays.asList(operation);
Enrichment enrichment = new Enrichment();
enrichment.setSourceField("foo");
enrichment.setDestinationField("bar");
enrichment.setEnrichmentName("baz");
enrichment.setDescription("Erich foo to bar with baz");
enrichment.setIgnoreDownstreamErrors(true);
enrichment.setOverwrite(false);
NluEnrichmentSentiment sentiment = new NluEnrichmentSentiment.Builder().document(true).build();
NluEnrichmentEmotion emotion = new NluEnrichmentEmotion.Builder().document(true).build();
NluEnrichmentEntities entities = new NluEnrichmentEntities.Builder().emotion(true).sentiment(true).model("WhatComesAfterQux").build();
NluEnrichmentKeywords keywords = new NluEnrichmentKeywords.Builder().emotion(true).sentiment(true).build();
NluEnrichmentSemanticRoles semanticRoles = new NluEnrichmentSemanticRoles.Builder().entities(true).build();
NluEnrichmentFeatures features = new NluEnrichmentFeatures.Builder().sentiment(sentiment).emotion(emotion).entities(entities).keywords(keywords).semanticRoles(semanticRoles).build();
EnrichmentOptions options = new EnrichmentOptions.Builder().features(features).build();
enrichment.setOptions(options);
List<Enrichment> updatedEnrichments = Arrays.asList(enrichment);
UpdateConfigurationOptions.Builder updateBuilder = new UpdateConfigurationOptions.Builder(environmentId, testConfig.getConfigurationId());
updateBuilder.name(updatedName);
updateBuilder.description(updatedDescription);
updateBuilder.conversions(updatedConversions);
updateBuilder.normalizations(updatedNormalizations);
updateBuilder.enrichments(updatedEnrichments);
Configuration updatedConfiguration = discovery.updateConfiguration(updateBuilder.build()).execute();
assertEquals(updatedName, updatedConfiguration.getName());
assertEquals(updatedDescription, updatedConfiguration.getDescription());
assertEquals(updatedConversions, updatedConfiguration.getConversions());
assertEquals(updatedNormalizations, updatedConfiguration.getNormalizations());
assertEquals(updatedEnrichments, updatedConfiguration.getEnrichments());
Date now = new Date();
assertTrue(fuzzyBefore(updatedConfiguration.getCreated(), start));
assertTrue(fuzzyBefore(updatedConfiguration.getUpdated(), now));
assertTrue(fuzzyAfter(updatedConfiguration.getUpdated(), start));
}
Aggregations