Better describe the transcription

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    omniai (1.1.1)
+    omniai (1.1.2)
       event_stream_parser
       http
       zeitwerk

lib/omniai/transcribe.rb

@@ -119,8 +119,8 @@ def process!
       response = request!
       raise HTTPError, response.flush unless response.status.ok?
 
-      data = @format.nil? || @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

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

lib/omniai/version.rb

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

spec/omniai/transcribe/transcription_spec.rb

@@ -1,17 +1,21 @@
 # frozen_string_literal: true
 
 RSpec.describe OmniAI::Transcribe::Transcription do
-  subject(:transcription) { described_class.new(format: OmniAI::Transcribe::Format::JSON, data: { 'text' => 'Hi!' }) }
-
-  describe '#format' do
-    it { expect(transcription.format).to eq('json') }
-  end
+  subject(:transcription) { described_class.new(text: 'Hi!', model: 'whisper', format: 'text') }
 
   describe '#text' do
     it { expect(transcription.text).to eq('Hi!') }
   end
 
+  describe '#model' do
+    it { expect(transcription.model).to eq('whisper') }
+  end
+
+  describe '#format' do
+    it { expect(transcription.format).to eq('text') }
+  end
+
   describe '#inspect' do
-    it { expect(transcription.inspect).to eq('#<OmniAI::Transcribe::Transcription text="Hi!" format="json">') }
+    it { expect(transcription.inspect).to eq('#<OmniAI::Transcribe::Transcription text="Hi!">') }
   end
 end

spec/omniai/transcribe_spec.rb

@@ -28,7 +28,9 @@ def path
   end
 
   describe '.process!' do
-    subject(:process!) { FakeTranscribe.process!(path, client:, model:) }
+    subject(:process!) { FakeTranscribe.process!(path, client:, model:, format:) }
+
+    let(:format) { described_class::Format::JSON }
 
     let(:client) { FakeClient.new(api_key: '...') }
     let(:model) { FakeTranscribe::Model::FAKE }
@@ -53,5 +55,17 @@ def path
 
       it { expect { process! }.to raise_error(OmniAI::HTTPError) }
     end
+
+    context 'when OK with a non-JSON format' do
+      before do
+        stub_request(:post, 'http://localhost:8080/transcribe')
+          .to_return(status: 200, body: 'Hi!')
+      end
+
+      let(:format) { described_class::Format::TEXT }
+
+      it { expect(process!).to be_a(OmniAI::Transcribe::Transcription) }
+      it { expect(process!.text).to eq('Hi!') }
+    end
   end
 end