genie_cli 0.2.1 → 0.2.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f9431dda686ce68a45532465d2d0f75baabb62e7dee4dc5278d51cdbd0cf00b7
-  data.tar.gz: 86a77baeb167e1bdf058d020368a9069599f44c69fee4d290830bc02fb83cf96
+  metadata.gz: 9b8833039cb511082479f5b8cbe3dee54620d908900ecbcd908c13f8c99992d4
+  data.tar.gz: f058b13fc6b3990c169d44b83119c4d4ce282dde8c7f4ae5de790f188e8dcf3b
 SHA512:
-  metadata.gz: 8cc795dbe839e79b4e2d859051f2bb7c7fdcbaad70c50d17ded96d581aca5fe15cf2482c6d13fd39657f35872e0f997deb7a1bda4f54ad9b89c619600508cffa
-  data.tar.gz: d9a87bb13bfd5e679baa1934904c348e6e957b7737211c06992d85439e1740cc52e2be77cd5820fd47279d5c04ae5c4582266e6df29afa906e812b8206f3879d
+  metadata.gz: 6b281e4cbfde5084204a0cfca381a018d707c8e08dfc006e088c7dc7544df57b19710fe8861a616c9cb2cd2e375489a88f553a90b3db2798efc745e0b52af1e3
+  data.tar.gz: f6d58af389ae61f318b97d89d6fae5c24a9f494d571c86b94a93820fff1e032d81f103298abad5ba6c3c8c98281569e75e56af7d5d3276f2d99cdba2e90e44ad

data/lib/genie/session.rb

@@ -8,7 +8,7 @@ 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
@@ -20,6 +20,8 @@ module Genie
       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)
@@ -32,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/version.rb

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

data/lib/genie.rb

@@ -1,3 +1,4 @@
+require "yaml"
 require "ruby_llm"
 require "dotenv/load"
 require "tty-command"

data/lib/tools/list_files.rb

@@ -4,7 +4,7 @@ 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)"
 
@@ -15,7 +15,7 @@ module Genie
       @ignore_paths = ignore_paths
     end
 
-    def execute(directory:, recursive: false, filter: nil)
+    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
@@ -26,15 +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
 
       # Apply ignore paths
-      listing.reject! { |entry| @ignore_paths.any? { |ignore_path| entry[:name].gsub(@base_path, '').start_with?(ignore_path) } }
+      listing.reject! { |entry| @ignore_paths.any? { |ignore_path| entry.start_with?(ignore_path) } }
 
-      Genie.output listing.map { |e| e[:name] }.join("\n") + "\n", color: :green
+      Genie.output listing.join("\n") + "\n", color: :green
 
-      listing
+      listing.join("\n")
     rescue => e
       Genie.output "Error: #{e.message}", color: :red
       { error: e.message }
@@ -45,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
 
@@ -59,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/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,7 +22,7 @@ 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
@@ -35,8 +33,7 @@ class TestListFiles < TLDR
 
     actual = t.execute(directory: "./test/data/sample_files")
 
-    expected = [{ name: "read_file_test.txt", type: "file" },
-                { name: "one.txt", type: "file" }]
+    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

metadata

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