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

tooluniverse/test/test_dataset_tool.py

@@ -0,0 +1,200 @@
+from tooluniverse import ToolUniverse
+
+# Step 1: Initialize tool universe
+tooluni = ToolUniverse()
+tooluni.load_tools()
+
+test_queries = [
+    # Test 1: Search for a specific drug by name
+    {"name": "drugbank_vocab_search", "arguments": {"query": "aspirin"}},
+    # Test 2: Exact match search for a DrugBank ID
+    {
+        "name": "drugbank_vocab_search",
+        "arguments": {
+            "query": "DB00001",
+            "search_fields": ["DrugBank ID"],
+            "exact_match": True,
+        },
+    },
+    # Test 3: Search in synonyms with case sensitivity
+    {
+        "name": "drugbank_vocab_search",
+        "arguments": {
+            "query": "Lepirudin",
+            "search_fields": ["Synonyms", "Common name"],
+            "case_sensitive": True,
+            "limit": 10,
+        },
+    },
+    # Test 4: Filter drugs that have CAS numbers
+    {
+        "name": "drugbank_vocab_filter",
+        "arguments": {"field": "CAS", "condition": "not_empty", "limit": 5},
+    },
+    # Test 5: Filter drugs starting with "DB000"
+    {
+        "name": "drugbank_vocab_filter",
+        "arguments": {
+            "field": "DrugBank ID",
+            "condition": "starts_with",
+            "value": "DB000",
+            "limit": 10,
+        },
+    },
+    # Test 6: Filter drugs containing "interferon" in common name
+    {
+        "name": "drugbank_vocab_filter",
+        "arguments": {
+            "field": "Common name",
+            "condition": "contains",
+            "value": "interferon",
+            "limit": 5,
+        },
+    },
+    # Test 7: Complex search with multiple fields
+    {
+        "name": "drugbank_vocab_search",
+        "arguments": {
+            "query": "insulin",
+            "search_fields": ["Common name", "Synonyms"],
+            "case_sensitive": False,
+            "exact_match": False,
+            "limit": 3,
+        },
+    },
+    # ---------- drugbank_full_search ----------
+    # 1. Partial common-name query
+    {
+        "name": "drugbank_full_search",
+        "arguments": {
+            "query": "acetylsalicylic",
+            "search_fields": ["name"],
+            "limit": 5,
+        },
+    },
+    # 2. Exact-match lookup by DrugBank ID
+    {
+        "name": "drugbank_full_search",
+        "arguments": {
+            "query": "DB00003",
+            "search_fields": ["drugbank_id"],
+            "exact_match": True,
+        },
+    },
+    # 3. Case-sensitive synonym search
+    {
+        "name": "drugbank_full_search",
+        "arguments": {
+            "query": "Olanzapine",
+            "search_fields": ["synonyms"],
+            "case_sensitive": True,
+            "limit": 3,
+        },
+    },
+    # ---------- drugbank_links_search ----------
+    # 4. Exact CAS-number lookup
+    {
+        "name": "drugbank_links_search",
+        "arguments": {
+            "query": "205923-56-4",
+            "search_fields": ["CAS Number"],
+            "exact_match": True,
+        },
+    },
+    # 5. Search by PharmGKB identifier
+    {
+        "name": "drugbank_links_search",
+        "arguments": {"query": "PA10040", "search_fields": ["PharmGKB ID"]},
+    },
+    # 6. Search by KEGG Drug ID
+    {
+        "name": "drugbank_links_search",
+        "arguments": {"query": "D03455", "search_fields": ["KEGG Drug ID"]},
+    },
+    # ---------- dict_search (DICTrank) ----------
+    # 7. Search by trade name
+    {
+        "name": "dict_search",
+        "arguments": {"query": "ZYPREXA", "search_fields": ["Trade Name"], "limit": 2},
+    },
+    # 8. Search by active ingredient (case-insensitive)
+    {
+        "name": "dict_search",
+        "arguments": {
+            "query": "trimethobenzamide",
+            "search_fields": ["Active Ingredient(s)"],
+            "case_sensitive": False,
+        },
+    },
+    # ---------- dili_search (DILIrank) ----------
+    # 9. Exact compound-name search
+    {
+        "name": "dili_search",
+        "arguments": {
+            "query": "acetaminophen",
+            "search_fields": ["Compound Name"],
+            "exact_match": True,
+        },
+    },
+    # 10. Partial compound-name query
+    {
+        "name": "dili_search",
+        "arguments": {
+            "query": "mercap",
+            "search_fields": ["Compound Name"],
+            "limit": 5,
+        },
+    },
+    # ---------- diqt_search (DIQTA) ----------
+    # 11. Exact DrugBank ID lookup
+    {
+        "name": "diqt_search",
+        "arguments": {
+            "query": "DB00637",
+            "search_fields": ["DrugBank ID"],
+            "exact_match": True,
+        },
+    },
+    # 12. Generic name search (case-insensitive)
+    {
+        "name": "diqt_search",
+        "arguments": {
+            "query": "dofetilide",
+            "search_fields": ["Generic/Proper Name(s)"],
+            "case_sensitive": False,
+        },
+    },
+]
+
+for idx, query in enumerate(test_queries):
+    print(
+        f"\n[{idx+1}] Running tool: {query['name']} with arguments: {query['arguments']}"
+    )
+    try:
+        result = tooluni.run(query)
+        print("✅ Success. Example output snippet:")
+
+        # Pretty print the result based on its structure
+        if isinstance(result, dict):
+            if "results" in result:
+                print(
+                    f"Found {result.get('total_results', result.get('total_matches', 0))} results"
+                )
+                if result["results"]:
+                    # Show first result
+                    first_result = result["results"][0]
+                    # print(f"First result: {first_result.get('Common name', 'N/A')} (ID: {first_result.get('DrugBank ID', 'N/A')})")
+                    print(f"First result: {first_result})")
+                    if "matched_fields" in first_result:
+                        print(f"Matched fields: {first_result['matched_fields']}")
+            elif "error" in result:
+                print(f"Error: {result['error']}")
+            else:
+                print(
+                    str(result)[:200] + "..." if len(str(result)) > 200 else str(result)
+                )
+        else:
+            print(str(result)[:200] + "..." if len(str(result)) > 200 else str(result))
+
+    except Exception as e:
+        print(f"❌ Error: {e}")

