gpt41-mini-zero ModelNames keys aligned
Browse files- config/chain_configs.py +3 -3
config/chain_configs.py
CHANGED
|
@@ -26,12 +26,12 @@ from config.llm_config import llms
|
|
| 26 |
|
| 27 |
# Note: The default LLM here is GPT-4o (low temp); the UI can override this choice.
|
| 28 |
|
| 29 |
-
# Model name constants - these match the keys in llm_config.llms
|
| 30 |
class ModelNames:
|
| 31 |
GPT4_ZERO = "GPT-4o (zero temp)"
|
| 32 |
GPT4_LOW = "GPT-4o (low temp)"
|
| 33 |
GPT4_MID = "GPT-4o (mid temp)"
|
| 34 |
-
|
| 35 |
O1_HIGH_REASONING = "o1 (high reasoning_effort)"
|
| 36 |
O3_MINI_HIGH_REASONING = "o3-mini (high reasoning_effort)"
|
| 37 |
|
|
@@ -39,7 +39,7 @@ chain_configs = {
|
|
| 39 |
"diagnoser": {
|
| 40 |
"class": DiagnoserChain,
|
| 41 |
"template_standardize": template_standardize_exercise,
|
| 42 |
-
"llm_standardize": llms[ModelNames.
|
| 43 |
"llm_4o_mini": llms[ModelNames.GPT4_LOW],
|
| 44 |
"llm_4o": llms[ModelNames.GPT4_LOW],
|
| 45 |
# 4 different diagnosis templates (to run in parallel:
|
|
|
|
| 26 |
|
| 27 |
# Note: The default LLM here is GPT-4o (low temp); the UI can override this choice.
|
| 28 |
|
| 29 |
+
# Model name constants - these should match the keys in llm_config.llms
|
| 30 |
class ModelNames:
|
| 31 |
GPT4_ZERO = "GPT-4o (zero temp)"
|
| 32 |
GPT4_LOW = "GPT-4o (low temp)"
|
| 33 |
GPT4_MID = "GPT-4o (mid temp)"
|
| 34 |
+
GPT41_MINI_ZERO = "GPT-4.1-mini (zero temp)"
|
| 35 |
O1_HIGH_REASONING = "o1 (high reasoning_effort)"
|
| 36 |
O3_MINI_HIGH_REASONING = "o3-mini (high reasoning_effort)"
|
| 37 |
|
|
|
|
| 39 |
"diagnoser": {
|
| 40 |
"class": DiagnoserChain,
|
| 41 |
"template_standardize": template_standardize_exercise,
|
| 42 |
+
"llm_standardize": llms[ModelNames.GPT41_MINI_ZERO], # Always fixed
|
| 43 |
"llm_4o_mini": llms[ModelNames.GPT4_LOW],
|
| 44 |
"llm_4o": llms[ModelNames.GPT4_LOW],
|
| 45 |
# 4 different diagnosis templates (to run in parallel:
|