openai-mapper 0.1.0

data/lib/openai/model.rb

@@ -0,0 +1,15 @@
+# typed: false
+
+module Openai
+  class Model < ::Openai::Resource::Api
+    def initialize
+      @path = "/v1/models"
+      super()
+    end
+
+    def request(model_id)
+      @response = @connection.get(path: [@path, model_id].join("/"))
+      @data = ::Openai::Mapper::Model.from_json(super())
+    end
+  end
+end

data/lib/openai/models.rb

@@ -0,0 +1,15 @@
+# typed: false
+
+module Openai
+  class Models < ::Openai::Resource::Api
+    def initialize
+      @path = "/v1/models"
+      super()
+    end
+
+    def request
+      @response = @connection.get(path: @path)
+      @data = ::Openai::Mapper::Models.from_json(super())
+    end
+  end
+end

data/lib/openai/request/chat/completion.rb

@@ -0,0 +1,24 @@
+# typed: false
+
+module Openai
+  module Request
+    module Chat
+      class Completion < ::Openai::Mapper::BaseMapper
+        attribute :model, ::Shale::Type::String
+        attribute :messages, ::Openai::Mapper::Message, collection: true
+        attribute :functions, ::Openai::Mapper::Functions, default: -> {}
+        attribute :function_call, ::Openai::Mapper::Functions, default: -> {}
+        attribute :temperature, ::Shale::Type::Float, default: -> { 1.0 } # Selector with top_p
+        attribute :top_p, ::Shale::Type::Float, default: -> { 1.0 } # Selector with temperature
+        attribute :n, ::Shale::Type::Integer, default: -> { 1 }
+        attribute :stream, ::Shale::Type::Boolean, default: -> { false }
+        attribute :stop, ::Shale::Type::String, default: -> {} # Selector of types
+        attribute :max_tokens, ::Shale::Type::Integer, default: -> { 5 }
+        attribute :presence_penalty, ::Shale::Type::Float, default: -> { 0 }
+        attribute :frequency_penalty, ::Shale::Type::Float, default: -> { 0 }
+        attribute :logit_bias, ::Shale::Type::String, default: -> {}
+        attribute :user, ::Shale::Type::String, default: -> {}
+      end
+    end
+  end
+end

data/lib/openai/request/images/create.rb

@@ -0,0 +1,15 @@
+# typed: false
+
+module Openai
+  module Request
+    module Images
+      class Create < ::Openai::Mapper::BaseMapper
+        attribute :prompt, ::Shale::Type::String
+        attribute :response_format, ::Shale::Type::String, default: -> { "url" }
+        attribute :n, ::Shale::Type::Integer, default: -> { 1 }
+        attribute :size, ::Shale::Type::String, default: -> {}
+        attribute :user, ::Shale::Type::String, default: -> {}
+      end
+    end
+  end
+end

data/lib/openai/request/images/edit.rb

@@ -0,0 +1,17 @@
+# typed: false
+
+module Openai
+  module Request
+    module Images
+      class Edit < ::Openai::Mapper::BaseMapper
+        attribute :image, ::Shale::Type::String
+        attribute :prompt, ::Shale::Type::String
+        attribute :mask, ::Shale::Type::String
+        attribute :n, ::Shale::Type::Integer, default: -> { 1 }
+        attribute :size, ::Shale::Type::String, default: -> { "1024x1024" }
+        attribute :response_format, ::Shale::Type::String, default: -> { "url" }
+        attribute :user, ::Shale::Type::String, default: -> {}
+      end
+    end
+  end
+end

data/lib/openai/request/images/variation.rb

@@ -0,0 +1,15 @@
+# typed: false
+
+module Openai
+  module Request
+    module Images
+      class Variation < ::Openai::Mapper::BaseMapper
+        attribute :image, ::Shale::Type::String
+        attribute :n, ::Shale::Type::Integer, default: -> { 1 }
+        attribute :response_format, ::Shale::Type::String, default: -> { "url" }
+        attribute :size, ::Shale::Type::String, default: -> { "1024x1024" }
+        attribute :user, ::Shale::Type::String, default: -> {}
+      end
+    end
+  end
+end

data/lib/openai/resource/api.rb