tooluniverse/test/test_disease_target_score.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+"""
+Simple test for disease-target score extraction tools.
+"""
+
+import sys
+import os
+
+# Add the src directory to the path
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "src"))
+
+from tooluniverse import ToolUniverse  # noqa: E402
+
+
+def test_disease_target_score_tools():
+    """Test the disease-target score extraction tools"""
+    print("Testing Disease-Target Score Tools...")
+
+    # Initialize ToolUniverse and load all tools
+    engine = ToolUniverse()
+    engine.load_tools()
+
+    # Test both generic and specific tools
+    tests = [
+        (
+            "generic tool",
+            "disease_target_score",
+            {"efoId": "EFO_0000339", "datasourceId": "impc"},
+        ),
+        (
+            "ChEMBL-specific tool",
+            "chembl_disease_target_score",
+            {"efoId": "EFO_0000339"},
+        ),
+    ]
+
+    for name, tool_name, args in tests:
+        print(f"Testing {name}...")
+        result = engine.run_one_function({"name": tool_name, "arguments": args})
+        print(f"\033[92m{result}\033[0m")
+        assert (
+            result and result.get("total_targets_with_scores", 0) > 0
+        ), f"{name} failed"
+        print(f"✅ Found {result['total_targets_with_scores']} targets")
+
+    print("🎉 All tests passed!")
+    return True
+
+
+if __name__ == "__main__":
+    try:
+        test_disease_target_score_tools()
+        print("SUCCESS")
+    except Exception as e:
+        print(f"FAILED: {e}")
+        sys.exit(1)

tooluniverse/test/test_drugbank_filter_examples.py

