active_genie 0.0.24 → 0.0.25

data/lib/active_genie/concerns/loggable.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module ActiveGenie
   module Concerns
     module Loggable
@@ -6,38 +8,24 @@ module ActiveGenie
       end
 
       module ClassMethods
-        def with_logging_context(context_method, observer_proc = nil)
+        def call_with_log_context(context_method)
           original_method = instance_method(:call)
-          
+
           define_method(:call) do |*args, **kwargs, &block|
             context = send(context_method, *args, **kwargs)
-            bound_observer = observer_proc ? ->(log) { instance_exec(log, &observer_proc) } : nil
-            
-            ActiveGenie::Logger.with_context(context, observer: bound_observer) do
+
+            ActiveGenie::Logger.with_context(context) do
               original_method.bind(self).call(*args, **kwargs, &block)
             end
           end
         end
-      end
-
-      def info(log)
-        ::ActiveGenie::Logger.info(log)
-      end
 
-      def error(log)
-        ::ActiveGenie::Logger.error(log)
-      end
-
-      def warn(log)
-        ::ActiveGenie::Logger.warn(log)
-      end
+        def logger(data)
+          log = ActiveGenie::Logger.call(data)
+          @config&.log&.call_observers(log)
 
-      def debug(log)
-        ::ActiveGenie::Logger.debug(log)
-      end
-
-      def trace(log)
-        ::ActiveGenie::Logger.trace(log)
+          log
+        end
       end
     end
   end

data/lib/active_genie/config/battle_config.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module ActiveGenie
+  module Config
+    class BattleConfig
+    end
+  end
+end

data/lib/active_genie/config/data_extractor_config.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module ActiveGenie
+  module Config
+    class DataExtractorConfig
+      attr_accessor :with_explanation, :min_accuracy, :verbose
+
+      def initialize
+        @with_explanation = true
+        @min_accuracy = 70
+        @verbose = false
+      end
+
+      def merge(config_params = {})
+        dup.tap do |config|
+          config.with_explanation = config_params[:with_explanation] if config_params[:with_explanation]
+          config.min_accuracy = config_params[:min_accuracy] if config_params[:min_accuracy]
+          config.verbose = config_params[:verbose] if config_params[:verbose]
+        end
+      end
+    end
+  end
+end

data/lib/active_genie/config/llm_config.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module ActiveGenie
+  module Config
+    class LlmConfig
+      attr_accessor :model, :temperature, :max_tokens, :max_retries, :retry_delay,
+                    :model_tier, :read_timeout, :open_timeout, :client
+      attr_reader :provider
+
+      def initialize
+        @model = nil
+        @provider = nil
+        @client = nil
+        @temperature = 0
+        @max_tokens = 4096
+        @max_retries = nil
+        @retry_delay = nil
+        @model_tier = 'lower_tier'
+        @read_timeout = nil
+        @open_timeout = nil
+      end
+
+      def provider=(provider)
+        @provider = provider&.to_s&.downcase&.strip&.to_sym
+      end
+
+      def merge(config_params = {})
+        dup.tap do |config|
+          config_params.each do |key, value|
+            config.send("#{key}=", value) if config.respond_to?("#{key}=")
+          end
+        end
+      end
+    end
+  end
+end

data/lib/active_genie/config/log_config.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module ActiveGenie
+  module Config
+    class LogConfig
+      def add_observer(observers: [], scope: nil, &block)
+        @observers ||= []
+
+        raise ArgumentError, 'Scope must be a hash' if !scope.nil? && !scope.is_a?(Hash)
+
+        @observers << { observer: block, scope: scope || {} } if block_given?
+        Array(observers).each do |observer|
+          @observers << { observer:, scope: scope || {} }
+        end
+      end
+
+      def remove_observer(observers)
+        Array(observers).each do |observer|
+          @observers.delete_if { |obs| obs[:observer] == observer }
+        end
+      end
+
+      def clear_observers
+        @observers = []
+      end
+
+      def call_observers(log)
+        Array(@observers).each do |obs|
+          next unless obs[:scope].all? { |key, value| log[key.to_sym] == value }
+
+          obs[:observer].call(log)
+        rescue StandardError => e
+          ActiveGenie::Logger.call(code: :observer_error, **obs, error: e.message)
+        end
+      end
+
+      def merge(config_params = {})
+        dup.tap do |config|
+          config.add_observer(config_params[:observers]) if config_params[:observers]
+        end
+      end
+    end
+  end
+end

