intelli_agent 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3fca7e6fcb2e2360ccd76c9d09506f73db72dd525dfd771322a19a6982eedd52
-  data.tar.gz: 2bcb9b1cc62b6dfb9a6c4ecc634dab04e9c1c27a2941dd2feb6a0f8aad858934
+  metadata.gz: 729fa0b1b8a80ceaeb2568429fa7cbbe167f63d865d6beb1283daa62eaf4ca21
+  data.tar.gz: 505c4a6e21b9e4520b21eca38aa19cac782bbe4701acb02539f6e276baf3e04d
 SHA512:
-  metadata.gz: 3de9eaddade5b5d5f16e385ef4cbff9bd7e7eade087b4bbc04c5c8ae2c3d0cc15ef79b518536d023f3153b62a4b748e53f735b7f33a338450db403d824c86f06
-  data.tar.gz: aa27675fcfc8c0bcab9701b1f0e8ca6ec09eb03f8c4fd141b9c0b63465c196f9bfcf85ad8c401eeb2c32e82930f310c3ae0fbac445f43f1e5d4dd6aa6c1622e7
+  metadata.gz: c395e1406aff8919b518122b75c28f666d1e7ecfcebb21951f04b92824d4edc165d25e4a8eb24cfccc1bf1f3c2934651dcd27dd33aac09a426f94394c0ea26f3
+  data.tar.gz: 5d0ac8a6696e167ea964c5fe8090b9d318b6742dbe15f88117f094e74332e8f472dad2ff2aeb6d47fa204de5b5e5f1cb9534cfa30aaa67eb7b6f285eed2e74c8

data/lib/{ai.rb → 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,7 +17,7 @@ module AI
     response.dig('choices', 0, 'message', 'content').strip
   end
 
-  def self.vision(prompt:, image_url:, response_format: nil)
+  def vision(prompt:, image_url:, response_format: nil)
     messages = [{ type: :text, text: prompt },
                 { type: :image_url, image_url: { url: image_url } }]
 
@@ -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/{ai/agent.rb → intelli_agent.rb}

@@ -1,10 +1,14 @@
-class AI::Agent
+require 'openai'
+require 'intelli_agent/ai'
+
+class IntelliAgent
+  extend AI
   attr_reader :assistant, :thread, :instructions, :vector_store_id
 
   def initialize(assistant_id: nil, thread_id: nil, thread_instructions: nil, vector_store_id: nil)
     @openai_client = OpenAI::Client.new
 
-    assistant_id ||= ENV.fetch('OPENAI_AKAPU_CORE_ASSISTANT_ID')
+    assistant_id ||= ENV.fetch('OPENAI_ASSISTANT_ID')
     @assistant = @openai_client.assistants.retrieve(id: assistant_id)
 
     thread_params = {}
@@ -19,17 +23,10 @@ class AI::Agent
     @instructions = thread_instructions || @assistant['instructions']
   end
 
-  def add_message(text, role: 'user')
-    @openai_client.messages.create(thread_id: @thread['id'], parameters: { role: role, content: text })
-  end
-
-  def messages
-    @openai_client.messages.list(thread_id: @thread['id'])
-  end
-
-  def last_message
-    messages['data'].first['content'].first['text']['value']
-  end
+  def add_message(text, role: 'user') = @openai_client.messages.create(thread_id: @thread['id'], parameters: { role: role, content: text })
+  def messages = @openai_client.messages.list(thread_id: @thread['id'])
+  def last_message = messages['data'].first['content'].first['text']['value']
+  def runs = @openai_client.runs.list(thread_id: @thread['id'])
 
   def run(instructions: nil, additional_instructions: nil, additional_message: nil, model: nil, tool_choice: nil)
     params = { assistant_id: @assistant['id'] }
@@ -64,8 +61,4 @@ class AI::Agent
       end
     end
   end
-
-  def runs
-    @openai_client.runs.list(thread_id: @thread['id'])
-  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: intelli_agent
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Gedean Dias
@@ -31,12 +31,9 @@ extensions: []
 extra_rdoc_files: []
 files:
 - README.md
-- lib/ai.rb
-- lib/ai.rb:Zone.Identifier
-- lib/ai/agent.rb
-- lib/ai/agent.rb:Zone.Identifier
-- lib/ai/agent_test.rb_
-- lib/ai/agent_test.rb_:Zone.Identifier
+- lib/intelli_agent.rb
+- lib/intelli_agent/agent_test.rb_
+- lib/intelli_agent/ai.rb
 homepage: https://github.com/gedean/intelli_agent
 licenses:
 - MIT