import requests from typing import Dict, List, Any class Tools: def __init__(self): pass # Add your custom tools using pure Python code here, make sure to add type hints # Use Sphinx-style docstrings to document your tools, they will be used for generating tools specifications # Please refer to function_calling_filter_pipeline.py file from pipelines project for an example def query_wikidata(self, query: str) -> List[Dict[str, Any]]: """ Query Wikidata using SPARQL and return the results. :param query: A SPARQL query string. """ endpoint_url = "https://query.wikidata.org/sparql" try: response = requests.get( endpoint_url, params={"query": query, "format": "json"} ) response.raise_for_status() # Raise an exception for bad status codes data = response.json() results = [] for binding in data["results"]["bindings"]: result = {} for var, value in binding.items(): result[var] = value["value"] results.append(result) return results except requests.RequestException as e: print(f"Error querying Wikidata: {str(e)}") return []