data/lib/active_genie/config/providers/anthropic_config.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require_relative './provider_base'
+
+module ActiveGenie
+  module Config
+    module Providers
+      # Configuration class for the Anthropic API client.
+      # Manages API keys, URLs, model selections, and client instantiation.
+      class AnthropicConfig < ProviderBase
+        NAME = :anthropic
+
+        # Retrieves the API key.
+        # Falls back to the ANTHROPIC_API_KEY environment variable if not set.
+        # @return [String, nil] The API key.
+        def api_key
+          @api_key || ENV['ANTHROPIC_API_KEY']
+        end
+
+        # Retrieves the base API URL for Anthropic API.
+        # Defaults to 'https://api.anthropic.com'.
+        # @return [String] The API base URL.
+        def api_url
+          @api_url || 'https://api.anthropic.com'
+        end
+
+        # Retrieves the Anthropic version.
+        # Defaults to '2023-06-01'.
+        # @return [String] The Anthropic version.
+        def anthropic_version
+          @anthropic_version || '2023-06-01'
+        end
+
+        # Retrieves the model name designated for the lower tier (e.g., cost-effective, faster).
+        # Defaults to 'claude-3-haiku'.
+        # @return [String] The lower tier model name.
+        def lower_tier_model
+          @lower_tier_model || 'claude-3-5-haiku-20241022'
+        end
+
+        # Retrieves the model name designated for the middle tier (e.g., balanced performance).
+        # Defaults to 'claude-3-sonnet'.
+        # @return [String] The middle tier model name.
+        def middle_tier_model
+          @middle_tier_model || 'claude-3-7-sonnet-20250219'
+        end
+
+        # Retrieves the model name designated for the upper tier (e.g., most capable).
+        # Defaults to 'claude-3-opus'.
+        # @return [String] The upper tier model name.
+        def upper_tier_model
+          @upper_tier_model || 'claude-3-opus-20240229'
+        end
+      end
+    end
+  end
+end

data/lib/active_genie/config/providers/deepseek_config.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require_relative './provider_base'
+
+module ActiveGenie
+  module Config
+    module Providers
+      # Configuration class for the DeepSeek API client.
+      # Manages API keys, organization IDs, URLs, model selections, and client instantiation.
+      class DeepseekConfig < ProviderBase
+        NAME = :deepseek
+
+        # Retrieves the API key.
+        # Falls back to the DEEPSEEK_API_KEY environment variable if not set.
+        # @return [String, nil] The API key.
+        def api_key
+          @api_key || ENV['DEEPSEEK_API_KEY']
+        end
+
+        # Retrieves the base API URL for DeepSeek API.
+        # Defaults to 'https://api.deepseek.com/v1'.
+        # @return [String] The API base URL.
+        def api_url
+          @api_url || 'https://api.deepseek.com/v1'
+        end
+
+        # Retrieves the model name designated for the lower tier (e.g., cost-effective, faster).
+        # Defaults to 'deepseek-chat'.
+        # @return [String] The lower tier model name.
+        def lower_tier_model
+          @lower_tier_model || 'deepseek-chat'
+        end
+
+        # Retrieves the model name designated for the middle tier (e.g., balanced performance).
+        # Defaults to 'deepseek-chat'.
+        # @return [String] The middle tier model name.
+        def middle_tier_model
+          @middle_tier_model || 'deepseek-chat'
+        end
+
+        # Retrieves the model name designated for the upper tier (e.g., most capable).
+        # Defaults to 'deepseek-reasoner'.
+        # @return [String] The upper tier model name.
+        def upper_tier_model
+          @upper_tier_model || 'deepseek-reasoner'
+        end
+      end
+    end
+  end
+end

