class CreateAgentSessions < ActiveRecord::Migration[7.1]
  def change
    create_table :agent_sessions do |t|
      t.integer :session_type, null: false
      t.references :subject, polymorphic: true, null: false, index: false
      t.references :result, polymorphic: true, index: false
      t.references :account, null: false, index: true
      t.references :assistant, null: false, index: true
      t.references :user, index: true
      t.string :llm_model
      t.float :credits_consumed
      t.jsonb :faq_ids, default: []
      t.jsonb :document_ids, default: []
      t.jsonb :scenario_ids, default: []
      t.jsonb :run_context, default: {}

      t.timestamps
    end

    add_index :agent_sessions, [:account_id, :session_type, :created_at]
    add_index :agent_sessions, [:account_id, :subject_type, :subject_id]
    add_index :agent_sessions, [:account_id, :result_type, :result_id]
  end
end