Class: Raif::Admin::ModelCompletionBatchesController

Inherits:
ApplicationController show all
Defined in:
app/controllers/raif/admin/model_completion_batches_controller.rb

Instance Method Summary collapse

Instance Method Details

#index ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'app/controllers/raif/admin/model_completion_batches_controller.rb', line 6

def index
  @selected_status = params[:status].presence || "all"
  @selected_llm_model_key = params[:llm_model_key].presence
  @selected_type = params[:type].presence

  @llm_model_keys = Raif::ModelCompletionBatch.distinct.order(:llm_model_key).pluck(:llm_model_key)
  @types = Raif::ModelCompletionBatch.distinct.order(:type).pluck(:type).compact

  batches = Raif::ModelCompletionBatch.order(created_at: :desc)

  if @selected_status != "all" && Raif::ModelCompletionBatch::STATUSES.include?(@selected_status)
    batches = batches.where(status: @selected_status)
  end

  batches = batches.where(llm_model_key: @selected_llm_model_key) if @selected_llm_model_key.present?
  batches = batches.where(type: @selected_type) if @selected_type.present?

  @pagy, @model_completion_batches = pagy(batches)

  @completion_counts_by_batch_id = Raif::ModelCompletion
    .where(raif_model_completion_batch_id: @model_completion_batches.map(&:id))
    .group(:raif_model_completion_batch_id)
    .count
end

#show ⇒ Object 



31
32
33
34
 
# File 'app/controllers/raif/admin/model_completion_batches_controller.rb', line 31

def show
  @model_completion_batch = Raif::ModelCompletionBatch.find(params[:id])
  @model_completions = @model_completion_batch.raif_model_completions.order(:id).to_a
end