genie_cli 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ae88ad33211ea966ef1f55bf6da8497ba8f669e46bec8ed5bf51e70dfc62b0a8
-  data.tar.gz: 4f349cad129aee7c7f8e336a8f7e2846c13caf74a2381bef1e5797b683dbacaa
+  metadata.gz: f9431dda686ce68a45532465d2d0f75baabb62e7dee4dc5278d51cdbd0cf00b7
+  data.tar.gz: 86a77baeb167e1bdf058d020368a9069599f44c69fee4d290830bc02fb83cf96
 SHA512:
-  metadata.gz: 6c6dd37077b01500c8bed0e33180cc36ec706d4bbeb66b0d8dbe7e15ac4366f8166d681c3635151ecaa5de724aede4ed82e17b6ab340824140d36e0d7b174e9d
-  data.tar.gz: 5b8d9186b2528f3635d825842e6b98c8b7594e2e69fd680a417d5be231e121d630982d048766d2b4a3840cb16d1a0e695677a24086b407ab162d8db9b25980d0
+  metadata.gz: 8cc795dbe839e79b4e2d859051f2bb7c7fdcbaad70c50d17ded96d581aca5fe15cf2482c6d13fd39657f35872e0f997deb7a1bda4f54ad9b89c619600508cffa
+  data.tar.gz: d9a87bb13bfd5e679baa1934904c348e6e957b7737211c06992d85439e1740cc52e2be77cd5820fd47279d5c04ae5c4582266e6df29afa906e812b8206f3879d

data/lib/genie/session.rb

@@ -15,9 +15,11 @@ module Genie
     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
 
       # Initialize the LLM chat with the specified model
       @chat = RubyLLM.chat(model: model)

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.1'
 end

data/lib/genie.rb

@@ -32,9 +32,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

@@ -8,6 +8,13 @@ module Genie
     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 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)
 
@@ -22,6 +29,9 @@ module Genie
         listing = listing.select { |entry| entry[:name].include?(filter) }
       end
 
+      # Apply ignore paths
+      listing.reject! { |entry| @ignore_paths.any? { |ignore_path| entry[:name].gsub(@base_path, '').start_with?(ignore_path) } }
+
       Genie.output listing.map { |e| e[:name] }.join("\n") + "\n", color: :green
 
       listing

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

@@ -28,4 +28,17 @@ class TestListFiles < TLDR
 
     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 = [{ name: "read_file_test.txt", type: "file" },
+                { name: "one.txt", type: "file" }]
+
+    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.1
 platform: ruby
 authors:
 - Jeff McFadden