Fix bugs preventing selective use of ai services

Author: azimux

CHANGELOG.md

@@ -1,4 +1,4 @@
-## [0.0.5] - 2025-03-04
+## [0.0.6] - 2025-03-04
 
 - Do not require all 3 services to function. Let user require which services they want to use.
 

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    foobara-ai (0.0.5)
+    foobara-ai (0.0.6)
       foobara-anthropic-api
       foobara-ollama-api
       foobara-open-ai-api

lib/foobara/ai.rb

@@ -7,10 +7,7 @@ module Ai
     module AnswerBot
       foobara_domain!
 
-      base_dir = "#{__dir__}/../../src/foobara/ai/answer_bot"
-      Util.require_directory "#{base_dir}/types"
-      Dir["#{base_dir}/domain_mappers/*.rb"].each { |f| require f }
-      Dir["#{base_dir}/*.rb"].each { |f| require f }
+      require_relative "../../src/foobara/ai/answer_bot/types/service_enum"
 
       ai_services = {}
 
@@ -26,12 +23,19 @@ module AnswerBot
 
         ai_services[service] = domain
         foobara_depends_on domain
-
-        path = Util.underscore(domain.foobara_domain_name)
-        Util.require_directory "#{__dir__}/../../src/foobara/ai/answer_bot/domain_mappers/#{path}"
       end
 
       AI_SERVICES = ai_services
+
+      base_dir = "#{__dir__}/../../src/foobara/ai/answer_bot"
+      Util.require_directory "#{base_dir}/types"
+      Dir["#{base_dir}/domain_mappers/*.rb"].each { |f| require f }
+      Dir["#{base_dir}/*.rb"].each { |f| require f }
+
+      AI_SERVICES.each_value do |domain|
+        path = Util.underscore(domain.foobara_domain_name)
+        Util.require_directory "#{base_dir}/domain_mappers/#{path}"
+      end
     end
   end
 end

src/foobara/ai/answer_bot/list_models.rb

@@ -1,3 +1,4 @@
+require_relative "types/service_enum"
 require_relative "types/model"
 
 module Foobara
@@ -8,11 +9,13 @@ class ListModels < Foobara::Command
 
         result [Types::Model]
 
-        LIST_COMMANDS = [
-          OpenAiApi::ListModels,
-          AnthropicApi::ListModels,
-          OllamaApi::ListLocalModels
-        ].freeze
+        LIST_COMMANDS = AI_SERVICES.values.map do |domain|
+          if domain == OllamaApi
+            OllamaApi::ListLocalModels
+          else
+            domain::ListModels
+          end
+        end
 
         depends_on(*LIST_COMMANDS)
 

version.rb

@@ -1,6 +1,6 @@
 module Foobara
   module Ai
-    VERSION = "0.0.5".freeze
+    VERSION = "0.0.6".freeze
 
     local_ruby_version = File.read("#{__dir__}/.ruby-version").chomp
     local_ruby_version_minor = local_ruby_version[/\A(\d+\.\d+)\.\d+\z/, 1]