use of com.ibm.watson.developer_cloud.discovery.v1.model.TrainingDataSet in project java-sdk by watson-developer-cloud.
the class DiscoveryServiceIT method deleteTrainingDataQueryIsSuccessful.
@Test
public void deleteTrainingDataQueryIsSuccessful() {
TrainingQuery query = createTestQuery(collectionId, "Query" + UUID.randomUUID().toString());
String queryId = query.getQueryId();
ListTrainingDataOptions.Builder listBuilder = new ListTrainingDataOptions.Builder(environmentId, collectionId);
TrainingDataSet trainingData = discovery.listTrainingData(listBuilder.build()).execute();
List<TrainingQuery> queryList = trainingData.getQueries();
boolean doesQueryExist = false;
for (TrainingQuery q : queryList) {
if (q.getQueryId().equals(queryId)) {
doesQueryExist = true;
break;
}
}
assertTrue(doesQueryExist);
DeleteTrainingDataOptions.Builder deleteBuilder = new DeleteTrainingDataOptions.Builder(environmentId, collectionId, queryId);
discovery.deleteTrainingData(deleteBuilder.build()).execute();
listBuilder = new ListTrainingDataOptions.Builder(environmentId, collectionId);
trainingData = discovery.listTrainingData(listBuilder.build()).execute();
queryList = trainingData.getQueries();
doesQueryExist = false;
for (TrainingQuery q : queryList) {
if (q.getQueryId().equals(queryId)) {
doesQueryExist = true;
break;
}
}
assertFalse(doesQueryExist);
}
use of com.ibm.watson.developer_cloud.discovery.v1.model.TrainingDataSet in project java-sdk by watson-developer-cloud.
the class DiscoveryServiceTest method listTrainingDataIsSuccessful.
@Test
public void listTrainingDataIsSuccessful() throws InterruptedException {
server.enqueue(jsonResponse(listTrainingDataResp));
ListTrainingDataOptions getRequest = new ListTrainingDataOptions.Builder(environmentId, collectionId).build();
TrainingDataSet response = discoveryService.listTrainingData(getRequest).execute();
RecordedRequest request = server.takeRequest();
assertEquals(TRAINING1_PATH, request.getPath());
assertEquals(GET, request.getMethod());
assertEquals(listTrainingDataResp, response);
}
use of com.ibm.watson.developer_cloud.discovery.v1.model.TrainingDataSet in project java-sdk by watson-developer-cloud.
the class DiscoveryServiceIT method deleteAllCollectionTrainingDataIsSuccessful.
@Test
public void deleteAllCollectionTrainingDataIsSuccessful() {
String collId = setupTestQueries(collectionId);
DeleteAllTrainingDataOptions.Builder deleteBuilder = new DeleteAllTrainingDataOptions.Builder(environmentId, collId);
discovery.deleteAllTrainingData(deleteBuilder.build()).execute();
ListTrainingDataOptions.Builder listBuilder = new ListTrainingDataOptions.Builder(environmentId, collId);
TrainingDataSet trainingData = discovery.listTrainingData(listBuilder.build()).execute();
assertEquals(trainingData.getQueries().size(), 0);
}
use of com.ibm.watson.developer_cloud.discovery.v1.model.TrainingDataSet in project java-sdk by watson-developer-cloud.
the class Discovery method listTrainingData.
/**
* Lists the training data for this collection.
*
* @param listTrainingDataOptions the {@link ListTrainingDataOptions} containing the options for the call
* @return a {@link ServiceCall} with a response type of {@link TrainingDataSet}
*/
public ServiceCall<TrainingDataSet> listTrainingData(ListTrainingDataOptions listTrainingDataOptions) {
Validator.notNull(listTrainingDataOptions, "listTrainingDataOptions cannot be null");
String[] pathSegments = { "v1/environments", "collections", "training_data" };
String[] pathParameters = { listTrainingDataOptions.environmentId(), listTrainingDataOptions.collectionId() };
RequestBuilder builder = RequestBuilder.get(RequestBuilder.constructHttpUrl(getEndPoint(), pathSegments, pathParameters));
builder.query(VERSION, versionDate);
return createServiceCall(builder.build(), ResponseConverterUtils.getObject(TrainingDataSet.class));
}
Aggregations