Update models/loader.py

models/loader.py

@@ -2,15 +2,14 @@
 import torch
 import os
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
+from backend.agents import ROLE_PROMPTS
 
-# Configuration for loading models
 QUANTIZATION_CONFIG = BitsAndBytesConfig(
     load_in_4bit=True,
     bnb_4bit_quant_type="nf4",
     bnb_4bit_compute_dtype=torch.bfloat16,
 )
 
-# Use a registry to map agent roles to specific models
 MODEL_REGISTRY = {
     "ceo": "Qwen/Qwen3-0.6B",
     "planner": "Qwen/Qwen3-0.6B",
@@ -21,6 +20,7 @@ MODEL_REGISTRY = {
     "worker_backend_coder": "Qwen/Qwen3-0.6B",
     "worker_front_end_coder": "Qwen/Qwen3-0.6B",
     "worker_tester": "Qwen/Qwen3-0.6B",
+    "code_analyst": "Qwen/Qwen3-0.6B"
 }
 _MODEL_CACHE = {}