@@ -0,0 +1,179 @@
+from tooluniverse import ToolUniverse
+from typing import Any, Dict, List
+import json
+
+# Initialize tool universe
+tooluni = ToolUniverse()
+tooluni.load_tools()
+
+print("=== DrugBank Filter Tool Usage Examples ===\n")
+
+# 详细的使用示例，展示如何使用各种过滤器
+filter_examples: List[Dict[str, Any]] = [
+    {
+        "name": "Example 1: Find all insulin-related drugs",
+        "query": {
+            "name": "drugbank_filter",
+            "arguments": {
+                "filters": {"Common name": {"contains": "insulin"}},
+                "limit": 5,
+            },
+        },
+    },
+    {
+        "name": "Example 2: Find drugs with specific DrugBank ID pattern",
+        "query": {
+            "name": "drugbank_filter",
+            "arguments": {
+                "filters": {"DrugBank ID": {"starts_with": "DB001"}},
+                "limit": 5,
+            },
+        },
+    },
+    {
+        "name": "Example 3: Find drugs with CAS numbers (quality check)",
+        "query": {
+            "name": "drugbank_filter",
+            "arguments": {"filters": {"CAS": {"not_empty": True}}, "limit": 3},
+        },
+    },
+    {
+        "name": "Example 4: Find interferon drugs with UNII identifiers",
+        "query": {
+            "name": "drugbank_filter",
+            "arguments": {
+                "filters": {
+                    "Common name": {"contains": "interferon"},
+                    "UNII": {"not_empty": True},
+                },
+                "limit": 3,
+            },
+        },
+    },
+    {
+        "name": "Example 5: Find drugs ending with 'acid'",
+        "query": {
+            "name": "drugbank_filter",
+            "arguments": {
+                "filters": {"Common name": {"ends_with": "acid"}},
+                "limit": 4,
+            },
+        },
+    },
+    {
+        "name": "Example 6: Simple string filter (aspirin in synonyms)",
+        "query": {
+            "name": "drugbank_filter",
+            "arguments": {"filters": {"Synonyms": "aspirin"}, "limit": 3},
+        },
+    },
+    {
+        "name": "Example 7: Exact drug name match",
+        "query": {
+            "name": "drugbank_filter",
+            "arguments": {
+                "filters": {"Common name": {"exact": "Lepirudin"}},
+                "limit": 5,
+            },
+        },
+    },
+    {
+        "name": "Example 8: Complex multi-field filter",
+        "query": {
+            "name": "drugbank_filter",
+            "arguments": {
+                "filters": {
+                    "DrugBank ID": {"starts_with": "DB00"},
+                    "CAS": {"not_empty": True},
+                    "Common name": {"contains": "alfa"},
+                },
+                "limit": 3,
+            },
+        },
+    },
+]
+
+for idx, example in enumerate(filter_examples, 1):
+    print(f"[{idx}] {example['name']}")
+    print("Filter Arguments:")
+    print(json.dumps(example["query"]["arguments"], indent=2, ensure_ascii=False))
+
+    try:
+        result = tooluni.run(example["query"])
+
+        if isinstance(result, dict) and "results" in result:
+            total = result.get("total_matches", 0)
+            returned = result.get("returned_results", 0)
+            print(f"✅ Success: Found {total} matches, showing {returned} results")
+
+            # Show first result details
+            if result["results"]:
+                first = result["results"][0]
+                print(
+                    f"   First result: {first.get('Common name', 'N/A')} (ID: {first.get('DrugBank ID', 'N/A')})"
+                )
+                if first.get("CAS"):
+                    print(f"   CAS: {first['CAS']}")
+                if first.get("UNII"):
+                    print(f"   UNII: {first['UNII']}")
+                if first.get("Synonyms"):
+                    synonyms = (
+                        first["Synonyms"][:100] + "..."
+                        if len(first["Synonyms"]) > 100
+                        else first["Synonyms"]
+                    )
+                    print(f"   Synonyms: {synonyms}")
+
+            # Show applied filters
+            if "applied_filters" in result:
+                print(f"   Applied filters: {', '.join(result['applied_filters'])}")
+
+        elif isinstance(result, dict) and "error" in result:
+            print(f"❌ Error: {result['error']}")
+        else:
+            print("❌ Unexpected result format")
+
+    except Exception as e:
+        print(f"❌ Exception: {e}")
+
+    print("-" * 80)
+
+print("\n=== Filter Usage Summary ===")
+print(
+    """
+Common Filter Patterns:
+
+1. Simple String Matching:
+   {"field_name": "search_text"}
+
+2. Contains Pattern:
+   {"field_name": {"contains": "text"}}
+
+3. Starts With Pattern:
+   {"field_name": {"starts_with": "prefix"}}
+
+4. Ends With Pattern:
+   {"field_name": {"ends_with": "suffix"}}
+
+5. Exact Match:
+   {"field_name": {"exact": "exact_value"}}
+
+6. Data Quality Check:
+   {"field_name": {"not_empty": true}}
+
+7. Multiple Conditions (AND logic):
+   {
+     "field1": {"contains": "text1"},
+     "field2": {"not_empty": true}
+   }
+
+Available Fields:
+- DrugBank ID
+- Common name
+- CAS
+- UNII
+- Synonyms
+- Accession Numbers
+- Standard InChI Key
+"""
+)

