Update crs_arena/battle_manager.py

crs_arena/battle_manager.py

@@ -14,21 +14,34 @@ from utils import get_crs_model
 
 # CRS models with their configuration files.
 CRS_MODELS = {
+    "crbcrs_redial": "data/arena/crs_config/CRB_CRS/crb_crs_redial.yaml",
     "kbrd_redial": "data/arena/crs_config/KBRD/kbrd_redial.yaml",
-    "kbrd_opendialkg": "data/arena/crs_config/KBRD/kbrd_opendialkg.yaml",
     "unicrs_redial": "data/arena/crs_config/UniCRS/unicrs_redial.yaml",
+    "kbrd_opendialkg": "data/arena/crs_config/KBRD/kbrd_opendialkg.yaml",
+    "chatgpt_redial": "data/arena/crs_config/ChatGPT/chatgpt_redial.yaml",
+    "barcor_opendialkg": "data/arena/crs_config/BARCOR/barcor_opendialkg.yaml",
     "unicrs_opendialkg": "data/arena/crs_config/UniCRS/unicrs_opendialkg.yaml",
     "barcor_redial": "data/arena/crs_config/BARCOR/barcor_redial.yaml",
-    "barcor_opendialkg": "data/arena/crs_config/BARCOR/barcor_opendialkg.yaml",
-    "chatgpt_redial": "data/arena/crs_config/ChatGPT/chatgpt_redial.yaml",
     "chatgpt_opendialkg": (
         "data/arena/crs_config/ChatGPT/chatgpt_opendialkg.yaml"
     ),
-    "crbcrs_redial": "data/arena/crs_config/CRB_CRS/crb_crs_redial.yaml",
 }
 
 CONVERSATION_COUNTS = defaultdict(int).fromkeys(CRS_MODELS.keys(), 0)
 
+# Temporary
+CONVERSATION_COUNTS.update(
+    {
+        "kbrd_redial": 7,
+        "kbrd_opendialkg": 7,
+        "unicrs_redial": 4,
+        "unicrs_opendialkg": 3,
+        "barcor_redial": 2,
+        "barcor_opendialkg": 2,
+    }
+)
+
+
 
 def get_crs_fighters() -> Tuple[CRSFighter, CRSFighter]:
     """Selects two CRS models for a battle.