Class: SqlGenius::Core::Ai::RewriteQuery

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

Overview

Suggests a rewritten version of a SQL query based on the schema context of the tables it references.

Instance Method Summary collapse

Constructor Details

#initialize(client, config, connection) ⇒ RewriteQuery

Returns a new instance of RewriteQuery.



9
10
11
12
13
 
# File 'lib/sql_genius/core/ai/rewrite_query.rb', line 9

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

Instance Method Details

#call(sql) ⇒ Object 



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

def call(sql)
  tables = SqlValidator.extract_table_references(sql, @connection)
  schema = SchemaContextBuilder.new(@connection).call(tables, detail: :basic)

  messages = [
    { role: "system", content: system_prompt(schema) },
    { role: "user",   content: sql },
  ]
  @client.chat(messages: messages)
end