Skip to content

Commit 1d912e8

Browse files
committed
Add Rake task for context relevancy auto eval
Copies the behaviour of the existing auto eval metrics and adds a Rake task which takes in a question and returns the context relevancy metric for it.
1 parent 03bbc7b commit 1d912e8

2 files changed

Lines changed: 23 additions & 0 deletions

File tree

lib/tasks/evaluation.rake

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -204,4 +204,20 @@ namespace :evaluation do
204204
abort e.message
205205
end
206206
end
207+
208+
desc "Run context relevancy evaluation for a user input"
209+
task generate_context_relevancy_evaluation: :environment do
210+
raise "Requires an INPUT env var" if ENV["INPUT"].blank?
211+
212+
begin
213+
result = AutoEvaluation::EvaluateAnswerFromQuestionMessage.call(
214+
evaluation_class: AutoEvaluation::ContextRelevancy,
215+
question_message: ENV["INPUT"],
216+
)
217+
218+
puts result.to_json
219+
rescue AutoEvaluation::EvaluateAnswerFromQuestionMessage::TaskFailedError => e
220+
abort e.message
221+
end
222+
end
207223
end

spec/lib/tasks/evaluation_spec.rb

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -588,4 +588,11 @@
588588
let(:evaluation_class) { AutoEvaluation::Coherence }
589589
end
590590
end
591+
592+
describe "generate_context_relevancy_evaluation" do
593+
it_behaves_like "a task that returns a ScoreResult" do
594+
let(:task_name) { "evaluation:generate_context_relevancy_evaluation" }
595+
let(:evaluation_class) { AutoEvaluation::ContextRelevancy }
596+
end
597+
end
591598
end

0 commit comments

Comments
 (0)