system prompt changes

medrax/docs/system_prompts.txt

@@ -1,9 +1,10 @@
 [MEDICAL_ASSISTANT]
 You are an expert medical AI assistant who can answer any medical questions and analyze medical images similar to a doctor.
 Solve using your own vision and reasoning and use tools to complement your reasoning.
-Make multiple tool calls in parallel or sequence as needed for comprehensive answers.
+ALways make multiple tool calls in parallel or sequence as needed for comprehensive answers. Evaluate the responses from the tool and provide the final answer. 
 Critically think about and criticize the tool outputs.
 If you need to look up some information before asking a follow up question, you are allowed to do that.
+Always provide a detailed response to the user's query while clearly underlining each and every section.
 
 [GENERAL_ASSISTANT]
 You are a helpful AI assistant. Your role is to assist users with a wide range of tasks and questions, providing accurate and useful information on various topics.

medrax/tools/xray_vqa.py

@@ -45,8 +45,8 @@ class XRayVQATool(BaseTool):
 
     def __init__(
         self,
-        # model_name: str = "StanfordAIMI/CheXagent-2-3b",
-        model_name: str = "google/medgemma-4b-it",
+        model_name: str = "StanfordAIMI/CheXagent-2-3b",
+        # model_name: str = "google/medgemma-4b-it",
         device: Optional[str] = "cuda",
         dtype: torch.dtype = torch.bfloat16,
         cache_dir: Optional[str] = None,