tooluniverse 1.0.9.1__py3-none-any.whl → 1.0.11__py3-none-any.whl

tooluniverse/worms_tool.py

@@ -0,0 +1,64 @@
+import requests
+import urllib.parse
+from typing import Any, Dict
+from .base_tool import BaseTool
+from .tool_registry import register_tool
+
+
+@register_tool("WoRMSRESTTool")
+class WoRMSRESTTool(BaseTool):
+    def __init__(self, tool_config: Dict):
+        super().__init__(tool_config)
+        self.base_url = "https://www.marinespecies.org/rest"
+        self.session = requests.Session()
+        self.session.headers.update({"Accept": "application/json"})
+        self.timeout = 30
+
+    def _build_url(self, args: Dict[str, Any]) -> str:
+        url = self.tool_config["fields"]["endpoint"]
+        for k, v in args.items():
+            url = url.replace(f"{{{k}}}", str(v))
+        return url
+
+    def run(self, arguments: Dict[str, Any]) -> Dict[str, Any]:
+        try:
+            # Build URL with proper encoding
+            query = arguments.get("query", "")
+            if not query:
+                return {"status": "error", "error": "Query parameter is required"}
+
+            # URL encode the query
+            encoded_query = urllib.parse.quote(query)
+            url = (
+                f"https://www.marinespecies.org/rest/AphiaRecordsByName/{encoded_query}"
+            )
+
+            response = self.session.get(url, timeout=self.timeout)
+            response.raise_for_status()
+
+            # Check if response is empty
+            if not response.text.strip():
+                return {
+                    "status": "success",
+                    "data": [],
+                    "url": url,
+                    "message": "No results found for this query",
+                }
+
+            data = response.json()
+
+            # WoRMS returns array of species, extract first few results
+            if isinstance(data, list) and len(data) > 0:
+                # Limit results to first 5 for better performance
+                limited_data = data[:5]
+                return {
+                    "status": "success",
+                    "data": limited_data,
+                    "url": url,
+                    "count": len(limited_data),
+                    "total_found": len(data),
+                }
+            else:
+                return {"status": "success", "data": data, "url": url}
+        except Exception as e:
+            return {"status": "error", "error": f"WoRMS API error: {str(e)}"}

{tooluniverse-1.0.9.1.dist-info → tooluniverse-1.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tooluniverse
-Version: 1.0.9.1
+Version: 1.0.11
 Summary: A comprehensive collection of scientific tools for Agentic AI, offering integration with the ToolUniverse SDK and MCP Server to support advanced scientific workflows.
 Author-email: Shanghua Gao <shanghuagao@gmail.com>
 Project-URL: Homepage, https://github.com/mims-harvard/ToolUniverse
@@ -29,7 +29,7 @@ Requires-Dist: sentence-transformers>=5.1.0
 Requires-Dist: fitz>=0.0.1.dev2
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: admet-ai>=1.2.0
-Requires-Dist: setuptools>=70.0.0
+Requires-Dist: setuptools<81.0.0,>=70.0.0
 Requires-Dist: pdfplumber>=0.11.0
 Requires-Dist: playwright>=1.55.0
 Requires-Dist: faiss-cpu>=1.12.0
@@ -39,6 +39,7 @@ Requires-Dist: aiohttp
 Requires-Dist: beautifulsoup4>=4.12.0
 Requires-Dist: markitdown[all]>=0.1.0
 Requires-Dist: psutil>=5.9.0
+Requires-Dist: ddgs>=9.0.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.0; extra == "dev"
@@ -86,8 +87,12 @@ Requires-Dist: kaleido>=0.2.1; extra == "visualization"
 Requires-Dist: scipy>=1.7.0; extra == "visualization"
 Requires-Dist: matplotlib>=3.5.0; extra == "visualization"
 Requires-Dist: networkx>=3.4.0; extra == "visualization"
+Provides-Extra: space
+Requires-Dist: huggingface_hub>=0.34.0; extra == "space"
+Requires-Dist: pyyaml>=6.0.0; extra == "space"
+Requires-Dist: requests>=2.32.0; extra == "space"
 Provides-Extra: all
-Requires-Dist: tooluniverse[dev,docs,graph,visualization]; extra == "all"
+Requires-Dist: tooluniverse[dev,docs,graph,space,visualization]; extra == "all"
 Dynamic: license-file
 
 # <img src="docs/_static/logo.png" alt="ToolUniverse Logo" height="28" style="vertical-align: middle; margin-right: 8px;" /> ToolUniverse: Democratizing AI scientists