@@ -0,0 +1,73 @@
+# typed: false
+# frozen_string_literal: true
+
+require "securerandom"
+
+module Openai
+  module Resource
+    class Api
+      attr_reader :connection, :path, :data, :response
+
+      def initialize
+        @file_source = ::File.join(__dir__, "../../../input/")
+        Excon.defaults[:ssl_verify_peer] = false
+        @connection = Excon.new(BASE_URL, headers: headers)
+      end
+
+      def request
+        parse_body
+      end
+
+      private
+
+      def ssl_options
+        {verify: true, verify_mode: 0}
+      end
+
+      def headers
+        {
+          "Content-Type" => "application/json",
+          "Authorization" => "Bearer #{ACCESS_TOKEN}"
+        }
+      end
+
+      def parse_body
+        @data = @response.body
+      end
+
+      def multipart_attachment(content_type, key, filename, data)
+        data.binmode if data.respond_to?(:binmode)
+        data.pos = 0 if data.respond_to?(:pos=)
+        body = "--#{@boundary}#{Excon::CR_NL}"
+        body << multipart_file_header(key, filename, content_type)
+        body << Excon::CR_NL
+        body << data.to_s
+        body << Excon::CR_NL
+        body
+      end
+
+      def multipart_file_header(key, filename, content_type)
+        body = %(Content-Disposition: form-data; name="#{key}"; filename="#{filename}"#{Excon::CR_NL})
+        body << "Content-Type: #{content_type}#{Excon::CR_NL}"
+        body
+      end
+
+      def multipart_field(key, value)
+        body = "--#{@boundary}#{Excon::CR_NL}"
+        body << %(Content-Disposition: form-data; name="#{key}"#{Excon::CR_NL})
+        body << Excon::CR_NL
+        body << value.to_s
+        body << Excon::CR_NL
+        body
+      end
+
+      def attach_image(key, filename, image, mime = "image/png")
+        multipart_attachment(mime, key, filename, image)
+      end
+
+      def read_image(image)
+        ::File.read(::File.join(@file_source, image))
+      end
+    end
+  end
+end

data/lib/openai/version.rb

@@ -0,0 +1,6 @@
+# typed: false
+# frozen_string_literal: true
+
+module Openai
+  VERSION = "0.1.0"
+end

data/lib/openai.rb

@@ -0,0 +1,28 @@
+# typed: false
+
+if ["test", "development"].include?(ENV["BUNDLE_ENV"])
+  require "dotenv"
+  Dotenv.load
+end
+
+require "zeitwerk"
+require "excon"
+require "shale"
+require "ox"
+require "tomlib"
+require "shale/adapter/ox"
+
+loader = Zeitwerk::Loader.for_gem
+loader.setup
+
+Shale.xml_adapter = Shale::Adapter::Ox
+Shale.toml_adapter = Tomlib
+
+module Openai
+  BASE_URL = "https://api.openai.com"
+  ACCESS_TOKEN = ENV.fetch("OPENAI_TOKEN")
+
+  class Error < StandardError; end
+end
+
+loader.eager_load

data/openai-mapper.gemspec

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+require_relative "lib/openai/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "openai-mapper"
+  spec.version = Openai::VERSION
+  spec.authors = ["Sergey Bezugliy"]
+  spec.email = ["s.bezugliy@gmail.com"]
+  spec.licenses    = ['MIT']
+
+  spec.summary = "OpenAI resources API client and object mapper"
+  spec.description = "OpenAI resources API client and object mapper"
+  spec.homepage = "https://github.com/sbezugliy/openai-mapper"
+  spec.required_ruby_version = ">= 2.6.0"
+
+  spec.metadata["allowed_push_host"] = "https://rubygems.org/"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/sbezugliy/openai-mapper"
+  spec.metadata["changelog_uri"] = "https://github.com/sbezugliy/openai-mapper/CHANGELOG.MD"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(__dir__) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      (File.expand_path(f) == __FILE__) ||
+        f.start_with?(*%w[bin/ test/ spec/ features/ .git .circleci appveyor Gemfile output])
+    end
+  end
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "excon"
+  spec.add_dependency "shale"
+  # spec.add_dependency "faraday_middleware"
+  # spec.add_dependency "async-http-faraday"
+  spec.add_dependency "rake"
+  spec.add_dependency "zeitwerk"
+  spec.add_dependency "mimemagic"
+  spec.add_dependency "ox"
+  spec.add_dependency "tomlib"
+  spec.add_dependency "pry"
+
+  # Uncomment to register a new dependency of your gem
+  spec.add_development_dependency "dotenv"
+  spec.add_development_dependency "rspec"
+  spec.add_development_dependency "rubocop", "~> 1.56.0"
+  spec.add_development_dependency "rubocop-capybara"
+  spec.add_development_dependency "rubocop-factory_bot"
+  spec.add_development_dependency "rubocop-performance", "~> 1.18.0"
+  spec.add_development_dependency "rubocop-rake"
+  spec.add_development_dependency "rubocop-rspec"
+  spec.add_development_dependency "rubocop-packs"
+  spec.add_development_dependency "simplecov"
+  spec.add_development_dependency "standard", "~> 1.30"
+  spec.add_development_dependency "standard-performance", "~> 1.1.0"
+  spec.add_development_dependency "shoulda-matchers"
+  spec.add_development_dependency "webmock"
+  spec.add_development_dependency "async-rspec", "~> 1.2"
+
+  spec.add_development_dependency "bundler-audit"
+  spec.add_development_dependency "dawnscanner"
+  spec.add_development_dependency "ruby_audit"
+
+  # For more information and examples about making a new gem, check out our
+  # guide at: https://bundler.io/guides/creating_gem.html
+end

data/sig/openai/mapper.rbs

@@ -0,0 +1,6 @@
+module Openai
+  module Mapper
+    VERSION: String
+    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+  end
+end