ragdoll-cli 0.0.2

data/lib/ragdoll/cli/commands/stats.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require 'json'
+
+module Ragdoll
+  module CLI
+    class Stats
+      def call(options)
+        client = StandaloneClient.new
+
+        puts 'Retrieving system statistics'
+        puts "Options: #{options.to_h}" unless options.to_h.empty?
+        puts
+
+        stats = client.stats
+
+        if stats.nil? || stats.empty?
+          puts 'No statistics available.'
+          return
+        end
+
+        case options[:format]
+        when 'json'
+          puts JSON.pretty_generate(stats)
+        when 'plain'
+          stats.each do |key, value|
+            puts "#{key.to_s.tr('_', ' ').capitalize}: #{value}"
+          end
+        else
+          # Table format (default)
+          puts 'System Statistics:'
+          puts
+          puts 'Metric'.ljust(30) + 'Value'
+          puts '-' * 50
+
+          stats.each do |key, value|
+            metric = key.to_s.tr('_', ' ').capitalize.ljust(30)
+            puts "#{metric}#{value}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ragdoll/cli/commands/status.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require 'json'
+
+module Ragdoll
+  module CLI
+    class Status
+      def call(id, options)
+        client = StandaloneClient.new
+
+        puts "Checking status for document ID: #{id}"
+        puts "Options: #{options.to_h}" unless options.to_h.empty?
+        puts
+
+        status = client.document_status(id: id)
+
+        if status.nil? || status.empty?
+          puts 'Document not found or no status available.'
+          return
+        end
+
+        case options[:format]
+        when 'json'
+          puts JSON.pretty_generate(status)
+        else
+          puts "Document ID: #{id}"
+          puts "Status: #{status[:status] || 'N/A'}"
+          puts "Embeddings Count: #{status[:embeddings_count] || 'N/A'}"
+          puts "Embeddings Ready: #{status[:embeddings_ready] || 'N/A'}"
+          puts "Message: #{status[:message] || 'No message available'}"
+        end
+      end
+    end
+  end
+end

data/lib/ragdoll/cli/commands/update.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module Ragdoll
+  module CLI
+    class Update
+      def call(id, options)
+        client = StandaloneClient.new
+
+        puts "Updating document ID: #{id}"
+        puts "Options: #{options.to_h}" unless options.to_h.empty?
+        puts
+
+        update_options = {}
+        update_options[:title] = options[:title] if options[:title]
+
+        if update_options.empty?
+          puts 'No updates provided. Use --title to update the document title.'
+          return
+        end
+
+        result = client.update_document(id: id, **update_options)
+
+        if result[:success]
+          puts "Document ID #{id} updated successfully."
+          puts result[:message] if result[:message]
+        else
+          puts "Failed to update document ID #{id}."
+          puts result[:message] if result[:message]
+        end
+      end
+    end
+  end
+end

data/lib/ragdoll/cli/configuration_loader.rb

