Class: SharedTools::Tools::DatabaseQueryTool

Inherits:
RubyLLM::Tool
  • Object
show all
Defined in:
lib/shared_tools/tools/database_query_tool.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(logger: nil) ⇒ DatabaseQueryTool

Returns a new instance of DatabaseQueryTool.

Parameters:

  • logger (Logger) (defaults to: nil) —

    optional logger



81
82
83
84
 
# File 'lib/shared_tools/tools/database_query_tool.rb', line 81

def initialize(logger: nil)
  @logger = logger || RubyLLM.logger
  @connection_cache = {}
end

Class Method Details

.name ⇒ Object 



8
 
# File 'lib/shared_tools/tools/database_query_tool.rb', line 8

def self.name = 'database_query'

Instance Method Details

#execute(query:, database: "default", limit: 100, timeout: 30, params: []) ⇒ Hash

Execute read-only database query

Parameters:

  • query (String) —

    SQL SELECT query to execute

  • database (String) (defaults to: "default") —

    Database configuration name

  • limit (Integer) (defaults to: 100) —

    Maximum rows to return (1-10000), default 100

  • timeout (Integer) (defaults to: 30) —

    Query timeout in seconds (1-300), default 30

  • params (Array) (defaults to: []) —

    Parameters for parameterized query

Returns:

  • (Hash) —

    Query results with success status



95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
 
# File 'lib/shared_tools/tools/database_query_tool.rb', line 95

def execute(query:, database: "default", limit: 100, timeout: 30, params: [])
  @logger.info("DatabaseQueryTool#execute database=#{database} limit=#{limit} timeout=#{timeout}")

  begin
    # Validate and sanitize inputs
    validate_query(query)
    limit = validate_limit(limit)
    timeout = validate_timeout(timeout)

    # Get or create database connection (cached for in-memory databases)
    db = get_connection(database, timeout)

    # Add LIMIT clause if not present
    limited_query = add_limit_to_query(query, limit)

    @logger.debug("Executing query: #{limited_query}")
    @logger.debug("With parameters: #{params.inspect}") if params && !params.empty?

    # Execute query with parameters
    start_time = Time.now
    results = if params && !params.empty?
      db[limited_query, *params].all
    else
      db[limited_query].all
    end
    execution_time = Time.now - start_time

    @logger.info("Query executed successfully: #{results.length} rows in #{execution_time.round(3)}s")

    {
      success:        true,
      query:          limited_query,
      row_count:      results.length,
      data:           results,
      database:       database,
      execution_time: execution_time.round(3),
      executed_at:    Time.now.iso8601
    }
  rescue Sequel::DatabaseError => e
    @logger.error("Database error: #{e.message}")
    {
      success:    false,
      error:      "Database error: #{e.message}",
      error_type: "database_error",
      query:      query,
      database:   database
    }
  rescue => e
    @logger.error("Query execution failed: #{e.message}")
    {
      success:    false,
      error:      e.message,
      error_type: e.class.name,
      query:      query,
      database:   database
    }
  end
end