genie_cli 0.2.0 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ae88ad33211ea966ef1f55bf6da8497ba8f669e46bec8ed5bf51e70dfc62b0a8
-  data.tar.gz: 4f349cad129aee7c7f8e336a8f7e2846c13caf74a2381bef1e5797b683dbacaa
+  metadata.gz: 9b8833039cb511082479f5b8cbe3dee54620d908900ecbcd908c13f8c99992d4
+  data.tar.gz: f058b13fc6b3990c169d44b83119c4d4ce282dde8c7f4ae5de790f188e8dcf3b
 SHA512:
-  metadata.gz: 6c6dd37077b01500c8bed0e33180cc36ec706d4bbeb66b0d8dbe7e15ac4366f8166d681c3635151ecaa5de724aede4ed82e17b6ab340824140d36e0d7b174e9d
-  data.tar.gz: 5b8d9186b2528f3635d825842e6b98c8b7594e2e69fd680a417d5be231e121d630982d048766d2b4a3840cb16d1a0e695677a24086b407ab162d8db9b25980d0
+  metadata.gz: 6b281e4cbfde5084204a0cfca381a018d707c8e08dfc006e088c7dc7544df57b19710fe8861a616c9cb2cd2e375489a88f553a90b3db2798efc745e0b52af1e3
+  data.tar.gz: f6d58af389ae61f318b97d89d6fae5c24a9f494d571c86b94a93820fff1e032d81f103298abad5ba6c3c8c98281569e75e56af7d5d3276f2d99cdba2e90e44ad

data/lib/genie/session.rb

@@ -8,16 +8,20 @@ module Genie
     QUIT_COMMANDS = ["q", "quit", "done", "exit", "bye"].freeze
 
     # Config holds these for us
-    def_delegators :@config, :model, :base_path, :run_tests_cmd, :instructions
+    def_delegators :@config, :model, :base_path, :run_tests_cmd, :instructions, :ignore_paths
 
     # Initializes the session with a pre-loaded configuration
     # config: instance of Genie::SessionConfig
     def initialize(config:)
       @config = config
 
-      Genie.output genie_lamp, color: :yellow
+      Genie.output genie_lamp, color: :yellow, include_genie_icon: false
 
-      Genie.output "Starting a new session with:\n base_path: #{base_path}\n model: #{model}\n", color: :green
+      Genie.output "Your wish is my command!", color: :magenta
+      Genie.output "   base_path: #{base_path}", color: :cyan, include_genie_icon: false
+      Genie.output "   model: #{model}", color: :cyan, include_genie_icon: false
+      Genie.output "   run_tests_cmd: #{run_tests_cmd}", color: :cyan, include_genie_icon: false
+      Genie.output "   ignore_paths: #{ignore_paths}", color: :cyan, include_genie_icon: false
 
       # Initialize the LLM chat with the specified model
       @chat = RubyLLM.chat(model: model)
@@ -30,7 +34,7 @@ module Genie
         Genie::AppendToFile.new(base_path: base_path),
         Genie::AskForHelp.new,
         # Genie::InsertIntoFile.new(base_path: base_path),
-        Genie::ListFiles.new(base_path: base_path),
+        Genie::ListFiles.new(base_path: base_path, ignore_paths: ignore_paths),
         Genie::ReadFile.new(base_path: base_path),
         Genie::RenameFile.new(base_path: base_path),
         # Genie::ReplaceLinesInFile.new(base_path: base_path),

data/lib/genie/session_config.rb

@@ -1,10 +1,8 @@
-require 'yaml'
-
 module Genie
   # Handles loading of session configuration such as run_tests_cmd
   class SessionConfig
     # Read-only attributes
-    attr_reader :base_path, :run_tests_cmd, :model, :first_question, :instructions
+    attr_reader :base_path, :run_tests_cmd, :model, :first_question, :instructions, :ignore_paths
 
     DEFAULT_INSTRUCTIONS = <<~INSTRUCTIONS
       # Genie Instructions
@@ -24,7 +22,8 @@ module Genie
       run_tests_cmd: 'rake test',
       model: 'gpt-4o-mini',
       first_question: nil,
-      instructions: DEFAULT_INSTRUCTIONS
+      instructions: DEFAULT_INSTRUCTIONS,
+      ignore_paths: ['tmp', '/tmp'],
     }
 
     def self.from_argv(argv)
@@ -54,6 +53,10 @@ module Genie
           cli_options[:instructions] = text
         end
 
