Class: SqlGenius::Core::Ai::MigrationRisk

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

Instance Method Summary collapse

Constructor Details

#initialize(client, config, connection) ⇒ MigrationRisk

Returns a new instance of MigrationRisk.



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

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

Instance Method Details

#call(migration_sql) ⇒ Object 



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

def call(migration_sql)
  tables = extract_table_names(migration_sql)
  schema = SchemaContextBuilder.new(@connection).call(tables, detail: :basic)
  schema_text = schema.to_s.empty? ? "Could not determine" : schema

  messages = [
    { role: "system", content: system_prompt },
    { role: "user",   content: "Migration:\n#{migration_sql}\n\nAffected Tables:\n#{schema_text}" },
  ]
  @client.chat(messages: messages)
end