intelli_agent 0.0.6 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 475b4ea33841f7026d1186e346d0939a40bf80277bada1f00c058f4e966bd513
-  data.tar.gz: 0fd82b217902cc662e163d08565afe0ff1eee10d6f690be1fb206fe47b216cdd
+  metadata.gz: cc52520455ecddb77608867d1b38fcc3e0d5bfa2ad40e6810af9d544b2e0c1fa
+  data.tar.gz: 7b9df2447bcfb0f842bdf3fdb10aa2d71d0fe6e38425bf0c901f482c7567fd5b
 SHA512:
-  metadata.gz: 6af16bf022d0bb540d9e7748bb3186277d18347facaa0983c29af055382d6d7a88492d8b1dae184b6c3f7544a54f57b773f611a05507a510490648c5a7922791
-  data.tar.gz: b0cb06584d055c8ea1350da73d71986b2897fb2ac6c9016d6c4b81cb1e40b49d361e80561c80dc9b6a6963edb60bedcb76a160b9dd7908fc2f05b05fb99197a0
+  metadata.gz: 87c3853788806cbdda92fe70ef0fdfc4e168fb36433c819ff25a3e94faa46ec28eb6d4700648f6e14d652c9c33048e2225ea2f94ee5ff0f187420487ef6603ea
+  data.tar.gz: 2f77388f90d653761fec0340f9902e2e9e898548e8bb178c8d5cd1d85a5c11701d7aa84504c34e088baca432823363df13840be454d2e157a3ed76c5d2f61301

data/lib/intelli_agent/agent.rb.disabled

@@ -0,0 +1,67 @@
+require 'openai'
+require 'intelli_agent/anthropic'
+require 'intelli_agent/openai'
+
+class IntelliAgent
+  extend OpenAI
+  extend Anthropic
+  
+  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_ASSISTANT_ID')
+    @assistant = @openai_client.assistants.retrieve(id: assistant_id)
+
+    thread_params = {}
+
+    # Only one vector store can be attached, according to the OpenAI API documentation
+    @vector_store_id = vector_store_id
+    thread_params = { tool_resources: { file_search: { vector_store_ids: [vector_store_id] } } } if @vector_store_id
+
+    thread_id ||= @openai_client.threads.create(parameters: thread_params)['id']
+    @thread = @openai_client.threads.retrieve(id: thread_id)
+
+    @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 })
+  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'] }
+
+    params[:instructions] = instructions || @instructions
+    params[:additional_instructions] = additional_instructions unless additional_instructions.nil?
+    params[:tool_choice] = tool_choice unless tool_choice.nil?
+
+    params[:additional_messages] = [{ role: :user, content: additional_message }] unless additional_message.nil?
+
+    params[:model] = model || @assistant['model']
+
+    run_id = @openai_client.runs.create(thread_id: @thread['id'], parameters: params)['id']
+
+    loop do
+      response = @openai_client.runs.retrieve(id: run_id, thread_id: @thread['id'])
+
+      case response['status']
+      when 'queued', 'in_progress', 'cancelling'
+        puts 'Status: Waiting AI Processing finish'
+        sleep 1
+      when 'completed'
+        puts last_message
+        break
+      when 'requires_action'
+        # Handle tool calls (see below)
+      when 'cancelled', 'failed', 'expired'
+        puts response['last_error'].inspect
+        break # or `exit`
+      else
+        puts "Unknown status response: #{status}"
+      end
+    end
+  end
+end

data/lib/intelli_agent/anthropic.rb

