WIP major refactor and features

simulator_tests/test_basic_conversation.py

@@ -37,6 +37,7 @@ class BasicConversationTest(BaseSimulatorTest):
                 {
                     "prompt": "Please use low thinking mode. Analyze this Python code and explain what it does",
                     "files": [self.test_files["python"]],
+                    "model": "flash",
                 },
             )
 
@@ -54,6 +55,7 @@ class BasicConversationTest(BaseSimulatorTest):
                     "prompt": "Please use low thinking mode. Now focus on the Calculator class specifically. Are there any improvements you'd suggest?",
                     "files": [self.test_files["python"]],  # Same file - should be deduplicated
                     "continuation_id": continuation_id,
+                    "model": "flash",
                 },
             )
 
@@ -69,6 +71,7 @@ class BasicConversationTest(BaseSimulatorTest):
                     "prompt": "Please use low thinking mode. Now also analyze this configuration file and see how it might relate to the Python code",
                     "files": [self.test_files["python"], self.test_files["config"]],
                     "continuation_id": continuation_id,
+                    "model": "flash",
                 },
             )