Add analyzeEntitySentiment sample

Author: Ace Nassri

language/analyze.js

@@ -215,6 +215,87 @@ function analyzeSyntaxInFile (bucketName, fileName) {
   // [END language_syntax_file]
 }
 
+function analyzeEntitySentimentOfText (text) {
+  // [START language_entity_sentiment_string]
+  // Imports the Google Cloud client library
+  const Language = require('@google-cloud/language').v1beta2();
+
+  // Instantiates a client
+  const language = Language.languageServiceClient();
+
+  // The text to analyze, e.g. "Hello, world!"
+  // const text = 'Hello, world!';
+
+  // Configure a request containing a string
+  const request = {
+    document: {
+      type: 'PLAIN_TEXT',
+      content: text
+    }
+  };
+
+  // Detects sentiment of entities in the document
+  language.analyzeEntitySentiment(request)
+    .then((results) => {
+      const entities = results[0].entities;
+
+      console.log(`Entities and sentiments:`)
+      entities.forEach((entity) => {
+        console.log(`  Name: ${entity.name}`);
+        console.log(`  Type: ${entity.type}`);
+        console.log(`  Score: ${entity.sentiment.score}`);
+        console.log(`  Magnitude: ${entity.sentiment.magnitude}`);
+      });
+    })
+    .catch((err) => {
+      console.error('ERROR:', err);
+    });
+  // [END language_entity_sentiment_string]
+}
+
+function analyzeEntitySentimentInFile (bucketName, fileName) {
+  // [START language_entity_sentiment_file]
+  // Imports the Google Cloud client libraries
+  const Language = require('@google-cloud/language').v1beta2();
+  const Storage = require('@google-cloud/storage');
+
+  // Instantiates the clients
+  const language = Language.languageServiceClient();
+  const storage = Storage();
+
+  // The name of the bucket where the file resides, e.g. "my-bucket"
+  // const bucketName = 'my-bucket';
+
+  // The name of the file to analyze, e.g. "file.txt"
+  // const fileName = 'file.txt';
+
+  // Configure a request containing a string
+  const request = {
+    document: {
+      type: 'PLAIN_TEXT',
+      gcsContentUri: `gs://${bucketName}/${fileName}`
+    }
+  };
+
+  // Detects sentiment of entities in the document
+  language.analyzeEntitySentiment(request)
+    .then((results) => {
+      const entities = results[0].entities;
+
+      console.log(`Entities and sentiments:`)
+      entities.forEach((entity) => {
+        console.log(`  Name: ${entity.name}`);
+        console.log(`  Type: ${entity.type}`);
+        console.log(`  Score: ${entity.sentiment.score}`);
+        console.log(`  Magnitude: ${entity.sentiment.magnitude}`);
+      });
+    })
+    .catch((err) => {
+      console.error('ERROR:', err);
+    });
+  // [END language_entity_sentiment_file]
+}
+
 require(`yargs`)
   .demand(1)
   .command(
@@ -253,12 +334,26 @@ require(`yargs`)
     {},
     (opts) => analyzeSyntaxInFile(opts.bucketName, opts.fileName)
   )
+  .command(
+    `entity-sentiment-text <text>`,
+    `Detects sentiment of the entities in a string.`,
+    {},
+    (opts) => analyzeEntitySentimentOfText(opts.text)
+  )
+  .command(
+    `entity-sentiment-file <bucketName> <fileName>`,
+    `Detects sentiment of the entities in a file in Google Cloud Storage.`,
+    {},
+    (opts) => analyzeEntitySentimentInFile(opts.bucketName, opts.fileName)
+  )
   .example(`node $0 sentiment-text "President Obama is speaking at the White House."`)
   .example(`node $0 sentiment-file my-bucket file.txt`, `Detects sentiment in gs://my-bucket/file.txt`)
   .example(`node $0 entities-text "President Obama is speaking at the White House."`)
   .example(`node $0 entities-file my-bucket file.txt`, `Detects entities in gs://my-bucket/file.txt`)
   .example(`node $0 syntax-text "President Obama is speaking at the White House."`)
   .example(`node $0 syntax-file my-bucket file.txt`, `Detects syntax in gs://my-bucket/file.txt`)
+  .example(`node $0 entity-sentiment-text "President Obama is speaking at the White House."`)
+  .example(`node $0 entity-sentiment-file my-bucket file.txt`, `Detects sentiment of entities in gs://my-bucket/file.txt`)
   .wrap(120)
   .recommendCommands()
   .epilogue(`For more information, see https://cloud.google.com/natural-language/docs`)

language/system-test/analyze.test.js

@@ -86,3 +86,21 @@ test('should analyze syntax in a file', async (t) => {
   t.true(output.includes(`President`));
   t.true(output.includes(`Obama`));
 });
+
+test(`should analyze entity sentiment in text`, async (t) => {
+  const output = await runAsync(`${cmd} entity-sentiment-text "${text}"`, cwd);
+  t.true(output.includes(`Entities and sentiments:`));
+  t.true(output.includes(`Obama`));
+  t.true(output.includes(`PERSON`));
+  t.true(output.includes(`Score: 0`));
+  t.true(output.includes(`Magnitude: 0`));
+});
+
+test('should analyze entity sentiment in a file', async (t) => {
+  const output = await runAsync(`${cmd} entity-sentiment-file ${bucketName} ${fileName}`, cwd);
+  t.true(output.includes(`Entities and sentiments:`));
+  t.true(output.includes(`Obama`));
+  t.true(output.includes(`PERSON`));
+  t.true(output.includes(`Score: 0`));
+  t.true(output.includes(`Magnitude: 0`));
+});