omniai 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 54cb0309adb2c49473342a47f964d2ddb8035a42d8d8fe1d9fa6960ac0987cf4
-  data.tar.gz: 5f6792e26172ad6383308cd98501d6b53f4607411efba7317d90c6525fbb2473
+  metadata.gz: 62a5db8ecb28e379812a5dce657588fb44c55b7af4b8a2463bdb712d91b9f0a9
+  data.tar.gz: 3d5a935c85975fb46a5062c8a4999c0f1fcdb187f2aa542863a4503791dc095b
 SHA512:
-  metadata.gz: f065076a409f5acc450eb020edbeb840d1f66ece51085b8493e2663132210e38cf02d12709e5932ecdeef88f50c8e61541c9b0d676281e7cd22e397ae9cb4f30
-  data.tar.gz: c44b00b48485d903c9f44f63137367fc71a0bf97c38d1d80274f8a51073339125ddbfeb1a427bbd459b496e4fe1c8f7921c5e5c7534af1499a2471621c3be6d5
+  metadata.gz: 4846e70507f51344862e5382ffcf5043b4896408eff86377d75af7439bd5cb6d865a19b84f67ae87be686c9c0be9e1005d3321896a3083ea2a0d208e051a2a6b
+  data.tar.gz: ab185c7be3efffc942c9eb104f4ad77bec9847718f0ee91456e2fd57cd1fd2ca3206fc73f601ba964084bb0c6aa3450da17c8280c53592fa363854d19640faae

data/Gemfile

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+source 'https://rubygems.org'
+
+gemspec
+
+gem 'rake'
+
+gem 'rspec'
+gem 'rspec_junit_formatter'
+gem 'rubocop'
+gem 'rubocop-rake'
+gem 'rubocop-rspec'

data/bin/console

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'bundler/setup'
+require 'omniai'
+
+require 'irb'
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,6 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install

data/lib/omniai/chat/completion.rb

@@ -0,0 +1,74 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    # An abstract class that provides an interface for chatting for various vendors (e.g. OpenAI::OpenAI::ChatGPT::Completion).
+    #
+    # Usage:
+    #
+    #    class OmniAI::OpenAI::ChatGPT::Completion < OmniAI::Chat::Completion
+    #      module Model
+    #        CHAT = "davinci"
+    #      end
+    #      def completion(messages, model:, temperature: 0.0, format: :text)
+    #      end
+    #    end
+    #
+    # Once defined, the subclass can be used to interface with the vendor's chat API as follows:
+    #
+    #    client.chat.completion(messages, model: "...", temperature: 0.0, format: :text)
+    class Completion
+      # @param client [OmniAI::Client] the client
+      # @param messages [String]
+      # @param model [String]
+      # @param temperature [Float]
+      # @param format [Symbol] either :text or :json
+      def initialize(client:, messages:, model:, temperature: 0.0, format: :text)
+        @client = client
+        @messages = messages
+        @model = model
+        @temperature = temperature
+        @format = format
+      end
+
+      # @param prompt [String, Message]
+      # @param model [String]
+      # @param format [Symbol] either :text or :json
+      # @param temperature [Float]
+      # @return [Hash]
+      # @raise [ExecutionError]
+      def process!
+        response = request!
+        raise HTTPError, response unless response.status.ok?
+
+        parse!(response:)
+      end
+
+      protected
+
+      # @param response [HTTP::Response]
+      # @return [OmniAI::Chat::Completion::Response]
+      def parse!(response:)
+        raise NotImplementedError, "#{self.class.name}#parse! undefined"
+      end
+
+      # @return [Hash]
+      def payload
+        raise NotImplementedError, "#{self.class.name}#payload undefined"
+      end
+
+      # @return [String]
+      def path
+        raise NotImplementedError, "#{self.class.name}#path undefined"
+      end
+
+      # @return [HTTP::Response]
+      def request!
+        @client
+          .connection
+          .accept(:json)
+          .post(path, json: payload)
+      end
+    end
+  end
+end

data/lib/omniai/chat.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module OmniAI
+  # An abstract class that provides an interface for chatting for various vendors (e.g. OpenAI::ChatGPT).
+  #
+  # Usage:
+  #
+  #   class OmniAI::OpenAI::Chat < OmniAI::Chat
+  #     def completion(messages, model:, temperature: 0.0, format: :text)
+  #       # TODO: implement
+  #     end
+  #   end
+  #
+  # Once defined, it can be used to interface with the vendor's chat API as follows:
+  #
+  #   client.chat.completion("...", model: "...", temperature: 0.0, format: :text)
+  #
+  # @param client [OmniAI::Client] the client
+  class Chat
+    def initialize(client:)
+      @client = client
+    end
+
+    # @raise [OmniAI::Error]
+    # @param messages [String]
+    # @param model [String]
+    # @param format [Symbol] either :text or :json
+    # @param temperature [Float]
+    # @return [OmniAI::Chat::Completion] an instance of OmniAI::Chat::Completion
+    def completion(messages, model:, temperature: 0.0, format: :text)
+      raise NotImplementedError, "#{self.class.name}#completion undefined"
+    end
+  end
+end

