omniai 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '08e84fed0ec1535369239cec24e1ba320714b006d63c41c70a3fab4ad9b27448'
-  data.tar.gz: 1b1c8c415873efad649481e9a54b2377e4a1aa3081c1fe17c03615f59d6c82e8
+  metadata.gz: 5d90d000047c670a17478146fa35fca9c993127067b01b11113cc46e739d15e2
+  data.tar.gz: d025855eb6ea4ae79bc7c126a2853fff7a6a0e21d597fa86b2efb6b8d1c3ddf1
 SHA512:
-  metadata.gz: bcebbc73ba9c7fc02aac3c49485bca5c9beea176e53d160fbddfb787442bac4dca7d643b220ee67a156fd9b464af273a8bedcf8201e3120c6671caf5fc9740a8
-  data.tar.gz: aa4a0d58e825d950b8241dfc1fb1da789a1efa3f81f7b7e67d09f8126696624f4a5d4f34dc50d913001ef1ee4f5b00e71f1ac6a478620aa8e0b38ed60d5f20b5
+  metadata.gz: 77247cd74f88fdaa527b377940f32cfaad187c26f8ae4c6f373c708c866ce5c956c0436194786acb474ae07f423ea83511df6bbd5029c91247aff4b5d8ee2891
+  data.tar.gz: c7d3a40a6c21fe1827b162e94eae0dd0d2f66014797e4d878fa6c7aacf17c443412fcefa2f4bc0ad008f7b64b41ab4e2d583109990d6418d77451f1ba67b4fbd

data/README.md

@@ -60,19 +60,57 @@ client = OmniAI::OpenAI::Client.new
 
 ### Chat
 
-Clients that support chat (e.g. Anthropic w/ "Claude", Google w/ "Gemini", Mistral w/ "LeChat", OpenAI w/ "ChatGPT", etc) can generate completions using either a basic or streaming API:
+Clients that support chat (e.g. Anthropic w/ "Claude", Google w/ "Gemini", Mistral w/ "LeChat", OpenAI w/ "ChatGPT", etc) generate completions using the following calls:
 
-#### Basic
+#### w/ a Simple Prompt
 
 ```ruby
 completion = client.chat.completion('Tell me a joke.')
 puts(completion.choice.message.content) # '...'
 ```
 
+#### w/ a Collection of Messages
+
+```ruby
+messages = [
+  {
+    role: 'system',
+    content: 'You are a helpful assistant with an expertise in geography.',
+  },
+  {
+    role: 'user',
+    content: 'What is the capital of Canada?',
+  },
+]
+completion = client.chat.completion(messages, model: '...', temperature: 0.7, format: :json)
+puts(completion.choice.message.content) # '...'
+```
+
+#### w/ a Collection of Files
+
+```ruby
+
+image_a_url = "https://images.unsplash.com/photo-1517849845537-4d257902454a?w=800&h=800&format=jpeg&fit=crop"
+image_b_url = "https://images.unsplash.com/photo-1537151625747-768eb6cf92b2?q=80&w=1024&h=1024&format=jpeg"
+
+message = {
+  role: 'user',
+  content: [
+    OmniAI::Chat::Content.new('What are in these images and are they different?'),
+    OmniAI::Chat::Content.new(image_a_url, type: :image),
+    OmniAI::Chat::Content.new(image_b_url, type: :image),
+  ]
+}
+
+completion = client.chat.completion(message)
+puts(completion.choice.message.content) # '...'
+```
+
 #### Streaming
 
 ```ruby
-client.chat.completion 'Tell me a joke.', stream: proc do |chunk|
+stream = proc do |chunk|
   print(chunk.choice.delta.content) # '...'
 end
+client.chat.completion('Tell me a joke.', stream:)
 ```

data/lib/omniai/chat/chunk.rb

@@ -31,6 +31,11 @@ module OmniAI
         @data['model']
       end
 
+      # @return [Array<OmniAI::Chat::Choice>]
+      def choices
+        @choices ||= @data['choices'].map { |data| Choice.new(data:) }
+      end
+
       # @param [index] [Integer]
       # @return [OmniAI::Chat::Delta]
       def choice(index: 0)

data/lib/omniai/chat/completion.rb

@@ -38,7 +38,7 @@ module OmniAI
 
       # @return [Array<OmniAI::Chat::Choice>]
       def choices
-        @choices ||= @data['choices'].map { |choice| Choice.new(data: choice) }
+        @choices ||= @data['choices'].map { |data| Choice.new(data:) }
       end
 
       # @param [index] [Integer] optional - default is 0

data/lib/omniai/chat/content.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module OmniAI
+  class Chat
+    # A file used for analysis.
+    class Content
+      attr_accessor :type, :value
+
+      # @param url [String]
+      # @param text [String]
+      # @param type [Symbol] :image / :video / :audio / :text
+      def initialize(value, type: :text)
+        @value = value
+        @type = type
+      end
+    end
+  end
+end

data/lib/omniai/config.rb

@@ -0,0 +1,6 @@
+module OmniAI
+  # A configuration for each agent w/ `api_key` / `host` / `logger`.
+  class Config
+    attr_accessor :api_key, :host, :logger
+  end
+end

data/lib/omniai/version.rb

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

data/lib/omniai.rb

@@ -16,7 +16,7 @@ module OmniAI
 
     # @param response [HTTP::Response]
     def initialize(response)
-      super("status=#{response.status.inspect} headers=#{response.headers.inspect} body=#{response.body.inspect}")
+      super("status=#{response.status.inspect} headers=#{response.headers.inspect} body=#{response.body}")
       @response = response
     end
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-06-12 00:00:00.000000000 Z
+date: 2024-06-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -69,11 +69,13 @@ files:
 - lib/omniai/chat/choice.rb
 - lib/omniai/chat/chunk.rb
 - lib/omniai/chat/completion.rb
+- lib/omniai/chat/content.rb
 - lib/omniai/chat/delta.rb
 - lib/omniai/chat/message.rb
 - lib/omniai/chat/request.rb
 - lib/omniai/chat/usage.rb
 - lib/omniai/client.rb
+- lib/omniai/config.rb
 - lib/omniai/version.rb
 homepage: https://github.com/ksylvest/omniai
 licenses: []