tooluniverse/test/test_efo.py

@@ -0,0 +1,31 @@
+from tooluniverse import ToolUniverse
+
+# Step 1: Initialize tool universe
+tooluni = ToolUniverse()
+tooluni.load_tools()
+
+# Step 2: Define test queries for the ChEMBLTool
+test_queries = [
+    {
+        "name": "OSL_get_efo_id_by_disease_name",
+        "arguments": {"disease": "Bardet Biedl syndrome"},
+    },
+    {
+        "name": "OSL_get_efo_id_by_disease_name",
+        "arguments": {"disease": "Bardet-Biedl syndrome"},
+    },
+]
+
+# Step 3: Run all test queries
+for idx, query in enumerate(test_queries):
+    # try:
+    print(
+        f"\n[{idx+1}] Running tool: {query['name']} with arguments: {query['arguments']}"
+    )
+    result = tooluni.run(query)
+    print("✅ Success. Example output snippet:")
+    print(
+        result if isinstance(result, dict) else str(result)
+    )  # Print snippet if result is big
+# except Exception as e:
+#     print(f"❌ Failed. Error: {str(e)}")

tooluniverse/test/test_enrichr_tool.py

@@ -0,0 +1,21 @@
+from tooluniverse import ToolUniverse
+
+# Step 1: Initialize tool universe
+tooluni = ToolUniverse()
+tooluni.load_tools()
+
+#
+test_queries = [
+    {
+        "name": "enrichr_gene_enrichment_analysis",
+        "arguments": {"gene_list": ["TP53", "USP7"]},
+    },
+]
+
+for idx, query in enumerate(test_queries):
+    print(
+        f"\n[{idx+1}] Running tool: {query['name']} with arguments: {query['arguments']}"
+    )
+    result = tooluni.run(query)
+    print("✅ Success. Example output snippet:")
+    print(result if isinstance(result, dict) else str(result))

tooluniverse/test/test_europe_pmc_tool.py

@@ -0,0 +1,20 @@
+from tooluniverse import ToolUniverse
+
+# ...existing code initializing ToolUniverse...
+tooluni = ToolUniverse()
+tooluni.load_tools()
+
+test_queries = [
+    {
+        "name": "EuropePMC_search_articles",
+        "arguments": {"query": "cancer immunotherapy", "limit": 3},
+    },
+]
+
+for idx, query in enumerate(test_queries):
+    print(
+        f"\n[{idx+1}] Running tool: {query['name']} with arguments: {query['arguments']}"
+    )
+    result = tooluni.run(query)
+    print("✅ Success. Example output snippet:")
+    print(result if isinstance(result, dict) else str(result))

tooluniverse/test/test_fda_adv.py

