kodo-bot 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 28db2b743e15b691e4c20712a38e822add6c2e559edf75780a9a82887695dfee
-  data.tar.gz: ddc6dc895a4b102506c2a68f5d497053dba8c6bf34ea09150ba47d25e3c8eab8
+  metadata.gz: f1b635240737c255adbfeae59a3a49414133f0b5a01897e92f44628334fe0b00
+  data.tar.gz: 04731bbe6a6419939afb2851c44ff328556a5ac3e2f22c8f9918a3a9e92a93b2
 SHA512:
-  metadata.gz: f475e0f066e08ae275093769a245a1a8e2d88147a8a0aa76985968365fd84cb12d530cfdee3c5afb69ca177f56d24437a725fe9ce4b451a3b61778c183d2745f
-  data.tar.gz: 703e9236a4338a9441e28aadb0d00ce2ab43e9f689fa79ff7fea63578e852c4682bd3ac4384706ae4f01d3acc376a1638791032def2f6d185b5cb2d2e4d53410
+  metadata.gz: 41c5de2afa4352882563f3b3ac56652ef1bad44729b449a5f74df910c876b31d4c34e13940cd2a6679a3192877b5370daaeb08f75b0b8085894ea4f47101bcd8
+  data.tar.gz: b4984ce2c8e2405b6706f631d3621650b8ad16b66d8fa80d209e0d1ad2cd65084eed0ebe49564a0a5eb79b7187bdd674677efe5a2a2052e5c53555277c96fe45

data/bin/kodo

@@ -1,32 +1,32 @@
 #!/usr/bin/env ruby
 # frozen_string_literal: true
 
-require_relative "../lib/kodo"
+require_relative '../lib/kodo'
 
 module Kodo
   class CLI
     COMMANDS = {
-      "start"     => "Start the Kodo daemon",
-      "chat"      => "Chat with Kodo directly in the terminal",
-      "memories"  => "List what Kodo remembers about you",
-      "reminders" => "List active reminders",
-      "status"    => "Show daemon status",
-      "version"   => "Show version",
-      "init"      => "Create default config and prompt files in ~/.kodo/",
-      "help"      => "Show this help"
+      'start' => 'Start the Kodo daemon',
+      'chat' => 'Chat with Kodo directly in the terminal',
+      'memories' => 'List what Kodo remembers about you',
+      'reminders' => 'List active reminders',
+      'status' => 'Show daemon status',
+      'version' => 'Show version',
+      'init' => 'Create default config and prompt files in ~/.kodo/',
+      'help' => 'Show this help'
     }.freeze
 
     def run(args = ARGV)
-      command = args.first || "help"
+      command = args.first || 'help'
 
       case command
-      when "start"     then start(args)
-      when "chat"      then chat
-      when "memories"  then memories
-      when "reminders" then reminders
-      when "init"      then init
-      when "version"   then puts "kodo v#{VERSION}"
-      when "status"    then status
+      when 'start'     then start(args)
+      when 'chat'      then chat
+      when 'memories'  then memories
+      when 'reminders' then reminders
+      when 'init'      then init
+      when 'version'   then puts "kodo v#{VERSION}"
+      when 'status'    then status
       else help
       end
     end
@@ -36,11 +36,9 @@ module Kodo
     def start(args)
       interval = nil
       args.each do |arg|
-        if arg.start_with?("--heartbeat-interval=")
-          interval = arg.split("=", 2).last.to_i
-        end
+        interval = arg.split('=', 2).last.to_i if arg.start_with?('--heartbeat-interval=')
       end
-      if (idx = args.index("--heartbeat-interval")) && !args[idx + 1]&.start_with?("--")
+      if (idx = args.index('--heartbeat-interval')) && !args[idx + 1]&.start_with?('--')
         interval = args[idx + 1]&.to_i
       end
 
@@ -48,28 +46,45 @@ module Kodo
       daemon.start!
     end
 
