tooluniverse 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

tooluniverse/graphql_tool.py

@@ -108,7 +108,7 @@ class OpentargetToolDrugNameMatch(GraphQLTool):
                 print("No drug name found in the arguments.")
                 return None
             drug_name_results = self.drug_generic_tool.run(name_arguments)
-            if 'openfda.generic_name' in drug_name_results:
+            if drug_name_results is not None and 'openfda.generic_name' in drug_name_results:
                 arguments[each_args] = drug_name_results['openfda.generic_name']
                 print("Found generic name. Trying with the generic name: ", arguments[each_args])
                 results = execute_query(endpoint_url=self.endpoint_url, query=self.query_schema, variables=arguments)

tooluniverse/openfda_tool.py

@@ -3,6 +3,7 @@ from .base_tool import BaseTool
 import copy
 import re
 import json
+import os
 
 def check_keys_present(api_capabilities_dict, keys):
     for key in keys:
@@ -219,7 +220,7 @@ class FDATool(BaseTool):
         if self.exists is None:
             self.exists = self.return_fields
         self.endpoint_url = endpoint_url
-        self.api_key = api_key
+        self.api_key = api_key or os.getenv('FDA_API_KEY')
 
     def run(self, arguments):
         arguments = copy.deepcopy(arguments)
@@ -390,7 +391,7 @@ class FDADrugLabelGetDrugGenericNameTool(FDADrugLabelTool):
                 }
         
 
-        from data.fda_drugs_with_brand_generic_names_for_tool import drug_list
+        from .data.fda_drugs_with_brand_generic_names_for_tool import drug_list
         
         self.brand_to_generic = {drug['brand_name']: drug['generic_name'] for drug in drug_list}
         self.generic_to_brand = {drug['generic_name']: drug['brand_name'] for drug in drug_list}

{tooluniverse-0.1.2.dist-info → tooluniverse-0.1.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: tooluniverse
-Version: 0.1.2
+Version: 0.1.4
 Summary: A collection of biomedical tools designed for use by Agentic AI.
 Home-page: https://github.com/mims-harvard/TxAgent
 Author: Shanghua Gao
@@ -11,11 +11,12 @@ License-File: LICENSE
 Requires-Dist: requests
 Requires-Dist: numpy
 Requires-Dist: graphql-core
+Dynamic: license-file
 
 # ToolUniverse
 
 [![ProjectPage](https://img.shields.io/badge/Page-TxAgent-red)](https://zitniklab.hms.harvard.edu/TxAgent)
-[![PaperLink](https://img.shields.io/badge/PDF-TxAgent-red)]()
+[![PaperLink](https://img.shields.io/badge/Arxiv-TxAgent-red)](https://arxiv.org/pdf/2503.10970)
 [![TxAgent-PIP](https://img.shields.io/badge/Pip-TxAgent-blue)](https://pypi.org/project/txagent/)
 [![ToolUniverse-PIP](https://img.shields.io/badge/Pip-ToolUniverse-blue)](https://pypi.org/project/tooluniverse/)
 [![TxAgent](https://img.shields.io/badge/Code-TxAgent-purple)](https://github.com/mims-harvard/TxAgent)
@@ -124,7 +125,15 @@ tooluni.run(query)
 ## Citation
 
 ```
-
+@misc{gao2025txagent,
+      title={TxAgent: An AI Agent for Therapeutic Reasoning Across a Universe of Tools}, 
+      author={Shanghua Gao and Richard Zhu and Zhenglun Kong and Ayush Noori and Xiaorui Su and Curtis Ginder and Theodoros Tsiligkaridis and Marinka Zitnik},
+      year={2025},
+      eprint={2503.10970},
+      archivePrefix={arXiv},
+      primaryClass={cs.AI},
+      url={https://arxiv.org/abs/2503.10970}, 
+}
 ```
 
 ## Contact

{tooluniverse-0.1.2.dist-info → tooluniverse-0.1.4.dist-info}/RECORD

@@ -1,8 +1,8 @@
 tooluniverse/__init__.py,sha256=j44hf13WM50HQJRi7aloeTG6y6872wJtfX8ydsaiJFI,652
 tooluniverse/base_tool.py,sha256=WMGgr02lhEsVzHgt7NIeynXQ2wZtu97yZI-lQWOSoSs,1145
 tooluniverse/execute_function.py,sha256=CmA8AIfyolkcooASD2_3pXgF1BqESU5_VUqdHSYiM_M,10139
-tooluniverse/graphql_tool.py,sha256=FpyP70ZiQkd031crECAhZ7CbXBor8x6S-n48VICnxVs,5086
-tooluniverse/openfda_tool.py,sha256=SPs4skPqwaC2RaZvqD1LA31VeQmMl4bOh7CkTpJKM5U,21017
+tooluniverse/graphql_tool.py,sha256=WU-9YAYbmfbfGwGDn7Y22DVju9t68CKuv_JNz_11NkM,5120
+tooluniverse/openfda_tool.py,sha256=s4ctA7ckID7VA1wRceRGknFV5zu9sR7fqTJ8oa9U0q4,21056
 tooluniverse/restful_tool.py,sha256=5HoTyCSd507D4Qjcqjd16o8Y4BFZ02Zm4cGuczIXWdw,4078
 tooluniverse/utils.py,sha256=zh1IopojPBIJzK_8eGytTmWlPckluQQZTpiCQONw-pY,6416
 tooluniverse/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -11,8 +11,8 @@ tooluniverse/data/fda_drugs_with_brand_generic_names_for_tool.py,sha256=Clb7kQpP
 tooluniverse/data/monarch_tools.json,sha256=I5KvamNZRvkKBaKbM6Ypo333m-RejQhqScX8h-uket4,3567
 tooluniverse/data/opentarget_tools.json,sha256=nNvl-vYqS8nLfyIvHO40Fsnx6HJxwInUuS0mrrh6xmY,66201
 tooluniverse/data/special_tools.json,sha256=tOI0TFpdjZBEE5Af1Vc4CeXQ0gAkz39xho9hDVBYR70,1518
-tooluniverse-0.1.2.dist-info/LICENSE,sha256=Unq9Y3buGL9J0vFop03FYZiVweytt1xdgLIh2n9ck2c,1115
-tooluniverse-0.1.2.dist-info/METADATA,sha256=jIX6oJVszWsBMomzMZjhbPnH2ymvaq8tD6z6BbyOE78,4742
-tooluniverse-0.1.2.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-tooluniverse-0.1.2.dist-info/top_level.txt,sha256=zZ8YeCJ5FAkEwdd_mxsFtSCQMBDgBdxrrmHo3RNBiWs,13
-tooluniverse-0.1.2.dist-info/RECORD,,
+tooluniverse-0.1.4.dist-info/licenses/LICENSE,sha256=Unq9Y3buGL9J0vFop03FYZiVweytt1xdgLIh2n9ck2c,1115
+tooluniverse-0.1.4.dist-info/METADATA,sha256=-qi9jrg4MLNop0QFi9ezycLwM-81Poo0J9CkAD4k7Lg,5220
+tooluniverse-0.1.4.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+tooluniverse-0.1.4.dist-info/top_level.txt,sha256=zZ8YeCJ5FAkEwdd_mxsFtSCQMBDgBdxrrmHo3RNBiWs,13
+tooluniverse-0.1.4.dist-info/RECORD,,

{tooluniverse-0.1.2.dist-info → tooluniverse-0.1.4.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (76.0.0)
+Generator: setuptools (77.0.3)
 Root-Is-Purelib: true
 Tag: py3-none-any