intelli_agent 0.2.1 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be996d76730695ba38477d0e748affd640eb03bf361892128305e668e32e1a43
-  data.tar.gz: d860d1779e054c68214bf69fe640f2420486e514dded5f0b2b9c328da09a0e32
+  metadata.gz: 8b165ba2e8f4c70fc7b6719f1761e89a44c7646d9018ad55142d0b9ff39c167a
+  data.tar.gz: d618dd1fe8e8cdcacab135f2a59a59aa823875e9e30ec39d4dce3649b962a5af
 SHA512:
-  metadata.gz: 46b6fe360ca8265f5d495b2853f4ea8e3fe9c5834e6e7e2cd072de2196b77e49483d8e67a74c334f526b12ed26d36bb66720241946c9de088b931ef7d5f4eacc
-  data.tar.gz: d61707f97959304260975fc8b9b17f0f1cf355345b2f4db5715db2675645fcb19bec6e521fdf4d5ec54e4f7f2d6f7dd7bcfc636cae885f8b67151df7070aa3a7
+  metadata.gz: c72e622cbba6b3302096f4f32522173b29e0bf7c8921fc75fd6bfb8f231b2bd804c2421f128a605381b7b94d7c43c241b2cfd20b32b86281673623af5fc5dae9
+  data.tar.gz: 3e74939dc68fa5bc007c3a785b327fcffbda1e71efb0fdaaae6fe82b594b5e15a6756a7418764d7ab4e690b0f218d8f8417ae01c1ade8738291495ba7a2234bb

data/lib/intelli_agent/openai.rb

@@ -8,6 +8,10 @@ module IntelliAgent::OpenAI
       dig('choices', 0, 'message', 'content')
     end
 
+    def message
+      dig('choices', 0, 'message')
+    end
+
     def content?
       !content.nil?
     end
@@ -19,11 +23,31 @@ module IntelliAgent::OpenAI
     def tool_calls?
       !tool_calls.nil?
     end
+
+    def functions
+      return if tool_calls.nil?
+      
+      functions = tool_calls.filter { |tool| tool['type'].eql? 'function' }
+      return if functions.empty?
+      
+      functions_list = []
+      functions.map.with_index do |function, function_index|
+        function_def = tool_calls.dig(function_index, 'function')
+        functions_list << { id: function['id'], name: function_def['name'], arguments: Oj.load(function_def['arguments'], symbol_keys: true) }
+      end
+
+      functions_list
+    end
+
+    def functions?
+      !functions.nil?
+    end
   end
 
   def self.embed(input, model: 'text-embedding-3-large')
     response = OpenAI::Client.new.embeddings(parameters: { input:, model: })
-    response.dig('data', 0, 'embedding')
+    def response.embedding = dig('data', 0, 'embedding')
+    response
   end
 
   def self.vision(prompt:, image_url:, model: :advanced, response_format: nil, max_tokens: MAX_TOKENS)
@@ -36,41 +60,22 @@ module IntelliAgent::OpenAI
 
     response = OpenAI::Client.new.chat(parameters:)
     
-    response.dig('choices', 0, 'message', 'content').strip
-  end  
+    def response.content = dig('choices', 0, 'message', 'content').strip
 
-  def self.single_prompt(prompt:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS, tools: nil)
-    model = select_model(model)
-    
-    parameters = { model:, messages: [{ role: 'user', content: prompt }], max_tokens: }
-    parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
-    parameters[:tools] = tools if tools
-
-    response = OpenAI::Client.new.chat(parameters:)
-    response.extend(ResponseExtender)
     response
-  end
-
-  def self.single_chat(system:, user:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS)
-    model = select_model(model)
-    parameters = { model:,
-                   messages: [
-                     { role: 'system', content: system },
-                     { role: 'user', content: user }
-                   ], max_tokens: }
+  end  
 
-    parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
-    parameters[:tools] = tools if tools
+  def self.single_prompt(prompt:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS, tools: nil, function_run_context: self)
+    chat(messages: [{ user: prompt }], model:, response_format:, max_tokens:, tools:, function_run_context:)
+  end
 
-    response = OpenAI::Client.new.chat(parameters:)
-    response.extend(ResponseExtender)
-    response
+  def self.single_chat(system:, user:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS, tools: nil, function_run_context: self)
+    chat(messages: [{ system: }, { user: }], model:, response_format:, max_tokens:, tools:, function_run_context:)
   end
 
-  def self.chat(messages:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS)
+  def self.chat(messages:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS, tools: nil, function_run_context: self)
     model = select_model(model)
-    
-    messages = determine_message_format(messages).eql?(:short_format) ? convert_message_to_standard_format(messages) : messages
+    messages = parse_messages(messages)
     
     parameters = { model:, messages:, max_tokens: }
     parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
@@ -78,6 +83,23 @@ module IntelliAgent::OpenAI
 
     response = OpenAI::Client.new.chat(parameters:)
     response.extend(ResponseExtender)
+
+    if response.functions?
+      parameters[:messages] << response.message
+
+      response.functions.each do |function|
+        parameters[:messages] << {
+          tool_call_id: function[:id],
+          role: :tool,
+          name: function[:name],
+          content: parameters[:function_run_context].send(function[:name], **function[:arguments])
+        }
+      end
+
+      response = OpenAI::Client.new.chat(parameters:)
+      response.extend(ResponseExtender)
+    end
+
     response
   end
 
@@ -94,21 +116,15 @@ module IntelliAgent::OpenAI
     end
   end
 
-  def self.determine_message_format(messages)
+  def self.parse_messages(messages)
     case messages
     in [{ role: String, content: String }, *]
-      :standard_format
-    in [{ system: String }, { user: String }, *]
-      :short_format
+      messages
     else
-      :unknown_format
+      messages.map do |msg|
+        role, content = msg.first
+        { role: role.to_s, content: content }
+      end
     end
   end
-
-  def self.convert_message_to_standard_format(messages)
-    messages.map do |msg|
-      role, content = msg.first
-      { role: role.to_s, content: content }
-    end
-  end  
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: intelli_agent
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.3
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2027-09-21 00:00:00.000000000 Z
+date: 2027-09-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -38,6 +38,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.3'
+- !ruby/object:Gem::Dependency
+  name: oj
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3'
 description: A helper layer over Anthropic and OpenAI API
 email: gedean.dias@gmail.com
 executables: []