Update tools/__init__.py

tools/__init__.py

@@ -1,6 +1,19 @@
-from .gemini_tool import GeminiTool
-from .umls_tool import UMLSLookupTool
-from .bioportal_tool import BioPortalLookupTool
-from .quantum_tool import QuantumTreatmentOptimizerTool
+# /home/user/app/tools/__init__.py
 
-__all__ = ["GeminiTool", "UMLSLookupTool", "BioPortalLookupTool", "QuantumTreatmentOptimizerTool"]
+from .bioportal_tool import BioPortalLookupTool, BioPortalInput
+from .gemini_tool import GeminiTool, GeminiInput
+from .umls_tool import UMLSLookupTool, UMLSInput
+# Corrected import for the quantum optimizer tool:
+from .quantum_treatment_optimizer_tool import QuantumTreatmentOptimizerTool, QuantumOptimizerInput
+
+# Optional: Define __all__ to control what `from tools import *` imports
+__all__ = [
+    "BioPortalLookupTool",
+    "BioPortalInput",
+    "GeminiTool",
+    "GeminiInput",
+    "UMLSLookupTool",
+    "UMLSInput",
+    "QuantumTreatmentOptimizerTool",
+    "QuantumOptimizerInput",
+]