riazmo commited on
Commit
8fdc878
·
verified ·
1 Parent(s): 1f495e7

Upload settings.py

Browse files
Files changed (1) hide show
  1. config/settings.py +9 -5
config/settings.py CHANGED
@@ -167,11 +167,15 @@ class Settings:
167
  "advisor": self.models.agent3_model,
168
  "generator": self.models.agent4_model,
169
 
170
- # Stage 2 New Architecture agents
171
- "brand_identifier": self.models.agent3_model, # Llama 70B - needs context understanding
172
- "benchmark_advisor": self.models.agent3_model, # Llama 70B - needs reasoning
173
- "best_practices_validator": self.models.agent3_model, # Llama 70B - needs judgment
174
- "head_synthesizer": self.models.agent3_model, # Llama 70B - needs synthesis
 
 
 
 
175
  "benchmark_extractor": self.models.agent2_model, # Phi-3.5 - structured extraction
176
  }
177
  return model_map.get(agent_name, self.models.fallback_model)
 
167
  "advisor": self.models.agent3_model,
168
  "generator": self.models.agent4_model,
169
 
170
+ # Stage 2 New Architecture agents — optimized model per role
171
+ # AURORA: Creative/visual reasoning for brand color analysis
172
+ "brand_identifier": os.getenv("BRAND_IDENTIFIER_MODEL", "Qwen/Qwen2.5-72B-Instruct"),
173
+ # ATLAS: 128K context for large benchmark data, strong comparative reasoning
174
+ "benchmark_advisor": os.getenv("BENCHMARK_ADVISOR_MODEL", "meta-llama/Llama-3.3-70B-Instruct"),
175
+ # SENTINEL: Methodical rule-following, precise judgment
176
+ "best_practices_validator": os.getenv("BEST_PRACTICES_MODEL", "Qwen/Qwen2.5-72B-Instruct"),
177
+ # NEXUS: 128K context for combined inputs, strong synthesis
178
+ "head_synthesizer": os.getenv("HEAD_SYNTHESIZER_MODEL", "meta-llama/Llama-3.3-70B-Instruct"),
179
  "benchmark_extractor": self.models.agent2_model, # Phi-3.5 - structured extraction
180
  }
181
  return model_map.get(agent_name, self.models.fallback_model)