@@ -0,0 +1,43 @@
+module IntelliAgent::Anthropic
+  BASIC_MODEL = 'claude-3-haiku-20240307' # ENV.fetch('Anthropic_BASIC_MODEL')
+  ADVANCED_MODEL = 'claude-3-5-sonnet-20240620' # ENV.fetch('Anthropic_ADVANCED_MODEL')
+
+  def self.single_prompt(prompt:, model: :basic, max_tokens: 1000)
+    model = select_model(model)
+
+    parameters = { model:, max_tokens:, messages: [{ role: 'user', content: prompt }] }
+
+    response = Anthropic::Client.new.messages(parameters:)
+    response.dig('content', 0, 'text').strip
+  end
+
+  def self.single_chat(system:, user:, model: :basic, max_tokens: 1000)
+    model = select_model(model)
+
+    parameters = { model:, system:, max_tokens:,
+                   messages: [ { role: 'user', content: user } ] }
+
+    response = Anthropic::Client.new.messages(parameters:)
+    response.dig('content', 0, 'text').strip
+  end
+
+  def self.chat(system:, messages:, model: :basic, max_tokens: 1000)
+    model = select_model(model)
+    
+    parameters = { model:, max_tokens:, system:, messages: }
+
+    response = Anthropic::Client.new.messages(parameters:)
+    response.dig('content', 0, 'text').strip
+  end
+
+  def self.select_model(model)
+    case model
+    when :basic
+      BASIC_MODEL
+    when :advanced
+      ADVANCED_MODEL
+    else
+      model
+    end
+  end  
+end

data/lib/intelli_agent/{ai.rb → openai.rb}

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

data/lib/intelli_agent.rb

@@ -1,64 +1,8 @@
 require 'openai'
-require 'intelli_agent/ai'
+require 'anthropic'
 
-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_ASSISTANT_ID')
-    @assistant = @openai_client.assistants.retrieve(id: assistant_id)
-
-    thread_params = {}
-
-    # Only one vector store can be attached, according to the OpenAI API documentation
-    @vector_store_id = vector_store_id
-    thread_params = { tool_resources: { file_search: { vector_store_ids: [vector_store_id] } } } if @vector_store_id
-
-    thread_id ||= @openai_client.threads.create(parameters: thread_params)['id']
-    @thread = @openai_client.threads.retrieve(id: thread_id)
-
-    @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 })
-  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'] }
-
-    params[:instructions] = instructions || @instructions
-    params[:additional_instructions] = additional_instructions unless additional_instructions.nil?
-    params[:tool_choice] = tool_choice unless tool_choice.nil?
-
-    params[:additional_messages] = [{ role: :user, content: additional_message }] unless additional_message.nil?
-
-    params[:model] = model || @assistant['model']
-
-    run_id = @openai_client.runs.create(thread_id: @thread['id'], parameters: params)['id']
-
-    loop do
-      response = @openai_client.runs.retrieve(id: run_id, thread_id: @thread['id'])
-
-      case response['status']
-      when 'queued', 'in_progress', 'cancelling'
-        puts 'Status: Waiting AI Processing finish'
-        sleep 1
-      when 'completed'
-        puts last_message
-        break
-      when 'requires_action'
-        # Handle tool calls (see below)
-      when 'cancelled', 'failed', 'expired'
-        puts response['last_error'].inspect
-        break # or `exit`
-      else
-        puts "Unknown status response: #{status}"
-      end
-    end
-  end
+module IntelliAgent
 end
+
+require 'intelli_agent/anthropic'
+require 'intelli_agent/openai'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: intelli_agent
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.1.0
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2027-07-23 00:00:00.000000000 Z
+date: 2027-08-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ruby-openai
@@ -24,7 +24,21 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '7.1'
-description: A helper layer over OpenAI API
+- !ruby/object:Gem::Dependency
+  name: anthropic
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.3'
+description: A helper layer over Anthropic and OpenAI API
 email: gedean.dias@gmail.com
 executables: []
 extensions: []
@@ -32,8 +46,10 @@ extra_rdoc_files: []
 files:
 - README.md
 - lib/intelli_agent.rb
+- lib/intelli_agent/agent.rb.disabled
 - lib/intelli_agent/agent_test.rb_
-- lib/intelli_agent/ai.rb
+- lib/intelli_agent/anthropic.rb
+- lib/intelli_agent/openai.rb
 homepage: https://github.com/gedean/intelli_agent
 licenses:
 - MIT
@@ -53,8 +69,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.16
+rubygems_version: 3.5.17
 signing_key: 
 specification_version: 4
-summary: A helper layer over OpenAI API
+summary: A helper layer over Anthropic and OpenAI API
 test_files: []