Class: SqlGenius::Core::Ai::IndexAdvisor

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

Instance Method Summary collapse

Constructor Details

#initialize(client, config, connection) ⇒ IndexAdvisor

Returns a new instance of IndexAdvisor.



7
8
9
10
11
 
# File 'lib/sql_genius/core/ai/index_advisor.rb', line 7

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

Instance Method Details

#call(sql, explain_rows) ⇒ Object 



13
14
15
16
17
18
19
20
21
22
23
 
# File 'lib/sql_genius/core/ai/index_advisor.rb', line 13

def call(sql, explain_rows)
  tables = SqlValidator.extract_table_references(sql, @connection)
  schema = SchemaContextBuilder.new(@connection).call(tables, detail: :with_cardinality)
  explain_text = explain_rows.map { |row| row.join(" | ") }.join("\n")

  messages = [
    { role: "system", content: system_prompt },
    { role: "user",   content: "Query:\n#{sql}\n\nEXPLAIN:\n#{explain_text}\n\nSchema:\n#{schema}" },
  ]
  @client.chat(messages: messages)
end