Class: Raif::Admin::ModelCompletionsController

Inherits:
ApplicationController show all
Includes:
Pagy::Backend
Defined in:
app/controllers/raif/admin/model_completions_controller.rb

Instance Method Summary collapse

Methods inherited from Raif::ApplicationController

#raif_current_user

Instance Method Details

#index ⇒ Object 



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

def index
  @selected_status = params[:status].present? ? params[:status].to_sym : :all
  @selected_llm_model_key = params[:llm_model_key].presence
  @llm_model_keys = Raif::ModelCompletion.distinct.order(:llm_model_key).pluck(:llm_model_key)

  model_completions = Raif::ModelCompletion.order(created_at: :desc)

  if @selected_status.present? && @selected_status != :all
    case @selected_status
    when :completed
      model_completions = model_completions.completed
    when :failed
      model_completions = model_completions.failed
    when :started
      model_completions = model_completions.started.where(completed_at: nil, failed_at: nil)
    when :pending
      model_completions = model_completions.where(started_at: nil)
    end
  end

  if @selected_llm_model_key.present?
    model_completions = model_completions.where(llm_model_key: @selected_llm_model_key)
  end

  @pagy, @model_completions = pagy(model_completions)
end

#show ⇒ Object 



35
36
37
 
# File 'app/controllers/raif/admin/model_completions_controller.rb', line 35

def show
  @model_completion = Raif::ModelCompletion.find(params[:id])
end