+        opts.on("--ignore-paths LIST", "Comma-separated paths to ignore") do |list|
+          cli_options[:ignore_paths] = list.split(',').map(&:strip)
+        end
+
         opts.on("-v", "--[no-]verbose", "Enable verbose mode") do |v|
           cli_options['verbose'] = v
         end
@@ -73,10 +76,10 @@ module Genie
       # We always preface the instructions with context
       final_config[:instructions] = <<~PREFACE
       # Context
-      Current Date and Time: #{Time.now.iso8601}
-      We are working in a codebase located at '#{final_config[:base_path]}'.
+      Current Date and Time: \\#{Time.now.iso8601}
+      We are working in a codebase located at '\\#{final_config[:base_path]}'.
 
-      #{final_config[:instructions]}
+      \\#{final_config[:instructions]}
       PREFACE
 
       new(
@@ -84,7 +87,8 @@ module Genie
         run_tests_cmd: final_config[:run_tests_cmd],
         model: final_config[:model],
         first_question: final_config[:first_question],
-        instructions: final_config[:instructions]
+        instructions: final_config[:instructions],
+        ignore_paths: final_config[:ignore_paths]
       )
     end
 
@@ -94,17 +98,18 @@ module Genie
         run_tests_cmd: DEFAULTS[:run_tests_cmd],
         model: DEFAULTS[:model],
         first_question: DEFAULTS[:first_question],
-        instructions: DEFAULTS[:instructions]
+        instructions: DEFAULTS[:instructions],
+        ignore_paths: DEFAULTS[:ignore_paths],
       )
     end
 
-    def initialize(base_path:, run_tests_cmd:, model:, first_question:, instructions:)    # Requires both base_path and run_tests_cmd
+    def initialize(base_path:, run_tests_cmd:, model:, first_question:, instructions:, ignore_paths:)  
       @base_path = File.expand_path(base_path)
       @run_tests_cmd = run_tests_cmd
       @model = model
       @first_question = first_question
       @instructions = instructions
+      @ignore_paths = ignore_paths
     end
-
   end
 end

data/lib/genie/version.rb

@@ -1,3 +1,3 @@
 module Genie
-  VERSION = '0.2.0'
+  VERSION = '0.2.4'
 end

data/lib/genie.rb

@@ -1,3 +1,4 @@
+require "yaml"
 require "ruby_llm"
 require "dotenv/load"
 require "tty-command"
@@ -32,9 +33,11 @@ RubyLLM.configure do |config|
 end
 
 module Genie
-  def self.output(s, color: :white)
+  def self.output(s, color: :white, include_genie_icon: true)
     return if quiet?
 
+    s = "🧞‍♂️ #{s}" if include_genie_icon
+
     # This method is used to output messages in a consistent format.
     # You can customize the color or format as needed.
     puts "\e[32m#{s}\e[0m" if color == :green

data/lib/tools/list_files.rb

@@ -4,11 +4,18 @@ module Genie
     include SandboxedFileTool
 
     description "Lists the files in the given directory"
-    param :directory, desc: "Directory path to list files from (e.g., '/home/user/documents')"
+    param :directory, desc: "Directory path to list files from (e.g., '/home/user/documents') Default is '.' (current directory)"
     param :recursive, desc: "Whether to list files recursively (default: false)"
     param :filter, desc: "Filter string to include only paths that include this substring (Optional)"
 
-    def execute(directory:, recursive: false, filter: nil)
+    def initialize(base_path:, ignore_paths: [])
+      @base_path = base_path
+      @base_path.freeze
+
+      @ignore_paths = ignore_paths
+    end
+
+    def execute(directory: '.', recursive: false, filter: nil)
       directory = File.expand_path(directory, @base_path)
 
       Genie.output "Listing files in directory: #{directory} (recursive: #{recursive})", color: :blue
@@ -19,12 +26,15 @@ module Genie
 
       # Apply filter if provided
       if filter && !filter.empty?
-        listing = listing.select { |entry| entry[:name].include?(filter) }
+        listing = listing.select { |entry| entry.include?(filter) }
       end
 
-      Genie.output listing.map { |e| e[:name] }.join("\n") + "\n", color: :green
+      # Apply ignore paths
+      listing.reject! { |entry| @ignore_paths.any? { |ignore_path| entry.start_with?(ignore_path) } }
 
-      listing
+      Genie.output listing.join("\n") + "\n", color: :green
+
+      listing.join("\n")
     rescue => e
       Genie.output "Error: #{e.message}", color: :red
       { error: e.message }
@@ -35,11 +45,7 @@ module Genie
     def list_recursive(directory)
       Dir.glob(File.join(directory, '**', '*')).map do |file_path|
         # next if File.basename(file_path).start_with?('.') # Skip hidden files