@@ -0,0 +1,95 @@
+# === openfda_test_all_tools.py ===
+
+from tooluniverse import ToolUniverse
+
+# Step 1: Initialize tool universe
+tooluni = ToolUniverse()
+tooluni.load_tools()
+
+# Step 2: Define all test queries for all tools
+test_queries = [
+    # # ====== Tests for FAERS_count_reactions_by_drug_event ======
+    # {"name": "FAERS_count_reactions_by_drug_event", "arguments": {"medicinalproduct": "LIPITOR"}},
+    # {"name": "FAERS_count_reactions_by_drug_event", "arguments": {"medicinalproduct": "LIPITOR", "patient.patientsex": "Male", "serious": "Serious"}},
+    # {"name": "FAERS_count_reactions_by_drug_event", "arguments": {"medicinalproduct": "ASPIRIN", "patient.patientagegroup": "Adult"}},
+    # {"name": "FAERS_count_reactions_by_drug_event", "arguments": {"medicinalproduct": "ADVIL", "patient.patientsex": "Female", "patient.patientagegroup": "Elderly", "occurcountry": "US", "serious": "Non-serious"}},
+    # {"name": "FAERS_count_reactions_by_drug_event", "arguments": {"medicinalproduct": "ADVIL", "seriousnessdeath": "Yes"}},
+    # # ====== Tests for FAERS_count_drugs_by_drug_event ======
+    # {"name": "FAERS_count_drugs_by_drug_event", "arguments": {"patient.patientsex": "Female"}},
+    # {"name": "FAERS_count_drugs_by_drug_event", "arguments": {"patient.patientagegroup": "Child"}},
+    # {"name": "FAERS_count_drugs_by_drug_event", "arguments": {"serious": "Serious"}},
+    # {"name": "FAERS_count_drugs_by_drug_event", "arguments": {"occurcountry": "CA", "serious": "Non-serious"}},
+    # # ====== Tests for FAERS_count_country_by_drug_event ======
+    # {"name": "FAERS_count_country_by_drug_event", "arguments": {"medicinalproduct": "METFORMIN"}},
+    # {"name": "FAERS_count_country_by_drug_event", "arguments": {"medicinalproduct": "INSULIN", "patient.patientsex": "Male"}},
+    # {"name": "FAERS_count_country_by_drug_event", "arguments": {"medicinalproduct": "ASPIRIN", "serious": "Serious"}},
+    # {"name": "FAERS_count_country_by_drug_event", "arguments": {"medicinalproduct": "STATIN", "patient.patientsex": "Female", "patient.patientagegroup": "Adult", "serious": "Non-serious"}},
+    # # ====== Tests for FAERS_count_reportercountry_by_drug_event ======
+    # {"name": "FAERS_count_reportercountry_by_drug_event", "arguments": {"medicinalproduct": "LANTUS"}},
+    # {"name": "FAERS_count_reportercountry_by_drug_event", "arguments": {"medicinalproduct": "AMLODIPINE", "patient.patientsex": "Unknown"}},
+    # {"name": "FAERS_count_reportercountry_by_drug_event", "arguments": {"medicinalproduct": "ATORVASTATIN", "patient.patientagegroup": "Elderly"}},
+    # {"name": "FAERS_count_reportercountry_by_drug_event", "arguments": {"medicinalproduct": "GABAPENTIN", "patient.patientsex": "Male", "patient.patientagegroup": "Adult", "serious": "Serious"}},
+    # # ====== Tests for FAERS_count_seriousness_by_drug_event ======
+    # {"name": "FAERS_count_seriousness_by_drug_event", "arguments": {"medicinalproduct": "NAPROXEN"}},
+    # {"name": "FAERS_count_seriousness_by_drug_event", "arguments": {"medicinalproduct": "ALBUTEROL", "patient.patientsex": "Female"}},
+    # {"name": "FAERS_count_seriousness_by_drug_event", "arguments": {"medicinalproduct": "PREDNISONE", "patient.patientagegroup": "Child"}},
+    # {"name": "FAERS_count_seriousness_by_drug_event", "arguments": {"medicinalproduct": "METHOTREXATE", "patient.patientsex": "Male", "patient.patientagegroup": "Adult", "occurcountry": "US"}},
+    # # ====== Tests for FAERS_count_outcomes_by_drug_event ======
+    # {"name": "FAERS_count_outcomes_by_drug_event", "arguments": {"medicinalproduct": "IBUPROFEN"}},
+    # {"name": "FAERS_count_outcomes_by_drug_event", "arguments": {"medicinalproduct": "IBUPROFEN", "patient.patientsex": "Female", "occurcountry": "GB"}},
+    # {"name": "FAERS_count_outcomes_by_drug_event", "arguments": {"medicinalproduct": "IBUPROFEN", "patient.patientsex": "Male", "patient.patientagegroup": "Adolescent"}},
+    # # ====== Tests for FAERS_count_drug_routes_by_event ======
+    # {"name": "FAERS_count_drug_routes_by_event", "arguments": {"medicinalproduct": "INSULIN", "serious": "Serious"}},
+    # {"name": "FAERS_count_drug_routes_by_event", "arguments": {"medicinalproduct": "MORPHINE", "serious": "Non-serious"}},
+    # # ====== Tests for FAERS_count_patient_age_distribution ======
+    # {"name": "FAERS_count_patient_age_distribution", "arguments": {"medicinalproduct": "PENICILLIN"}},
+    # {"name": "FAERS_count_patient_age_distribution", "arguments": {"medicinalproduct": "ACETAMINOPHEN"}},
+    # ====== Tests for FAERS_count_death_related_by_drug ======
+    # {"name": "FAERS_count_death_related_by_drug", "arguments": {"medicinalproduct": "FENTANYL"}},
+    # {"name": "FAERS_count_death_related_by_drug", "arguments": {"medicinalproduct": "WARFARIN"}},
+    # # ====== Tests for FAERS_count_additive_adverse_reactions ======
+    # {"name": "FAERS_count_additive_adverse_reactions", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"]}},
+    # {"name": "FAERS_count_additive_adverse_reactions", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"], "patient.patientsex": "Male"}},
+    # {"name": "FAERS_count_additive_adverse_reactions", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"], "serious": "Serious"}},
+    # # ====== Tests for FAERS_count_additive_event_reports_by_country ======
+    # {"name": "FAERS_count_additive_event_reports_by_country", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"]}},
+    # {"name": "FAERS_count_additive_event_reports_by_country", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"], "serious": "Non-serious"}},
+    # # ====== Tests for FAERS_count_additive_reports_by_reporter_country ======
+    # {"name": "FAERS_count_additive_reports_by_reporter_country", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"]}},
+    # {"name": "FAERS_count_additive_reports_by_reporter_country", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"], "patient.patientagegroup": "Elderly"}},
+    # # ====== Tests for FAERS_count_additive_seriousness_classification ======
+    # {"name": "FAERS_count_additive_seriousness_classification", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"]}},
+    # {"name": "FAERS_count_additive_seriousness_classification", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"], "occurcountry": "CA"}},
+    # ====== Tests for FAERS_count_additive_reaction_outcomes ======
+    # {"name": "FAERS_count_additive_reaction_outcomes", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"]}},
+    # {"name": "FAERS_count_additive_reaction_outcomes", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"]}},
+    # ====== Tests for FAERS_count_additive_administration_routes ======
+    # {"name": "FAERS_count_additive_administration_routes", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"]}},
+    # {"name": "FAERS_count_additive_administration_routes", "arguments": {"medicinalproducts": ["LIPITOR", "ASPIRIN"], "serious": "Non-serious"}},
+    #            {'name': 'FAERS_count_additive_seriousness_classification', 'arguments': {'medicinalproducts': ['Dysport'], 'occurcountry': 'US', 'patient.patientagegroup': 'Adult'}},
+    # {'name': 'FAERS_count_additive_adverse_reactions', 'arguments': {'medicinalproducts': ['atorvastatin calcium'], 'serious': 'Yes', 'seriousnessdeath': 'Yes'}},
+    {
+        "name": "FAERS_count_additive_adverse_reactions",
+        "arguments": {
+            "medicinalproducts": ["Antara"],
+            "patient.patientsex": "Male",
+            "patient.patientagegroup": "Elderly",
+            "serious": "Yes",
+            "seriousnessdeath": "No",
+        },
+    },
+]
+
+# Step 3: Run all test queries
+for idx, query in enumerate(test_queries):
+    try:
+        print(
+            f"\n[{idx+1}] Running tool: {query['name']} with arguments: {query['arguments']}"
+        )
+        result = tooluni.run(query)
+        print("✅ Success. Example output snippet:")
+        print(
+            result if isinstance(result, dict) else str(result)[:500]
+        )  # Print snippet if result is big
+    except Exception as e:
+        print(f"❌ Failed. Error: {str(e)}")