intelli_agent 0.1.4 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cb2113329a8f68d1433d8d54846c73eb620818041f617f5140afd00247c53dec
-  data.tar.gz: ff64fce4f52a30f3d1289925f92ab30767d4e1ee54f75f96e702f32ae4e5354c
+  metadata.gz: be996d76730695ba38477d0e748affd640eb03bf361892128305e668e32e1a43
+  data.tar.gz: d860d1779e054c68214bf69fe640f2420486e514dded5f0b2b9c328da09a0e32
 SHA512:
-  metadata.gz: 784acf6fd048f8f521ebcb4dad4381b48db1e79a32b424d98c790b8abf74ddbf1f3a6b69ae2dffe67dd68a731dda295aa7df627689ba5672837d2be4fe0a0b0e
-  data.tar.gz: 4ead6219753043957a4728b8808b57a0829a4954671f06895002ffc47c1087e61124a9c1d37a7328fa203e0a1ad5541570247513f79b8b477518ba704bd4f44d
+  metadata.gz: 46b6fe360ca8265f5d495b2853f4ea8e3fe9c5834e6e7e2cd072de2196b77e49483d8e67a74c334f526b12ed26d36bb66720241946c9de088b931ef7d5f4eacc
+  data.tar.gz: d61707f97959304260975fc8b9b17f0f1cf355345b2f4db5715db2675645fcb19bec6e521fdf4d5ec54e4f7f2d6f7dd7bcfc636cae885f8b67151df7070aa3a7

data/lib/intelli_agent/openai.rb

@@ -1,24 +1,31 @@
 module IntelliAgent::OpenAI
   BASIC_MODEL = ENV.fetch('OPENAI_BASIC_MODEL', 'gpt-4o-mini')
   ADVANCED_MODEL = ENV.fetch('OPENAI_ADVANCED_MODEL', 'gpt-4o-2024-08-06')
-  MAX_TOKENS = ENV.fetch('OPENAI_MAX_TOKENS', 16383)
+  MAX_TOKENS = ENV.fetch('OPENAI_MAX_TOKENS', 16_383).to_i
+
+  module ResponseExtender
+    def content
+      dig('choices', 0, 'message', 'content')
+    end
+
+    def content?
+      !content.nil?
+    end
+
+    def tool_calls
+      dig('choices', 0, 'message', 'tool_calls')
+    end
+
+    def tool_calls?
+      !tool_calls.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')
   end
 
-  def self.single_prompt(prompt:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS)
-    model = select_model(model)
-    
-    parameters = { model:, messages: [{ role: 'user', content: prompt }], max_tokens: }
-    parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
-    
-    response = OpenAI::Client.new.chat(parameters:)
-
-    response.dig('choices', 0, 'message', 'content').strip
-  end
-
   def self.vision(prompt:, image_url:, model: :advanced, response_format: nil, max_tokens: MAX_TOKENS)
     model = select_model(model)
     messages = [{ type: :text, text: prompt },
@@ -30,6 +37,18 @@ module IntelliAgent::OpenAI
     response = OpenAI::Client.new.chat(parameters:)
     
     response.dig('choices', 0, 'message', 'content').strip
+  end  
+
+  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)
@@ -41,10 +60,11 @@ module IntelliAgent::OpenAI
                    ], 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.dig('choices', 0, 'message', 'content').strip
+    response.extend(ResponseExtender)
+    response
   end
 
   def self.chat(messages:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS)
@@ -54,10 +74,11 @@ module IntelliAgent::OpenAI
     
     parameters = { model:, messages:, 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.dig('choices', 0, 'message', 'content').strip
+    response.extend(ResponseExtender)
+    response
   end
 
   def self.models = OpenAI::Client.new.models.list

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: intelli_agent
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.2.1
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2027-09-02 00:00:00.000000000 Z
+date: 2027-09-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -69,7 +69,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.18
+rubygems_version: 3.5.19
 signing_key: 
 specification_version: 4
 summary: A helper layer over Anthropic and OpenAI API