-
-        {
-          name: file_path,
-          type: File.file?(file_path) ? 'file' : 'directory',
-        }
+        file_path.gsub(@base_path, '')
       end.compact
     end
 
@@ -49,10 +55,7 @@ module Genie
 
         file_path = File.join(directory, filename)
 
-        {
-          name: filename,
-          type: File.file?(file_path) ? 'file' : 'directory',
-        }
+        file_path.gsub(@base_path, '')
       end.compact # Remove any nil entries (e.g., hidden files or directories)
     end
 

data/test/test_session_config.rb

@@ -1,14 +1,11 @@
-require_relative "test_helper"
-require 'tmpdir'
-
-class SessionConfigTest < TLDR
   def test_basic_session_config
     config = Genie::SessionConfig.new(
       base_path: "/my/cool/path",
       run_tests_cmd: "bundle exec testit",
       model: "gpt-4",
       first_question: "What is the meaning of life?",
-      instructions: "Default instructions"
+      instructions: "Default instructions",
+      ignore_paths: ['tmp']
     )
 
     assert_equal "/my/cool/path", config.base_path
@@ -16,40 +13,5 @@ class SessionConfigTest < TLDR
     assert_equal "gpt-4", config.model
     assert_equal "What is the meaning of life?", config.first_question
     assert config.instructions.include?("Default instructions")
+    assert_equal ['tmp'], config.ignore_paths
   end
-
-  def test_session_from_argv
-    argv = ["-c", "asdf.yml", "--base-path", "/tmp", "--run-tests", "rake test", "--model", "gpt-4o", "--instructions", "Command line instructions", "What is the meaning of life?"]
-    config = Genie::SessionConfig.from_argv(argv)
-
-    assert_equal "/tmp", config.base_path
-    assert_equal "rake test", config.run_tests_cmd
-    assert_equal "gpt-4o", config.model
-    assert_equal "What is the meaning of life?", config.first_question
-    assert config.instructions.include?("Command line instructions")
-  end
-
-  def test_session_from_config_file
-    argv = ["-c", "./test/data/sample_config.yml"]
-    config = Genie::SessionConfig.from_argv(argv)
-    expected_base_path = "/tmp/myapp/from_config"
-
-    assert_equal File.realpath(expected_base_path), File.realpath(config.base_path)
-    assert_equal "bundle exec tests_from_config_ex", config.run_tests_cmd
-    assert_equal "test_model_from_config", config.model
-    assert_equal nil, config.first_question
-    assert config.instructions.include?("Instructions from config file")
-  end
-
-  def test_default_instructions
-    config = Genie::SessionConfig.default
-    default_instructions = config.instructions
-    assert_includes default_instructions, "Genie coding assistant"
-    assert_includes default_instructions, "Test Driven Development"
-    assert_includes default_instructions, "tools available"
-    assert_includes default_instructions, "write tests first"
-    assert_includes default_instructions, "do not have access to any files outside"
-    assert_includes default_instructions, "do not have access to the internet"
-  end
-
-end

data/test/tools/test_list_files.rb

@@ -4,9 +4,7 @@ class TestListFiles < TLDR
   def test_list_files
     actual = Genie::ListFiles.new(base_path: File.expand_path("../../", __dir__)).execute(directory: "./test/data/sample_files")
 
-    expected = [{ name: "read_file_test.txt", type: "file" },
-                { name: "one.txt", type: "file" },
-                { name: "a_dir", type: "directory" }]
+    expected = "/test/data/sample_files/read_file_test.txt\n/test/data/sample_files/one.txt\n/test/data/sample_files/a_dir"
 
     assert_equal expected, actual
   end
@@ -24,8 +22,20 @@ class TestListFiles < TLDR
     t = Genie::ListFiles.new(base_path: base_path)
     actual = t.execute(directory: "./test/data/sample_files", filter: "one")
 
-    expected = [{ name: "one.txt", type: "file" }]
+    expected = "/test/data/sample_files/one.txt"
 
     assert_equal expected, actual
   end
+
+  def test_list_files_with_ignore_paths
+    base_path = File.expand_path("../../", __dir__)
+    t = Genie::ListFiles.new(base_path: base_path, ignore_paths: ["a_dir"])
+
+    actual = t.execute(directory: "./test/data/sample_files")
+
+    expected = "/test/data/sample_files/read_file_test.txt\n/test/data/sample_files/one.txt\n/test/data/sample_files/a_dir"
+
+    assert_equal expected, actual
+  end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: genie_cli
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.4
 platform: ruby
 authors:
 - Jeff McFadden