shared_tools 0.4.2 → 0.4.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c72583fc2d491abce837a4c42268770fa5385c96800c28bd846a2b6cd12aa506
-  data.tar.gz: 41aefa6cb69ceea26c66d2c68bc46602d08e824213737ad432bdcf73bf3fef92
+  metadata.gz: 679fe640e745e716a1ec2c9cb77fd9af47b67b09ddd5caba23ef9b6fbbf50d24
+  data.tar.gz: afa308531e5afc205b244771db0ed2e8965e5532d5f642d81c3284819db6f7b6
 SHA512:
-  metadata.gz: 77eb4bb5382359ef3132b376ddeca5f244edf9f4e4eab2d54bf7c9a0eff49825c75a0fbf5efca5ce0f737b799c782d34f3c1f91ac3a1e42dbd9c7d14b6a3cb16
-  data.tar.gz: e342c4d27481ce0bc72631753f64b6dfc047b70d3848b48be21affec7baa0360f46537dfbb86061cac756818df64f93e384c3f1e0048be83aee620ef1e585593
+  metadata.gz: 87264d8b31d6b016e56ac361fa528a73071163f4dfaa4d562e877e20b7d011aa85a0de033212aa326fe9e8983940893d046237eab043c5ee6f61306dc126c21b
+  data.tar.gz: eee3b6364a669539ff7d1ffb290e1255416c0a20f32f68873a4d69061dcf41e5675a23b2ecb3fcacaf7714f6ad0b69d726cf3577642bceef611895574870d44c

data/CHANGELOG.md

@@ -2,6 +2,18 @@
 
 ## Unreleased
 
+### [0.4.4] - 2026-06-06
+
+#### Tools
+- Added `SearchCodebaseTool` (`tools/search_codebase_tool.rb`) — read-only codebase search using ripgrep (`rg`) with grep fallback; supports `path`, `extension`, and `max_results` parameters; returns matches with file path and line number; no authorization prompt required
+- Enhanced `ShellEvalTool`: added optional `workdir` parameter to set the working directory for command execution; validates the directory exists before running
+- Enhanced `ShellEvalTool`: output is now truncated at 5000 bytes (stdout) and 2000 bytes (stderr) to prevent oversized tool responses; truncated output is marked with `[truncated]`
+- Moved `require "open3"` to the top of `ShellEvalTool` instead of inline inside `execute`
+
+#### Tests
+- Added `test/shared_tools/tools/search_codebase_tool_test.rb` — 16 tests covering search, extension filtering, `max_results` capping, truncation flag, error cases, and tool-key reporting
+- Added 4 tests to `test/shared_tools/tools/eval/shell_eval_tool_test.rb` covering `workdir` execution, missing-directory error, stdout truncation, and non-truncation of short output
+
 ### [0.4.2] - 2026-03-27
 
 #### MCP Clients

data/lib/shared_tools/tools/calculator_tool.rb

@@ -1,12 +1,22 @@
 # calculator_tool.rb - Safe mathematical calculator
 require 'ruby_llm/tool'
-require 'dentaku'
+
+DENTAKU_AVAILABLE = begin
+  require 'dentaku'
+  true
+rescue LoadError, Gem::LoadError
+  false
+end
 
 module SharedTools
   module Tools
     class CalculatorTool < RubyLLM::Tool
       def self.name = 'calculator'
 
+      def available?
+        DENTAKU_AVAILABLE
+      end
+
       description <<~'DESCRIPTION'
         Perform advanced mathematical calculations with comprehensive error handling and validation.
         This tool supports basic arithmetic operations, parentheses, and common mathematical functions.

data/lib/shared_tools/tools/database_query_tool.rb

@@ -1,12 +1,22 @@
 # database_query_tool.rb - Safe database query execution
 require 'ruby_llm/tool'
-require 'sequel'
+
+SEQUEL_AVAILABLE = begin
+  require 'sequel'
+  true
+rescue LoadError, Gem::LoadError
+  false
+end
 
 module SharedTools
   module Tools
     class DatabaseQueryTool < RubyLLM::Tool
       def self.name = 'database_query'
 