data/lib/active_genie/config/providers/google_config.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+require_relative './provider_base'
+
+module ActiveGenie
+  module Config
+    module Providers
+      # Configuration class for the Google Generative Language API client.
+      # Manages API keys, URLs, model selections, and client instantiation.
+      class GoogleConfig < ProviderBase
+        NAME = :google
+
+        # Retrieves the API key.
+        # Falls back to the GENERATIVE_LANGUAGE_GOOGLE_API_KEY environment variable if not set.
+        # @return [String, nil] The API key.
+        def api_key
+          @api_key || ENV['GENERATIVE_LANGUAGE_GOOGLE_API_KEY'] || ENV['GEMINI_API_KEY']
+        end
+
+        # Retrieves the base API URL for Google Generative Language API.
+        # Defaults to 'https://generativelanguage.googleapis.com'.
+        # @return [String] The API base URL.
+        def api_url
+          # NOTE: Google Generative Language API uses a specific path structure like /v1beta/models/{model}:generateContent
+          # The base URL here should be just the domain part.
+          @api_url || 'https://generativelanguage.googleapis.com'
+        end
+
+        # Retrieves the model name designated for the lower tier (e.g., cost-effective, faster).
+        # Defaults to 'gemini-2.0-flash-lite'.
+        # @return [String] The lower tier model name.
+        def lower_tier_model
+          @lower_tier_model || 'gemini-2.0-flash-lite'
+        end
+
+        # Retrieves the model name designated for the middle tier (e.g., balanced performance).
+        # Defaults to 'gemini-2.0-flash'.
+        # @return [String] The middle tier model name.
+        def middle_tier_model
+          @middle_tier_model || 'gemini-2.0-flash'
+        end
+
+        # Retrieves the model name designated for the upper tier (e.g., most capable).
+        # Defaults to 'gemini-2.5-pro-experimental'.
+        # @return [String] The upper tier model name.
+        def upper_tier_model
+          @upper_tier_model || 'gemini-2.5-pro-experimental'
+        end
+      end
+    end
+  end
+end

data/lib/active_genie/config/providers/openai_config.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require_relative './provider_base'
+
+module ActiveGenie
+  module Config
+    module Providers
+      # Configuration class for the OpenAI API client.
+      # Manages API keys, organization IDs, URLs, model selections, and client instantiation.
+      class OpenaiConfig < ProviderBase
+        NAME = :openai
+
+        # Retrieves the API key.
+        # Falls back to the OPENAI_API_KEY environment variable if not set.
+        # @return [String, nil] The API key.
+        def api_key
+          @api_key || ENV['OPENAI_API_KEY']
+        end
+
+        # Retrieves the base API URL for OpenAI API.
+        # Defaults to 'https://api.openai.com/v1'.
+        # @return [String] The API base URL.
+        def api_url
+          @api_url || 'https://api.openai.com/v1'
+        end
+
+        # Retrieves the model name designated for the lower tier (e.g., cost-effective, faster).
+        # Defaults to 'gpt-4o-mini'.
+        # @return [String] The lower tier model name.
+        def lower_tier_model
+          @lower_tier_model || 'gpt-4.1-mini'
+        end
+
+        # Retrieves the model name designated for the middle tier (e.g., balanced performance).
+        # Defaults to 'gpt-4o'.
+        # @return [String] The middle tier model name.
+        def middle_tier_model
+          @middle_tier_model || 'gpt-4.1'
+        end
+
+        # Retrieves the model name designated for the upper tier (e.g., most capable).
+        # Defaults to 'o1-preview'.
+        # @return [String] The upper tier model name.
+        def upper_tier_model
+          @upper_tier_model || 'o3-mini'
+        end
+      end
+    end
+  end
+end

data/lib/active_genie/config/providers/provider_base.rb

