openai-client 0.2.1 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: beca324f47af00b9a3783c51af2b17c37ffe9334ab604e0843f2c07d6b637014
-  data.tar.gz: fe550ad70a671ffd2b1bb7dfecdfbbeffa303f6b5201ce6379e1b663230bebd5
+  metadata.gz: cc51a3de3eacf9128058627f6775f3258f6b400dc34064a4194bde9f7da0f477
+  data.tar.gz: f78be5f020154fe2d316b1438eb14e2b11fab21b6001df78bd231ecc58d32269
 SHA512:
-  metadata.gz: cc43ebb61aaf697ba274b39dd838a30c6d79c1fe4e46a3618e49c0f2a82ae0d8809c26ce3ac6d8835d260be73487883d53c1842385cddc098eadd4ba44d2e0a0
-  data.tar.gz: 8fb583dacd100893af86455720c121450bd574af12f5afd7f4bd1f3b21da14461ea5fca0ee4a4a4f402ca8fb197f8c0ef5359c99d46d45222d0abadb55d634d0
+  metadata.gz: c8ac94f2fadb74327367411a8998117a794a2f5aa2cb99e87806403224bd98b35486f5244a3e9d23c8290746d141f4232079372cf667abfd92e66e86651cd2d3
+  data.tar.gz: cd40f73c1a77570f13243213a8b67fcb05ca2f4c196c9501fc9fc498b105f15abe61629a854dfe6a8302c9eeceaab75e32d9e877a632e91580a9f0d4c12b266e

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    openai-client (0.1.0)
+    openai-client (0.3.0)
       faraday (~> 1.8)
       faraday_middleware (~> 1.2)
 

data/README.md

@@ -1,23 +1,43 @@
 # Openai::Client
+
 This gem is a wrapper for calling the OpenAI and GPT-3 APIs.
 
