nlpcloud 1.0.7 → 1.0.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c8a08233de83602a883f76f24b1be58b8d6fa909b9194b3b1deeb2d9018d4987
-  data.tar.gz: 73830d3e197ff7727371d66f24f24b6829322ec04dafae682fcb3a03750716a3
+  metadata.gz: a39b2bf884144622e741cf6de5e6ec38a9d19ef1379dcc54d1c6d5b8a4f4aeeb
+  data.tar.gz: 0b15ca98c9731449ee6ecab93d298eeb0e673ecac512643c359e282ea2958649
 SHA512:
-  metadata.gz: ffb8e5c89f39e339c1163679b3596b3567018c52e51b626351e186162a13d5bf7ac0258387dba2190d856faa9139788a3871a2fe07efbb76b8ca26d09db2a8af
-  data.tar.gz: 3216fbd124155653a20ea661a8fae938ddd8305b43bd08495b360b91cb57eb1d81a55b6aa774be2e36722f0cd1edbc644bcf6a1abb482eefb352500e1d151bd1
+  metadata.gz: 299e5bddc247032e76dd556bad12fa86dc1b88170335061c2d4f7be468e5ccfff317677549066b88266935746fcdec4a4a7e37928df919c329718f56285d0ffc
+  data.tar.gz: fb23ed98fa5e1baa7177f7d66020135ac9252737c9f7f30e7a32da97fa59030a8795297d31a830e4fb674be7eef510e288ffdce6ba242508cb3b14bfd7a68453

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": 'nlploud-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,26 @@ 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, top_k: nil, top_p: nil,
+                   temperature: nil, repetition_penalty: nil, length_penalty: 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,
+        'top_k' => top_k,
+        'top_p' => top_p,
+        'temperature' => temperature,
+        'repetition_penalty' => repetition_penalty,
+        'length_penalty' => length_penalty
+      }
+      response = RestClient.post("#{@root_url}/generation", payload.to_json, @headers)
+      JSON.parse(response.body)
+    end
+
     def sentiment(text)
       payload = {
         'text' => text
@@ -59,6 +85,30 @@ module NLPCloud
       JSON.parse(response.body)
     end
 
+    def translation(text)
+      payload = {
+        'text' => text
+      }
+      response = RestClient.post("#{@root_url}/translation", payload.to_json, @headers)
+      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.7
+  version: 1.0.11
 platform: ruby
 authors:
 - Julien Salinas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-03-04 00:00:00.000000000 Z
+date: 2021-08-24 00:00:00.000000000 Z
 dependencies: []
-description: 'Ruby client for the NLP Cloud API. NLP Cloud serves high performance
-  pre-trained 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: []