-    def chat
+    def chat # rubocop:disable Metrics/MethodLength
       Config.ensure_home_dir!
       assembler = PromptAssembler.new
       assembler.ensure_default_files!
-      LLM.configure!(Kodo.config)
 
       passphrase = Kodo.config.memory_encryption? ? Kodo.config.memory_passphrase : nil
 
-      puts "🥁 Kodo v#{VERSION} — direct chat mode"
-      puts "   Model: #{Kodo.config.llm_model}"
-      puts "   Type your message and press Enter. Ctrl+C to quit.\n\n"
-
       memory = Memory::Store.new(passphrase: passphrase)
       audit = Memory::Audit.new
       knowledge = Memory::Knowledge.new(passphrase: passphrase)
       reminder_store = Memory::Reminders.new(passphrase: passphrase)
+
+      secrets_store = Secrets::Store.new(passphrase: Kodo.config.secrets_passphrase)
+      broker = Secrets::Broker.new(store: secrets_store, audit: audit)
+      search_provider = resolve_chat_search_provider(broker)
+
+      LLM.configure!(Kodo.config, broker: broker)
+
+      puts "🥁 Kodo v#{VERSION} — direct chat mode"
+      puts "   Model: #{Kodo.config.llm_model}"
+      puts "   Type your message and press Enter. Ctrl+C to quit.\n\n"
+
+      router = nil
+
+      on_secret_stored = lambda do |_secret_name|
+        search_provider = resolve_chat_search_provider(broker)
+        router.reload_tools!(search_provider: search_provider)
+        LLM.configure!(Kodo.config, broker: broker)
+      end
+
       router = Router.new(
         memory: memory,
         audit: audit,
         prompt_assembler: assembler,
         knowledge: knowledge,
-        reminders: reminder_store
+        reminders: reminder_store,
+        search_provider: search_provider,
+        broker: broker,
+        on_secret_stored: on_secret_stored
       )
       console = Channels::Console.new
       console.connect!
@@ -80,10 +95,10 @@ module Kodo
         break if input.nil? || input.empty?
 
         message = Message.new(
-          channel_id: "console",
+          channel_id: 'console',
           sender: :user,
           content: input,
-          metadata: { chat_id: "console" }
+          metadata: { chat_id: 'console' }
         )
 
         response = with_spinner { router.route(message, channel: console) }
@@ -100,15 +115,15 @@ module Kodo
 
       active = knowledge.all_active
       if active.empty?
-        puts "Kodo has no stored memories yet."
-        puts "Chat with Kodo and it will remember things you tell it."
+        puts 'Kodo has no stored memories yet.'
+        puts 'Chat with Kodo and it will remember things you tell it.'
         return
       end
 
       puts "Kodo's memories (#{active.length} facts):"
-      puts ""
+      puts ''
 
-      grouped = active.group_by { |f| f["category"] }
+      grouped = active.group_by { |f| f['category'] }
       Memory::Knowledge::VALID_CATEGORIES.each do |cat|
         facts = grouped[cat]
         next unless facts&.any?
@@ -118,7 +133,7 @@ module Kodo
           puts "    - #{f['content']}"
           puts "      id: #{f['id']}  source: #{f['source']}  #{f['created_at']}"
         end
-        puts ""
+        puts ''
       end
     end
 
@@ -127,20 +142,20 @@ module Kodo
       passphrase = Kodo.config.memory_encryption? ? Kodo.config.memory_passphrase : nil
       reminder_store = Memory::Reminders.new(passphrase: passphrase)
 
-      active = reminder_store.all_active.sort_by { |r| r["due_at"] }
+      active = reminder_store.all_active.sort_by { |r| r['due_at'] }
       if active.empty?
-        puts "No active reminders."
+        puts 'No active reminders.'
         return
       end
 
       puts "Active reminders (#{active.length}):"
-      puts ""
+      puts ''
 
       active.each do |r|
         puts "  - #{r['content']}"
         puts "    due: #{r['due_at']}  id: #{r['id']}"
       end
-      puts ""
+      puts ''
     end
 
     def init
