intelli_agent 0.2.9 → 0.2.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 586990e6d1d8fa1fb890bb9530d7f175f4f403ed309ca111517755ed5e07c2f3
-  data.tar.gz: 669c3765fd006605eb86b93a31b90aa4e6101a50820e8e3dd9eaeec54f17f2d0
+  metadata.gz: e94ab4d8cf68071308ab502086f092fc7424622de20420d00bba18451cd10705
+  data.tar.gz: f74579964f8d1a40a7d3f26f64b079ada32d202f2cf6c08f15e615e24ded8ee7
 SHA512:
-  metadata.gz: 7af7fe5d68898a89ace33fd71a5eeb519566760900c675782d5620ff29c56013b3e6b92e7074b006668cafa825e5b3ce48db5666b61e5140e14903b54874e739
-  data.tar.gz: dbb26d71d8187090d151436a8ec2229950b13f93f87c228c42cfaaa77ec98762f75f86e9696f1f9a843b0848cfde92a2ca9cd3afdab613cb51d693193ac17fb8
+  metadata.gz: fbc1c8b705f7050b8f9bf18aa8aef380559e38019bde3911f598b4546ac6cdd2d2596b158502509663058cdfa8d052cd4e3dc992973f813739f86bd9336930bc
+  data.tar.gz: b82503246fe4365c46fad3528522cd4ee1acf8d7ecb9a72d3c4975acfde835031fc0a6556cb85944fecf3806bcf1de8c75d9b1fa04a03c21d7f8b2398ac989fe

data/lib/intelli_agent/openai.rb

@@ -58,42 +58,32 @@ module IntelliAgent::OpenAI
     response
   end
 
-  def self.vision(prompt:, image_url:, model: :gpt_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 }], max_tokens: }
-    parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
-
-    response = OpenAI::Client.new.chat(parameters:)
-    
-    def response.content = dig('choices', 0, 'message', 'content').strip
-
-    response
+  def self.vision(prompt:, image_url:, model: :gpt_advanced, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil)
+    message_content = [{ type: :text, text: prompt }, { type: :image_url, image_url: { url: image_url } }]
+    chat(messages: [{ role: :user, content: message_content }], model:, response_format:, max_tokens:, store:, tools:, auto_run_functions:, function_context:)    
   end  
 
-  def self.single_prompt(prompt:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, tools: nil, auto_run_functions: false, function_context: nil)
+  def self.single_prompt(prompt:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil)
     chat(messages: [{ user: prompt }], model:, response_format:, max_tokens:, store:, tools:, auto_run_functions:, function_context:)
   end
 
-  def self.single_chat(system:, user:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, tools: nil, auto_run_functions: false, function_context: nil)
+  def self.single_chat(system:, user:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil)
     chat(messages: [{ system: }, { user: }], model:, response_format:, max_tokens:, store:, tools:, auto_run_functions:, function_context:)
   end
 
-  def self.chat(messages:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, tools: nil, auto_run_functions: false, function_context: nil)
+  def self.chat(messages:, model: :gpt_basic, response_format: nil, max_tokens: MAX_TOKENS, store: true, metadata: nil, tools: nil, auto_run_functions: false, function_context: nil)
     model = select_model(model)
-
-    # o1 models doesn't support max_tokens, instead max_completion_tokens
     is_o1_model = model.start_with?('o1')
-    max_completion_tokens = max_tokens if is_o1_model
 
     messages = parse_messages(messages)
     
     parameters = { model:, messages:, store: }
+    parameters[:metadata] = metadata if metadata
+
 
-    parameters[:max_completion_tokens] = max_completion_tokens if is_o1_model
-    parameters[:max_tokens] = max_completion_tokens unless is_o1_model
+    # o1 family models doesn't support max_tokens params. Instead, use max_completion_tokens
+    parameters[:max_completion_tokens] = max_tokens if is_o1_model
+    parameters[:max_tokens] = max_tokens unless is_o1_model
 
     parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
     parameters[:tools] = tools if tools
@@ -133,7 +123,7 @@ module IntelliAgent::OpenAI
 
   def self.parse_messages(messages)
     case messages
-    in [{ role: String, content: String }, *]
+    in [{ role: String | Symbol, content: String | Array }, *]
       messages
     else
       messages.map do |msg|

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: intelli_agent
 version: !ruby/object:Gem::Version
-  version: 0.2.9
+  version: 0.2.11
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2027-10-03 00:00:00.000000000 Z
+date: 2024-10-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -84,7 +84,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.20
+rubygems_version: 3.5.21
 signing_key: 
 specification_version: 4
 summary: A helper layer over Anthropic and OpenAI API