Add tool to echo the indicators used by the LLM

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

services.py

@@ -250,3 +250,23 @@ def get_wdi_data(
             data=_simplify_wdi_data(all_data),
             note=note,
         )
+
+
+def used_indicators(indicator_ids: list[str] | str) -> list[str]:
+    """Returns the list of indicator ids (idno) that have been used by the LLM.
+
+    Args:
+        indicator_ids: A list of indicator ids (idno) that have been used by the LLM.
+
+    Returns:
+        A list of indicator ids (idno) that have been used by the LLM.
+    """
+
+    if isinstance(indicator_ids, str):
+        indicator_ids = indicator_ids.replace(" ", "").split(",")
+
+    hf_send_post(
+        dict(method="used_indicators", params=dict(indicator_ids=indicator_ids))
+    )
+
+    return indicator_ids

wdi_mcp_gradio.py

@@ -70,6 +70,19 @@ def get_wdi_data(indicator_id: str, country_codes_str: str, date: str, per_page:
     )
 
 
+def used_indicators(indicator_ids: list[str] | str):
+    """Returns the list of indicator ids (idno) that have been used by the LLM.
+
+    Args:
+        indicator_ids: A list of indicator ids (idno) that have been used by the LLM.
+
+    Returns:
+        A list of indicator ids (idno) that have been used by the LLM.
+    """
+
+    return services.used_indicators(indicator_ids=indicator_ids)
+
+
 def build_interface():
     # --- Build the Gradio interface ---
 
@@ -161,6 +174,24 @@ def build_interface():
                 outputs=data_output,
             )
 
+        with gr.Tab("Used Indicators"):
+            gr.Markdown(
+                "Returns the list of indicator ids (idno) that have been used by the LLM."
+            )
+            indicator_ids_input = gr.Textbox(
+                label="Indicator IDs",
+                placeholder="e.g. NY.GDP.MKTP.CD, SP.POP.TOTL",
+                lines=1,
+            )
+            used_indicators_btn = gr.Button("Get Used Indicators")
+            used_indicators_output = gr.JSON(label="Used Indicators (list)")
+
+            used_indicators_btn.click(
+                fn=used_indicators,
+                inputs=indicator_ids_input,
+                outputs=used_indicators_output,
+            )
+
     return demo