@@ -148,64 +163,77 @@ module Kodo
       PromptAssembler.new.ensure_default_files!
 
       puts "✅ Kodo home directory: #{Kodo.home_dir}"
-      puts ""
-      puts "   Created files:"
-      puts "   📋 config.yml     — LLM provider and channel settings"
-      puts "   🎭 persona.md     — personality and tone (make Kodo yours)"
-      puts "   👤 user.md        — tell Kodo about yourself"
-      puts "   💓 pulse.md       — what to notice during idle beats"
-      puts "   🌱 origin.md      — first-run onboarding conversation"
-      puts ""
-      puts "   Quick start:"
-      puts "   1. Set an LLM API key (e.g. ANTHROPIC_API_KEY, OPENAI_API_KEY)"
-      puts "   2. Set TELEGRAM_BOT_TOKEN (get one from @BotFather on Telegram)"
-      puts "   3. Enable Telegram in ~/.kodo/config.yml"
+      puts ''
+      puts '   Created files:'
+      puts '   📋 config.yml     — LLM provider and channel settings'
+      puts '   🎭 persona.md     — personality and tone (make Kodo yours)'
+      puts '   👤 user.md        — tell Kodo about yourself'
+      puts '   💓 pulse.md       — what to notice during idle beats'
+      puts '   🌱 origin.md      — first-run onboarding conversation'
+      puts ''
+      puts '   Quick start:'
+      puts '   1. Set an LLM API key (e.g. ANTHROPIC_API_KEY, OPENAI_API_KEY)'
+      puts '   2. Set TELEGRAM_BOT_TOKEN (get one from @BotFather on Telegram)'
+      puts '   3. Enable Telegram in ~/.kodo/config.yml'
       puts "   4. Edit ~/.kodo/persona.md to customize Kodo's personality"
-      puts "   5. Run: kodo start"
-      puts ""
-      puts "   Supported LLM providers:"
-      puts "   Anthropic, OpenAI, Gemini, DeepSeek, Mistral, Ollama,"
-      puts "   OpenRouter, Perplexity, xAI, and any OpenAI-compatible API"
+      puts '   5. Run: kodo start'
+      puts ''
+      puts '   Supported LLM providers:'
+      puts '   Anthropic, OpenAI, Gemini, DeepSeek, Mistral, Ollama,'
+      puts '   OpenRouter, Perplexity, xAI, and any OpenAI-compatible API'
     end
 
     def status
       puts "🥁 Kodo v#{VERSION}"
-      puts ""
+      puts ''
       puts "   Home:   #{Kodo.home_dir}"
-      puts "   Config: #{File.exist?(Config.config_path) ? "✅" : "❌"} #{Config.config_path}"
-      puts ""
+      puts "   Config: #{File.exist?(Config.config_path) ? '✅' : '❌'} #{Config.config_path}"
+      puts ''
 
       # Prompt files
-      puts "   Prompt files:"
+      puts '   Prompt files:'
       %w[persona.md user.md pulse.md origin.md].each do |f|
         path = File.join(Kodo.home_dir, f)
-        status = File.exist?(path) ? "✅" : "  "
+        status = File.exist?(path) ? '✅' : '  '
         puts "   #{status} #{f}"
       end
-      puts ""
+      puts ''
 
       # Provider keys
-      puts "   LLM providers:"
+      puts '   LLM providers:'
       {
-        "ANTHROPIC_API_KEY"  => "Anthropic",
-        "OPENAI_API_KEY"     => "OpenAI",
-        "GEMINI_API_KEY"     => "Gemini",
-        "OLLAMA_API_BASE"    => "Ollama"
+        'ANTHROPIC_API_KEY' => 'Anthropic',
+        'OPENAI_API_KEY' => 'OpenAI',
+        'GEMINI_API_KEY' => 'Gemini',
+        'OLLAMA_API_BASE' => 'Ollama'
       }.each do |env, name|
