Class: SqlGenius::Core::Ai::Optimization

Inherits:
Object
  • Object
show all
Defined in:
lib/sql_genius/core/ai/optimization.rb

Overview

Analyses a SQL query + its EXPLAIN output and asks the AI client for optimization suggestions.

Construct with:

connection - a Core::Connection implementation
client     - a Core::Ai::Client
config     - the Core::Ai::Config

Call:

.call(sql, explain_rows, allowed_tables)
  explain_rows - Array of arrays OR a pre-formatted String
  -> Hash with "suggestions" key

Instance Method Summary collapse

Constructor Details

#initialize(connection, client, config) ⇒ Optimization

Returns a new instance of Optimization.



19
20
21
22
23
 
# File 'lib/sql_genius/core/ai/optimization.rb', line 19

def initialize(connection, client, config)
  @connection = connection
  @client = client
  @config = config
end

Instance Method Details

#call(sql, explain_rows, allowed_tables) ⇒ Object 



25
26
27
28
29
30
31
32
33
 
# File 'lib/sql_genius/core/ai/optimization.rb', line 25

def call(sql, explain_rows, allowed_tables)
  schema = build_schema_description(allowed_tables)
  messages = [
    { role: "system", content: system_prompt(schema) },
    { role: "user", content: user_prompt(sql, explain_rows) },
  ]

  @client.chat(messages: messages)
end