lib/gen_ai/chat/base.rb in gen-ai-0.4.0.alpha.3 vs lib/gen_ai/chat/base.rb in gen-ai-0.4.0

@@ -4,26 +4,29 @@
   class Chat
     class Base < GenAI::Base
       USER_ROLE = 'user'
       ASSISTANT_ROLE = 'assistant'
 
+      attr_reader :history, :default_options
+
       def initialize(provider:, token:, options: {})
         @history = []
+        @default_options = {}
         @model = GenAI::Language.new(provider, token, options: options)
       end
 
       def start(history: [], context: nil, examples: [])
-        @history = build_history(history.map(&:deep_symbolize_keys!), context, examples.map(&:deep_symbolize_keys!))
+        @history = build_history(history.map(&:deep_symbolize_keys), context, examples.map(&:deep_symbolize_keys))
       end
 
       def message(message, options = {})
-        if @history.size == 1
+        if @history.size == 1 && @history.first[:role] != 'system'
           append_to_message(message)
         else
           append_to_history({ role: USER_ROLE, content: message })
         end
 
-        response = @model.chat(@history.dup, options)
+        response = @model.chat(@history.dup, default_options.merge(options).compact)
         append_to_history({ role: ASSISTANT_ROLE, content: response.value })
         response
       end
 
       private