gpt3 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 16be7ac400fbba7f5377773709e5e55d32c7c57dfb50be45692f1a5b093b16af
+  data.tar.gz: b76bdcc17aac130051f106f6a1fcaeaa0519263ee67bbf56376e6ca0b781587b
+SHA512:
+  metadata.gz: daba59d99cffe0db10b42bd1dabf6321be3a03be61b5a87141215fcd3dc8b478b2e994bee95f6591157f36f8fe7f5206da4c751992f678e85237be24bd1f81d5
+  data.tar.gz: db7a6d5ce762ace7d952e3977c41b9b68cb5adf41d471061a7bb2953b967ddb96bbf232931beafc462aa64f39c04e9c99a80e3c5efcc6ec737207b6a1d105956

data/lib/gpt3.rb

@@ -0,0 +1,78 @@
+class GPT3
+  # TODO do these need to declared as dependencies somewhere in the gemspec?
+  require 'net/http'
+  require 'uri'
+  require 'json'
+
+  API_VERSION = 1
+  ENGINE      = 'davinci'
+  BASE_URL    = "https://api.openai.com/v#{API_VERSION}/engines/#{ENGINE}"
+  SECRET_KEY  = ENV.fetch('OPENAI_SK', 'sk-FuRLCnrvrnkcOZoprk259QGjfLWTl1epWw9heo8s')
+
+  class Completion
+    DEFAULT_TEMPERATURE = 0.5
+    # defaults:
+    # {
+    #   "prompt": "Once upon a time",
+    #   "max_tokens": 5,
+    #   "temperature": 1,
+    #   "top_p": 1,
+    #   "n": 1,
+    #   "stream": false,
+    #   "logprobs": null,
+    #   "stop": "\n"
+    # }
+
+    attr_accessor :data
+    def initialize(*params)
+      self.data = params
+    end
+
+    def self.create(prompt, max_tokens: 512, engine: ENGINE, temperature: DEFAULT_TEMPERATURE, n: 1, stop: nil, verbose: false)
+      endpoint = URI.parse("#{GPT3::BASE_URL}/completions")
+      header   = {
+        'Content-Type':  'application/json',
+        'Authorization': 'Bearer ' + SECRET_KEY
+      }
+      data     = {
+        'prompt':      prompt,
+        'max_tokens':  max_tokens,
+        'temperature': temperature,
+        'stop':        stop,
+        'n':           n
+      }
+
+      if verbose
+        puts "+ Creating GPT-3 Completion for the following prompt (with temperature=#{temperature}):"
+        puts "| " + prompt.gsub("\n", "\n| ")
+        puts "+" + ("-" * 50)
+      end
+
+      # Create the HTTP objects
+      https         = Net::HTTP.new(endpoint.host, endpoint.port)
+      https.use_ssl = true
+      request       = Net::HTTP::Post.new(endpoint.request_uri, header)
+      request.body  = data.to_json
+
+      # Send the request
+      response = https.request(request)
+
+      # Parse it and return formatted API results
+      JSON.parse(response.body)
+    end
+
+    def self.search
+      raise "Not Implemented"
+    end
+  end
+
+  class Engine
+    def self.list
+      raise "Not Implemented"
+    end
+
+    def self.retrieve(engine)
+      raise "Not Implemented"
+    end
+  end
+end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: gpt3
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Andrew Brown
+- Indent Labs
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-07-14 00:00:00.000000000 Z
+dependencies: []
+description: An unofficial wrapper for using the GPT-3 Sandbox API in Ruby.
+email: andrew@indentlabs.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/gpt3.rb
+homepage: https://rubygems.org/gems/gpt3
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: An unofficial wrapper for using the GPT-3 Sandbox API in Ruby.
+test_files: []