+      def available?
+        SEQUEL_AVAILABLE
+      end
+
       description <<~'DESCRIPTION'
         Execute safe, read-only database queries with automatic connection management and security controls.
         This tool is designed for secure data retrieval operations only, restricting access to SELECT statements

data/lib/shared_tools/tools/eval/shell_eval_tool.rb

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require "open3"
 
 module SharedTools
   module Tools
@@ -7,13 +8,18 @@ module SharedTools
       # @example
       #   tool = SharedTools::Tools::Eval::ShellEvalTool.new
       #   tool.execute(command: "ls -la")
+      #   tool.execute(command: "ls -la", workdir: "/tmp")
       class ShellEvalTool < ::RubyLLM::Tool
+        STDOUT_MAX = 5000
+        STDERR_MAX = 2000
+
         def self.name = 'eval_shell'
 
         description "Execute a shell command safely and return the result."
 
         params do
           string :command, description: "The shell command to execute"
+          string :workdir, description: "Working directory for the command (default: current directory)", required: false
         end
 
         # @param logger [Logger] optional logger
@@ -22,9 +28,10 @@ module SharedTools
         end
 
         # @param command [String] shell command to execute
+        # @param workdir [String, nil] working directory for the command
         #
         # @return [Hash] execution result
-        def execute(command:)
+        def execute(command:, workdir: nil)
           @logger.info("Requesting permission to execute command: '#{command}'")
 
           if command.strip.empty?
@@ -33,7 +40,12 @@ module SharedTools
             return { error: error_msg }
           end
 
-          # Show user the command and ask for confirmation
+          if workdir
+            unless File.directory?(workdir)
+              return { error: "Working directory not found: #{workdir}" }
+            end
+          end
+
           allowed = SharedTools.execute?(tool: self.class.to_s, stuff: command)
 
           unless allowed
@@ -43,9 +55,11 @@ module SharedTools
 
           @logger.info("Executing command: '#{command}'")
 
-          # Use Open3 for safer command execution with proper error handling
-          require "open3"
-          stdout, stderr, status = Open3.capture3(command)
+          opts = workdir ? { chdir: workdir } : {}
+          stdout, stderr, status = Open3.capture3(command, **opts)
+
+          stdout = truncate(stdout, STDOUT_MAX)
+          stderr = truncate(stderr, STDERR_MAX)
 
           if status.success?
             @logger.debug("Command execution completed successfully with #{stdout.bytesize} bytes of output")
@@ -58,6 +72,13 @@ module SharedTools
           @logger.error("Command execution failed: #{e.message}")
           { error: e.message }
         end
+
+        private
+
+        def truncate(str, max)
+          return str if str.length <= max
+          str.slice(0, max) + "\n[truncated]"
+        end
       end
     end
   end

data/lib/shared_tools/tools/search_codebase_tool.rb

