Update agent.py

agent.py

@@ -198,9 +198,11 @@ class FlagRiskInput(BaseModel):
     urgency: str          = Field("High")
 
 # ── Tool Implementations ──────────────────────────────────────────────────────
-@tool("order_lab_test", args_schema=LabOrderInput, description="Place an order for a laboratory test.")
+@tool("order_lab_test", args_schema=LabOrderInput)
 def order_lab_test(test_name: str, reason: str, priority: str = "Routine") -> str:
-    """Place an order for a laboratory test."""
+    """
+    Place an order for a laboratory test.
+    """
     logger.info(f"Ordering lab test: {test_name}, reason: {reason}, priority: {priority}")
     return json.dumps({
         "status": "success",
@@ -208,7 +210,7 @@ def order_lab_test(test_name: str, reason: str, priority: str = "Routine") -> st
         "details": f"Reason: {reason}"
     })
 
-@tool("prescribe_medication", args_schema=PrescriptionInput, description="Prepare a medication prescription.")
+@tool("prescribe_medication", args_schema=PrescriptionInput)
 def prescribe_medication(
     medication_name: str,
     dosage: str,
@@ -217,7 +219,9 @@ def prescribe_medication(
     duration: str,
     reason: str
 ) -> str:
-    """Prepare a medication prescription."""
+    """
+    Prepare a medication prescription.
+    """
     logger.info(f"Preparing prescription: {medication_name} {dosage}, route: {route}, freq: {frequency}")
     return json.dumps({
         "status": "success",
@@ -225,13 +229,15 @@ def prescribe_medication(
         "details": f"Duration: {duration}. Reason: {reason}"
     })
 
-@tool("check_drug_interactions", args_schema=InteractionCheckInput, description="Check for drug–drug interactions and allergy risks.")
+@tool("check_drug_interactions", args_schema=InteractionCheckInput)
 def check_drug_interactions(
     potential_prescription: str,
     current_medications: Optional[List[str]] = None,
     allergies: Optional[List[str]] = None
 ) -> str:
-    """Check for drug–drug interactions and allergy risks."""
+    """
+    Check for drug–drug interactions and allergy risks.
+    """
     logger.info(f"Checking interactions for: {potential_prescription}")
     warnings: List[str] = []
     pm = [m.lower().strip() for m in (current_medications or []) if m]
@@ -266,9 +272,11 @@ def check_drug_interactions(
     )
     return json.dumps({"status": status, "message": message, "warnings": warnings})
 
-@tool("flag_risk", args_schema=FlagRiskInput, description="Flag a clinical risk with given urgency.")
+@tool("flag_risk", args_schema=FlagRiskInput)
 def flag_risk(risk_description: str, urgency: str = "High") -> str:
-    """Flag a clinical risk with given urgency."""
+    """
+    Flag a clinical risk with given urgency.
+    """
     logger.info(f"Flagging risk: {risk_description} (urgency={urgency})")
     return json.dumps({
         "status": "flagged",