intelli_agent 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cc52520455ecddb77608867d1b38fcc3e0d5bfa2ad40e6810af9d544b2e0c1fa
-  data.tar.gz: 7b9df2447bcfb0f842bdf3fdb10aa2d71d0fe6e38425bf0c901f482c7567fd5b
+  metadata.gz: 5215cbeeb5f66df2815f4259cad370a962e2fc9cb8df39c55af6b1b73531f47b
+  data.tar.gz: 9d41844c690e1ceb733d099ce73fffaba9efbe53d233abcd91963202ac9ea2fe
 SHA512:
-  metadata.gz: 87c3853788806cbdda92fe70ef0fdfc4e168fb36433c819ff25a3e94faa46ec28eb6d4700648f6e14d652c9c33048e2225ea2f94ee5ff0f187420487ef6603ea
-  data.tar.gz: 2f77388f90d653761fec0340f9902e2e9e898548e8bb178c8d5cd1d85a5c11701d7aa84504c34e088baca432823363df13840be454d2e157a3ed76c5d2f61301
+  metadata.gz: d26c18ba4f481df7f2d1fc6517dc4195e1f59de1ff130795cf94e75d7c721456464566ee743740cd68728f9e62aea4feb655d5da5a020e16016b10341f03ff9d
+  data.tar.gz: 35aedfcd5871657e690e7bb0dbfc4cf208b766872ed882d18ca6c22f9103b0fc94e70b3889f964908012e09ae0b6472357c29714fbc6e0fda7f6b59fbe8aaf64

data/lib/intelli_agent/openai.rb

@@ -1,58 +1,75 @@
-# In the future, this became a bus to more than one AI provider
 module IntelliAgent::OpenAI
-  BASIC_MODEL = 'gpt-4o-mini' # ENV.fetch('OPENAI_BASIC_MODEL')
-  ADVANCED_MODEL = 'gpt-4o' # ENV.fetch('OPENAI_ADVANCED_MODEL')
+  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:)
-    response.dig('choices', 0, 'message', 'content').strip
+
+    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)
+  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:)
 
-    response.dig('choices', 0, 'message', 'content').strip
+    if response_format.nil?
+      response.dig('choices', 0, 'message', 'content').strip
+    else
+      response
+    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)
 
     response = OpenAI::Client.new.chat(parameters:)
-    response.dig('choices', 0, 'message', 'content').strip
+    if response_format.nil?
+      response.dig('choices', 0, 'message', 'content').strip
+    else
+      response
+    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:)
-    response.dig('choices', 0, 'message', 'content').strip
+    if response_format.nil?
+      response.dig('choices', 0, 'message', 'content').strip
+    else
+      response
+    end
   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.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2027-08-05 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