Class: Raif::Admin::PromptStudio::TasksController

Inherits:
BaseController show all
Defined in:
app/controllers/raif/admin/prompt_studio/tasks_controller.rb

Instance Method Summary collapse

Methods inherited from Raif::ApplicationController

#raif_current_user

Instance Method Details

#create ⇒ Object 



27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
 
# File 'app/controllers/raif/admin/prompt_studio/tasks_controller.rb', line 27

def create
  original_task = Raif::Task.find(params[:source_task_id])

  unless prompt_studio_runs_enabled?
    redirect_to raif.admin_prompt_studio_task_path(original_task), alert: t("raif.admin.prompt_studio.common.runs_disabled")
    return
  end

  llm_model_key = params[:llm_model_key]

  unless llm_model_key.present? && Raif.available_llm_keys.map(&:to_s).include?(llm_model_key)
    redirect_to raif.admin_prompt_studio_task_path(original_task), alert: t("raif.admin.prompt_studio.tasks.rerun.invalid_model")
    return
  end

  new_task = original_task.class.new(
    creator: original_task.creator,
    source: original_task,
    llm_model_key: llm_model_key,
    available_model_tools: original_task.available_model_tools,
    run_with: original_task.run_with,
    prompt_studio_run: true,
    started_at: Time.current
  )
  new_task.assign_attributes(original_task.prompt_studio_task_attributes)
  new_task.save!
  Raif::PromptStudioTaskRunJob.perform_later(task: new_task)

  redirect_to raif.admin_prompt_studio_task_path(new_task)
rescue StandardError => e
  new_task&.update(failed_at: Time.current) unless new_task&.failed_at?
  redirect_to raif.admin_prompt_studio_task_path(original_task || params[:source_task_id]),
    alert: t("raif.admin.prompt_studio.tasks.rerun.error", message: e.message)
end

#index ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
 
# File 'app/controllers/raif/admin/prompt_studio/tasks_controller.rb', line 7

def index
  @task_types = Raif::Task.distinct.pluck(:type).sort
  @selected_type = params[:task_type] if params[:task_type].present?
  @llm_model_keys = Raif::Task.where(type: @selected_type).distinct.pluck(:llm_model_key).compact.sort if @selected_type.present?

  if @selected_type.present?
    tasks = apply_filters(Raif::Task.where(type: @selected_type).completed).includes(:raif_model_completion).order(created_at: :desc)
    @pagy, @tasks = pagy(tasks)
  end

  @show_batch_runs = prompt_studio_runs_enabled? && @selected_type.present? && @tasks.present?
end

#show ⇒ Object 



20
21
22
23
24
25
 
# File 'app/controllers/raif/admin/prompt_studio/tasks_controller.rb', line 20

def show
  @task = Raif::Task.find(params[:id])
  @comparison = build_prompt_comparison(@task)
  @original_task = @task.source if @task.prompt_studio_run? && @task.source.is_a?(Raif::Task)
  @available_llm_keys = Raif.available_llm_keys.map(&:to_s).sort
end