pplx-api-ruby 0.3.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: fc5b50652bf8d4622aca8f4aaa8fec78b35c74aa7e3685a673b55343e5e3cfed
+  data.tar.gz: 14a0cf1d709a231c2e5f9b59972a9c8e17ab70f56e5fc59cae6439a86c919e9f
+SHA512:
+  metadata.gz: de45182af9a65838961b1638115caf3cdff39e62ec4c207d8142cf8ed11456631e3dea462204db9a150948361e738341aa2dc2229b9a83578fe7a704f00decea
+  data.tar.gz: 1aba8a8502cdb39ebf476a23a9e8fc655114f575922f7b3b0d6f66ebcda1ab4ebb97c693ae3d6c26ba5bc67afc7ebbb2200b431de1d28f7b84a7eb7c058b721f

data/lib/pplx_api/client.rb

@@ -0,0 +1,43 @@
+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 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
+
+    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
+
+    case response
+    when Net::HTTPSuccess
+      JSON.parse(response.body)
+    when Net::HTTPUnprocessableEntity
+      raise "Validation Error"
+    else
+      raise "Unknown Error: #{response.code}"
+    end
+  end
+end

data/lib/pplx_api/configuration.rb

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

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification
+name: pplx-api-ruby
+version: !ruby/object:Gem::Version
+  version: 0.3.0
+platform: ruby
+authors:
+- John Paul
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2023-10-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: One of the fastest ways to access Mistral 7B, Llama2 13B, Code Llama
+  34B, Llama2 70B, replit-code-v1.5-3b models.
+email:
+- johnarpaul@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/pplx_api/client.rb
+- lib/pplx_api/configuration.rb
+homepage: https://github.com/Hyperaide/pplx-api-ruby
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/Hyperaide/pplx-api-ruby
+  source_code_uri: https://github.com/Hyperaide/pplx-api-ruby
+  changelog_uri: https://github.com/Hyperaide/pplx-api-ruby
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.6.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.15
+signing_key:
+specification_version: 4
+summary: API wrapper for pplx-api.
+test_files: []