nlpcloud 1.0.14 → 1.0.17

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5a625122c70741c70a1d3bf335696ade3bfc453f069f85c57737ee457509ba18
-  data.tar.gz: 496cc60651689612fd01f6d73d22242ec559eb139241debd7fe6367cb725cfda
+  metadata.gz: ba0ea165110cdb347d71a03eecfee3f90683d64526af1f26ee2b01a90818fd09
+  data.tar.gz: daf9745041d2cc720d5df0ef37275f44b320f8a760f985d264ae8173f287c150
 SHA512:
-  metadata.gz: 76b87c80b41bd0ce5c6640c82565f93dcdcf7dc0689784c82020f88bafc95503ef97531e1491e03cd63f73d45a731feff04e2c6d776d05c3fc2c90f009a10632
-  data.tar.gz: '04281a7294ffebc7bb448d190cb617aaba7755d7d6b461e4d353eb4120083127c96db5c42dee219df36a76710a4f4ca61b5836c41a75b1ed2670e13277ce49e8'
+  metadata.gz: 51c84014d9fbf0816ee3727ff013a809aa104be22cb79463a47b0ad2cdfd7b1458e2068ff92c153758e666fd2f349d1e99ba70170f3c8b3d60882153d1d320d0
+  data.tar.gz: d86e7c59c918f9c1d95b36e696eac83a1627096fd0a09c795802092c4f83ebfbe5a402fd482b98c63a8fefb63a02cd76ae1feb2fa5b2ead83bac5ed2cac1e656

data/lib/nlpcloud.rb

@@ -9,15 +9,19 @@ module NLPCloud
 
   # Client requests the API.
   class Client
-    def initialize(model, token, gpu: false)
+    def initialize(model, token, gpu: false, lang: '')
       @headers = {
         'Authorization' => "Token #{token}",
         'Content-Type' => 'application/json',
         'User-Agent' => 'nlpcloud-ruby-client'
       }
 
-      @root_url = if gpu
+      @root_url = if gpu && (lang != '')
+                    "#{BASE_URL}/#{API_VERSION}/gpu/#{lang}/#{model}"
+                  elsif gpu && (lang == '')
                     "#{BASE_URL}/#{API_VERSION}/gpu/#{model}"
+                  elsif !gpu && (lang != '')
+                    "#{BASE_URL}/#{API_VERSION}/#{lang}/#{model}"
                   else
                     "#{BASE_URL}/#{API_VERSION}/#{model}"
                   end
@@ -44,7 +48,7 @@ module NLPCloud
     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)
+                   temperature: nil, repetition_penalty: nil, length_penalty: nil, bad_words: nil, remove_end_sequence: nil)
       payload = {
         'text' => text,
         'min_length' => min_length,
@@ -62,7 +66,8 @@ module NLPCloud
         'temperature' => temperature,
         'repetition_penalty' => repetition_penalty,
         'length_penalty' => length_penalty,
-        'bad_words' => bad_words
+        'bad_words' => bad_words,
+        'remove_end_sequence' => remove_end_sequence
       }
       response = RestClient.post("#{@root_url}/generation", payload.to_json, @headers)
       JSON.parse(response.body)
@@ -109,6 +114,14 @@ module NLPCloud
       JSON.parse(response.body)
     end
 
+    def semantic_similarity(sentences)
+      payload = {
+        'sentences' => sentences
+      }
+      response = RestClient.post("#{@root_url}/semantic_similarity", payload.to_json, @headers)
+      JSON.parse(response.body)
+    end
+
     def tokens(text)
       payload = {
         'text' => text
@@ -133,6 +146,14 @@ module NLPCloud
       JSON.parse(response.body)
     end
 
+    def embeddings(sentences)
+      payload = {
+        'sentences' => sentences
+      }
+      response = RestClient.post("#{@root_url}/embeddings", payload.to_json, @headers)
+      JSON.parse(response.body)
+    end
+
     def lib_versions
       response = RestClient.get("#{@root_url}/versions", @headers)
       JSON.parse(response.body)

metadata

@@ -1,20 +1,21 @@
 --- !ruby/object:Gem::Specification
 name: nlpcloud
 version: !ruby/object:Gem::Version
-  version: 1.0.14
+  version: 1.0.17
 platform: ruby
 authors:
 - Julien Salinas
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-01-10 00:00:00.000000000 Z
+date: 2022-03-01 00:00:00.000000000 Z
 dependencies: []
 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.'
+  machine translation, language detection, semantic similarity, tokenization, POS
+  tagging, embeddings, 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: []