refactored tools

benchmarking/cli.py

@@ -135,7 +135,7 @@ def main():
                            help="Output directory for results (default: benchmark_results)")
     run_parser.add_argument("--max-questions", type=int, 
                            help="Maximum number of questions to process (default: all)")
-    run_parser.add_argument("--temperature", type=float, default=0.7, 
+    run_parser.add_argument("--temperature", type=float, default=1, 
                            help="Model temperature for response generation (default: 0.7)")
     run_parser.add_argument("--top-p", type=float, default=0.95, 
                            help="Top-p nucleus sampling parameter (default: 0.95)")

medrax/tools/__init__.py

@@ -3,14 +3,11 @@
 from .classification import *
 from .report_generation import *
 from .segmentation import *
-from .xray_vqa import *
-from .llava_med import *
+from .vqa import *
 from .grounding import *
 from .generation import *
 from .dicom import *
 from .utils import *
 from .rag import *
-from .web_browser import *
+from .browsing import *
 from .python_tool import *
-from .medsam2 import *
-

medrax/tools/browsing/__init__.py

@@ -0,0 +1,13 @@
+"""Web browsing tools for MedRAX2 medical agents."""
+
+from .duckduckgo import DuckDuckGoSearchTool, WebSearchInput
+from .web_browser import WebBrowserTool, WebBrowserSchema, SearchQuerySchema, VisitUrlSchema
+
+__all__ = [
+    "DuckDuckGoSearchTool",
+    "WebSearchInput",
+    "WebBrowserTool", 
+    "WebBrowserSchema",
+    "SearchQuerySchema",
+    "VisitUrlSchema"
+] 

medrax/tools/segmentation/__init__.py

@@ -0,0 +1,12 @@
+"""Medical image segmentation tools for MedRAX2."""
+
+from .segmentation import ChestXRaySegmentationTool, ChestXRaySegmentationInput, OrganMetrics
+from .medsam2 import MedSAM2Tool, MedSAM2Input
+
+__all__ = [
+    "ChestXRaySegmentationTool",
+    "ChestXRaySegmentationInput",
+    "OrganMetrics",
+    "MedSAM2Tool",
+    "MedSAM2Input"
+] 

medrax/tools/vqa/__init__.py

@@ -0,0 +1,11 @@
+"""Visual question answering tools for medical images."""
+
+from .xray_vqa import XRayVQATool, XRayVQAToolInput
+from .llava_med import LlavaMedTool, LlavaMedInput
+
+__all__ = [
+    "XRayVQATool",
+    "XRayVQAToolInput",
+    "LlavaMedTool",
+    "LlavaMedInput"
+]