console_agent 0.9.0 → 0.11.0

data/lib/console_agent/configuration.rb

@@ -1,6 +1,6 @@
 module ConsoleAgent
   class Configuration
-    PROVIDERS = %i[anthropic openai].freeze
+    PROVIDERS = %i[anthropic openai local].freeze
 
     PRICING = {
       'claude-sonnet-4-6' => { input: 3.0 / 1_000_000, output: 15.0 / 1_000_000 },
@@ -20,7 +20,9 @@ module ConsoleAgent
                   :storage_adapter, :memories_enabled,
                   :session_logging, :connection_class,
                   :admin_username, :admin_password,
-                  :authenticate
+                  :authenticate,
+                  :slack_bot_token, :slack_app_token, :slack_channel_ids,
+                  :local_url, :local_model, :local_api_key
 
     def initialize
       @provider     = :anthropic
@@ -40,6 +42,58 @@ module ConsoleAgent
       @admin_username   = nil
       @admin_password   = nil
       @authenticate     = nil
+      @safety_guards    = nil
+      @slack_bot_token  = nil
+      @slack_app_token  = nil
+      @slack_channel_ids = nil
+      @local_url        = 'http://localhost:11434'
+      @local_model      = 'qwen2.5:7b'
+      @local_api_key    = nil
+    end
+
+    def safety_guards
+      @safety_guards ||= begin
+        require 'console_agent/safety_guards'
+        SafetyGuards.new
+      end
+    end
+
+    # Register a custom safety guard by name with an around-block.
+    #
+    #   config.safety_guard :mailers do |&execute|
+    #     ActionMailer::Base.perform_deliveries = false
+    #     execute.call
+    #   ensure
+    #     ActionMailer::Base.perform_deliveries = true
+    #   end
+    def safety_guard(name, &block)
+      safety_guards.add(name, &block)
+    end
+
+    # Register a built-in safety guard by name.
+    # Available: :database_writes, :http_mutations, :mailers
+    #
+    # Options:
+    #   allow: Array of strings or regexps to allowlist for this guard.
+    #     - :http_mutations  → hosts (e.g. "s3.amazonaws.com", /googleapis\.com/)
+    #     - :database_writes → table names (e.g. "console_agent_sessions")
+    def use_builtin_safety_guard(name, allow: nil)
+      require 'console_agent/safety_guards'
+      guard_name = name.to_sym
+      case guard_name
+      when :database_writes
+        safety_guards.add(:database_writes, &BuiltinGuards.database_writes)
+      when :http_mutations
+        safety_guards.add(:http_mutations, &BuiltinGuards.http_mutations)
+      when :mailers
+        safety_guards.add(:mailers, &BuiltinGuards.mailers)
+      else
+        raise ConfigurationError, "Unknown built-in safety guard: #{name}. Available: database_writes, http_mutations, mailers"
+      end
+
+      if allow
+        Array(allow).each { |key| safety_guards.allow(guard_name, key) }
+      end
     end
 
     def resolved_api_key
@@ -50,6 +104,8 @@ module ConsoleAgent
         ENV['ANTHROPIC_API_KEY']
       when :openai
         ENV['OPENAI_API_KEY']
+      when :local
+        @local_api_key || 'no-key'
       end
     end
 
@@ -61,6 +117,8 @@ module ConsoleAgent
         'claude-sonnet-4-6'
       when :openai
         'gpt-5.3-codex'
+      when :local
+        @local_model
       end
     end
 
@@ -78,17 +136,27 @@ module ConsoleAgent
         'claude-opus-4-6'
       when :openai
         'gpt-5.3-codex'
+      when :local
+        @local_model
       end
     end
 
+    def resolved_timeout
+      @provider == :local ? [@timeout, 300].max : @timeout
+    end
+
     def validate!
       unless PROVIDERS.include?(@provider)
         raise ConfigurationError, "Unknown provider: #{@provider}. Valid: #{PROVIDERS.join(', ')}"
       end
 
-      unless resolved_api_key
-        env_var = @provider == :anthropic ? 'ANTHROPIC_API_KEY' : 'OPENAI_API_KEY'
-        raise ConfigurationError, "No API key. Set config.api_key or #{env_var} env var."
+      if @provider == :local
+        raise ConfigurationError, "No local_url configured for :local provider." unless @local_url && !@local_url.empty?
+      else
+        unless resolved_api_key
+          env_var = @provider == :anthropic ? 'ANTHROPIC_API_KEY' : 'OPENAI_API_KEY'
+          raise ConfigurationError, "No API key. Set config.api_key or #{env_var} env var."
+        end
       end
     end
   end