intelli_agent 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c76bcdd7573088ad3dcc72e5737a72b4bd352c1568c05d849f01007e409d5a09
-  data.tar.gz: 4c84207e8e0a56741825b4a8d8b37ffd8ac9be22025b17b100bca4c2872b0bf6
+  metadata.gz: 5215cbeeb5f66df2815f4259cad370a962e2fc9cb8df39c55af6b1b73531f47b
+  data.tar.gz: 9d41844c690e1ceb733d099ce73fffaba9efbe53d233abcd91963202ac9ea2fe
 SHA512:
-  metadata.gz: be84356ca55dcae68905bbd3b6547b764499406429b980c21acf48d2dc175433c47e85a51728f6b73194c0c7907db0e8d004dda6030966f5c84d4aecb040de12
-  data.tar.gz: 254c5b1c8db19b4b75278813f3a7faae286ec9bc7d1930b2b973cc5b7a0c867c88bac74f9ba7d0abd87fa6e0c206617a537c242681335177d5c6abb0aaeec021
+  metadata.gz: d26c18ba4f481df7f2d1fc6517dc4195e1f59de1ff130795cf94e75d7c721456464566ee743740cd68728f9e62aea4feb655d5da5a020e16016b10341f03ff9d
+  data.tar.gz: 35aedfcd5871657e690e7bb0dbfc4cf208b766872ed882d18ca6c22f9103b0fc94e70b3889f964908012e09ae0b6472357c29714fbc6e0fda7f6b59fbe8aaf64

data/lib/intelli_agent/openai.rb

@@ -1,20 +1,19 @@
-# In the future, this became a bus to more than one AI provider
 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)
 
   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)
+  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 }] }
-
+    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?
@@ -24,12 +23,12 @@ module IntelliAgent::OpenAI
     end
   end
 
-  def self.vision(prompt:, image_url:, model: :advanced, response_format: nil)
+  def self.vision(prompt:, image_url:, model: :advanced, response_format: nil, max_tokens: MAX_TOKENS)
     model = select_model(model)
     messages = [{ type: :text, text: prompt },
                 { type: :image_url, image_url: { url: image_url } }]
 
-    parameters = { model: model, messages: [{ role: :user, content: messages }] }
+    parameters = { model: model, messages: [{ role: :user, content: messages }], max_tokens: }
     parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
 
     response = OpenAI::Client.new.chat(parameters:)
@@ -41,13 +40,13 @@ module IntelliAgent::OpenAI
     end
   end
 
-  def self.single_chat(system:, user:, model: :basic, response_format: nil)
+  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: }
 
     parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
 
@@ -59,9 +58,10 @@ module IntelliAgent::OpenAI
     end
   end
 
-  def self.chat(messages:, model: :basic, response_format: nil)
+  def self.chat(messages:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS)
     model = select_model(model)
-    parameters = { model:, messages: }
+    
+    parameters = { model:, messages:, max_tokens: }
     parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
 
     response = OpenAI::Client.new.chat(parameters:)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: intelli_agent
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2027-08-07 00:00:00.000000000 Z
+date: 2027-08-27 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.17
+rubygems_version: 3.5.18
 signing_key: 
 specification_version: 4
 summary: A helper layer over Anthropic and OpenAI API