active_genie 0.0.12 → 0.0.19

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

@@ -0,0 +1,54 @@
+require_relative '../../clients/anthropic_client'
+require_relative './base_config'
+
+module ActiveGenie
+  module Configuration::Providers
+    # Configuration class for the Anthropic API client.
+    # Manages API keys, URLs, model selections, and client instantiation.
+    class AnthropicConfig < BaseConfig
+      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
+
+      # Lazily initializes and returns an instance of the AnthropicClient.
+      # Passes itself (the config object) to the client's constructor.
+      # @return [ActiveGenie::Clients::AnthropicClient] The client instance.
+      def client
+        @client ||= ::ActiveGenie::Clients::AnthropicClient.new(self)
+      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

data/lib/active_genie/configuration/providers/base_config.rb

@@ -0,0 +1,85 @@
+module ActiveGenie
+  module Configuration::Providers
+    class BaseConfig
+      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

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

@@ -0,0 +1,54 @@
+require_relative '../../clients/openai_client'
+require_relative './base_config'
+
+module ActiveGenie
+  module Configuration::Providers
+    # Configuration class for the DeepSeek API client.
+    # Manages API keys, organization IDs, URLs, model selections, and client instantiation.
+    class DeepseekConfig < BaseConfig
+      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
+
+      # 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
+        @client ||= ::ActiveGenie::Clients::OpenaiClient.new(self)
+      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

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

@@ -0,0 +1,56 @@
+require_relative '../../clients/google_client'
+require_relative './base_config'
+
+module ActiveGenie
+  module Configuration::Providers
+    # Configuration class for the Google Generative Language API client.
+    # Manages API keys, URLs, model selections, and client instantiation.
+    class GoogleConfig < BaseConfig
+      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
+
+      # Lazily initializes and returns an instance of the GoogleClient.
+      # Passes itself (the config object) to the client's constructor.
+      # @return [ActiveGenie::Clients::GoogleClient] The client instance.
+      def client
+        @client ||= ::ActiveGenie::Clients::GoogleClient.new(self)
+      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

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

@@ -0,0 +1,54 @@
+require_relative '../../clients/openai_client'
+require_relative './base_config'
+
+module ActiveGenie
+  module Configuration::Providers
+    # Configuration class for the OpenAI API client.
+    # Manages API keys, organization IDs, URLs, model selections, and client instantiation.
+    class OpenaiConfig < BaseConfig
+      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
+
+      # 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
+        @client ||= ::ActiveGenie::Clients::OpenaiClient.new(self)
+      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-4o-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-4o'
+      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 || 'o1-preview'
+      end
+    end
+  end
+end

data/lib/active_genie/configuration/providers_config.rb

@@ -5,8 +5,9 @@ module ActiveGenie::Configuration
       @default = nil
     end
 
-    def register(name, provider_class)
+    def register(provider_class)
       @all ||= {}
+      name = provider_class::NAME
       @all[name] = provider_class.new
       define_singleton_method(name) do
         instance_variable_get("@#{name}") || instance_variable_set("@#{name}", @all[name])
@@ -16,11 +17,12 @@ module ActiveGenie::Configuration
     end
 
     def default
-      @default || @all.values.first
+      @default || @all.values.find { |p| p.api_key }.class::NAME
     end
 
-    def all
-      @all
+    def valid
+      valid_provider_keys = @all.keys.select { |k| @all[k].valid? }
+      @all.slice(*valid_provider_keys)
     end
 
     def to_h(config = {})
@@ -32,6 +34,7 @@ module ActiveGenie::Configuration
     end
 
     private
+
     attr_writer :default
   end
 end

data/lib/active_genie/configuration/runtime_config.rb

@@ -0,0 +1,35 @@
+module ActiveGenie::Configuration
+  class RuntimeConfig
+    attr_writer :max_tokens, :temperature, :model, :provider, :api_key, :max_retries
+
+    def max_tokens
+      @max_tokens ||= 4096
+    end
+
+    def temperature
+      @temperature ||= 0.1
+    end
+
+    def model
+      @model
+    end
+
+    def provider
+      @provider ||= ActiveGenie.configuration.providers.default
+    end
+
+    def api_key
+      @api_key
+    end
+
+    def max_retries
+      @max_retries ||= 3
+    end
+
+    def to_h(config = {})
+      {
+        max_tokens:, temperature:, model:, provider:, api_key:, max_retries:,
+      }.merge(config)
+    end
+  end
+end

data/lib/active_genie/configuration.rb

@@ -1,6 +1,10 @@
 require_relative 'configuration/providers_config'
-require_relative 'configuration/openai_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'
 
 module ActiveGenie
   module Configuration
@@ -9,7 +13,10 @@ module ActiveGenie
     def providers
       @providers ||= begin 
         p = ProvidersConfig.new
-        p.register(:openai, ActiveGenie::Configuration::OpenaiConfig)
+        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
     end
