search_entity
Find Wikidata entity IDs by entering search queries to retrieve structured data identifiers for entities like people, places, and concepts.
Instructions
Search for a Wikidata entity ID by its query.
Args:
query (str): The query to search for. The query should be unambiguous enough to uniquely identify the entity.
Returns:
str: The Wikidata entity ID corresponding to the given query."
Input Schema
TableJSON Schema
| Name | Required | Description | Default |
|---|---|---|---|
| query | Yes |
Implementation Reference
- src/server.py:38-49 (handler)The handler function for the 'search_entity' tool, decorated with @server.tool() for registration. It invokes the helper function search_wikidata to perform the Wikidata entity search.@server.tool() async def search_entity(query: str) -> str: """ Search for a Wikidata entity ID by its query. Args: query (str): The query to search for. The query should be unambiguous enough to uniquely identify the entity. Returns: str: The Wikidata entity ID corresponding to the given query." """ return await search_wikidata(query, is_entity=True)
- src/server.py:12-35 (helper)Helper function that executes the HTTP request to Wikidata's search API to find entity or property IDs based on the query.async def search_wikidata(query: str, is_entity: bool = True) -> str: """ Search for a Wikidata item or property ID by its query. """ params = { "action": "query", "list": "search", "srsearch": query, "srnamespace": 0 if is_entity else 120, "srlimit": 1, # TODO: add a parameter to limit the number of results? "srqiprofile": "classic_noboostlinks" if is_entity else "classic", "srwhat": "text", "format": "json", } async with httpx.AsyncClient() as client: response = await client.get(WIKIDATA_URL, headers=HEADER, params=params) response.raise_for_status() try: title = response.json()["query"]["search"][0]["title"] title = title.split(":")[-1] return title except KeyError: return "No results found. Consider changing the search term."