use of com.google.cloud.speech.v1p1beta1.PhraseSet in project java-speech by googleapis.
the class AdaptationClientTest method listPhraseSetTest.
@Test
public void listPhraseSetTest() throws Exception {
PhraseSet responsesElement = PhraseSet.newBuilder().build();
ListPhraseSetResponse expectedResponse = ListPhraseSetResponse.newBuilder().setNextPageToken("").addAllPhraseSets(Arrays.asList(responsesElement)).build();
mockAdaptation.addResponse(expectedResponse);
LocationName parent = LocationName.of("[PROJECT]", "[LOCATION]");
ListPhraseSetPagedResponse pagedListResponse = client.listPhraseSet(parent);
List<PhraseSet> resources = Lists.newArrayList(pagedListResponse.iterateAll());
Assert.assertEquals(1, resources.size());
Assert.assertEquals(expectedResponse.getPhraseSetsList().get(0), resources.get(0));
List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
Assert.assertEquals(1, actualRequests.size());
ListPhraseSetRequest actualRequest = ((ListPhraseSetRequest) actualRequests.get(0));
Assert.assertEquals(parent.toString(), actualRequest.getParent());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
use of com.google.cloud.speech.v1p1beta1.PhraseSet in project java-speech by googleapis.
the class AdaptationClientTest method listPhraseSetTest2.
@Test
public void listPhraseSetTest2() throws Exception {
PhraseSet responsesElement = PhraseSet.newBuilder().build();
ListPhraseSetResponse expectedResponse = ListPhraseSetResponse.newBuilder().setNextPageToken("").addAllPhraseSets(Arrays.asList(responsesElement)).build();
mockAdaptation.addResponse(expectedResponse);
String parent = "parent-995424086";
ListPhraseSetPagedResponse pagedListResponse = client.listPhraseSet(parent);
List<PhraseSet> resources = Lists.newArrayList(pagedListResponse.iterateAll());
Assert.assertEquals(1, resources.size());
Assert.assertEquals(expectedResponse.getPhraseSetsList().get(0), resources.get(0));
List<AbstractMessage> actualRequests = mockAdaptation.getRequests();
Assert.assertEquals(1, actualRequests.size());
ListPhraseSetRequest actualRequest = ((ListPhraseSetRequest) actualRequests.get(0));
Assert.assertEquals(parent, actualRequest.getParent());
Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
use of com.google.cloud.speech.v1p1beta1.PhraseSet in project java-speech by googleapis.
the class SpeechModelAdaptationBeta method transcribeWithModelAdaptation.
/**
* Transcribe with model adaptation
*
* @param projectId your project id
* @param location the region
* @param gcsUri the path to the audio file
*/
public static void transcribeWithModelAdaptation(String projectId, String location, String gcsUri, String customClassId, String phraseSetId) throws Exception {
// the "close" method on the client to safely clean up any remaining background resources.
try (AdaptationClient adaptationClient = AdaptationClient.create()) {
// Create `PhraseSet` and `CustomClasses` to create custom lists of similar
// items that are likely to occur in your input data.
// The parent resource where the custom class and phrase set will be created.
LocationName parent = LocationName.of(projectId, location);
// Create the custom class
CreateCustomClassRequest classRequest = CreateCustomClassRequest.newBuilder().setParent(parent.toString()).setCustomClassId(customClassId).setCustomClass(CustomClass.newBuilder().addItems(ClassItem.newBuilder().setValue("sushido")).addItems(ClassItem.newBuilder().setValue("altura")).addItems(ClassItem.newBuilder().setValue("taneda")).build()).build();
CustomClass classResponse = adaptationClient.createCustomClass(classRequest);
// Create the phrase set
CreatePhraseSetRequest phraseRequest = CreatePhraseSetRequest.newBuilder().setParent(parent.toString()).setPhraseSetId(phraseSetId).setPhraseSet(PhraseSet.newBuilder().setBoost(10).addPhrases(Phrase.newBuilder().setValue(String.format("Visit restaurants like %s%n", customClassId))).build()).build();
PhraseSet phraseResponse = adaptationClient.createPhraseSet(phraseRequest);
// Next section shows how to use the newly created custom class and phrase set
// to send a transcription request with speech adaptation
// Speech adaptation configuration
SpeechAdaptation speechAdaptation = SpeechAdaptation.newBuilder().addCustomClasses(classResponse).addPhraseSets(phraseResponse).build();
// the "close" method on the client to safely clean up any remaining background resources.
try (SpeechClient speechClient = SpeechClient.create()) {
// The path to the audio file to transcribe
// gcsUri URI for audio file in Cloud Storage, e.g. gs://[BUCKET]/[FILE]
// Builds the sync recognize request
RecognitionConfig config = RecognitionConfig.newBuilder().setEncoding(AudioEncoding.FLAC).setSampleRateHertz(16000).setLanguageCode("en-US").setAdaptation(// Set the adaptation object
speechAdaptation).build();
RecognitionAudio audio = RecognitionAudio.newBuilder().setUri(gcsUri).build();
// Performs speech recognition on the audio file.
RecognizeResponse response = speechClient.recognize(config, audio);
List<SpeechRecognitionResult> results = response.getResultsList();
for (SpeechRecognitionResult result : results) {
// There can be several alternative transcripts for a given chunk of speech. Just use the
// first (most likely) one here.
SpeechRecognitionAlternative alternative = result.getAlternativesList().get(0);
System.out.printf("Adapted Transcription: %s%n", alternative.getTranscript());
}
}
} catch (ApiException e) {
System.out.println("Client Interaction Error: \n" + e.toString());
}
}
Aggregations