jlengrand/dialogflow-fun
1
0
Fork 0
mirror of https://github.com/jlengrand/dialogflow-fun.git synced 2026-03-10 08:11:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Now using composite names

Browse Source
This commit is contained in:
Julien Lengrand-Lambert
2019-10-16 16:03:43 +02:00
parent c5bc41edb3
commit 6d9adf88ad
5 changed files with 92 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

100
dialogflow-fun-java/src/main/java/CreateAgent.java
View File

@@ -1,123 +1,31 @@
import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collector;
import java.util.stream.Collectors;
import com.google.cloud.dialogflow.v2beta1.EntityType;
import com.google.cloud.dialogflow.v2beta1.EntityTypesClient;
import com.google.cloud.dialogflow.v2beta1.ProjectAgentName;
import com.google.cloud.dialogflow.v2beta1.Agent;
import com.google.cloud.dialogflow.v2beta1.AgentsClient;
import com.google.cloud.dialogflow.v2beta1.ProjectName;
import com.google.cloud.dialogflow.v2beta1.EntityType.Entity;
import com.google.cloud.dialogflow.v2beta1.EntityType.Kind;
import data.ReadNames;
public class CreateAgent {
private static final String CONTACT = "contact";
private static final String GOOGLE_AUTH_ENV_NAME = "GOOGLE_APPLICATION_CREDENTIALS";
private static final String PROJECT_NAME = "dialogflow-fun";
private static final String AGENT_NAME = "dialogflow-fun-agent";
private static final String AGENT_ID = "ac522b80-e75b-40cd-9493-269fbb4ef634";
private static final String ENTITY_TYPE_ID = "Developer";
private static ReadNames namesReader = new ReadNames();
private static EntityCreator entityCreator = new EntityCreator();
public static void main(String[] args) {
CreateAgent createAgent = new CreateAgent();
public static void main(String[] args) throws IOException {
if (!googleAuthOk()) {
System.out.println("Error with Google Authentication. Exiting...");
System.exit(0);
}
deleteAllEntityTypes();
createContactEntityType();
entityCreator.deleteAllEntityTypes();
entityCreator.createContactEntityType();
System.out.println("Done");
}
public static void deleteAllEntityTypes() {
try {
EntityTypesClient entityTypesClient = EntityTypesClient.create();
ProjectAgentName parent = ProjectAgentName.of(PROJECT_NAME);
for (EntityType entityType : entityTypesClient.listEntityTypes(parent).iterateAll()) {
System.out.println("Deleting " + entityType.getDisplayName() + " with name " + entityType.getName());
entityTypesClient.deleteEntityType(entityType.getName());
}
} catch (IOException e) {
e.printStackTrace();
}
}
public static void createContactEntityType() {
try {
EntityTypesClient entityTypesClient = EntityTypesClient.create();
ProjectAgentName parent = ProjectAgentName.of(PROJECT_NAME);
List<String> names = namesReader.getListOfNames();
List<Entity> contactEntities = names.stream().map(n -> Entity.newBuilder().setValue(n).build())
.collect(Collectors.toList());
EntityType contactEntityType = EntityType.newBuilder().setDisplayName(CONTACT).setKind(Kind.KIND_MAP)
.addAllEntities(contactEntities).build();
entityTypesClient.createEntityType(parent, contactEntityType);
} catch (IOException e) {
e.printStackTrace();
}
}
public static void getEntitiesInfo() {
// ENTITY TYPES
// https://googleapis.dev/java/google-cloud-clients/latest/index.html?com/google/cloud/dialogflow/v2/package-summary.html
// * **System** - entities that are defined by the Dialogflow API for common
// data types such as date, time, currency, and so on. A system entity is
// represented by the `EntityType` type.
// * **Developer** - entities that are defined by you that represent actionable
// data that is meaningful to your application. For example, you could define a
// `pizza.sauce` entity for red or white pizza sauce, a `pizza.cheese` entity
// for the different types of cheese on a pizza, a `pizza.topping` entity for
// different toppings, and so on. A developer entity is represented by the
// `EntityType` type.
// * **User** - entities that are built for an individual user such as
// favorites, preferences, playlists, and so on. A user entity is represented by
// the [SessionEntityType][google.cloud.dialogflow.v2.SessionEntityType] type.
System.out.println("==========");
System.out.println("Getting Entities info");
try {
EntityTypesClient entityTypesClient = EntityTypesClient.create();
ProjectAgentName parent = ProjectAgentName.of(PROJECT_NAME);
for (EntityType entityType : entityTypesClient.listEntityTypes(parent).iterateAll()) {
System.out.println(entityType.getDisplayName());
List<Entity> entities = entityType.getEntitiesList();
entities.forEach(CreateAgent::printEntity);
}
} catch (IOException e) {
e.printStackTrace();
}
System.out.println("==========");
}
public static void printEntity(Entity entity) {
System.out.println(entity.getValue());
}
public static void getAgentInfo() {
System.out.println("==========");
System.out.println("Getting Agent info");

76
dialogflow-fun-java/src/main/java/EntityCreator.java Normal file
View File

@@ -0,0 +1,76 @@
import java.io.IOException;
import java.util.List;
import java.util.stream.Collectors;
import com.google.cloud.dialogflow.v2beta1.EntityType;
import com.google.cloud.dialogflow.v2beta1.EntityTypesClient;
import com.google.cloud.dialogflow.v2beta1.ProjectAgentName;
import com.google.cloud.dialogflow.v2beta1.EntityType.Entity;
import com.google.cloud.dialogflow.v2beta1.EntityType.Kind;
import data.ReadNames;
public class EntityCreator {
private static final String CONTACT_ENTITY_NAME = "names_composite";
private static final String PROJECT_NAME = "dialogflow-fun";
private ReadNames namesReader = new ReadNames();
public void createContactEntityType() {
try {
EntityTypesClient entityTypesClient = EntityTypesClient.create();
ProjectAgentName parent = ProjectAgentName.of(PROJECT_NAME);
List<String> names = namesReader.getListOfNames();
List<Entity> contactEntities = names.stream().map(n -> Entity.newBuilder().setValue(n).build())
.collect(Collectors.toList());
EntityType contactEntityType = EntityType.newBuilder().setDisplayName(CONTACT_ENTITY_NAME)
.setKind(Kind.KIND_MAP).addAllEntities(contactEntities).build();
entityTypesClient.createEntityType(parent, contactEntityType);
} catch (IOException e) {
e.printStackTrace();
}
}
public void deleteAllEntityTypes() throws IOException {
EntityTypesClient entityTypesClient = EntityTypesClient.create();
ProjectAgentName parent = ProjectAgentName.of(PROJECT_NAME);
for (EntityType entityType : entityTypesClient.listEntityTypes(parent).iterateAll()) {
System.out.println("Deleting " + entityType.getDisplayName() + " with name " + entityType.getName());
entityTypesClient.deleteEntityType(entityType.getName());
}
}
public void getEntitiesInfo() {
System.out.println("==========");
System.out.println("Getting Entities info");
try {
EntityTypesClient entityTypesClient = EntityTypesClient.create();
ProjectAgentName parent = ProjectAgentName.of(PROJECT_NAME);
for (EntityType entityType : entityTypesClient.listEntityTypes(parent).iterateAll()) {
System.out.println(entityType.getDisplayName());
List<Entity> entities = entityType.getEntitiesList();
entities.forEach(EntityCreator::printEntity);
}
} catch (IOException e) {
e.printStackTrace();
}
System.out.println("==========");
}
public static void printEntity(Entity entity) {
System.out.println(entity.getValue());
}
}

2
dialogflow-fun-java/src/main/java/data/ReadNames.java
View File

@@ -8,7 +8,7 @@ import java.util.Scanner;
public class ReadNames {
private static final String NAMES_FILE_LOCATION = "names";
private static final String NAMES_FILE_LOCATION = "names2";
public List<String> getListOfNames() throws FileNotFoundException {

2
dialogflow-fun-java/src/main/resources/names
View File

@@ -247,4 +247,4 @@ Allyn Claire  
Magdalene Michna  
Leticia Jorgenson  
Jaleesa Dally  
Candace Tootle   
Candace Tootle   

10
dialogflow-fun-java/src/main/resources/names_composite Normal file
View File

@@ -0,0 +1,10 @@
mum
dad
my mum
my dad
my @sys.person:person
uncle @sys.person:person
aunt @sys.person:person
grandma @sys.person:person
grandpa @sys.person:person
@sys.person:person