@@ -0,0 +1,89 @@
+# frozen_string_literal: true
+
+module ActiveGenie
+  module Config
+    module Providers
+      class ProviderBase
+        NAME = :unknown
+
+        attr_writer :api_key, :organization, :api_url, :client,
+                    :lower_tier_model, :middle_tier_model, :upper_tier_model
+
+        # Maps a symbolic tier (:lower_tier, :middle_tier, :upper_tier) to a specific model name.
+        # Falls back to the lower_tier_model if the tier is nil or unrecognized.
+        # @param tier [Symbol, String, nil] The symbolic tier name.
+        # @return [String] The corresponding model name.
+        def tier_to_model(tier)
+          {
+            lower_tier: lower_tier_model,
+            middle_tier: middle_tier_model,
+            upper_tier: upper_tier_model
+          }[tier&.to_sym] || lower_tier_model
+        end
+
+        # Returns a hash representation of the configuration.
+        # @param config [Hash] Additional key-value pairs to merge into the hash.
+        # @return [Hash] The configuration settings as a hash.
+        def to_h(config = {})
+          {
+            name: NAME,
+            api_key:,
+            api_url:,
+            lower_tier_model:,
+            middle_tier_model:,
+            upper_tier_model:,
+            **config
+          }
+        end
+
+        # Validates the configuration.
+        # @return [Boolean] True if the configuration is valid, false otherwise.
+        def valid?
+          api_key && api_url
+        end
+
+        # Retrieves the API key.
+        # Falls back to the OPENAI_API_KEY environment variable if not set.
+        # @return [String, nil] The API key.
+        def api_key
+          raise NotImplementedError, 'Subclasses must implement this method'
+        end
+
+        # Retrieves the base API URL for OpenAI API.
+        # Defaults to 'https://api.openai.com/v1'.
+        # @return [String] The API base URL.
+        def api_url
+          raise NotImplementedError, 'Subclasses must implement this method'
+        end
+
+        # Lazily initializes and returns an instance of the OpenaiClient.
+        # Passes itself (the config object) to the client's constructor.
+        # @return [ActiveGenie::Clients::OpenaiClient] The client instance.
+        def client
+          raise NotImplementedError, 'Subclasses must implement this method'
+        end
+
+        # Retrieves the model name designated for the lower tier (e.g., cost-effective, faster).
+        # Defaults to 'gpt-4o-mini'.
+        # @return [String] The lower tier model name.
+        def lower_tier_model
+          raise NotImplementedError, 'Subclasses must implement this method'
+        end
+
+        # Retrieves the model name designated for the middle tier (e.g., balanced performance).
+        # Defaults to 'gpt-4o'.
+        # @return [String] The middle tier model name.
+        def middle_tier_model
+          raise NotImplementedError, 'Subclasses must implement this method'
+        end
+
+        # Retrieves the model name designated for the upper tier (e.g., most capable).
+        # Defaults to 'o1-preview'.
+        # @return [String] The upper tier model name.
+        def upper_tier_model
+          raise NotImplementedError, 'Subclasses must implement this method'
+        end
+      end
+    end
+  end
+end

data/lib/active_genie/config/providers_config.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+module ActiveGenie
+  module Config
+    class ProvidersConfig
+      def initialize
+        @all = {}
+        @default = nil
+      end
+
+      attr_reader :all
+
+      def default
+        @default || valid.keys.first
+      end
+
+      def default=(provider)
+        normalized_provider = provider.to_s.downcase.strip
+        @default = normalized_provider.size > 0 ? normalized_provider : valid.keys.first
+      end
+
+      def valid
+        valid_provider_keys = @all.keys.select { |k| @all[k].valid? }
+        @all.slice(*valid_provider_keys)
+      end
+
+      def add(provider_classes)
+        @all ||= {}
+        Array(provider_classes).each do |provider|
+          name = provider::NAME
+          remove([name]) if @all.key?(name)
+
+          @all[name] = provider.new
+        end
+
+        self
+      end
+
+      def remove(provider_classes)
+        Array(provider_classes).each do |provider|
+          @all.delete(provider::NAME)
+        end
+
+        self
+      end
+
+      def merge(config_params = {})
+        dup.tap do |config|
+          config.add(config_params[:providers]) if config_params[:providers]
+        end
+      end
+
+      def method_missing(m, *args, &block)
+        @all[m] || super
+      end
+
+      def respond_to_missing?(m, include_private = false)
+        @all.key?(m) || super
+      end
+    end
+  end
+end

