omniai 1.0.6 → 1.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 172ad18b8fbe97f2e8d23bf8624efea92591910f6d05eed5295789555f60e6d7
-  data.tar.gz: dc415cd3f63f293bc2d64cff39c3ee3c094f734d7a0fb330a1b25adf772f7016
+  metadata.gz: 1971cb590311266b342596c44ff56398999b8657e9f6cf2b8d2afffb8b91257b
+  data.tar.gz: 7866818c3dd9c1af3b03a351ecdc38c527ef1d08bb877533a9f9159bce8d5ed5
 SHA512:
-  metadata.gz: 1dc885877d697bf3c2053e8facbdc1a7329a8b3ae0804753e0e1daefcc9f37bcc7dd19de11c748fccdf4cc9be032cbbb1128558a8d437a125ab41ca44402d7ce
-  data.tar.gz: cb96cde1ff5203f08646217bfae1c0032a25f492cbb6b1565f34ea9a5a905c0626b6bfab5e651793cd853b34eaf0a95130ad8a41e2f8a77429c9faf0feb58cc5
+  metadata.gz: 2c93b8a97fc2ca09765e9c2c537c15491a7344cc505e6204addd972dbb2c7ee2219213d2892745df48bdd4c0193bf3d80391142da2462de437290fc3a6d442db
+  data.tar.gz: 0b7d13dfde39f89c78a57ea3b03549262fb7b9962d127a01e89ca3785402098b6c0e34fc957145615d279a14f4d644769c25a2d3f833aa9c1136397d6b7384b3

data/Gemfile

@@ -11,3 +11,5 @@ gem 'rspec_junit_formatter'
 gem 'rubocop'
 gem 'rubocop-rake'
 gem 'rubocop-rspec'
+gem 'simplecov'
+gem 'webmock'

data/README.md

@@ -66,7 +66,7 @@ Clients that support chat (e.g. Anthropic w/ "Claude", Google w/ "Gemini", Mistr
 
 ```ruby
 completion = client.chat('Tell me a joke.')
-puts(completion.choice.message.content) # '...'
+completion.choice.message.content # '...'
 ```
 
 #### w/ a Collection of Messages
@@ -77,13 +77,10 @@ messages = [
     role: 'system',
     content: 'You are a helpful assistant with an expertise in geography.',
   },
-  {
-    role: 'user',
-    content: 'What is the capital of Canada?',
-  },
+  'What is the capital of Canada?'
 ]
 completion = client.chat(messages, model: '...', temperature: 0.7, format: :json)
-puts(completion.choice.message.content) # '...'
+completion.choice.message.content
 ```
 
 #### w/ a Collection of Files
@@ -103,7 +100,7 @@ message = {
 }
 
 completion = client.chat(message)
-puts(completion.choice.message.content) # '...'
+completion.choice.message.content
 ```
 
 #### Streaming

data/lib/omniai/chat/choice.rb

@@ -4,28 +4,23 @@ module OmniAI
   class Chat
     # A choice returned by the API.
     class Choice
-      attr_accessor :data
+      attr_accessor :index, :delta, :message
 
       # @param data [Hash]
-      def initialize(data:)
-        @data = data
-      end
-
-      # @return [Integer]
-      def index
-        @data['index']
-      end
+      # @return [OmniAI::Chat::Choice]
+      def self.for(data:)
+        index = data['index']
+        delta = Delta.for(data: data['delta']) if data['delta']
+        message = Message.for(data: data['message']) if data['message']
 
-      # @return [OmniAI::Chat::Delta]
-      def delta
-        Delta.new(data: @data['delta']) if @data['delta']
+        new(index:, delta:, message:)
       end
 
-      # @return [OmniAI::Chat::Message]
-      def message
-        return unless @data['message']
-
-        @message ||= Message.for(role: @data['message'])
+      # @param data [Hash]
+      def initialize(index:, delta:, message:)
+        @index = index
+        @delta = delta
+        @message = message
       end
     end
   end

data/lib/omniai/chat/completion.rb

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

data/lib/omniai/chat/delta.rb

@@ -4,22 +4,27 @@ module OmniAI
   class Chat
     # A delta returned by the API.
     class Delta
-      attr_accessor :data
+      attr_accessor :role, :content
 
-      # @param content [Integer]
-      # @param role [String]
-      def initialize(data:)
-        @data = data
+      # @param data [Hash]
+      # @return [OmniAI::Chat::Message]
+      def self.for(data:)
+        content = data['content'] || data[:content]
+        role = data['role'] || data[:role]
+
+        new(content:, role: role || Role::USER)
       end
 
-      # @return [String, nil]
-      def role
-        @data['role']
+      # @param content [String]
+      # @param role [String]
+      def initialize(content:, role: nil)
+        @content = content
+        @role = role
       end
 
-      # @return [String, nil]
-      def content
-        @data['content']
+      # @return [String]
+      def inspect
+        "#<#{self.class.name} role=#{role.inspect} content=#{content.inspect}>"
       end
     end
   end

data/lib/omniai/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.0.6
+  version: 1.0.8
 platform: ruby
 authors:
 - Kevin Sylvestre