File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -105,16 +105,6 @@ namespace :evaluation do
105105 raise "Requires an INPUT env var" if ENV [ "INPUT" ] . blank?
106106
107107 result = AutoEvaluation ::TopicTagger . call ( ENV [ "INPUT" ] )
108-
109- # This is a temporary solution to ensure that the evaluation repo topic tagger evaluations continue to work
110- # since primary_topic is a required field on the GenerateInput class. This will be addressed in a follow up bit
111- # of work.
112- if result . status == "error"
113- result = result . to_h . merge (
114- primary_topic : "invalid_tool_output" ,
115- )
116- end
117-
118108 puts ( result . to_json )
119109 end
120110
Original file line number Diff line number Diff line change 431431 . to output ( "#{ result . to_json } \n " ) . to_stdout
432432 end
433433 end
434-
435- context "when the result has an error status" do
436- it "sets the primary_topic to 'invalid_tool_output'" do
437- ClimateControl . modify ( INPUT : input ) do
438- allow ( AutoEvaluation ::TopicTagger ) . to receive ( :call )
439- . with ( input )
440- . and_return (
441- AutoEvaluation ::TopicTagger ::Result . new (
442- status : "error" ,
443- primary_topic : nil ,
444- secondary_topic : nil ,
445- metrics : { } ,
446- llm_response : { } ,
447- error_message : "LLM did not return valid JSON that conformed to the schema." ,
448- ) ,
449- )
450-
451- expected_result = {
452- status : "error" ,
453- primary_topic : "invalid_tool_output" ,
454- secondary_topic : nil ,
455- metrics : { } ,
456- llm_response : { } ,
457- error_message : "LLM did not return valid JSON that conformed to the schema." ,
458- } . to_json
459-
460- expect { Rake ::Task [ task_name ] . invoke }
461- . to output ( "#{ expected_result } \n " ) . to_stdout
462- end
463- end
464- end
465434 end
466435
467436 describe "batch_process" do
You can’t perform that action at this time.
0 commit comments