intelli_agent 0.2.12 → 0.2.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 37f7cac456bcc0441d34d9f0e9901dfd49d04b3fde889ab44498bd1a735dcbd0
-  data.tar.gz: 0e3cb4e3597410c51db19ee141d273839a1ea8945c99029d422ce85c247874c7
+  metadata.gz: c91be5552b13825b657164e1ac9836e88fbdd98e99930ba7aececb54cb6e273a
+  data.tar.gz: f80215a13d785d1d50f835148be4f842a6c058f9f8445b94928feeb4f64b6e01
 SHA512:
-  metadata.gz: 47133255200fa5ee3cac2d91db4293348eadd243c11d0893f65280c7ecc378a39db2196cfd81b52343273e1cdc36e738bfa469b5c670e4bb5840c494e5f1cde1
-  data.tar.gz: cfc9cbf181b2971f51e44cf056bebd5bd0cd344d7561dfea2824249db4cc7c629100dade0fe0013ba29538d11a520e68ded5d811729585b0e7db2dd7f12fef9c
+  metadata.gz: 184e9e14319cfa84487eb3919ffb0b27a1f1b273464fd9410992d1c0467bbfbf2f68eabf7e31e3002f26feecd5107d37cae2acbfafdf7a3473ec1a0076a1699a
+  data.tar.gz: 8a3facc0c1bc891a457573e20b4581422565dd970cf666e1a848e46e226b5fd7203130e2740859c2c6393ef3826314ff914bfab0811c79af9845ed5192359ad9

data/lib/intelli_agent/openai/messages.rb

@@ -0,0 +1,41 @@
+class IntelliAgent::OpenAI::Messages < Array
+  # VALID_ROLES = %i[system user assistant function tool].freeze
+
+  def initialize messages = nil
+    super parse_messages(messages)
+  end
+
+  def add(message) = concat(parse_messages(message))
+
+=begin
+The output format should be an array of hashes with the keys `role` and `content`.
+
+The input can be:
+  - A simple hash `{ user: 'user message' }`, which should be transformed into `{ role: 'user', content: 'user message' }`
+  - A hash with multiple keys `{system: 'message', user: 'user message' }`, which should be transformed into `[{ role: 'system', content: 'message' }, { role: 'user', content: 'user message' }]`
+  - An array of hashes `[{ system: 'message' }, { user: 'user message' }]`, which should be transformed into `[{ role: 'system', content: 'message' }, { role: 'user', content: 'user message' }]`
+
+Important note: the content can be either a string or an array.
+=end
+  private
+  def parse_messages(messages)
+    return [] if messages.nil?
+
+    messages = [messages] unless messages.is_a?(Array)
+
+    messages.flat_map do |msg|
+      if msg.is_a?(Hash)
+        if msg.keys.size == 1
+          role, content = msg.first
+          { role: role.to_s, content: content }
+        elsif msg.key?(:role) && msg.key?(:content)
+          { role: msg[:role].to_s, content: msg[:content] }
+        else
+          msg.map { |role, content| { role: role.to_s, content: content } }
+        end
+      else
+        raise ArgumentError, "Invalid message format: #{msg}"
+      end
+    end
+  end
+end

data/lib/intelli_agent/openai/messages_helper copy.rb

@@ -0,0 +1,40 @@
+class IntelliAgent::OpenAI::Messages < Array
+  VALID_ROLES = %i[system user assistant tool].freeze
+
+  def initialize(messages = [])
+    @messages = self.class.parse_messages(messages)
+  end
+
+  def add(message) = @messages.concat(self.class.parse_messages(message))
+
+=begin
+  O Formato de saída deve ser um array de hashes com as chaves `role` e `content`.
+
+  A entrada pode ser:
+    - um hash simples { user: 'mensagem do usuário' }, deve ser transformado em { role: 'user', content: 'mensagem do usuário' }
+    - um hash com várias chaves {system: 'mensagem', user: 'mensagem do usuário' }, deve ser transformado em [{ role: 'system', content: 'mensagem'}, { role: 'user', content: 'mensagem do usuário' }] 
+    - um array de hashes [{ system: 'mensagem'}, { user: 'mensagem do usuário' }], deve ser transformado em [{ role: 'system', content: 'mensagem'}, { role: 'user', content: 'mensagem do usuário' }]
+  Obervação imporante: o content pode ser uma string ou um array
+=end
+
+  def self.parse_messages(messages)
+    return [] if messages.empty?
+
+    messages = [messages] unless messages.is_a?(Array)
+
+    messages.flat_map do |msg|
+      if msg.is_a?(Hash)
+        if msg.keys.size == 1
+          role, content = msg.first
+          { role: role.to_s, content: content }
+        elsif msg.key?(:role) && msg.key?(:content)
+          { role: msg[:role].to_s, content: msg[:content] }
+        else
+          msg.map { |role, content| { role: role.to_s, content: content } }
+        end
+      else
+        raise ArgumentError, "Invalid message format: #{msg}"
+      end
+    end
+  end
+end

data/lib/intelli_agent/openai.rb

@@ -75,7 +75,7 @@ module IntelliAgent::OpenAI
     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
@@ -120,16 +120,4 @@ module IntelliAgent::OpenAI
       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

data/lib/intelli_agent.rb

@@ -6,3 +6,4 @@ end
 
 require 'intelli_agent/anthropic'
 require 'intelli_agent/openai'
+require 'intelli_agent/openai/messages'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: intelli_agent
 version: !ruby/object:Gem::Version
-  version: 0.2.12
+  version: 0.2.13
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-10-10 00:00:00.000000000 Z
+date: 2024-10-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -65,6 +65,8 @@ files:
 - lib/intelli_agent/agent_test.rb_
 - lib/intelli_agent/anthropic.rb
 - lib/intelli_agent/openai.rb
+- lib/intelli_agent/openai/messages.rb
+- lib/intelli_agent/openai/messages_helper copy.rb
 homepage: https://github.com/gedean/intelli_agent
 licenses:
 - MIT