omniai 1.1.0 → 1.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7a31351fb7cc352c108f55c3df8d57020bf2d520b935dce75eb87e62e4a0737c
-  data.tar.gz: a8ca504f40836a0072dcf4549767632a19bb6497da53390404fc1399ae264522
+  metadata.gz: facf65abfc03bc2fedc0572bbdebe5baca776da08b3eba9f4bde4b082bf037a7
+  data.tar.gz: 265613b4ab126f34d68d079241d8f4d2eae96c3e5c4f083aa9ff2761003a5678
 SHA512:
-  metadata.gz: 76f22ab7edf30b0f5aedbe071d7a1ff0fd5c1b2a5ab1f5a36273b9041c62a27796a510a4c73fd52f27606ff4a5e8d84fbb46cea03a6949da71ada89cdc62daba
-  data.tar.gz: 73d4718a9a593e5c520f48f8a1fa6c06560a2f3447c609f1a85ecbbda9c7d55dc45a6bec9d57de5054b1936be1d21266bbcb3324096a8337db1cf05d7359f09a
+  metadata.gz: a689d6591eabb97d47fea58a73c2b7fcc133c62aea51180bffe846976086b3819099708757cae85e20c5db3e38964b9dc7c0dc1cbc9fa8b076bc322a56dec4cb
+  data.tar.gz: bc80184100a4ca2888fb8a0ee3b93ac8d9d301e875e541e692b9a08564eb85fa59d58fcd88e35c4d72142624621ef73174b276c8da95ae53bfcabd1fc1ae776d

data/README.md

@@ -80,7 +80,7 @@ messages = [
   'What is the capital of Canada?'
 ]
 completion = client.chat(messages, model: '...', temperature: 0.7, format: :json)
-completion.choice.message.content
+completion.choice.message.content  # '...'
 ```
 
 #### w/ a Collection of Files
@@ -100,7 +100,7 @@ message = {
 }
 
 completion = client.chat(message)
-completion.choice.message.content
+completion.choice.message.content  # '...'
 ```
 
 #### Streaming
@@ -117,5 +117,6 @@ client.chat('Tell me a joke.', stream:)
 Clients that support chat (e.g. OpenAI w/ "Whisper", etc) convert recordings to text via the following calls:
 
 ```ruby
-client.transcribe(file.path)
+transcription = client.transcribe(file.path)
+transcription.text # '...'
 ```

data/lib/omniai/client.rb

@@ -56,7 +56,7 @@ module OmniAI
     # @param format [Symbol] :text, :srt, :vtt, or :json (default)
     #
     # @return text [OmniAI::Transcribe::Transcription]
-    def transcribe(file, model:, language: nil, prompt: nil, temperature: nil, format: Transcription::Format::JSON)
+    def transcribe(file, model:, language: nil, prompt: nil, temperature: nil, format: nil)
       raise NotImplementedError, "#{self.class.name}#speak undefined"
     end
   end

data/lib/omniai/transcribe/transcription.rb

@@ -4,22 +4,20 @@ module OmniAI
   class Transcribe
     # A transcription returned by the API.
     class Transcription
-      attr_accessor :data, :format
+      attr_accessor :text, :model, :format
 
-      # @param data [Hash]
-      def initialize(data:, format:)
-        @data = data
+      # @param text [String]
+      # @param model [String]
+      # @param format [String]
+      def initialize(text:, model:, format:)
+        @text = text
+        @model = model
         @format = format
       end
 
-      # @return [String]
-      def text
-        @data['text']
-      end
-
       # @return [String]
       def inspect
-        "#<#{self.class} text=#{text.inspect} format=#{format.inspect}>"
+        "#<#{self.class} text=#{text.inspect}>"
       end
     end
   end

data/lib/omniai/transcribe.rb

@@ -119,8 +119,8 @@ module OmniAI
       response = request!
       raise HTTPError, response.flush unless response.status.ok?
 
-      data = @format.eql?(Format::JSON) ? response.parse : { text: String(response.body) }
-      Transcription.new(format: @format, data:)
+      text = @format.nil? || @format.eql?(Format::JSON) ? response.parse['text'] : String(response.body)
+      Transcription.new(text:, model: @model, format: @format)
     end
 
     protected

data/lib/omniai/version.rb

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

metadata

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