aia 0.9.4 → 0.9.6

data/lib/aia/chat_processor_service.rb

@@ -94,21 +94,22 @@ module AIA
 
     def determine_operation_type
       mode = AIA.config.client.model.modalities
-      if mode.supports?(:text_to_image)
+
+      if mode.text_to_image?
         :text_to_image
-      elsif mode.supports?(:image_to_text)
+      elsif mode.image_to_text?
         :image_to_text
-      elsif mode.supports?(:audio_to_text)
+      elsif mode.audio_to_text?
         :audio_to_text
-      elsif mode.supports?(:text_to_audio)
+      elsif mode.text_to_audio?
         :text_to_audio
-      elsif mode.supports?(:audio_to_audio)
+      elsif mode.audio_to_audio?
         :audio_to_audio
-      elsif mode.supports?(:image_to_image)
+      elsif mode.image_to_image?
         :image_to_image
-      elsif mode.supports?(:audio_to_image)
+      elsif mode.audio_to_image?
         :audio_to_image
-      elsif mode.supports?(:image_to_audio)
+      elsif mode.image_to_audio?
         :image_to_audio
       else
         :text_to_text

data/lib/aia/directive_processor.rb

@@ -299,16 +299,16 @@ module AIA
     desc "All Available models or query on [partial LLM or provider name] Examples: //llms ; //llms openai ; //llms claude"
     def available_models( args=nil, context_manager=nil)
       query     = args
-      header    = "Available LLMs"
-
-      if query
-        header += " for #{query.join(' and ')}"
-      end
+      header    = "\nAvailable LLMs"
+      header   += " for #{query.join(' and ')}" if query
 
       puts header + ':'
+      puts
+
+      q1 = query.select{|q| q.include?('_to_')}.map{|q| ':'==q[0] ? q[1...] : q}
+      q2 = query.reject{|q| q.include?('_to_')}
 
-      q1 = query.select{|q| !q.start_with?(':')}
-      q2 = query.select{|q|  q.start_with?(':')}
+      counter = 0
 
       RubyLLM.models.all.each do |llm|
         inputs  = llm.modalities.input.join(',')
@@ -316,17 +316,25 @@ module AIA
         entry   = "- #{llm.id} (#{llm.provider}) #{inputs} to #{outputs}"
 
         if query.nil? || query.empty?
+          counter += 1
           puts entry
           next
         end
 
         show_it = true
-        q1.each{|q| show_it &&= entry.include?(q)}
-        q2.each{|q| show_it &&= llm.modalities.supports?(q)}
+        q1.each{|q| show_it &&= llm.modalities.send("#{q}?")}
+        q2.each{|q| show_it &&= entry.include?(q)}
 
-        puts entry if show_it
+        if show_it
+          counter += 1
+          puts entry
+        end
       end
 
+      puts if counter > 0
+      puts "#{counter} LLMs matching your query"
+      puts
+
       ""
     end
     alias_method :am,         :available_models

data/lib/aia/ruby_llm_adapter.rb

@@ -96,17 +96,17 @@ module AIA
       modes = @chat.model.modalities
 
       # TODO: Need to consider how to handle multi-mode models
-      if modes.supports? :text_to_text
+      if modes.text_to_text?
         text_to_text(prompt)
 
-      elsif modes.supports? :image_to_text
+      elsif modes.image_to_text?
         image_to_text(prompt)
-      elsif modes.supports? :text_to_image
+      elsif modes.text_to_image?
         text_to_image(prompt)
 
-      elsif modes.supports? :text_to_audio
+      elsif modes.text_to_audio?
         text_to_audio(prompt)
-      elsif modes.supports? :audio_to_text
+      elsif modes.audio_to_text?
         audio_to_text(prompt)
 
       else

data/lib/aia/utility.rb

@@ -21,7 +21,7 @@ module AIA
          (O)       using #{AIA.config.adapter} (v#{RubyLLM::VERSION})
        __||__    \\) model db was last refreshed on
      [/______\\]  /    #{AIA.config.last_refresh}