@@ -0,0 +1,144 @@
+# frozen_string_literal: true
+
+require 'yaml'
+require 'fileutils'
+
+module Ragdoll
+  module CLI
+    class ConfigurationLoader
+      DEFAULT_CONFIG_PATH = File.expand_path('~/.ragdoll/config.yml')
+
+      def initialize
+        @config_path = ENV['RAGDOLL_CONFIG'] || DEFAULT_CONFIG_PATH
+      end
+
+
+      def load
+        config = load_config_file
+        configure_ragdoll(config)
+      end
+
+
+      def create_default_config
+        ensure_config_directory
+
+        default_config = {
+          'llm_provider' => 'openai',
+          'embedding_model' => 'text-embedding-3-small',
+          'chunk_size' => 1000,
+          'chunk_overlap' => 200,
+          'search_similarity_threshold' => 0.7,
+          'max_search_results' => 10,
+          'database_config' => {
+            'adapter' => 'postgresql',
+            'database' => 'ragdoll_development',
+            'host' => 'localhost',
+            'port' => 5432,
+            'username' => ENV['USER'] || 'postgres',
+            'auto_migrate' => true
+          },
+          'log_level' => 'warn',
+          'log_file' => File.expand_path('~/.ragdoll/ragdoll.log')
+        }
+
+        File.write(@config_path, YAML.dump(default_config))
+        default_config
+      end
+
+
+      def config_exists?
+        File.exist?(@config_path)
+      end
+
+      attr_reader :config_path
+
+      private
+
+      def load_config_file
+        return create_default_config unless config_exists?
+
+        YAML.load_file(@config_path)
+      rescue StandardError => e
+        puts "Warning: Could not load config file #{@config_path}: #{e.message}"
+        puts 'Using default configuration.'
+        create_default_config
+      end
+
+
+      def configure_ragdoll(config)
+        Ragdoll::Core.configure do |ragdoll_config|
+          # LLM and embedding configuration
+          if ragdoll_config.respond_to?(:llm_provider=)
+            ragdoll_config.llm_provider = config['llm_provider']&.to_sym || :openai
+          end
+          if ragdoll_config.respond_to?(:embedding_model=)
+            ragdoll_config.embedding_model = config['embedding_model'] || 'text-embedding-3-small'
+          end
+
+          # Processing settings
+          if ragdoll_config.respond_to?(:chunk_size=)
+            ragdoll_config.chunk_size = config['chunk_size'] || 1000
+          end
+          if ragdoll_config.respond_to?(:chunk_overlap=)
+            ragdoll_config.chunk_overlap = config['chunk_overlap'] || 200
+          end
+          if ragdoll_config.respond_to?(:search_similarity_threshold=)
+            ragdoll_config.search_similarity_threshold = config['search_similarity_threshold'] || 0.7
+          end
+          if ragdoll_config.respond_to?(:max_search_results=)
+            ragdoll_config.max_search_results = config['max_search_results'] || 10
+          end
+
+          # Database configuration - avoid direct setting if method doesn't exist
+          db_config = config['database_config'] || {
+            'adapter' => 'postgresql',
+            'database' => 'ragdoll_development',
+            'host' => 'localhost',
+            'port' => 5432,
+            'username' => ENV['USER'] || 'postgres',
+            'auto_migrate' => true
+          }
+          # Set database configuration only if the method exists
+          ragdoll_config.database_config = db_config if ragdoll_config.respond_to?(:database_config=)
+
+          # Logging configuration
+          if ragdoll_config.respond_to?(:log_level=)
+            ragdoll_config.log_level = config['log_level']&.to_sym || :warn
+          end
+          if ragdoll_config.respond_to?(:log_file=)
+            ragdoll_config.log_file = config['log_file'] || File.expand_path('~/.ragdoll/ragdoll.log')
+          end
+
+          # Set API keys from environment variables or config
+          if ragdoll_config.respond_to?(:openai_api_key=)
+            ragdoll_config.openai_api_key = ENV['OPENAI_API_KEY'] || config.dig('api_keys', 'openai')
+          end
+          if ragdoll_config.respond_to?(:anthropic_api_key=)
+            ragdoll_config.anthropic_api_key = ENV['ANTHROPIC_API_KEY'] || config.dig('api_keys',
+                                                                                      'anthropic')
+          end
+          if ragdoll_config.respond_to?(:google_api_key=)
+            ragdoll_config.google_api_key = ENV['GOOGLE_API_KEY'] || config.dig('api_keys', 'google')
+          end
+          if ragdoll_config.respond_to?(:azure_api_key=)
+            ragdoll_config.azure_api_key = ENV['AZURE_OPENAI_API_KEY'] || config.dig('api_keys', 'azure')
+          end
+          if ragdoll_config.respond_to?(:huggingface_api_key=)
+            ragdoll_config.huggingface_api_key = ENV['HUGGINGFACE_API_KEY'] || config.dig('api_keys',
+                                                                                          'huggingface')
+          end
+
+          if config.dig('api_keys', 'ollama_url') && ragdoll_config.respond_to?(:ollama_url=)
+            ragdoll_config.ollama_url = config.dig('api_keys', 'ollama_url')
+          end
+        end
+      end
+
+
+      def ensure_config_directory
+        config_dir = File.dirname(@config_path)
+        FileUtils.mkdir_p(config_dir) unless Dir.exist?(config_dir)
+      end
+    end
+  end
+end

data/lib/ragdoll/cli/standalone_client.rb

@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+module Ragdoll
+  module CLI
+    class StandaloneClient
+      include DebugMe
+      def add_document(path, **options)
+        Ragdoll.add_document(path: path, **options)
+      end
+
+
+      def document_status(id)
+        Ragdoll.document_status(id: id)
+      end
+
+
+      def get_document(id)
+        Ragdoll.get_document(id: id)
+      end
+
+
+      def update_document(id, **options)
+        Ragdoll.update_document(id: id, **options)
+      end
+
+
+      def delete_document(id)
+        Ragdoll.delete_document(id: id)
+      end
+
+
+      def list_documents(**options)
+        Ragdoll.list_documents(**options)
+      end
+
+
+      def search(query, **options)
+        Ragdoll.search(query: query, **options)
+      end
+
+
+      def get_context(query, **options)
+        Ragdoll.get_context(query: query, **options)
+      end
+
+
+      def enhance_prompt(prompt, **options)
+        Ragdoll.enhance_prompt(prompt: prompt, **options)
+      end
+
+
+      def stats
+        Ragdoll.stats
+      end
+
+
+      def healthy?
+        Ragdoll.healthy?
+      end
+
+
+      def configuration
+        Ragdoll.configuration
+      end
+
+
+    end
+  end
+end

data/lib/ragdoll/cli/version.rb

@@ -0,0 +1,8 @@
+# ragdoll/cli/version.rb
+# frozen_string_literal: true
+
+module Ragdoll
+  module CLI
+    VERSION = "0.0.2"
+  end
+end