Search in sources :

Example 16 with EntityType

use of com.google.cloud.dialogflow.v2.EntityType in project java-dialogflow by googleapis.

the class EntityTypesClientTest method listEntityTypesTest4.

@Test
public void listEntityTypesTest4() throws Exception {
    EntityType responsesElement = EntityType.newBuilder().build();
    ListEntityTypesResponse expectedResponse = ListEntityTypesResponse.newBuilder().setNextPageToken("").addAllEntityTypes(Arrays.asList(responsesElement)).build();
    mockEntityTypes.addResponse(expectedResponse);
    String parent = "parent-995424086";
    String languageCode = "languageCode-2092349083";
    ListEntityTypesPagedResponse pagedListResponse = client.listEntityTypes(parent, languageCode);
    List<EntityType> resources = Lists.newArrayList(pagedListResponse.iterateAll());
    Assert.assertEquals(1, resources.size());
    Assert.assertEquals(expectedResponse.getEntityTypesList().get(0), resources.get(0));
    List<AbstractMessage> actualRequests = mockEntityTypes.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    ListEntityTypesRequest actualRequest = ((ListEntityTypesRequest) actualRequests.get(0));
    Assert.assertEquals(parent, actualRequest.getParent());
    Assert.assertEquals(languageCode, actualRequest.getLanguageCode());
    Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Also used : AbstractMessage(com.google.protobuf.AbstractMessage) ListEntityTypesPagedResponse(com.google.cloud.dialogflow.v2.EntityTypesClient.ListEntityTypesPagedResponse) Test(org.junit.Test)

Example 17 with EntityType

use of com.google.cloud.dialogflow.v2.EntityType in project java-dialogflow by googleapis.

the class EntityTypesClientTest method listEntityTypesTest2.

@Test
public void listEntityTypesTest2() throws Exception {
    EntityType responsesElement = EntityType.newBuilder().build();
    ListEntityTypesResponse expectedResponse = ListEntityTypesResponse.newBuilder().setNextPageToken("").addAllEntityTypes(Arrays.asList(responsesElement)).build();
    mockEntityTypes.addResponse(expectedResponse);
    String parent = "parent-995424086";
    ListEntityTypesPagedResponse pagedListResponse = client.listEntityTypes(parent);
    List<EntityType> resources = Lists.newArrayList(pagedListResponse.iterateAll());
    Assert.assertEquals(1, resources.size());
    Assert.assertEquals(expectedResponse.getEntityTypesList().get(0), resources.get(0));
    List<AbstractMessage> actualRequests = mockEntityTypes.getRequests();
    Assert.assertEquals(1, actualRequests.size());
    ListEntityTypesRequest actualRequest = ((ListEntityTypesRequest) actualRequests.get(0));
    Assert.assertEquals(parent, actualRequest.getParent());
    Assert.assertTrue(channelProvider.isHeaderSent(ApiClientHeaderProvider.getDefaultApiClientHeaderKey(), GaxGrpcProperties.getDefaultApiClientHeaderPattern()));
}
Also used : AbstractMessage(com.google.protobuf.AbstractMessage) ListEntityTypesPagedResponse(com.google.cloud.dialogflow.v2.EntityTypesClient.ListEntityTypesPagedResponse) Test(org.junit.Test)

Example 18 with EntityType

use of com.google.cloud.dialogflow.v2.EntityType in project java-dialogflow-cx by googleapis.

the class ITSystemTest method setUp.

@BeforeClass
public static void setUp() throws IOException, ExecutionException, InterruptedException {
    /* create agents */
    agentsClient = AgentsClient.create();
    Agent agent = Agent.newBuilder().setDisplayName(DISPLAY_NAME).setDescription(DESCRIPTION).setTimeZone(AGENT_TIME_ZONE).setDefaultLanguageCode(DEFAULT_LANGUAGE_CODE).setStartFlow(ID).setEnableStackdriverLogging(true).build();
    Agent response = agentsClient.createAgent(PARENT, agent);
    agentName = response.getName();
    /* create entity types */
    entityTypesClient = EntityTypesClient.create();
    EntityType entityType = EntityType.newBuilder().setAutoExpansionMode(EntityType.AutoExpansionMode.AUTO_EXPANSION_MODE_DEFAULT).setAutoExpansionModeValue(1).setDisplayName(DISPLAY_NAME).setEnableFuzzyExtraction(true).setKind(EntityType.Kind.KIND_LIST).build();
    CreateEntityTypeRequest request = CreateEntityTypeRequest.newBuilder().setParent(agentName).setEntityType(entityType).build();
    EntityType entityTypeResponse = entityTypesClient.createEntityType(request);
    entityTypesName = entityTypeResponse.getName();
    /* create flows */
    flowsClient = FlowsClient.create();
    Flow flow = Flow.newBuilder().setNluSettings(NLUSETTINGS).setDisplayName(DISPLAY_NAME).setDescription(DESCRIPTION).build();
    CreateFlowRequest createFlowRequest = CreateFlowRequest.newBuilder().setParent(agentName).setFlow(flow).build();
    Flow flowResponse = flowsClient.createFlow(createFlowRequest);
    flowName = flowResponse.getName();
    Flow trainFlow = Flow.newBuilder().setNluSettings(NLUSETTINGS).setDisplayName(DISPLAY_NAME2).setDescription(DESCRIPTION).build();
    CreateFlowRequest createTrainFlowRequest = CreateFlowRequest.newBuilder().setParent(agentName).setFlow(trainFlow).build();
    Flow trainFlowResponse = flowsClient.createFlow(createTrainFlowRequest);
    trainFlowName = trainFlowResponse.getName();
    /* create intents */
    intentsClient = IntentsClient.create();
    Intent intent = Intent.newBuilder().setDisplayName(DISPLAY_NAME).setPriority(1).build();
    CreateIntentRequest createIntentRequest = CreateIntentRequest.newBuilder().setIntent(intent).setParent(agentName).build();
    Intent intentResponse = intentsClient.createIntent(createIntentRequest);
    intentsName = intentResponse.getName();
    /* create pages */
    pagesClient = PagesClient.create();
    Page page = Page.newBuilder().setDisplayName(DISPLAY_NAME).build();
    CreatePageRequest createPageRequest = CreatePageRequest.newBuilder().setPage(page).setParent(flowName).build();
    Page pageResponse = pagesClient.createPage(createPageRequest);
    pageName = pageResponse.getName();
    /* create session */
    sessionsClient = SessionsClient.create();
    /* create transition route groups */
    transitionRouteGroupsClient = TransitionRouteGroupsClient.create();
    TransitionRouteGroup transitionRouteGroup = TransitionRouteGroup.newBuilder().setDisplayName(DISPLAY_NAME).build();
    CreateTransitionRouteGroupRequest transitionRouteGroupRequest = CreateTransitionRouteGroupRequest.newBuilder().setParent(flowName).setTransitionRouteGroup(transitionRouteGroup).build();
    TransitionRouteGroup transitionRouteGroupResponse = transitionRouteGroupsClient.createTransitionRouteGroup(transitionRouteGroupRequest);
    transitionRouteGroupName = transitionRouteGroupResponse.getName();
    /* create version */
    versionsClient = VersionsClient.create();
    Version version = Version.newBuilder().setStateValue(2).setDisplayName(DISPLAY_NAME).build();
    CreateVersionRequest createVersionRequest = CreateVersionRequest.newBuilder().setParent(flowName).setVersion(version).build();
    Version versionResponse = versionsClient.createVersionAsync(createVersionRequest).get();
    versionName = versionResponse.getName();
}
Also used : Agent(com.google.cloud.dialogflow.cx.v3beta1.Agent) CreateEntityTypeRequest(com.google.cloud.dialogflow.cx.v3beta1.CreateEntityTypeRequest) CreatePageRequest(com.google.cloud.dialogflow.cx.v3beta1.CreatePageRequest) Intent(com.google.cloud.dialogflow.cx.v3beta1.Intent) Page(com.google.cloud.dialogflow.cx.v3beta1.Page) CreateTransitionRouteGroupRequest(com.google.cloud.dialogflow.cx.v3beta1.CreateTransitionRouteGroupRequest) Flow(com.google.cloud.dialogflow.cx.v3beta1.Flow) EntityType(com.google.cloud.dialogflow.cx.v3beta1.EntityType) Version(com.google.cloud.dialogflow.cx.v3beta1.Version) CreateFlowRequest(com.google.cloud.dialogflow.cx.v3beta1.CreateFlowRequest) TransitionRouteGroup(com.google.cloud.dialogflow.cx.v3beta1.TransitionRouteGroup) CreateVersionRequest(com.google.cloud.dialogflow.cx.v3beta1.CreateVersionRequest) CreateIntentRequest(com.google.cloud.dialogflow.cx.v3beta1.CreateIntentRequest) BeforeClass(org.junit.BeforeClass)

Example 19 with EntityType

use of com.google.cloud.dialogflow.v2.EntityType in project java-dialogflow-cx by googleapis.

the class ITSystemTest method getEntityTypeTest.

@Test
public void getEntityTypeTest() {
    GetEntityTypeRequest request = GetEntityTypeRequest.newBuilder().setName(entityTypesName).build();
    EntityType entityType = entityTypesClient.getEntityType(request);
    assertEntityTypesDetails(entityType);
}
Also used : EntityType(com.google.cloud.dialogflow.cx.v3beta1.EntityType) GetEntityTypeRequest(com.google.cloud.dialogflow.cx.v3beta1.GetEntityTypeRequest) Test(org.junit.Test)

Example 20 with EntityType

use of com.google.cloud.dialogflow.v2.EntityType in project osmosis by openstreetmap.

the class PbfBlobDecoder method buildRelationMembers.

private void buildRelationMembers(org.openstreetmap.osmosis.core.domain.v0_6.Relation relation, List<Long> memberIds, List<Integer> memberRoles, List<Osmformat.Relation.MemberType> memberTypes, PbfFieldDecoder fieldDecoder) {
    List<RelationMember> members = relation.getMembers();
    // Ensure parallel lists are of equal size.
    if ((memberIds.size() != memberRoles.size()) || (memberIds.size() != memberTypes.size())) {
        throw new OsmosisRuntimeException("Number of member ids (" + memberIds.size() + "), member roles (" + memberRoles.size() + "), and member types (" + memberTypes.size() + ") don't match");
    }
    Iterator<Long> memberIdIterator = memberIds.iterator();
    Iterator<Integer> memberRoleIterator = memberRoles.iterator();
    Iterator<Osmformat.Relation.MemberType> memberTypeIterator = memberTypes.iterator();
    // Build up the list of relation members for the way. The member ids are
    // delta encoded meaning that each id is stored as a delta against
    // the previous one.
    long memberId = 0;
    while (memberIdIterator.hasNext()) {
        Osmformat.Relation.MemberType memberType = memberTypeIterator.next();
        memberId += memberIdIterator.next();
        EntityType entityType;
        RelationMember member;
        if (memberType == Osmformat.Relation.MemberType.NODE) {
            entityType = EntityType.Node;
        } else if (memberType == Osmformat.Relation.MemberType.WAY) {
            entityType = EntityType.Way;
        } else if (memberType == Osmformat.Relation.MemberType.RELATION) {
            entityType = EntityType.Relation;
        } else {
            throw new OsmosisRuntimeException("Member type of " + memberType + " is not supported.");
        }
        member = new RelationMember(memberId, entityType, fieldDecoder.decodeString(memberRoleIterator.next()));
        members.add(member);
    }
}
Also used : Osmformat(crosby.binary.Osmformat) OsmosisRuntimeException(org.openstreetmap.osmosis.core.OsmosisRuntimeException) EntityType(org.openstreetmap.osmosis.core.domain.v0_6.EntityType) RelationMember(org.openstreetmap.osmosis.core.domain.v0_6.RelationMember)

Aggregations

EntityType (org.openstreetmap.osmosis.core.domain.v0_6.EntityType)12 Test (org.junit.Test)9 RelationMember (org.openstreetmap.osmosis.core.domain.v0_6.RelationMember)9 EntityType (com.google.cloud.dialogflow.cx.v3beta1.EntityType)4 ListEntityTypesPagedResponse (com.google.cloud.dialogflow.v2.EntityTypesClient.ListEntityTypesPagedResponse)4 AbstractMessage (com.google.protobuf.AbstractMessage)4 EntityType (org.hypertrace.entity.type.service.v1.EntityType)4 Test (org.junit.jupiter.api.Test)4 EntityType (com.google.cloud.dialogflow.v2.EntityType)3 ServiceException (com.google.protobuf.ServiceException)3 JSONDocument (org.hypertrace.core.documentstore.JSONDocument)3 EntityType (org.hypertrace.entity.type.service.v2.EntityType)3 Relation (org.openstreetmap.osmosis.core.domain.v0_6.Relation)3 InvalidProtocolBufferException (com.google.protobuf.InvalidProtocolBufferException)2 Osmformat (crosby.binary.Osmformat)2 IOException (java.io.IOException)2 Document (org.hypertrace.core.documentstore.Document)2 SingleValueKey (org.hypertrace.core.documentstore.SingleValueKey)2 BeforeClass (org.junit.BeforeClass)2 Agent (com.google.cloud.dialogflow.cx.v3beta1.Agent)1