+* [Installation](#installation)
+* [Usage](#usage)
+* [OpenAI Models API](#openai-models-api)
+* [OpenAI Completions API](#openai-completions-api)
+* [OpenAI Edits API](#openai-edits-api)
+* [OpenAI Image API](#openai-image-api)
+  * [Create an Image](#create-an-image)
+  * [Create an Image Edit](#create-an-image-edit)
+  * [Create an Image Variation](#create-an-image-variation)
+
+
+
 ## Installation
+
 Add this line to your application's Gemfile:
+
 ```ruby
 gem 'openai-client'
 ```
+
 And then execute:
+
 ```bash
 bundle
 ```
+
 Or install it yourself as:
+
 ```bash
 gem install openai-client
 ```
 
 ## Usage
-- API key (`access_token`) https://beta.openai.com/account/api-keys.
-- Organization ID (if needed) https://beta.openai.com/account/org-settings.
+
+- API key (`access_token`) <https://beta.openai.com/account/api-keys>.
+- Organization ID (if needed) <https://beta.openai.com/account/org-settings>.
 
 ```ruby
 require 'openai-client'
@@ -26,7 +46,11 @@ Openai::Client.configure do |c|
   c.access_token    = 'access_token'
   c.organization_id = 'organization_id' # optional
 end
+```
+
+## OpenAI Models API
 
+```ruby
 # Models
 Openai::Client.models.list
 
@@ -34,8 +58,93 @@ Openai::Client.models.list
 Openai::Client.models.find(model_id)
 ```
 
+## OpenAI Completions API
+
+```ruby
+request_body = {
+  model: 'text-davinci-003',
+  prompt: 'Say this is a test',
+  max_tokens: 7,
+  temperature: 0,
+  top_p: 1,
+  n: 1,
+  stream: false,
+  logprobs: nil,
+  stop: "\n"
+}
+Openai::Client.completions.create(request_body)
+```
+
+[Request body documentation](https://platform.openai.com/docs/api-reference/completions/create)
+
+## OpenAI Edits API
+
+```ruby
+request_body = {
+  model: 'text-davinci-edit-001',
+  input: 'What day of the wek is it?',
+  instruction: 'Fix the spelling mistakes'
+}
+Openai::Client.edits.create(request_body)
+```
+
+[Request body documentation](https://platform.openai.com/docs/api-reference/edits/create)
+
+## OpenAI Image API
+
+### Create an Image
+
+```ruby
+request_body = {
+  prompt: 'A cute baby sea otter',
+  n: 1,                  # between 1 and 10
+  size: '1024x1024',     # 256x256, 512x512, or 1024x1024
+  response_format: 'url' # url or b64_json
+}
+response = Openai::Client.images.create(request_body)
+```
+
+[Request body documentation](https://platform.openai.com/docs/api-reference/images/create)
+
+### Create an Image Edit
+
+```ruby
+request_body = {
+  image: '/absolute/path/to/image/you/want/to/change/img.png'
+  mask: '/absolute/path/to/mask.png'
+  prompt: 'A cute baby sea otter wearing a beret',
+  n: 1,                  # between 1 and 10
+  size: '1024x1024',     # 256x256, 512x512, or 1024x1024
+  response_format: 'url' # url or b64_json
+}
+response = Openai::Client.images.edit(request_body)
+```
+
+- `image` - must be a valid PNG file, less than 4MB, and square. If mask is not provided, image must have transparency, which will be used as the mask.
+- `mask` - an additional image whose fully transparent areas (e.g. where alpha is zero) indicate where image should be edited. Must be a valid PNG file, less than 4MB, and have the same dimensions as image.
+
+[Request body documentation](https://platform.openai.com/docs/api-reference/images/create-edit)
+
+### Create an Image Variation
+
+```ruby
+request_body = {
+  image: '/absolute/path/to/image.png'
+  n: 1,                  # between 1 and 10
+  size: '1024x1024',     # 256x256, 512x512, or 1024x1024
+  response_format: 'url' # url or b64_json
+}
+response = Openai::Client.images.variations(request_body)
+```
+
+- `image` - must be a valid PNG file, less than 4MB, and square.
+
+[Request body documentation](https://platform.openai.com/docs/api-reference/images/create-variation)
+
 ## Contributing
-Bug reports and pull requests are welcome on GitHub at https://github.com/itikhonenko/openai-client.
+
+Bug reports and pull requests are welcome on GitHub at <https://github.com/itikhonenko/openai-client>.
 
 ## License
-The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/lib/openai/client/completions.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Openai
+  module Client
+    class Completions
+      PATH = 'completions'
+
+      # @api public
+      # Public: Makes an API call to create a completion.
+      #
+      # @param [Hash] body request body
+      #
+      # @return [Hash] a hash with a completion
+      def create(body)
+        Http.new.post(PATH, body).body
+      rescue Faraday::Error
+        nil
+      end
+    end
+  end
+end

data/lib/openai/client/edits.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module Openai
+  module Client
+    class Edits
+      PATH = 'edits'
+
+      # @api public
+      # Public: Makes an API call to create edit.
+      #
+      # @param [Hash] body request body
+      #
+      # @return [Hash] a hash with edit
+      def create(body)
+        Http.new.post(PATH, body).body
+      rescue Faraday::Error
+        nil
+      end
+    end
+  end
+end

data/lib/openai/client/http.rb

@@ -8,6 +8,7 @@ module Openai
           conn.response :json
           conn.response :raise_error
           conn.request :json
+          conn.request :multipart
         end
         @logger = Openai::Client.configuration.logger
       end
@@ -42,6 +43,24 @@ module Openai
         log_error(e) && raise
       end
 
+      # @api public
+      # Public: Makes a multipart request using the Faraday HTTP Client.
+      #
+      # @param [String] path API path
+      # @param [Hash] body request body
+      #
+      # @raise [Faraday::Error] on failure API call
+      #
+      # @return [Faraday::Response] instance of Faraday::Response class
+      def multipart_post(path, body = {})
+        connection.post(path, body) do |request|
+          request.headers['Content-Type'] = 'multipart/form-data'
+          request.options.timeout = 300
+        end
+      rescue Faraday::Error => e
+        log_error(e) && raise
+      end
+
       private
 
       attr_reader :connection, :logger

data/lib/openai/client/images.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+module Openai
+  module Client
+    class Images
+      CREATE_PATH    = 'images/generations'
+      EDIT_PATH      = 'images/edits'
+      VARIATION_PATH = 'images/variations'
+
+      # @api public
+      # Public: Makes an API call to create an image given a prompt.
+      #
+      # @param [Hash] body request body
+      #
+      # @return [Hash] an image
+      def create(body)
+        Http.new.post(CREATE_PATH, body).body
+      rescue Faraday::Error
+        nil
+      end
+
+      # @api public
+      # Public: Makes an API call to create an edited or extended image given an original image and a prompt.
+      #
+      # @param [Hash] body request body
+      #
+      # @return [Hash] an edited or extended image
+      def edit(body)
+        Http.new.multipart_post(EDIT_PATH, upload_io(body, %i[image mask])).body
+      rescue Faraday::Error
+        nil
+      end
+
+      # @api public
+      # Public: Makes an API call to creates a variation of a given image.
+      #
+      # @param [Hash] body request body
+      #
+      # @return [Hash] a variation of a given image
+      def variations(body)
+        Http.new.multipart_post(VARIATION_PATH, upload_io(body, %i[image])).body
+      rescue Faraday::Error
+        nil
+      end
+
+      private
+
+      # @api private
+      # Internal: Creates the open IO object for the uploaded file.
+      #
+      # @param [Hash] data request body
+      # @param [Array] param_names parameter names
+      #
+      # @return [Hash] request body
+      def upload_io(data, param_names)
+        param_names.each do |param|
+          data[param] = Faraday::FilePart.new(data[param], 'rb') if data[param]
+        end
+        data
+      end
+    end
+  end
+end

data/lib/openai/client/models.rb

@@ -11,7 +11,7 @@ module Openai
       # @return [Hash] a hash with models
       def list
         Http.new.get(PATH).body
-      rescue StandardError
+      rescue Faraday::Error
         nil
       end
 
@@ -23,7 +23,7 @@ module Openai
       # @return [Hash] found model or nil
       def find(id)
         Http.new.get("#{PATH}/#{id}").body
-      rescue StandardError
+      rescue Faraday::Error
         nil
       end
     end

data/lib/openai/client/version.rb

@@ -2,6 +2,6 @@
 
 module Openai
   module Client
-    VERSION = '0.2.1'
+    VERSION = '0.4.0'
   end
 end

data/lib/openai/client.rb

@@ -9,12 +9,15 @@ require 'openai/client/configuration'
 require 'openai/client/configurable'
 require 'openai/client/http'
 require 'openai/client/models'
+require 'openai/client/edits'
+require 'openai/client/completions'
+require 'openai/client/images'
 
 module Openai
   module Client
     extend Configurable
 
-    ATTRS = ['models'].freeze
+    ATTRS = ['models', 'edits', 'completions', 'images'].freeze
 
     class << self
       ATTRS.each do |attr|

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: openai-client
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.4.0
 platform: ruby
 authors:
 - Ihor Tykhonenko
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-02-06 00:00:00.000000000 Z
+date: 2023-02-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -54,12 +54,14 @@ files:
 - README.md
 - Rakefile
 - lib/openai/client.rb
+- lib/openai/client/completions.rb
 - lib/openai/client/configurable.rb
 - lib/openai/client/configuration.rb
+- lib/openai/client/edits.rb
 - lib/openai/client/http.rb
+- lib/openai/client/images.rb
 - lib/openai/client/models.rb
 - lib/openai/client/version.rb
-- openai-client-0.1.0.gem
 - sig/openai/client.rbs
 homepage: https://github.com/itikhonenko/openai-client
 licenses: