intelli_agent 0.1.3 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9cb37d7cb52a2574978fbf164f5b9eb0c92e3b71a91114c897e3bae81f51d9ef
-  data.tar.gz: e52d588661951e958e2d3551f2cfa92f622dacf9a84a00f50590db18d8fb0905
+  metadata.gz: 4a8a694f4bb6ba84140e24bdf44962629eb63ff75b8a9ff8bc1ab5ef416db5cb
+  data.tar.gz: a3ef08dc21d823b19d597f8b1355328f55d6f767ddb21a99fe3bbac5cf9256b3
 SHA512:
-  metadata.gz: 7320d62158d94eca1db5ba269fb1a38e2c611a9772104cf718c2aa602407ff1648bcd0daeea2581c32478121fff351cb08881b8b02d8f14e4be67cbe0e726f40
-  data.tar.gz: a74ddef7f64bc99e3d87518ee97988dcf47f56a8d9cb3ef28bd6fed1e739ecf41ba94b4822eb1590f1bffda1badfbc5d28ec8433d9bf5e6ffaafab86274fde5d
+  metadata.gz: 7602007cab2f23363d6ed87c9f22d1d262325e2080eb8dc5f2f7e0b8d599d98f7e8666348e3e477660acbf01a2f63bfc98199cd8f4eb5afb143165ef79058b74
+  data.tar.gz: 49bbef1d5c157d9b4f51aede33b6bc83858a53b4c10281132b0dee3b06117459a144ab3c386496ac23a7c0078fb6f6079d9f0e0906898af1cb2f365d99ffe7af

data/lib/intelli_agent/openai.rb

@@ -1,28 +1,23 @@
 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 tool_calls
+      dig('choices', 0, 'message', 'tool_calls')
+    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:)
-
-    if response_format.nil?
-      response.dig('choices', 0, 'message', 'content').strip
-    else
-      response
-    end
-  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 },
@@ -32,12 +27,20 @@ module IntelliAgent::OpenAI
     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  
 
-    if response_format.nil?
-      response.dig('choices', 0, 'message', 'content').strip
-    else
-      response
-    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)
@@ -49,13 +52,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:)
-    if response_format.nil?
-      response.dig('choices', 0, 'message', 'content').strip
-    else
-      response
-    end
+    response.extend(ResponseExtender)
+    response
   end
 
   def self.chat(messages:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS)
@@ -65,13 +66,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:)
-    if response_format.nil?
-      response.dig('choices', 0, 'message', 'content').strip
-    else
-      response
-    end
+    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.3
+  version: 0.2.0
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2027-09-01 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