Support agent messages to simplify multi-turn

Gemfile.lock

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

lib/omniai/chat/prompt.rb

@@ -144,6 +144,24 @@ def system(content = nil, &)
       def user(content = nil, &)
         message(content, role: Role::USER, &)
       end
+
+      # @example
+      #   prompt.agent('the capital of Canada is Ottawa.')
+      #
+      # @example
+      #   prompt.agent do |message|
+      #     message.text 'the capital of Canada is Ottawa.'
+      #   end
+      #
+      # @param content [String, nil]
+      #
+      # @yield [builder]
+      # @yieldparam builder [Message::Builder]
+      #
+      # @return [Message]
+      def agent(content = nil, &)
+        message(content, role: Role::AGENT, &)
+      end
     end
   end
 end

lib/omniai/version.rb

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

spec/omniai/chat/prompt_spec.rb

@@ -75,6 +75,20 @@
     end
   end
 
+  describe '#agent' do
+    context 'with some text' do
+      let(:message) { prompt.user('The capital of Canada is Ottawa.') }
+
+      it { expect { message }.to(change { prompt.messages.size }) }
+    end
+
+    context 'with a block' do
+      let(:message) { prompt.user { |message| message.text('The capital of Canada is Ottawa.') } }
+
+      it { expect { message }.to(change { prompt.messages.size }) }
+    end
+  end
+
   describe '#serialize' do
     subject(:serialize) { prompt.serialize }