omniai 1.3.1 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 28e6e2a736d78cbc0135308eb28320a85a319b78a888cd96fa3320456c3c0222
-  data.tar.gz: '044786d21ba8f2316599836af153c6dec80d055a0cbcf1e843c000296d1df350'
+  metadata.gz: 8b3d66712c9a2fd460abf00c29b056b87ff8cdd08c8b38d69ece0755d8c76578
+  data.tar.gz: f0fdf7b20e3f3cc3c4e2bee6d21b68bd917d9267c9f1297dec91f2f8bccc71e9
 SHA512:
-  metadata.gz: c260f1b52fe5ffec70dc5c6755426c9798062bb234da303d76ec21319d450433e95f03ca787fcfa8cd966899235416d5a83580ffd11c24615d8a58df8d64efbc
-  data.tar.gz: 0acc9d0556d0bf8de430ca3a52977c88c6a43edfe017226d1831b4ccc034d2ba3bcde38fad112d61d8910bfc9a9ee440999354919a8d4c3cb614a0d0fc3e2314
+  metadata.gz: 98c6c3380181aaf7ad858968b998adab49f8cd200d94755ff6cf79423378dd401c4c538147aeb416b023c47fe7f1806a8742a9486429e90bcc487eb8944cba52
+  data.tar.gz: f251bb7436e9536275a285508f63964693a067378f55a72b3cf01fb9cb801f55ef96fd8a64c9b4e5790c6c400ee6adcee64f6dc5dc413e92b42d5f869afce45f

data/README.md