-        s = ENV[env] ? "✅ set" : "  not set"
-        puts "   #{s.start_with?("✅") ? "✅" : "  "} #{name.ljust(12)} (#{env})"
+        s = ENV[env] ? '✅ set' : '  not set'
+        puts "   #{s.start_with?('✅') ? '✅' : '  '} #{name.ljust(12)} (#{env})"
+      end
+      puts ''
+
+      puts "   Telegram: #{ENV['TELEGRAM_BOT_TOKEN'] ? '✅ set' : '❌ TELEGRAM_BOT_TOKEN missing'}"
+    end
+
+    def resolve_chat_search_provider(broker)
+      # Try broker first (handles both stored secrets and env vars)
+      if broker.available?('tavily_api_key')
+        api_key = broker.fetch('tavily_api_key', requestor: 'search')
+        return Search::Tavily.new(api_key: api_key) if api_key
       end
-      puts ""
 
-      puts "   Telegram: #{ENV["TELEGRAM_BOT_TOKEN"] ? "✅ set" : "❌ TELEGRAM_BOT_TOKEN missing"}"
+      # Fall back to config-based resolution
+      Kodo.config.search_provider_instance
+    rescue Kodo::Error
+      nil
     end
 
     def with_spinner
-      frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
+      frames = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏']
       done = false
       spinner = Thread.new do
         i = 0
-        while !done
+        until done
           print "\r\e[36m#{frames[i % frames.length]} thinking...\e[0m"
           i += 1
           sleep 0.1
@@ -221,21 +249,21 @@ module Kodo
 
     def help
       puts "🥁 Kodo v#{VERSION} — your personal AI agent"
-      puts ""
-      puts "Usage: kodo <command> [options]"
-      puts ""
+      puts ''
+      puts 'Usage: kodo <command> [options]'
+      puts ''
       COMMANDS.each do |cmd, desc|
-        puts "  %-12s %s" % [cmd, desc]
+        puts format('  %-12s %s', cmd, desc)
       end
-      puts ""
-      puts "Options:"
-      puts "  --heartbeat-interval=N   Set heartbeat interval in seconds (default: 60)"
-      puts ""
-      puts "Prompt files in ~/.kodo/:"
+      puts ''
+      puts 'Options:'
+      puts '  --heartbeat-interval=N   Set heartbeat interval in seconds (default: 60)'
+      puts ''
+      puts 'Prompt files in ~/.kodo/:'
       puts "  persona.md    Your agent's personality and tone"
-      puts "  user.md       Tell Kodo about yourself"
-      puts "  pulse.md      What to notice during idle beats"
-      puts "  origin.md     First-run onboarding conversation"
+      puts '  user.md       Tell Kodo about yourself'
+      puts '  pulse.md      What to notice during idle beats'
+      puts '  origin.md     First-run onboarding conversation'
     end
   end
 end

data/config/default.yml

@@ -35,6 +35,14 @@ channels:
     enabled: false
     bot_token_env: TELEGRAM_BOT_TOKEN
 
+# Web search — enables web_search and fetch_url tools.
+# Disabled by default. Set provider to enable.
+# search:
+#   provider: tavily
+#   providers:
+#     tavily:
+#       api_key_env: TAVILY_API_KEY
+
 memory:
   encryption: false
   passphrase_env: KODO_PASSPHRASE

data/lib/kodo/config.rb

@@ -1,48 +1,55 @@
 # frozen_string_literal: true
 
