Update args limit

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

mcp_client.py

@@ -39,6 +39,8 @@ SYSTEM_PROMPT = f"""You are a helpful assistant. Today is {datetime.now().strfti
 
 You **do not** have prior knowledge of the World Development Indicators (WDI) data. Instead, you must rely entirely on the tools available to you to answer the user's questions.
 
+When responding you must always plan the steps and enumerate all the tools that you plan to use to answer the user's query.
+
 ### Your Instructions:
 
 1. **Tool Use Only**:

wdi_mcp_server.py

@@ -13,11 +13,11 @@ mcp = FastMCP("wdi_data_mcp")
 async def search_relevant_indicators(
     query: str, top_k: int = 1
 ) -> dict[str, list[services.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).
+    """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). You, as an LLM, must always get at least `top_k=20` for better recall.
 
     Args:
         query: The search query by the user or one formulated by an LLM based on the user's prompt.
-        top_k: The number of shortlisted indicators that will be returned that are semantically related to the query.
+        top_k: The number of shortlisted indicators that will be returned that are semantically related to the query. IMPORTANT: You, as an LLM, must ALWAYS set this argument to at least 20.
 
     Returns:
         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.