omniai 0.0.9 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bab309e1be1b9acc7671c341b8e50cd0ec257ac22b254899b3661e428432f290
-  data.tar.gz: 7968c811a16cf77172b087a03693641dbca83f2ba683e5eff30eda05ad4817e4
+  metadata.gz: ba57f22a167440e6c6606d9bd1cf098673a11164b088a21f2a0e4d5b05b9689d
+  data.tar.gz: 2adbbf886759d517634f99e95670a532747afec94630e6df226d2d52c52f4330
 SHA512:
-  metadata.gz: b22be38a54e83cb76c45e94d739290ee2530e51d81fd892e8a749deba3a54f88d9f7ec99a159df2688dadd7f5be9ce0a6ad03acfc7d0344e7ae4910e910f3bce
-  data.tar.gz: a5a11dd87da47b2f5ee4a49d9a37bbfe55ccb1ff3b0a6da48544d076eb850e925343775ea90cbdf6a30f309bcb2d3e783be21798d7deecfc2762dca80a886393
+  metadata.gz: f08050fdae89870b429b70f0ebd4c8ed422beecdca7a201fd41bc9653b3f25dc277a9d25a6a9ca95a2abd7d95e7595a4b00e1282bf31e811b51a898eb38a5c0f
+  data.tar.gz: 5e256941913b412d2c7025cd20cfe836c786cf9556d8341b2b8106728eed0798236b64bacd06933127737a0bc69fae8e1d4b1cd42b2c3664091b25036e325a24

data/README.md

@@ -22,9 +22,11 @@ gem install omniai-openai
 
 ## Usage
 
+OmniAI implements APIs for a number of popular clients by default. A client can be initialized using the specific gem (e.g. `omniai-openai` for `OmniAI::OpenAI`). Vendor specific docs can be found within each repo.
+
 ### Client
 
-#### Anthropic (Claude)
+#### [OmniAI::Anthropic](https://github.com/ksylvest/omniai-anthropic)
 
 ```ruby
 require 'omniai-anthropic'
@@ -32,7 +34,7 @@ require 'omniai-anthropic'
 client = OmniAI::Anthropic::Client.new
 ```
 
-#### Google (Gemini)
+#### [OmniAI::Google](https://github.com/ksylvest/omniai-google)
 
 ```ruby
 require 'omniai-google'
@@ -40,7 +42,7 @@ require 'omniai-google'
 client = OmniAI::Google::Client.new
 ```
 
-#### Mistral (LeChat)
+#### [OmniAI::Mistral](https://github.com/ksylvest/omniai-mistral)
 
 ```ruby
 require 'omniai-mistral'
@@ -48,18 +50,18 @@ require 'omniai-mistral'
 client = OmniAI::Mistral::Client.new
 ```
 
-#### OpenAI (ChatGPT)
+#### [OmniAI::OpenAI](https://github.com/ksylvest/omniai-openai)
 
 ```ruby
 require 'omniai-openai'
 
 client = OmniAI::OpenAI::Client.new
-completion = client.chat.completion('Tell me a joke.')
-completion.choice.message.content # '...'
 ```
 
 ### 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:
+
 #### Basic
 
 ```ruby
@@ -70,7 +72,7 @@ puts(completion.choice.message.content) # '...'
 #### Streaming
 
 ```ruby
-stream = proc do |chunk|
+client.chat.completion 'Tell me a joke.', stream: proc do |chunk|
   print(chunk.choice.delta.content) # '...'
 end
 ```

data/lib/omniai/chat/chunk.rb

@@ -2,8 +2,8 @@
 
 module OmniAI
   class Chat
-    # A delta returned by the API.
-    class CompletionChunk
+    # A chunk returned by the API.
+    class Chunk
       attr_accessor :data
 
       # @param data [Hash]

data/lib/omniai/chat/completion.rb

@@ -2,7 +2,7 @@
 
 module OmniAI
   class Chat
-    # An class wrapping the response completion. A vendor may override if custom behavior is required.
+    # A completion returned by the API.
     class Completion
       attr_accessor :data
 

data/lib/omniai/chat/request.rb

@@ -62,7 +62,7 @@ module OmniAI
           parser.feed(chunk) do |_, data|
             break if data.eql?('[DONE]')
 
-            chunk = OmniAI::OpenAI::Chat::Chunk.new(data: data.parse)
+            chunk = OmniAI::OpenAI::Chat::Chunk.new(data: JSON.parse(data))
             @stream.call(chunk)
           end
         end

data/lib/omniai/chat/usage.rb

@@ -2,7 +2,7 @@
 
 module OmniAI
   class Chat
-    # An class wrapping the response usage. A vendor may override if custom behavior is required.
+    # A usage returned by the API.
     class Completion
       attr_accessor :data
 

data/lib/omniai/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.1.1
 platform: ruby
 authors:
 - Kevin Sylvestre
@@ -72,7 +72,6 @@ files:
 - lib/omniai/chat/delta.rb
 - lib/omniai/chat/message.rb
 - lib/omniai/chat/request.rb
-- lib/omniai/chat/stream.rb
 - lib/omniai/chat/usage.rb
 - lib/omniai/client.rb
 - lib/omniai/version.rb

data/lib/omniai/chat/stream.rb

@@ -1,28 +0,0 @@
-# frozen_string_literal: true
-
-module OmniAI
-  class Chat
-    # A processor for streaming back events in chunks.
-    class Stream
-      LINE_REGEX = /data:\s*(?<data>.*)\n\n/
-
-      def initialize
-        @buffer = String.new
-      end
-
-      # @yield [data] a parsed hash
-      # @param [String] chunk
-      def process!(chunk)
-        @buffer << chunk
-
-        while (line = @buffer.slice!(LINE_REGEX))
-          match = LINE_REGEX.match(line)
-          data = match[:data]
-          break if data.eql?('[DONE]')
-
-          yield JSON.parse(data)
-        end
-      end
-    end
-  end
-end