LLMによる自然言語を用いたDB問い合わせ機能の実装例

VertexAiApi::Gemini.new(
    model: "gemini-2.5-pro",
    temperature: 0,
    thinking_budget: -1, # 動的思考をオンにする
    max_output_tokens: 2048,
    fallback_models: ["gemini-2.5-flash"],
    json: true,
    response_schema: response_schema
)

  def response_schema
    {
      type: "object",
      properties: {
        sql_query: {
          type: "string",
          description: "生成されたSQL SELECTクエリ"
        },
        explanation: {
          type: "string",
          description: "クエリが何をするかの人間が読める説明"
        },
        confidence: {
          type: "integer",
          minimum: 0,
          maximum: 100,
          description: "生成されたSQLの信頼度 (0-100)"
        },
        warnings: {
          type: "array",
          items: { type: "string" },
          description: "クエリに関する警告や制限事項"
        }
      },
      required: ["sql_query", "explanation", "confidence"],
      additionalProperties: false
    }
  end

あなたはPostgreSQLのエキスパートで、自然言語から安全で正確なSELECTクエリを生成します。

重要な安全ルール:
- SELECT文のみを生成する
- INSERT、UPDATE、DELETE、DROP、CREATE、ALTER、TRUNCATEは絶対に使用しない
- システムテーブル (pg_*、information_schema) にはアクセスしない
- 結果は最大1000行に制限する
- テーブルを結合する際は適切なJOIN構文を使用する
- 可能な場合は常にパラメータ化クエリを使用する

データベーススキーマ:
#{format_schema_context(schema_context)}

ユーザークエリ: "#{query}"

この質問に答えるPostgreSQL SELECTクエリを生成してください。以下を含めてください:
1. SQLクエリ (SELECTのみ)
2. クエリが何をするかの明確な説明
3. 信頼度 (0-100)
4. 警告や制限事項

指定されたJSON形式で応答してください。

class SchemaContextGenerator
  # ActiveRecordのイントロスペクションを使用してデータベーススキーマコンテキストを動的に生成
  # これは、自然言語からSQLへの変換に使用するために、
  # すべてのテーブルとその関係の完全な概要を提供します

  def self.generate_context
    {
      tables: generate_tables_info,
      relationships: generate_relationships,
    }
  end

  def self.generate_tables_info
    ActiveRecord::Base.connection.tables.filter_map do |table_name|
      # Rails内部テーブルとシステムテーブルをスキップ
      next if skip_table?(table_name)

      columns = ActiveRecord::Base.connection.columns(table_name)

      {
        name: table_name,
        columns: columns.filter_map { |col| format_column(col) },
      }
    end
  end

  def self.generate_relationships
    relationships = {}

    ActiveRecord::Base.connection.tables.each do |table_name|
      next if skip_table?(table_name)

      foreign_keys = ActiveRecord::Base.connection.foreign_keys(table_name)
      relationships[table_name] = foreign_keys.map do |fk|
        {
          column: fk.column,
          references_table: fk.to_table,
          references_column: fk.primary_key,
          name: fk.name
        }
      end
    end

    relationships
  end

  def self.skip_table?(table_name)
    # Rails内部テーブルとシステムテーブルをスキップ
    internal_tables = %w[
      schema_migrations
      ar_internal_metadata
    ]

    internal_tables.include?(table_name)
  end

  def self.format_column(column)
    # クエリで公開すべきでない機密列をスキップ
    sensitive_columns = %w[
      encrypted_password
    ]

    return nil if sensitive_columns.any? { |sensitive| column.name.include?(sensitive) }

    {
      name: column.name,
      type: column.type.to_s,
      null: column.null,
      default: column.default,
      comment: column.comment
    }
  end
end

Table: articles
  Columns:
    id: integer
    user_id: integer
    slug: string
    title: string (nullable)
    ...

Table: ...
  Columns:
    ...

RELATIONSHIPS:
  articles.user_id -> users.id
  ...