data/lib/omniai/client.rb

@@ -1,27 +1,38 @@
 # frozen_string_literal: true
 
-# Usage:
-#
-#   require "omniai"
-#   require "omniai-anthropic"
-#   require "omniai-google"
-#   require "omniai-mistral"
-#   require "omniai-openai"
-#
-#   anthropic = OmniAI::Anthropic.new(api_key: ENV.fetch("OPENAI_API_KEY"))
-#   google = OmniAI::Google.new(api_key: ENV.fetch("OPENAI_API_KEY"))
-#   mistral = OmniAI::Mistral.new(api_key: ENV.fetch("OPENAI_API_KEY"))
-#   openai = OmniAI::OpenAI.new(api_key: ENV.fetch("OPENAI_API_KEY"))
+module OmniAI
+  # An abstract class that must be subclassed (e.g. OmniAI::OpenAI::Client).
+  #
+  # Usage:
+  #
+  #   class OmniAI::OpenAI::Client < OmniAI::Client
+  #     def initialize(api_key: ENV.fetch('OPENAI_API_KEY'))
+  #       super
+  #     end
+  #
+  #     @return [OmniAI::OpenAI::Chat]
+  #     def chat
+  #       # TODO: implement
+  #     end
+  #   end
+  class Client
+    class Error < StandardError; end
 
-module OmniAI::Client
-  class Error < StandardError; end
+    attr_accessor :api_key
 
-  # @param api_key [String] The API key to use for requests
-  def initialize(api_key:)
-    @api_key = api_key
-  end
+    # @param api_key [String]
+    def initialize(api_key:)
+      @api_key = api_key
+    end
+
+    # @return [HTTP::Client]
+    def connection
+      HTTP.auth("Bearer #{api_key}")
+    end
 
-  def http
-    HTTP.auth("Bearer #{@api_key}")
+    # @return [OmniAI::Chat] an instance of OmniAI::Chat
+    def chat
+      raise NotImplementedError, "#{self.class.name}#chat undefined"
+    end
   end
 end

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = "0.0.1"
+  VERSION = '0.0.2'
 end

data/lib/omniai.rb

@@ -1,7 +1,23 @@
 # frozen_string_literal: true
 
-require_relative "omniai/version"
+require 'http'
+
+require_relative 'omniai/version'
+require_relative 'omniai/client'
+require_relative 'omniai/chat'
+require_relative 'omniai/chat/completion'
 
 module OmniAI
   class Error < StandardError; end
+
+  # An error that wraps an HTTP::Response for non-OK requests.
+  class HTTPError < Error
+    attr_accessor :response
+
+    # @param response [HTTP::Response]
+    def initialize(response)
+      super("status=#{response.status} headers=#{response.headers} body=#{response.body}")
+      @response = response
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Kevin Sylvestre
@@ -24,53 +24,31 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-- !ruby/object:Gem::Dependency
-  name: rspec
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-- !ruby/object:Gem::Dependency
-  name: rubocop
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-description: Gives a consistent interface for interacting with OpenAI's ChatGPT, Google's
-  Gemini, Anthropic's Claude, Mistral's LeChat, and more.
+description: An interface for OpenAI's ChatGPT, Google's Gemini, Anthropic's Claude,
+  Mistral's LeChat, etc.
 email:
 - kevin@ksylvest.com
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- ".rspec"
-- ".rubocop.yml"
+- Gemfile
 - README.md
-- Rakefile
+- bin/console
+- bin/setup
 - lib/omniai.rb
+- lib/omniai/chat.rb
+- lib/omniai/chat/completion.rb
+- lib/omniai/chat/completion/response.rb
 - lib/omniai/client.rb
 - lib/omniai/version.rb
-- sig/omniai.rbs
 homepage: https://github.com/ksylvest/omniai
 licenses: []
-metadata: {}
+metadata:
+  homepage_uri: https://github.com/ksylvest/omniai
+  source_code_uri: https://github.com/ksylvest/omniai
+  changelog_uri: https://github.com/ksylvest/omniai/releases
+  rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []
 require_paths:
@@ -79,7 +57,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.0.0
+      version: 3.3.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="

data/.rspec

@@ -1,3 +0,0 @@
---format documentation
---color
---require spec_helper

data/.rubocop.yml

@@ -1,2 +0,0 @@
-AllCops:
-  TargetRubyVersion: 3.0

data/Rakefile

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
-
-RSpec::Core::RakeTask.new(:spec)
-
-require "rubocop/rake_task"
-
-RuboCop::RakeTask.new
-
-task default: %i[spec rubocop]

data/sig/omniai.rbs

@@ -1,3 +0,0 @@
-module OmniAI
-  VERSION: String
-end