@@ -193,3 +193,38 @@ tempfile = client.speak('The quick brown fox jumps over a lazy dog.', voice: 'HA
 tempfile.close
 tempfile.unlink
 ```
+
+## CLI
+
+OmniAI packages a basic command line interface (CLI) to allow for exploration of various APIs. A detailed CLI documentation can be found via help:
+
+```bash
+omniai --help
+```
+
+### Chat
+
+#### w/ a Prompt
+
+```bash
+omniai chat "What is the coldest place on earth?"
+```
+
+```
+The coldest place on earth is Antarctica.
+```
+
+#### w/o a Prompt
+
+```bash
+omniai chat --provider="openai" --model="gpt-4" --temperature="0.5"
+```
+
+```
+Type 'exit' or 'quit' to abort.
+# What is the warmet place on earth?
+```
+
+```
+The warmest place on earth is Africa.
+```

data/exe/omniai

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'omniai'
+
+cli = OmniAI::CLI.new
+cli.parse

data/lib/omniai/chat.rb

@@ -45,7 +45,7 @@ module OmniAI
     # @param client [OmniAI::Client] the client
     # @param model [String] required
     # @param temperature [Float, nil] optional
-    # @param stream [Proc, nil] optional
+    # @param stream [Proc, IO, nil] optional
     # @param format [Symbol, nil] optional - :json
     def initialize(messages, client:, model:, temperature: nil, stream: nil, format: nil)
       @messages = messages
@@ -97,7 +97,14 @@ module OmniAI
     def stream!(response:)
       raise Error, "#{self.class.name}#stream! unstreamable" unless @stream
 
-      Stream.new(response:).stream! { |chunk| @stream.call(chunk) }
+      Stream.new(response:).stream! do |chunk|
+        case @stream
+        when IO then @stream << chunk
+        else @stream.call(chunk)
+        end
+      end
+
+      @stream.flush if @stream.is_a?(IO)
     end
 
     # @return [Array<Hash>]

data/lib/omniai/cli/base_handler.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class CLI
+    # A generic handler for CLI commands (e.g. 'omnia chat').
+    class BaseHandler
+      # @param stdin [IO] an optional stream for stdin
+      # @param stdout [IO] an optional stream for stdout
+      # @param provider [String] an optional provider (defaults to 'openai')
+      # @param argv [Array<String>]
+      def self.handle!(argv:, stdin: $stdin, stdout: $stdout, provider: 'openai')
+        new(stdin:, stdout:, provider:).handle!(argv:)
+      end
+
+      # @param stdin [IO] an optional stream for stdin
+      # @param stdout [IO] an optional stream for stdout
+      # @param provider [String] an optional provider (defaults to 'openai')
+      def initialize(stdin: $stdin, stdout: $stdout, provider: 'openai')
+        @stdin = stdin
+        @stdout = stdout
+        @provider = provider
+        @args = {}
+      end
+
+      # @param argv [Array<String>]
+      def handle!(argv:)
+        raise NotImplementedError, "#{self.class}#handle! undefined"
+      end
+
+      private
+
+      # @return [OmniAI::Client]
+      def client
+        @client ||= OmniAI::Client.find(provider: @provider)
+      end
+    end
+  end
+end

data/lib/omniai/cli/chat_handler.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class CLI
+    # Used for CLI usage of 'omnia chat'.
+    class ChatHandler < BaseHandler
+      def handle!(argv:)
+        parser.parse!(argv)
+
+        if argv.empty?
+          listen!
+        else
+          chat(prompt: argv.join(' '))
+        end
+      end
+
+      private
+
+      def listen!
+        @stdout.puts('Type "exit" or "quit" to leave.')
+
+        loop do
+          @stdout.print('# ')
+          prompt = @stdin.gets&.chomp
+
+          break if prompt.nil? || prompt.match?(/\A(exit|quit)\z/i)
+
+          chat(prompt:)
+        rescue Interrupt
+          break
+        end
+      end
+
+      # @param prompt [String]
+      def chat(prompt:)
+        client.chat(prompt, **@args, stream: @stdout)
+      end
+
+      # @return [OptionParser]
+      def parser
+        OptionParser.new do |options|
+          options.banner = 'usage: omniai chat [options] "<prompt>"'
+
+          options.on('-h', '--help', 'help') do
+            @stdout.puts(options)
+            exit
+          end
+
+          options.on('-p', '--provider=PROVIDER', 'provider') { |provider| @provider = provider }
+          options.on('-m', '--model=MODEL', 'model') { |model| @args[:model] = model }
+          options.on('-t', '--temperature=TEMPERATURE', Float, 'temperature') do |temperature|
+            @args[:temperature] = temperature
+          end
+          options.on('-f', '--format=FORMAT', 'format') { |format| @args[:format] = format.intern }
+        end
+      end
+    end
+  end
+end

data/lib/omniai/cli.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+require 'optparse'
+
+module OmniAI
+  # Used when interacting with the suite from the command line interface (CLI).
+  #
+  # Usage:
+  #
+  #   cli = OmniAI::CLI.new
+  #   cli.parse
+  class CLI
+    ChatArgs = Struct.new(:provider, :model, :temperature)
+
+    # @param in [IO] a stream
+    # @param out [IO] a stream
+    # @param provider [String] a provider
+    def initialize(stdin: $stdin, stdout: $stdout, provider: 'openai')
+      @stdin = stdin
+      @stdout = stdout
+      @provider = provider
+      @args = {}
+    end
+
+    def parse(argv = ARGV)
+      parser.order!(argv)
+      command = argv.shift
+      return if command.nil?
+
+      case command
+      when 'chat' then ChatHandler.handle!(stdin: @stdin, stdout: @stdout, provider: @provider, argv:)
+      else raise Error, "unsupported command=#{command.inspect}"
+      end
+    end
+
+    private
+
+    # @return [OptionParser]
+    def parser
+      OptionParser.new do |options|
+        options.banner = 'usage: omniai [options] <command> [<args>]'
+
+        options.on('-h', '--help', 'help') do
+          @stdout.puts(options)
+          exit
+        end
+
+        options.on('-v', '--version', 'version') do
+          @stdout.puts(VERSION)
+          exit
+        end
+
+        options.on('-p', '--provider=PROVIDER', 'provider (default="openai")') do |provider|
+          @provider = provider
+        end
+
+        options.separator <<~COMMANDS
+          commands:
+            chat
+        COMMANDS
+      end
+    end
+  end
+end

data/lib/omniai/client.rb

@@ -16,8 +16,6 @@ module OmniAI
   #     end
   #   end
   class Client
-    class Error < StandardError; end
-
     # @return [String, nil]
     attr_accessor :api_key
 
@@ -30,6 +28,68 @@ module OmniAI
     # @return [Integer, nil]
     attr_accessor :timeout
 
+    # Initialize a client for Anthropic. This method requires the provider if it is undefined.
+    #
+    # @raise [OmniAI::Error] if the provider is not defined and the gem is not installed
+    # @return [Class<OmniAI::Client>]
+    def self.anthropic
+      require 'omniai/anthropic' unless defined?(OmniAI::Anthropic::Client)
+      OmniAI::Anthropic::Client
+    rescue LoadError
+      raise Error, "requires 'omniai-anthropic': `gem install omniai-anthropic`"
+    end
+
+    # Lookup the `OmniAI::Google::Client``. This method requires the provider if it is undefined.
+    #
+    # @raise [OmniAI::Error] if the provider is not defined and the gem is not installed
+    # @return [Class<OmniAI::Client>]
+    def self.google
+      require 'omniai/google' unless defined?(OmniAI::Google::Client)
+      OmniAI::Google::Client
+    rescue LoadError
+      raise Error, "requires 'omniai-google': `gem install omniai-google`"
+    end
+
+    # Initialize a client for Mistral. This method requires the provider if it is undefined.
+    #
+    # @raise [OmniAI::Error] if the provider is not defined and the gem is not installed
+    # @return [Class<OmniAI::Client>]
+    def self.mistral
+      require 'omniai/mistral' unless defined?(OmniAI::Mistral::Client)
+      OmniAI::Mistral::Client
+    rescue LoadError
+      raise Error, "requires 'omniai-mistral': `gem install omniai-mistral`"
+    end
+
+    # Initialize a client for OpenAI. This method requires the provider if it is undefined.
+    #
+    # @raise [OmniAI::Error] if the provider is not defined and the gem is not installed
+    # @return [Class<OmniAI::Client>]
+    def self.openai
+      require 'omniai/openai' unless defined?(OmniAI::OpenAI::Client)
+      OmniAI::OpenAI::Client
+    rescue LoadError
+      raise Error, "requires 'omniai-openai': `gem install omniai-openai`"
+    end
+
+    # Initialize a client by provider (e.g. 'openai'). This method attempts to require the provider.
+    #
+    # @raise [OmniAI::Error] if the provider is not defined and the gem is not installed
+    # @param provider [String] required (e.g. 'anthropic', 'google', 'mistral', 'openai', etc)
+    # @return [OmniAI::Client]
+    def self.find(provider:, **)
+      klass =
+        case provider
+        when 'anthropic' then anthropic
+        when 'google' then google
+        when 'mistral' then mistral
+        when 'openai' then openai
+        else raise Error, "unknown provider=#{provider.inspect}"
+        end
+
+      klass.new(**)
+    end
+
     # @param api_key [String, nil] optional
     # @param host [String, nil] optional - supports for customzing the host of the client (e.g. 'http://localhost:8080')
     # @param logger [Logger, nil] optional

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = '1.3.1'
+  VERSION = '1.4.0'
 end

data/lib/omniai.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require 'logger'
 require 'event_stream_parser'
 require 'http'
 require 'uri'
@@ -8,6 +9,7 @@ require 'zeitwerk'
 loader = Zeitwerk::Loader.for_gem
 loader.inflector.inflect 'omniai' => 'OmniAI'
 loader.inflector.inflect 'url' => 'URL'
+loader.inflector.inflect 'cli' => 'CLI'
 loader.setup
 
 module OmniAI

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.3.1
+  version: 1.4.0
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-06-21 00:00:00.000000000 Z
+date: 2024-07-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -56,7 +56,8 @@ description: An interface for OpenAI's ChatGPT, Google's Gemini, Anthropic's Cla
   Mistral's LeChat, etc.
 email:
 - kevin@ksylvest.com
-executables: []
+executables:
+- omniai
 extensions: []
 extra_rdoc_files: []
 files:
@@ -64,6 +65,7 @@ files:
 - README.md
 - bin/console
 - bin/setup
+- exe/omniai
 - lib/omniai.rb
 - lib/omniai/chat.rb
 - lib/omniai/chat/chunk.rb
@@ -78,6 +80,9 @@ files:
 - lib/omniai/chat/message_choice.rb
 - lib/omniai/chat/stream.rb
 - lib/omniai/chat/usage.rb
+- lib/omniai/cli.rb
+- lib/omniai/cli/base_handler.rb
+- lib/omniai/cli/chat_handler.rb
 - lib/omniai/client.rb
 - lib/omniai/config.rb
 - lib/omniai/instrumentation.rb