nlpcloud 1.0.9 → 1.0.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ab6ad951a1b80b36984a5b46365fe4ee273aa9d8bfc488b13007649d322860fb
-  data.tar.gz: 3f03bc5ec2be3f45f8e84271ba2ae506804ad9a038f6778f22bf372009429b6b
+  metadata.gz: 5530e1eddfaa48e0547b69a03e6cdb60227db948d0a105a828ec3510f0413a80
+  data.tar.gz: 9b43269b64b3837523a63e3864100ff310008390ad0206af9217f33f7de43dd4
 SHA512:
-  metadata.gz: bd06ca4a59eeeea5a8b00355d0d6a986505c4c17b666d535c17854ce10fa9b3f7f01480cc3efd1b2c7f29a9fe67005e022a6c37b2dbcbada289dec05f6fc3538
-  data.tar.gz: b6552e52e42447424c88573caca8e4d9bdee93c4ec2f5ab5e180bf1dfc3da2d71bd4a187aa41a2bf34af02a9c9fa8dcb61887bdc7ee916cb5700120252341ca6
+  metadata.gz: d174e5a6af642c508f59f273e02e38765de36afcb8792c8312f47f2295f5773d63540d641895bdd4364ca01158e0374e5610786909ab9c1a20fe2c41aaecf60d
+  data.tar.gz: f412ad06bdfadad1b1e0bcf8408e20c51ccfa99753756411ab1d235c2ad9a11702f1e32287a5a46c333ff55ef63fe609ed011747d27e10b6de01ceecc3f533b4

data/lib/nlpcloud.rb

@@ -9,11 +9,17 @@ module NLPCloud
 
   # Client requests the API.
   class Client
-    def initialize(model, token)
+    def initialize(model, token, gpu: false)
       @headers = {
-        'Authorization' => "Token #{token}"
+        'Authorization' => "Token #{token}",
+        'User-Agent' => 'nlpcloud-ruby-client'
       }
-      @root_url = "#{BASE_URL}/#{API_VERSION}/#{model}"
+
+      @root_url = if gpu
+                    "#{BASE_URL}/#{API_VERSION}/gpu/#{model}"
+                  else
+                    "#{BASE_URL}/#{API_VERSION}/#{model}"
+                  end
     end
 
     def entities(text)
@@ -24,7 +30,7 @@ module NLPCloud
       JSON.parse(response.body)
     end
 
-    def classification(text, labels, multi_class)
+    def classification(text, labels, multi_class: nil)
       payload = {
         'text' => text,
         'labels' => labels,
@@ -34,6 +40,33 @@ module NLPCloud
       JSON.parse(response.body)
     end
 
+    def generation(text, min_length: nil, max_length: nil, length_no_input: nil,
+                   end_sequence: nil, remove_input: nil, do_sample: nil, num_beams: nil, early_stopping: nil,
+                   no_repeat_ngram_size: nil, num_return_sequences: nil, top_k: nil, top_p: nil,
+                   temperature: nil, repetition_penalty: nil, length_penalty: nil, bad_words: nil)
+      payload = {
+        'text' => text,
+        'min_length' => min_length,
+        'max_length' => max_length,
+        'length_no_input' => length_no_input,
+        'end_sequence' => end_sequence,
+        'remove_input' => remove_input,
+        'do_sample' => do_sample,
+        'num_beams' => num_beams,
+        'early_stopping' => early_stopping,
+        'no_repeat_ngram_size' => no_repeat_ngram_size,
+        'num_return_sequences' => num_return_sequences,
+        'top_k' => top_k,
+        'top_p' => top_p,
+        'temperature' => temperature,
+        'repetition_penalty' => repetition_penalty,
+        'length_penalty' => length_penalty,
+        'bad_words' => bad_words
+      }
+      response = RestClient.post("#{@root_url}/generation", payload.to_json, @headers)
+      JSON.parse(response.body)
+    end
+
     def sentiment(text)
       payload = {
         'text' => text
@@ -67,6 +100,22 @@ module NLPCloud
       JSON.parse(response.body)
     end
 
+    def langdetection(text)
+      payload = {
+        'text' => text
+      }
+      response = RestClient.post("#{@root_url}/langdetection", payload.to_json, @headers)
+      JSON.parse(response.body)
+    end
+
+    def tokens(text)
+      payload = {
+        'text' => text
+      }
+      response = RestClient.post("#{@root_url}/tokens", payload.to_json, @headers)
+      JSON.parse(response.body)
+    end
+
     def dependencies(text)
       payload = {
         'text' => text

metadata

@@ -1,19 +1,20 @@
 --- !ruby/object:Gem::Specification
 name: nlpcloud
 version: !ruby/object:Gem::Version
-  version: 1.0.9
+  version: 1.0.13
 platform: ruby
 authors:
 - Julien Salinas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-04-27 00:00:00.000000000 Z
+date: 2021-09-28 00:00:00.000000000 Z
 dependencies: []
-description: 'Ruby client for the NLP Cloud API. NLP Cloud serves high performance
-  pre-trained or custom models for NER, sentiment-analysis, classification, summarization,
-  question answering, and POS tagging, ready for production, served through a REST
-  API. More details here: https://nlpcloud.io. Documentation: https://docs.nlpcloud.io.'
+description: 'NLP Cloud serves high performance pre-trained or custom models for NER,
+  sentiment-analysis, classification, summarization, text generation, question answering,
+  machine translation, language detection, tokenization, POS tagging, and dependency
+  parsing. It is ready for production, served through a REST API. This is the Ruby
+  client for the API. More details here: https://nlpcloud.io. Documentation: https://docs.nlpcloud.io.'
 email: all@juliensalinas.com
 executables: []
 extensions: []