Update app.py

app.py

@@ -50,11 +50,13 @@ os.environ.update({
     "TOKENIZERS_PARALLELISM": "false",
     "CUDA_LAUNCH_BLOCKING": "1"
 })
+
 current_dir = os.path.dirname(os.path.abspath(__file__))
 src_path = os.path.abspath(os.path.join(current_dir, "src"))
 sys.path.insert(0, src_path)
 
 from txagent.txagent import TxAgent
+
 # Initialize cache with 10GB limit
 cache = Cache(file_cache_dir, size_limit=10 * 1024**3)
 
@@ -235,6 +237,7 @@ def clean_response(text: str) -> str:
         (re.compile(r"Retrieving tools.*?\.", re.IGNORECASE), ""),
         (re.compile(r"I will start by retrieving.*?\.", re.IGNORECASE), ""),
         (re.compile(r"This requires reviewing.*?\.", re.IGNORECASE), ""),
+        (re.compile(r"Given the context, it is important to review.*?\.", re.IGNORECASE), ""),
         (re.compile(r"\s+"), " "),
         (re.compile(r"[^\w\s\.\,\(\)\-]"), ""),
         (re.compile(r"(No missed diagnoses identified\.)\s*\1+", re.IGNORECASE), r"\1"),  # Deduplicate
@@ -339,9 +342,6 @@ def init_agent():
         step_rag_num=4,
         seed=100,
         additional_default_tools=[],
-        disable_tools=True,  # Disable tools to avoid unnecessary calls
-        max_retries=2,  # Limit retries to prevent loops
-        max_tokens=4096,  # Increase token limit for complex inputs
     )
     agent.init_model()