intelli_agent 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c76bcdd7573088ad3dcc72e5737a72b4bd352c1568c05d849f01007e409d5a09
-  data.tar.gz: 4c84207e8e0a56741825b4a8d8b37ffd8ac9be22025b17b100bca4c2872b0bf6
+  metadata.gz: 9cb37d7cb52a2574978fbf164f5b9eb0c92e3b71a91114c897e3bae81f51d9ef
+  data.tar.gz: e52d588661951e958e2d3551f2cfa92f622dacf9a84a00f50590db18d8fb0905
 SHA512:
-  metadata.gz: be84356ca55dcae68905bbd3b6547b764499406429b980c21acf48d2dc175433c47e85a51728f6b73194c0c7907db0e8d004dda6030966f5c84d4aecb040de12
-  data.tar.gz: 254c5b1c8db19b4b75278813f3a7faae286ec9bc7d1930b2b973cc5b7a0c867c88bac74f9ba7d0abd87fa6e0c206617a537c242681335177d5c6abb0aaeec021
+  metadata.gz: 7320d62158d94eca1db5ba269fb1a38e2c611a9772104cf718c2aa602407ff1648bcd0daeea2581c32478121fff351cb08881b8b02d8f14e4be67cbe0e726f40
+  data.tar.gz: a74ddef7f64bc99e3d87518ee97988dcf47f56a8d9cb3ef28bd6fed1e739ecf41ba94b4822eb1590f1bffda1badfbc5d28ec8433d9bf5e6ffaafab86274fde5d

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,12 @@ 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: }
+    
+    messages = determine_message_format(messages).eql?(:short_format) ? convert_message_to_standard_format(messages) : messages
+    
+    parameters = { model:, messages:, max_tokens: }
     parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
 
     response = OpenAI::Client.new.chat(parameters:)
@@ -84,4 +86,22 @@ module IntelliAgent::OpenAI
       model
     end
   end
+
+  def self.determine_message_format(messages)
+    case messages
+    in [{ role: String, content: String }, *]
+      :standard_format
+    in [{ system: String }, { user: String }, *]
+      :short_format
+    else
+      :unknown_format
+    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.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2027-08-07 00:00:00.000000000 Z
+date: 2027-09-01 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