gpt41-mini-zero ModelNames keys aligned again
Browse files- config/chain_configs.py +4 -4
config/chain_configs.py
CHANGED
|
@@ -55,7 +55,7 @@ chain_configs = {
|
|
| 55 |
"distractors": {
|
| 56 |
"class": DistractorsChain,
|
| 57 |
"template_standardize": template_standardize_exercise,
|
| 58 |
-
"llm_standardize": llms[ModelNames.
|
| 59 |
"template_distractors_brainstorm_1": template_distractors_brainstorm_1,
|
| 60 |
"template_distractors_brainstorm_2": template_distractors_brainstorm_2,
|
| 61 |
"llm_brainstorm_1": llms[ModelNames.GPT4_LOW],
|
|
@@ -66,13 +66,13 @@ chain_configs = {
|
|
| 66 |
"learning_objectives": {
|
| 67 |
"class": LearningObjectivesChain,
|
| 68 |
"template_standardize": template_standardize_studytext,
|
| 69 |
-
"llm_standardize": llms[ModelNames.
|
| 70 |
"template_gen_prompt_a": template_gen_prompt_a,
|
| 71 |
"template_gen_prompt_b": template_gen_prompt_b,
|
| 72 |
"default_llm_a": llms[ModelNames.O1_HIGH_REASONING],
|
| 73 |
"default_llm_b": llms[ModelNames.O3_MINI_HIGH_REASONING],
|
| 74 |
"template_sanitize": template_sanitize_learning_objectives,
|
| 75 |
-
"llm_sanitize": llms[ModelNames.
|
| 76 |
},
|
| 77 |
"fluster": {
|
| 78 |
"class": FlusterWritingChain,
|
|
@@ -84,7 +84,7 @@ chain_configs = {
|
|
| 84 |
"template_refine_fluster": template_refine_fluster,
|
| 85 |
"llm_refine": llms[ModelNames.GPT4_ZERO],
|
| 86 |
"template_sanitize": template_sanitize_fluster,
|
| 87 |
-
"llm_sanitize": llms[ModelNames.
|
| 88 |
"template_structurize": template_isolate_exercises,
|
| 89 |
"llm_structurize": llms[ModelNames.GPT4_ZERO],
|
| 90 |
"template_fix_exercise": template_fix_exercise,
|
|
|
|
| 55 |
"distractors": {
|
| 56 |
"class": DistractorsChain,
|
| 57 |
"template_standardize": template_standardize_exercise,
|
| 58 |
+
"llm_standardize": llms[ModelNames.GPT41_MINI_ZERO], # Always fixed
|
| 59 |
"template_distractors_brainstorm_1": template_distractors_brainstorm_1,
|
| 60 |
"template_distractors_brainstorm_2": template_distractors_brainstorm_2,
|
| 61 |
"llm_brainstorm_1": llms[ModelNames.GPT4_LOW],
|
|
|
|
| 66 |
"learning_objectives": {
|
| 67 |
"class": LearningObjectivesChain,
|
| 68 |
"template_standardize": template_standardize_studytext,
|
| 69 |
+
"llm_standardize": llms[ModelNames.GPT41_MINI_ZERO], # Always fixed
|
| 70 |
"template_gen_prompt_a": template_gen_prompt_a,
|
| 71 |
"template_gen_prompt_b": template_gen_prompt_b,
|
| 72 |
"default_llm_a": llms[ModelNames.O1_HIGH_REASONING],
|
| 73 |
"default_llm_b": llms[ModelNames.O3_MINI_HIGH_REASONING],
|
| 74 |
"template_sanitize": template_sanitize_learning_objectives,
|
| 75 |
+
"llm_sanitize": llms[ModelNames.GPT41_MINI_ZERO],
|
| 76 |
},
|
| 77 |
"fluster": {
|
| 78 |
"class": FlusterWritingChain,
|
|
|
|
| 84 |
"template_refine_fluster": template_refine_fluster,
|
| 85 |
"llm_refine": llms[ModelNames.GPT4_ZERO],
|
| 86 |
"template_sanitize": template_sanitize_fluster,
|
| 87 |
+
"llm_sanitize": llms[ModelNames.GPT41_MINI_ZERO],
|
| 88 |
"template_structurize": template_isolate_exercises,
|
| 89 |
"llm_structurize": llms[ModelNames.GPT4_ZERO],
|
| 90 |
"template_fix_exercise": template_fix_exercise,
|