BtB-ExpC commited on
Commit
f7e044d
·
1 Parent(s): eec7865

gpt41-mini-zero ModelNames keys aligned

Browse files
Files changed (1) hide show
  1. 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
- GPT4_MINI_ZERO = "GPT-4o-mini (zero temp)"
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.GPT4_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:
 
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: