tooluniverse 0.1.0__tar.gz

tooluniverse-0.1.0/PKG-INFO

@@ -0,0 +1,54 @@
+Metadata-Version: 2.2
+Name: tooluniverse
+Version: 0.1.0
+Summary: A collection of biomedical tools designed for use by Agentic AI.
+Home-page: https://github.com/mims-harvard/TxAgent
+Author: Shanghua Gao
+Author-email: shanghuagao@gmail.com
+Requires-Python: >=3.6
+Description-Content-Type: text/markdown
+Requires-Dist: requests
+Requires-Dist: numpy
+Requires-Dist: graphql
+
+# ToolUniverse
+
+ToolUniverse is a collection of biomedical tools designed for use by Agentic AI.
+
+# Install
+
+### Local install
+
+```
+python -m pip install . --no-cache-dir
+```
+
+### Install from PIP
+
+```
+pip install tooluniverse
+```
+
+### Usage
+
+Get all tools
+
+```
+from tooluniverse import ToolUniverse
+tooluni = ToolUniverse()
+tooluni.load_tools()
+tool_name_list, tool_desc_list = tooluni.refresh_tool_name_desc()
+print(tool_name_list)
+print(tool_desc_list)
+```
+
+Function call to a tool
+
+```
+from tooluniverse import ToolUniverse
+tooluni = ToolUniverse()
+tooluni.load_tools()
+tooluni.refresh_tool_name_desc()
+query = {"name": "get_indications_by_drug_name", "arguments": {"drug_name": "KISUNLA"}}
+tooluni.run(query)
+```

tooluniverse-0.1.0/README.md

@@ -0,0 +1,41 @@
+# ToolUniverse
+
+ToolUniverse is a collection of biomedical tools designed for use by Agentic AI.
+
+# Install
+
+### Local install
+
+```
+python -m pip install . --no-cache-dir
+```
+
+### Install from PIP
+
+```
+pip install tooluniverse
+```
+
+### Usage
+
+Get all tools
+
+```
+from tooluniverse import ToolUniverse
+tooluni = ToolUniverse()
+tooluni.load_tools()
+tool_name_list, tool_desc_list = tooluni.refresh_tool_name_desc()
+print(tool_name_list)
+print(tool_desc_list)
+```
+
+Function call to a tool
+
+```
+from tooluniverse import ToolUniverse
+tooluni = ToolUniverse()
+tooluni.load_tools()
+tooluni.refresh_tool_name_desc()
+query = {"name": "get_indications_by_drug_name", "arguments": {"drug_name": "KISUNLA"}}
+tooluni.run(query)
+```

tooluniverse-0.1.0/pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools", "wheel"]
+build-backend = "setuptools.build_meta"

tooluniverse-0.1.0/setup.cfg

@@ -0,0 +1,31 @@
+[metadata]
+name = tooluniverse
+version = 0.1.0
+description = A collection of biomedical tools designed for use by Agentic AI.
+author = Shanghua Gao
+author_email = shanghuagao@gmail.com
+url = https://github.com/mims-harvard/TxAgent
+long_description = file: README.md
+long_description_content_type = text/markdown
+
+[options]
+packages = find:
+package_dir = 
+	= src
+python_requires = >=3.6
+install_requires = 
+	requests
+	numpy
+	graphql
+
+[options.packages.find]
+where = src
+
+[options.package_data]
+tooluniverse = 
+	data/*
+
+[egg_info]
+tag_build = 
+tag_date = 0
+

tooluniverse-0.1.0/src/tooluniverse/__init__.py

@@ -0,0 +1,17 @@
+from .execute_function import ToolUniverse
+from .restful_tool import MonarchTool, MonarchDiseasesForMultiplePhenoTool
+from .graphql_tool import OpentargetTool, OpentargetGeneticsTool, OpentargetToolDrugNameMatch
+from .openfda_tool import FDADrugLabelTool, FDADrugLabelSearchTool, FDADrugLabelSearchIDTool, FDADrugLabelGetDrugGenericNameTool
+
+__all__ = [
+    "ToolUniverse",
+    "MonarchTool",
+    "MonarchDiseasesForMultiplePhenoTool",
+    "OpentargetTool",
+    "OpentargetGeneticsTool",
+    "OpentargetToolDrugNameMatch",
+    "FDADrugLabelTool",
+    "FDADrugLabelSearchTool",
+    "FDADrugLabelSearchIDTool",
+    "FDADrugLabelGetDrugGenericNameTool"
+]

tooluniverse-0.1.0/src/tooluniverse/base_tool.py

@@ -0,0 +1,32 @@
+from .utils import extract_function_call_json, evaluate_function_call
+
+class BaseTool:
+    def __init__(self, tool_config):
+        self.tool_config = tool_config
+
+    def run(self):
+        pass
+
+    def check_function_call(self, function_call_json):
+        if isinstance(function_call_json, str):
+            function_call_json = extract_function_call_json(function_call_json)
+        if function_call_json is not None:
+            return evaluate_function_call(self.tool_config, function_call_json)
+        else:
+            return False, "Invalid JSON string of function call"
+
+    def get_required_parameters(self):
+        """
+        Retrieve required parameters from the endpoint definition.
+        Returns:
+        list: List of required parameters for the given endpoint.
+        """
+        required_params = []
+        parameters = self.tool_config.get('parameter', {}).get('properties', {})
+
+        # Check each parameter to see if it is required
+        for param, details in parameters.items():
+            if details.get('required', False):
+                required_params.append(param.lower())
+
+        return required_params