-require "yaml"
-require "fileutils"
+require 'yaml'
+require 'fileutils'
+require 'securerandom'
 
 module Kodo
   class Config
     # Map of Kodo config key → RubyLLM config setter name
     PROVIDER_KEY_MAP = {
-      "anthropic"  => "anthropic",
-      "openai"     => "openai",
-      "gemini"     => "gemini",
-      "deepseek"   => "deepseek",
-      "mistral"    => "mistral",
-      "openrouter" => "openrouter",
-      "perplexity" => "perplexity",
-      "xai"        => "xai"
+      'anthropic' => 'anthropic',
+      'openai' => 'openai',
+      'gemini' => 'gemini',
+      'deepseek' => 'deepseek',
+      'mistral' => 'mistral',
+      'openrouter' => 'openrouter',
+      'perplexity' => 'perplexity',
+      'xai' => 'xai'
     }.freeze
 
     DEFAULTS = {
-      "daemon" => {
-        "port" => 7377,
-        "heartbeat_interval" => 60
+      'daemon' => {
+        'port' => 7377,
+        'heartbeat_interval' => 60
       },
-      "llm" => {
-        "model" => "claude-sonnet-4-6",
-        "utility_model" => "claude-haiku-4-5-20251001",
-        "providers" => {
-          "anthropic" => { "api_key_env" => "ANTHROPIC_API_KEY" }
+      'llm' => {
+        'model' => 'claude-sonnet-4-6',
+        'utility_model' => 'claude-haiku-4-5-20251001',
+        'providers' => {
+          'anthropic' => { 'api_key_env' => 'ANTHROPIC_API_KEY' }
         }
       },
-      "channels" => {
-        "telegram" => {
-          "enabled" => false,
-          "bot_token_env" => "TELEGRAM_BOT_TOKEN"
+      'channels' => {
+        'telegram' => {
+          'enabled' => false,
+          'bot_token_env' => 'TELEGRAM_BOT_TOKEN'
         }
       },
-      "memory" => {
-        "encryption" => false,
-        "passphrase_env" => "KODO_PASSPHRASE",
-        "store" => "file"
+      'memory' => {
+        'encryption' => false,
+        'passphrase_env' => 'KODO_PASSPHRASE',
+        'store' => 'file'
       },
-      "logging" => {
-        "level" => "info",
-        "audit" => true
+      'search' => {
+        'provider' => nil,
+        'providers' => {
+          'tavily' => { 'api_key_env' => 'TAVILY_API_KEY' }
+        }
+      },
+      'logging' => {
+        'level' => 'info',
+        'audit' => true
       }
     }.freeze
 
@@ -61,23 +68,23 @@ module Kodo
       end
 
       def config_path
-        File.join(Kodo.home_dir, "config.yml")
+        File.join(Kodo.home_dir, 'config.yml')
       end
 
       def ensure_home_dir!
         dirs = [
           Kodo.home_dir,
-          File.join(Kodo.home_dir, "memory", "conversations"),
-          File.join(Kodo.home_dir, "memory", "knowledge"),
-          File.join(Kodo.home_dir, "memory", "reminders"),
-          File.join(Kodo.home_dir, "memory", "audit"),
-          File.join(Kodo.home_dir, "skills")
+          File.join(Kodo.home_dir, 'memory', 'conversations'),
+          File.join(Kodo.home_dir, 'memory', 'knowledge'),
+          File.join(Kodo.home_dir, 'memory', 'reminders'),
+          File.join(Kodo.home_dir, 'memory', 'audit'),
+          File.join(Kodo.home_dir, 'skills')
         ]
         dirs.each { |d| FileUtils.mkdir_p(d) }
 
-        unless File.exist?(config_path)
-          File.write(config_path, YAML.dump(DEFAULTS))
-        end
+        return if File.exist?(config_path)
+
+        File.write(config_path, YAML.dump(DEFAULTS))
       end
 
       private
@@ -94,20 +101,20 @@ module Kodo
     end
 
     # --- Daemon ---
-    def port               = data.dig("daemon", "port")
-    def heartbeat_interval = data.dig("daemon", "heartbeat_interval")
+    def port               = data.dig('daemon', 'port')
+    def heartbeat_interval = data.dig('daemon', 'heartbeat_interval')
 
     # --- LLM ---
-    def llm_model = data.dig("llm", "model")
-    def utility_model = data.dig("llm", "utility_model") || llm_model
+    def llm_model = data.dig('llm', 'model')
+    def utility_model = data.dig('llm', 'utility_model') || llm_model
 
     # Returns a hash of { "provider_name" => "actual_api_key" } for all configured providers
     def llm_api_keys
-      providers = data.dig("llm", "providers") || {}
+      providers = data.dig('llm', 'providers') || {}
       keys = {}
 
       providers.each do |provider, settings|
-        env_var = settings["api_key_env"]
+        env_var = settings['api_key_env']
         next unless env_var
 
         key = ENV[env_var]
@@ -117,23 +124,21 @@ module Kodo
         end
       end
 
-      if keys.empty?
-        raise Error, "No LLM API keys found. Set at least one provider key (e.g. ANTHROPIC_API_KEY)"
-      end
+      raise Error, 'No LLM API keys found. Set at least one provider key (e.g. ANTHROPIC_API_KEY)' if keys.empty?
 
       keys
     end
 
     # Optional: Ollama base URL for local models
     def ollama_api_base
-      data.dig("llm", "providers", "ollama", "api_base") || ENV["OLLAMA_API_BASE"]
+      data.dig('llm', 'providers', 'ollama', 'api_base') || ENV['OLLAMA_API_BASE']
     end
 
     # --- Memory / Encryption ---
-    def memory_encryption? = data.dig("memory", "encryption") == true
+    def memory_encryption? = data.dig('memory', 'encryption') == true
 
     def memory_passphrase_env
-      data.dig("memory", "passphrase_env") || "KODO_PASSPHRASE"
+      data.dig('memory', 'passphrase_env') || 'KODO_PASSPHRASE'
     end
 
     def memory_passphrase
@@ -141,21 +146,72 @@ module Kodo
       if memory_encryption? && (passphrase.nil? || passphrase.empty?)
         raise Error, "Memory encryption is enabled but #{memory_passphrase_env} is not set"
       end
+
       passphrase
     end
 
+    # --- Secrets ---
+    def secrets_passphrase
+      passphrase_path = File.join(Kodo.home_dir, '.passphrase')
+      if File.exist?(passphrase_path)
+        File.read(passphrase_path).strip
+      else
+        passphrase = SecureRandom.hex(32)
+        FileUtils.mkdir_p(Kodo.home_dir)
+        File.write(passphrase_path, passphrase)
+        File.chmod(0o600, passphrase_path)
+        passphrase
+      end
+    end
+
     # --- Logging ---
-    def log_level       = data.dig("logging", "level")&.to_sym || :info
-    def audit_enabled?  = data.dig("logging", "audit") != false
+    def log_level       = data.dig('logging', 'level')&.to_sym || :info
+    def audit_enabled?  = data.dig('logging', 'audit') != false
 
     # --- Channels ---
     def telegram_bot_token
-      env_var = data.dig("channels", "telegram", "bot_token_env")
+      env_var = data.dig('channels', 'telegram', 'bot_token_env')
       ENV.fetch(env_var) { raise Error, "Missing environment variable: #{env_var}" }
     end
 
     def telegram_enabled?
-      data.dig("channels", "telegram", "enabled") == true
+      data.dig('channels', 'telegram', 'enabled') == true
+    end
+
+    # --- Search ---
+    def search_provider
+      data.dig('search', 'provider')
+    end
+
+    def search_configured?
+      !search_provider.nil? && !search_provider.empty?
+    end
+
+    def search_api_key
+      return nil unless search_configured?
+
+      provider = search_provider
+      env_var = data.dig('search', 'providers', provider, 'api_key_env')
+      return nil unless env_var
+
+      ENV[env_var]
+    end
+
+    def search_provider_instance
+      return nil unless search_configured?
+
+      case search_provider
+      when 'tavily'
+        api_key = search_api_key
+        unless api_key
+          raise Error,
+                "Tavily API key not set (#{data.dig('search', 'providers', 'tavily', 'api_key_env')})"
+        end
+
+        Search::Tavily.new(api_key: api_key)
+      else
+        raise Error, "Unknown search provider: #{search_provider}"
+      end
     end
   end
 end