-    / \\__AI__/ \\/      #{AIA.config.tool_paths.empty? ? 'I forgot my toolbox' : 'I brought some tools'}
+    / \\__AI__/ \\/      #{AIA.config.tool_paths.empty? ? 'You can share my tools' : 'I will also use your tools'}
    /    /__\\
   (\\   /____\\   #{AIA.config.tool_paths.empty? ? '' : 'My Toolbox contains:'}
         ROBOT

data/lib/aia.rb

@@ -5,15 +5,19 @@
 # provides an interface for interacting with AI models and managing prompts.
 
 require 'ruby_llm'
+require 'ruby_llm/mcp'
+require 'shared_tools'
+require 'shared_tools/ruby_llm'
 require 'prompt_manager'
 
+
 require 'debug_me'
 include DebugMe
 $DEBUG_ME = false
 DebugMeDefaultOptions[:skip1] = true
 
 require_relative 'extensions/openstruct_merge'    # adds self.merge self.get_value
-require_relative 'extensions/ruby_llm/modalities' # adds model.modalities.supports? :text-to-text etc.
+require_relative 'extensions/ruby_llm/modalities' # adds model.modalities.text_to_text? etc.
 
 require_relative 'refinements/string.rb'        # adds #include_any? #include_all?
 

data/lib/extensions/ruby_llm/modalities.rb

@@ -1,26 +1,34 @@
 # lib/extensions/ruby_llm/modalities.rb
-# A models "modes" are often expressed in terms like:
-#   text-to-text
-#   text_to_audio
-#   audio to image
-#   image2image
-# This new supports? method tests the models modalities against
-# these common expressions
 
 class RubyLLM::Model::Modalities
-  def supports?(query_mode)
-    parts = query_mode
-              .to_s
-              .downcase
-              .split(/2|-to-| to |_to_/)
-              .map(&:strip)
-
-    if 2 == parts.size
-      input.include?(parts[0]) && output.include?(parts[1])
-    elsif 1 == parts.size
-      input.include?(parts[0]) || output.include?(parts[0])
-    else
-      false
-    end
-  end
+  #
+  def text_to_text? = input.include?('text') && output.include?('text')
+  def text_to_embeddings? = input.include?('text') && output.include?('embeddings')
+  def text_to_audio? = input.include?('text') && output.include?('audio')
+  def text_to_image? = input.include?('text') && output.include?('image')
+  def text_to_moderation? = input.include?('text') && output.include?('moderation')
+  #
+  def image_to_text? = input.include?('image') && output.include?('text')
+  def image_to_embeddings? = input.include?('image') && output.include?('embeddings')
+  def image_to_audio? = input.include?('image') && output.include?('audio')
+  def image_to_image? = input.include?('image') && output.include?('image')
+  def image_to_moderation? = input.include?('image') && output.include?('moderation')
+  #
+  def pdf_to_text? = input.include?('pdf') && output.include?('text')
+  def pdf_to_embeddings? = input.include?('pdf') && output.include?('embeddings')
+  def pdf_to_audio? = input.include?('pdf') && output.include?('audio')
+  def pdf_to_image? = input.include?('pdf') && output.include?('image')
+  def pdf_to_moderation? = input.include?('pdf') && output.include?('moderation')
+  #
+  def audio_to_text? = input.include?('audio') && output.include?('text')
+  def audio_to_embeddings? = input.include?('audio') && output.include?('embeddings')
+  def audio_to_audio? = input.include?('audio') && output.include?('audio')
+  def audio_to_image? = input.include?('audio') && output.include?('image')
+  def audio_to_moderation? = input.include?('audio') && output.include?('moderation')
+  #
+  def file_to_text? = input.include?('file') && output.include?('text')
+  def file_to_embeddings? = input.include?('file') && output.include?('embeddings')
+  def file_to_audio? = input.include?('file') && output.include?('audio')
+  def file_to_image? = input.include?('file') && output.include?('image')
+  def file_to_moderation? = input.include?('file') && output.include?('moderation')
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aia
 version: !ruby/object:Gem::Version
-  version: 0.9.4
+  version: 0.9.6
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -43,28 +43,42 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.5.4
+        version: 0.5.6
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.5.4
+        version: 0.5.6
 - !ruby/object:Gem::Dependency
   name: ruby_llm
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.3.0
+        version: 1.3.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.3.0
+        version: 1.3.1
+- !ruby/object:Gem::Dependency
+  name: ruby_llm-mcp
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: reline
   requirement: !ruby/object:Gem::Requirement
@@ -79,6 +93,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: shared_tools
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: shellwords
   requirement: !ruby/object:Gem::Requirement
@@ -292,6 +320,7 @@ files:
 - examples/tools/list_files.rb
 - examples/tools/read_file.rb
 - examples/tools/run_shell_command.rb
+- images/aia.png
 - justfile
 - lib/aia.rb
 - lib/aia/aia_completion.bash