Update server

Signed-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>

wdi_mcp_server.py

@@ -114,7 +114,9 @@ class DetailedOutput(SearchOutput):
 
 
 @mcp.tool()
-async def search_relevant_indicators(query: str, top_k: int = 1) -> list[SearchOutput]:
+async def search_relevant_indicators(
+    query: str, top_k: int = 1
+) -> dict[str, list[SearchOutput] | str]:
     """Search for a shortlist of relevant indicators from the World Development Indicators (WDI) given the query. The search ranking may not be optimal, so the LLM may use this as shortlist and pick the most relevant from the list (if any).
 
     Args:
@@ -122,13 +124,16 @@ async def search_relevant_indicators(query: str, top_k: int = 1) -> list[SearchO
         top_k: The number of shortlisted indicators that will be returned that are semantically related to the query.
 
     Returns:
-        List of objects with keys indicator code/idno and name.
+        A dictionary with keys `indicators` and `note`. The `indicators` key contains a list of indicator objects with keys indicator code/idno and name. The `note` key contains a note about the search.
     """
 
-    return [
-        SearchOutput(**out)
-        for out in get_top_k(query=query, top_k=top_k, fields=["idno", "name"])
-    ]
+    return {
+        "indicators": [
+            SearchOutput(**out)
+            for out in get_top_k(query=query, top_k=top_k, fields=["idno", "name"])
+        ],
+        "note": "IMPORTANT: Let the user know that the search is not exhaustive. The search is based on the semantic similarity of the query to the indicator definitions. It may not be optimal and the LLM may use this as shortlist and pick the most relevant from the list (if any).",
+    }
 
 
 @mcp.tool()