data/lib/active_genie/config/ranking_config.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+module ActiveGenie
+  module Config
+    class RankingConfig
+      attr_accessor :score_variation_threshold
+
+      def initialize
+        @score_variation_threshold = 30
+      end
+
+      def merge(config_params = {})
+        dup.tap do |config|
+          if config_params[:score_variation_threshold]
+            config.score_variation_threshold = config_params[:score_variation_threshold]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/active_genie/config/scoring_config.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module ActiveGenie
+  module Config
+    class ScoringConfig
+    end
+  end
+end

data/lib/active_genie/configuration.rb

@@ -1,42 +1,65 @@
-require_relative 'configuration/providers_config'
-require_relative 'configuration/providers/openai_config'
-require_relative 'configuration/providers/google_config'
-require_relative 'configuration/providers/anthropic_config'
-require_relative 'configuration/providers/deepseek_config'
-require_relative 'configuration/log_config'
-require_relative 'configuration/runtime_config'
+# frozen_string_literal: true
+
+require_relative 'config/providers_config'
+require_relative 'config/log_config'
+require_relative 'config/ranking_config'
+require_relative 'config/scoring_config'
+require_relative 'config/data_extractor_config'
+require_relative 'config/battle_config'
+require_relative 'config/llm_config'
 
 module ActiveGenie
-  module Configuration
-    module_function
+  class Configuration
+    def log
+      @log ||= Config::LogConfig.new
+    end
 
     def providers
-      @providers ||= begin 
-        p = ProvidersConfig.new
-        p.register(ActiveGenie::Configuration::Providers::OpenaiConfig)
-        p.register(ActiveGenie::Configuration::Providers::GoogleConfig)
-        p.register(ActiveGenie::Configuration::Providers::AnthropicConfig)
-        p.register(ActiveGenie::Configuration::Providers::DeepseekConfig)
-        p
-      end
+      @providers ||= Config::ProvidersConfig.new
     end
 
-    def log
-      @log ||= LogConfig.new
+    def ranking
+      @ranking ||= Config::RankingConfig.new
+    end
+
+    def scoring
+      @scoring ||= Config::ScoringConfig.new
+    end
+
+    def data_extractor
+      @data_extractor ||= Config::DataExtractorConfig.new
     end
 
-    def runtime
-      @runtime ||= RuntimeConfig.new
+    def battle
+      @battle ||= Config::BattleConfig.new
     end
 
-    def to_h(configs = {})
-      normalized_configs = configs.dig(:runtime) ? configs : { runtime: configs }
+    def llm
+      @llm ||= Config::LlmConfig.new
+    end
+
+    def merge(config_params = {})
+      return config_params if config_params.is_a?(Configuration)
+
+      new_configuration = dup
+
+      %w[log providers llm ranking scoring data_extractor battle].each do |key|
+        config = new_configuration.send(key)
+
+        next unless config.respond_to?(:merge)
 
-      {
-        providers: providers.to_h(normalized_configs.dig(:providers) || {}),
-        log: log.to_h(normalized_configs.dig(:log) || {}),
-        runtime: runtime.to_h(normalized_configs.dig(:runtime) || {})
-      }
+        new_config = if config_params.key?(key)
+                       config.merge(config_params[key])
+                     else
+                       config.merge(config_params)
+                     end
+
+        new_configuration.send("#{key}=", new_config)
+      end
+
+      new_configuration
     end
+
+    attr_writer :log, :providers, :ranking, :scoring, :data_extractor, :battle, :llm
   end
 end