@@ -18,10 +25,17 @@ module ActiveGenie
       @log ||= LogConfig.new
     end
 
+    def runtime
+      @runtime ||= RuntimeConfig.new
+    end
+
     def to_h(configs = {})
+      normalized_configs = configs.dig(:runtime) ? configs : { runtime: configs }
+
       {
-        providers: providers.to_h(configs.dig(:providers) || {}),
-        log: log.to_h(configs.dig(:log) || {})
+        providers: providers.to_h(normalized_configs.dig(:providers) || {}),
+        log: log.to_h(normalized_configs.dig(:log) || {}),
+        runtime: runtime.to_h(normalized_configs.dig(:runtime) || {})
       }
     end
   end

data/lib/active_genie/data_extractor/basic.rb

@@ -37,21 +37,34 @@ module ActiveGenie::DataExtractor
         {  role: 'system', content: PROMPT },
         {  role: 'user', content: @text }
       ]
+
+      properties = data_to_extract_with_explaination
+
       function = {
         name: 'data_extractor',
         description: 'Extract structured and typed data from user messages.',
-        schema: {
+        parameters: {
           type: "object",
-          properties: data_to_extract_with_explaination
+          properties:,
+          required: properties.keys
         }
       }
 
-      ::ActiveGenie::Clients::UnifiedClient.function_calling(
+      result = ::ActiveGenie::Clients::UnifiedClient.function_calling(
         messages,
         function,
         model_tier: 'lower_tier',
         config: @config
       )
+
+      ActiveGenie::Logger.debug({
+        code: :data_extractor,
+        text: @text[0..30],
+        data_to_extract: @data_to_extract,
+        extracted_data: result
+      })
+
+      result
     end
 
     private

data/lib/active_genie/data_extractor.rb

@@ -10,6 +10,10 @@ module ActiveGenie
       Basic.call(...)
     end
 
+    def call(...)
+      Basic.call(...)
+    end
+
     def from_informal(...)
       FromInformal.call(...)
     end

data/lib/active_genie/logger.rb

@@ -5,56 +5,55 @@ module ActiveGenie
   module Logger
     module_function
 
-    def with_context(context)
+    def with_context(context, observer: nil)
       @context ||= {}
+      @observers ||= []
       begin
         @context = @context.merge(context)
+        @observers << observer if observer
         yield if block_given?
       ensure
         @context.delete_if { |key, _| context.key?(key) }
+        @observers.delete(observer)
       end
     end
 
     def info(log)
-      save(log, level: :info)
+      call(log, level: :info)
     end
 
     def error(log)
-      save(log, level: :error)
+      call(log, level: :error)
     end
 
     def warn(log)
-      save(log, level: :warn)
+      call(log, level: :warn)
     end
 
     def debug(log)
-      save(log, level: :debug)
+      call(log, level: :debug)
     end
 
     def trace(log)
-      save(log, level: :trace)
+      call(log, level: :trace)
     end
 
-    def save(data, level: :info)
-      log = @context.merge(data || {})
-      log[:timestamp] = Time.now
-      log[:level] = level.to_s.upcase
-      log[:process_id] = Process.pid
-      config_log_level = LOG_LEVELS[log.dig(:config, :log_level)] || LOG_LEVELS[:info]
+    def call(data, level: :info)
+      log = {
+        **(@context || {}),
+        **(data || {}),
+        timestamp: Time.now,
+        level: level.to_s.upcase,
+        process_id: Process.pid
+      }
 
-      FileUtils.mkdir_p('logs')
-      File.write('logs/active_genie.log', "#{JSON.generate(log)}\n", mode: 'a')
-      if config_log_level >= LOG_LEVELS[level]
-        $stdout.puts log
-      else
-        $stdout.print '.'
-      end
+      append_to_file(log)
+      output(log, level)
+      call_observers(log)
 
       log
     end
 
-    private
-
     # Log Levels
     # 
     # LOG_LEVELS defines different levels of logging within the application. 
@@ -68,5 +67,25 @@ module ActiveGenie
     LOG_LEVELS = { info: 0, error: 0, warn: 1, debug: 2, trace: 3 }.freeze
 
     attr_accessor :context
+    
+    def append_to_file(log)
+      FileUtils.mkdir_p('logs')
+      File.write('logs/active_genie.log', "#{JSON.generate(log)}\n", mode: 'a')
+    end
+
+    def output(log, level)
+      config_log_level = LOG_LEVELS[log.dig(:config, :log_level)] || LOG_LEVELS[:info]
+      if config_log_level >= LOG_LEVELS[level]
+        $stdout.puts log
+      else
+        $stdout.print '.'
+      end
+    end
+
+    def call_observers(log)
+      return if @observers.nil? || @observers.size.zero?
+
+      @observers.each { |observer| observer.call(log) }
+    end
   end
 end