feat: sanitize response by escaping braces to prevent formatting errors & add exception

patientsim/checker.py

@@ -68,8 +68,11 @@ class CheckerAgent:
             raise ValueError(colorstr("red", f"Invalid visiting type: {self.visit_type}. Supported types: {', '.join(VISIT_TYPE)}"))
 
     def __call__(self, response: str, **kwargs) -> str:
+        # Escape braces in `response` before substitution to prevent unintended
+        # format-string expansion if the LLM output contains `{...}` patterns.
+        safe_response = response.replace("{", "{{").replace("}", "}}")
         return self.client(
-            user_prompt=self.prompt_template.format(response=response),
+            user_prompt=self.prompt_template.format(response=safe_response),
             using_multi_turn=False,
             verbose=False,
             temperature=self.temperature,

patientsim/patient.py

@@ -187,7 +187,13 @@ class PatientAgent:
             'reminder': reminder_desc,
             'sentence_limit': sentence_limit,
         }
-        prompt_kwargs.update(self.patient_conditions)
+        # Escape braces in patient data values to prevent accidental format-string
+        # expansion when a field contains text like "{diagnosis}" or "{key}".
+        safe_conditions = {
+            k: str(v).replace("{", "{{").replace("}", "}}")
+            for k, v in self.patient_conditions.items()
+        }
+        prompt_kwargs.update(safe_conditions)
         system_prompt = system_prompt_template.format(**prompt_kwargs)
         prompt_valid_check(system_prompt, self.patient_conditions)
         return system_prompt

patientsim/utils/common_utils.py

@@ -45,7 +45,7 @@ def detect_op_termination(text: str) -> bool:
     try:
         pattern = re.compile(r'Answer:\s*\d+\.\s*(.+)')
         return bool(pattern.search(text))
-    except:
+    except Exception:
         return False
 
 
@@ -54,7 +54,7 @@ def str_to_datetime(iso_time: Union[str, datetime]) -> datetime:
         if isinstance(iso_time, str):
             return datetime.fromisoformat(iso_time)
         return iso_time
-    except:
+    except Exception:
         raise ValueError(colorstr("red", f"`iso_time` must be str or date format, but got {type(iso_time)}"))
 
 
@@ -63,7 +63,7 @@ def datetime_to_str(iso_time: Union[str, datetime], format: str) -> str:
         if not isinstance(iso_time, str):
             return iso_time.strftime(format)
         return iso_time
-    except:
+    except Exception:
         raise ValueError(colorstr("red", f"`iso_time` must be str or date format, but got {type(iso_time)}"))