intelli_agent 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f1af98056d1cfee93c37a849051ec08bbd94ff1b08e06c51df43459d0da2b39f
-  data.tar.gz: 558d8b96efc731bb945d73a3d107a017147be910d57fcde9b9e84e0e20fcd42b
+  metadata.gz: 55a857044336782f57062c01ab7f1bed094698d47cf5c118d2e883f01c56134d
+  data.tar.gz: '098ef34bea5cfaa036ea484056d37f12cf849914c3593ad951f895e0113a1ff6'
 SHA512:
-  metadata.gz: 1a59b52b56baea277a307efc5d9fa6b6f209292fec80842378001f2394084e1db3b2ee4e47d197aa75366e7da85b9941885566e67ccd26de03069f0a64582427
-  data.tar.gz: a0779b693af7c7ac8856f1fae3dbdb91bb0ceceef1cb3e70c97b675dc47f1d60f22ece30e6447ff7a7a3ff9ed40abc99a96ce6768c169a2f3dbe6d81f3998e5e
+  metadata.gz: 201c21f58d7c5060ed8ac4a4dfd28349a6a1d0e61569eb9e7130d42572c299394668a7a3dc8b4b3b3af340e9ef77b6bbb4f315c563509434ceafc5fd86011b3b
+  data.tar.gz: f476bd10218da6111030b878793eb908070e3c3cff77f63c6b8082f9ac67486de83d3c7ef732dd5713a87b5633db0170fb90ab8b716aca149104e8c60a5b93ec

data/lib/intelli_agent/ai.rb

@@ -1,14 +1,14 @@
 # In the future, this became a bus to more than one AI provider
 module AI
-  BASIC_MODEL = ENV.fetch('OPENAI_BASIC_MODEL')
-  ADVANCED_MODEL = ENV.fetch('OPENAI_ADVANCED_MODEL')
+  BASIC_MODEL = 'gpt-4o-mini' # ENV.fetch('OPENAI_BASIC_MODEL')
+  ADVANCED_MODEL = 'gpt-4o' # ENV.fetch('OPENAI_ADVANCED_MODEL')
 
-  def self.embed(input, model: 'text-embedding-3-large')
+  def 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: AI::BASIC_MODEL, response_format: nil)
+  def single_prompt(prompt:, model: AI::BASIC_MODEL, response_format: nil)
     parameters = { model:, messages: [{ role: 'user', content: prompt }] }
 
     parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
@@ -17,11 +17,11 @@ module AI
     response.dig('choices', 0, 'message', 'content').strip
   end
 
-  def self.vision(prompt:, image_url:, response_format: nil)
+  def vision(prompt:, image_url:, model: AI::BASIC_MODEL, response_format: nil)
     messages = [{ type: :text, text: prompt },
                 { type: :image_url, image_url: { url: image_url } }]
 
-    parameters = { model: AI::ADVANCED_MODEL, messages: [{ role: :user, content: messages }] }
+    parameters = { model: model, messages: [{ role: :user, content: messages }] }
     parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
 
     response = OpenAI::Client.new.chat(parameters:)
@@ -29,7 +29,7 @@ module AI
     response.dig('choices', 0, 'message', 'content').strip
   end
 
-  def self.single_chat(system:, user:, model: AI::BASIC_MODEL, response_format: nil)
+  def single_chat(system:, user:, model: AI::BASIC_MODEL, response_format: nil)
     parameters = { model:,
                    messages: [
                      { role: 'system', content: system },
@@ -42,7 +42,7 @@ module AI
     response.dig('choices', 0, 'message', 'content').strip
   end
 
-  def self.chat(messages, model: AI::BASIC_MODEL, response_format: nil)
+  def chat(messages, model: AI::BASIC_MODEL, response_format: nil)
     parameters = { model:, messages: }
     parameters[:response_format] = { type: 'json_object' } if response_format.eql?(:json)
 
@@ -50,7 +50,7 @@ module AI
     response.dig('choices', 0, 'message', 'content').strip
   end
 
-  def self.models
+  def models
     OpenAI::Client.new.models.list
   end
 end

data/lib/intelli_agent.rb

@@ -1,4 +1,4 @@
-#require_relative './intelli_agent/ai'
+require 'openai'
 require 'intelli_agent/ai'
 
 class IntelliAgent

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: intelli_agent
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - Gedean Dias