nlpcloud 1.0.8 → 1.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 10c0a4d3c5d521504f79e2f5fff0299d84555508e86257a7b110a3cb6d4483ae
-  data.tar.gz: 64d3bee78530167352fc1b91c5d617844e44b237d714ca52c53119c0f7e8123c
+  metadata.gz: 0d15c8d10001f6858a03294421913df01689dc776786f22292ec4f02f0d73a83
+  data.tar.gz: 3b1bfc5e7dd10844cb6f714e4d6749579c81528fd5599cf76b52129677f2d510
 SHA512:
-  metadata.gz: 4e23aae57e6591d27673d8a20eec8f9b47149a3c1d24448eac02d5892b64fefa0a105d664da04150548b10033c57c3eabc4f5db9b5a730553b6d72525c314fc0
-  data.tar.gz: 34de7fba883dbc73ac045b227ff227d917458dc9190ca77ca259e6232e803ef181b0a640c463e1820eda0534db74b8de2893ec7c353bebe5c92a2293bca60405
+  metadata.gz: 701204c9a11f8000d618ffd9bab25f922e2607d48790931669843371927601336aa0d06ed9a8e4f911d7bc00a0c4f2e829b23d57c230530d941157f5c61367cb
+  data.tar.gz: cc2d2bab211b74a071aab5f9262360b81517d0f82bad2b8081981ffeb6f35ac3e3f1fb2e3546d57c9b7fa2623bf4b137069bd4cb27a9fa6ef2592a3ada1da294

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,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.8
+  version: 1.0.12
 platform: ruby
 authors:
 - Julien Salinas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-04-26 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 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: []
@@ -42,7 +43,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
+rubygems_version: 3.1.4
 signing_key: 
 specification_version: 4
 summary: Ruby client for the NLP Cloud API