Class: Leva::DatasetOptimizationsController

Inherits:
ApplicationController show all
Defined in:
app/controllers/leva/dataset_optimizations_controller.rb

Instance Method Summary collapse

Instance Method Details

#create ⇒ void

This method returns an undefined value.

POST /datasets/:dataset_id/optimization Starts the prompt optimization job with progress tracking



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'app/controllers/leva/dataset_optimizations_controller.rb', line 23

def create
  opt_params = optimization_params

  @optimization_run = @dataset.optimization_runs.create!(
    prompt_name: opt_params[:prompt_name],
    mode: opt_params[:mode],
    model: opt_params[:model],
    optimizer: opt_params[:optimizer],
    status: :pending
  )

  PromptOptimizationJob.perform_later(optimization_run_id: @optimization_run.id)

  redirect_to optimization_run_path(@optimization_run)
end

#new ⇒ void

This method returns an undefined value.

GET /datasets/:dataset_id/optimization/new Shows the prompt optimization form



10
11
12
13
14
15
16
17
18
 
# File 'app/controllers/leva/dataset_optimizations_controller.rb', line 10

def new
  @record_count = @dataset.dataset_records.count
  @prompt_optimizer = PromptOptimizer.new(dataset: @dataset)
  @can_optimize = @prompt_optimizer.can_optimize?
  @records_needed = @prompt_optimizer.records_needed
  @modes = PromptOptimizer::MODES
  @models = PromptOptimizer.available_models
  @optimizers = PromptOptimizer::OPTIMIZERS
end