pplx-api-ruby 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a69c3c1d912e83e05011f7b2700152fba3da470d13043d24abf8d509f78428c4
-  data.tar.gz: 6021094fa697e64fedaa3e4dcf6a3afb27311c1560f1577f143b4b7d8fcdb124
+  metadata.gz: a3b5a7154c0b6d374d52e2375bc70d23402da6f4bc26b397dabf24232dfebf15
+  data.tar.gz: a195eeb01c658e3d796284f0054fdda32e15c4440129ea868fb77fa0b114bbb8
 SHA512:
-  metadata.gz: f3de955ddabab0ad10c6681520b78c0adba858bc605bf50e2c5fba483c465c7bc9cbfeb7c21987d23f1112fb8a1079b3b32f663d7e0eafeaa898f5d788b23033
-  data.tar.gz: 73b22b8ab14d4c25ffe59490a84192907683e07ee02b204dd711b6d4769386fdaa997f0543a77107957c1489902909ed2f7b336dae8db05894f219433edb2628
+  metadata.gz: 730498b6a5605df3c2b30a864db9feb6d8559fc975978fbef57d59fce41049d30bfdd000d57645c3f21f2a71703d135442ea2edf02c2d234f9f30e1eaa75b924
+  data.tar.gz: 26db3484d88b6194cfffd68c32c58cd3d6bc7d46e2e37fed14bdf56c71601d6f736a0b5d43739fd8cd34ee9620cc6f1c2e93cbce194ca635964c9691e62c92f3

data/lib/pplx_api/client.rb

@@ -1,43 +1,45 @@
-class Client
-  API_ENDPOINT = "https://api.perplexity.ai/chat/completions".freeze
+module PplxApi
+  class Client
+    API_ENDPOINT = "https://api.perplexity.ai/chat/completions".freeze
 
-  def initialize(api_key: nil)
-    @uri = URI(API_ENDPOINT)
-    @api_key = api_key || PplxApi.config.api_key
-  end
+    def initialize(api_key: nil)
+      @uri = URI(API_ENDPOINT)
+      @api_key = api_key || PplxApi.config.api_key
+    end
 
-  def generate_response(model: 'mistral-7b-instruct', messages:, max_tokens: nil, temperature: nil, top_p: nil, top_k: nil, stream: false, presence_penalty: nil, frequency_penalty: nil)
-    headers = {
-      "Content-Type" => "application/json",
-      "Authorization" => "Bearer #{@api_key}"
-    }
+    def generate_response(model: 'mistral-7b-instruct', messages:, max_tokens: nil, temperature: nil, top_p: nil, top_k: nil, stream: false, presence_penalty: nil, frequency_penalty: nil)
+      headers = {
+        "Content-Type" => "application/json",
+        "Authorization" => "Bearer #{@api_key}"
+      }
 
-    body = {
-      model: model,
-      messages: messages,
-      max_tokens: max_tokens,
-      temperature: temperature,
-      top_p: top_p,
-      top_k: top_k,
-      stream: stream,
-      presence_penalty: presence_penalty,
-      frequency_penalty: frequency_penalty
-    }.compact
+      body = {
+        model: model,
+        messages: messages,
+        max_tokens: max_tokens,
+        temperature: temperature,
+        top_p: top_p,
+        top_k: top_k,
+        stream: stream,
+        presence_penalty: presence_penalty,
+        frequency_penalty: frequency_penalty
+      }.compact
 
-    request = Net::HTTP::Post.new(@uri, 'Content-Type' => 'application/json')
-    request.body = body.to_json
+      request = Net::HTTP::Post.new(@uri, 'Content-Type' => 'application/json')
+      request.body = body.to_json
 
-    response = Net::HTTP.start(@uri.hostname, @uri.port, use_ssl: true) do |http|
-      http.request(request)
-    end
+      response = Net::HTTP.start(@uri.hostname, @uri.port, use_ssl: true) do |http|
+        http.request(request)
+      end
 
-    case response
-    when Net::HTTPSuccess
-      JSON.parse(response.body)
-    when Net::HTTPUnprocessableEntity
-      raise "Validation Error"
-    else
-      raise "Unknown Error: #{response.code}"
+      case response
+      when Net::HTTPSuccess
+        JSON.parse(response.body)
+      when Net::HTTPUnprocessableEntity
+        raise "Validation Error"
+      else
+        raise "Unknown Error: #{response.code}"
+      end
     end
   end
 end

data/lib/pplx_api/configuration.rb

@@ -1,13 +1,15 @@
-class Configuration
-  attr_accessor :api_key
-end
+module PplxApi
+  class Configuration
+    attr_accessor :api_key
+  end
 
-@config = Configuration.new
+  @config = Configuration.new
 
-def self.config
-  @config
-end
+  def self.config
+    @config
+  end
 
-def self.configure
-  yield(@config)
+  def self.configure
+    yield(@config)
+  end
 end

data/lib/pplx_api/version.rb

@@ -1,3 +1,3 @@
 module PplxApi
-  VERSION = "0.4.0".freeze
+  VERSION = "0.5.0".freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pplx-api-ruby
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - John Paul