@@ -0,0 +1,91 @@
+# frozen_string_literal: true
+
+require "open3"
+
+module SharedTools
+  module Tools
+    # Search for a term across files using ripgrep (rg) with grep as fallback.
+    # Read-only — no authorization prompt required.
+    #
+    # @example
+    #   tool = SharedTools::Tools::SearchCodebaseTool.new
+    #   tool.execute(term: "def execute")
+    #   tool.execute(term: "RubyLLM", extension: "rb", max_results: 20)
+    class SearchCodebaseTool < ::RubyLLM::Tool
+      MAX_RESULTS_CAP = 500
+
+      def self.name = 'search_codebase'
+
+      description "Search for a term or pattern across files. Uses ripgrep (rg) when available, falls back to grep. Returns matching lines with file path and line number."
+
+      params do
+        string  :term,        description: "The search term or regex pattern to find"
+        string  :path,        description: "Directory to search in (default: current directory)", required: false
+        string  :extension,   description: "File extension to restrict the search to, without the dot (e.g. 'rb', 'js')", required: false
+        integer :max_results, description: "Maximum number of matching lines to return (default: 50, max: 500)", required: false
+      end
+
+      def initialize(logger: nil)
+        @logger = logger || RubyLLM.logger
+      end
+
+      def execute(term:, path: ".", extension: nil, max_results: 50)
+        @logger.info("SearchCodebaseTool#execute term=#{term.inspect} path=#{path} extension=#{extension.inspect}")
+
+        return { error: "Search term cannot be empty" } if term.strip.empty?
+
+        max_results = [[max_results.to_i, 1].max, MAX_RESULTS_CAP].min
+        search_path = File.expand_path(path)
+
+        return { error: "Path not found: #{path}" } unless File.exist?(search_path)
+
+        stdout, _stderr, _status = run_search(term: term, path: search_path, extension: extension)
+
+        all_lines  = stdout.lines.map(&:chomp).reject(&:empty?)
+        matches    = all_lines.first(max_results)
+        truncated  = all_lines.size > max_results
+
+        {
+          matches:   matches,
+          count:     matches.size,
+          truncated: truncated,
+          tool:      ripgrep_available? ? "rg" : "grep"
+        }
+      rescue => e
+        @logger.error("SearchCodebaseTool error: #{e.message}")
+        { error: e.message }
+      end
+
+      private
+
+      def run_search(term:, path:, extension:)
+        if ripgrep_available?
+          run_ripgrep(term: term, path: path, extension: extension)
+        else
+          run_grep(term: term, path: path, extension: extension)
+        end
+      end
+
+      def run_ripgrep(term:, path:, extension:)
+        args = ["rg", "--no-heading", "-n"]
+        args += ["-g", "*.#{extension}"] if extension
+        args << term
+        args << path
+        Open3.capture3(*args)
+      end
+
+      def run_grep(term:, path:, extension:)
+        args = ["grep", "-r", "-n"]
+        args += ["--include=*.#{extension}"] if extension
+        args << term
+        args << path
+        Open3.capture3(*args)
+      end
+
+      def ripgrep_available?
+        return @ripgrep_available unless @ripgrep_available.nil?
+        @ripgrep_available = system("which", "rg", out: File::NULL, err: File::NULL)
+      end
+    end
+  end
+end

data/lib/shared_tools/tools/weather_tool.rb

@@ -1,12 +1,22 @@
 # weather_tool.rb - API integration example
 require 'ruby_llm/tool'
-require 'openweathermap'
+
+OPENWEATHERMAP_AVAILABLE = begin
+  require 'openweathermap'
+  true
+rescue LoadError, Gem::LoadError
+  false
+end
 
 module SharedTools
   module Tools
     class WeatherTool < RubyLLM::Tool
       def self.name = 'weather_tool'
 
+      def available?
+        OPENWEATHERMAP_AVAILABLE
+      end
+
       description <<~'DESCRIPTION'
         Retrieve comprehensive current weather information for any city worldwide using the OpenWeatherMap API.
         This tool provides real-time weather data including temperature, atmospheric conditions, humidity,

data/lib/shared_tools/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SharedTools
-  VERSION = "0.4.2"
+  VERSION = "0.4.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: shared_tools
 version: !ruby/object:Gem::Version
-  version: 0.4.2
+  version: 0.4.4
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -65,6 +65,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: bigdecimal
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
 - !ruby/object:Gem::Dependency
   name: dentaku
   requirement: !ruby/object:Gem::Requirement
@@ -72,7 +86,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-  type: :runtime
+  type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
@@ -86,7 +100,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-  type: :runtime
+  type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
@@ -100,7 +114,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-  type: :runtime
+  type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
@@ -385,6 +399,7 @@ files:
 - lib/shared_tools/tools/notification/mac_driver.rb
 - lib/shared_tools/tools/notification/null_driver.rb
 - lib/shared_tools/tools/notification_tool.rb
+- lib/shared_tools/tools/search_codebase_tool.rb
 - lib/shared_tools/tools/secure_tool_template.rb
 - lib/shared_tools/tools/system_info_tool.rb
 - lib/shared_tools/tools/version.rb
@@ -415,7 +430,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.9
+rubygems_version: 4.0.13
 specification_version: 4
 summary: Shared utilities and AI tools for Ruby applications with configurable logging
 test_files: []