intelli_agent 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5215cbeeb5f66df2815f4259cad370a962e2fc9cb8df39c55af6b1b73531f47b
-  data.tar.gz: 9d41844c690e1ceb733d099ce73fffaba9efbe53d233abcd91963202ac9ea2fe
+  metadata.gz: cb2113329a8f68d1433d8d54846c73eb620818041f617f5140afd00247c53dec
+  data.tar.gz: ff64fce4f52a30f3d1289925f92ab30767d4e1ee54f75f96e702f32ae4e5354c
 SHA512:
-  metadata.gz: d26c18ba4f481df7f2d1fc6517dc4195e1f59de1ff130795cf94e75d7c721456464566ee743740cd68728f9e62aea4feb655d5da5a020e16016b10341f03ff9d
-  data.tar.gz: 35aedfcd5871657e690e7bb0dbfc4cf208b766872ed882d18ca6c22f9103b0fc94e70b3889f964908012e09ae0b6472357c29714fbc6e0fda7f6b59fbe8aaf64
+  metadata.gz: 784acf6fd048f8f521ebcb4dad4381b48db1e79a32b424d98c790b8abf74ddbf1f3a6b69ae2dffe67dd68a731dda295aa7df627689ba5672837d2be4fe0a0b0e
+  data.tar.gz: 4ead6219753043957a4728b8808b57a0829a4954671f06895002ffc47c1087e61124a9c1d37a7328fa203e0a1ad5541570247513f79b8b477518ba704bd4f44d

data/lib/intelli_agent/openai.rb

@@ -16,11 +16,7 @@ module IntelliAgent::OpenAI
     
     response = OpenAI::Client.new.chat(parameters:)
 
-    if response_format.nil?
-      response.dig('choices', 0, 'message', 'content').strip
-    else
-      response
-    end
+    response.dig('choices', 0, 'message', 'content').strip
   end
 
   def self.vision(prompt:, image_url:, model: :advanced, response_format: nil, max_tokens: MAX_TOKENS)
@@ -32,12 +28,8 @@ module IntelliAgent::OpenAI
     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
+    
+    response.dig('choices', 0, 'message', 'content').strip
   end
 
   def self.single_chat(system:, user:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS)
@@ -51,25 +43,21 @@ module IntelliAgent::OpenAI
     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
+
+    response.dig('choices', 0, 'message', 'content').strip
   end
 
   def self.chat(messages:, model: :basic, response_format: nil, max_tokens: MAX_TOKENS)
     model = select_model(model)
     
+    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:)
-    if response_format.nil?
-      response.dig('choices', 0, 'message', 'content').strip
-    else
-      response
-    end
+    
+    response.dig('choices', 0, 'message', 'content').strip
   end
 
   def self.models = OpenAI::Client.new.models.list
@@ -84,4 +72,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.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2027-08-27 00:00:00.000000000 Z
+date: 2027-09-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai