lib/intelli_agent/openai.rb in intelli_agent-0.2.12 vs lib/intelli_agent/openai.rb in intelli_agent-0.2.13

@@ -73,11 +73,11 @@
 
   def self.chat(messages:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil)
     model = select_model(model)
     is_o1_model = model.start_with?('o1')
 
-    messages = parse_messages(messages)
+    messages = IntelliAgent::OpenAI::Messages.new(messages) unless messages.is_a?(IntelliAgent::OpenAI::Messages)
     
     parameters = { model:, messages:, store: }
     parameters[:metadata] = metadata if metadata
 
 
@@ -116,20 +116,8 @@
       O1_BASIC_MODEL
     when :o1_advanced
       O1_ADVANCED_MODEL
     else
       model
-    end
-  end
-
-  def self.parse_messages(messages)
-    case messages
-    in [{ role: String | Symbol, content: String | Array }, *]
-      messages
-    else
-      messages.map do |msg|
-        role, content = msg.first
-        { role: role.to_s, content: content }
-      end
     end
   end
 end