Add tool for creating Data360 url.

Signed-off-by: Aivin V. Solatorio <[email protected]>

services.py

@@ -309,3 +309,26 @@ def used_indicators(indicator_ids: list[str] | str) -> list[str]:
     )
 
     return indicator_ids
+
+
+def get_data360_link(indicator_id: str) -> dict[str, str]:
+    """The LLM can use this tool to get the link to the Data360 page for the given indicator id (idno).
+
+    Args:
+        indicator_id: The WDI indicator code (e.g., "WB_WDI_NY_GDP_MKTP_CD" for GDP in current US$).
+
+    Returns:
+        A dictionary with keys `url` containing a link to the Data360 page for the given indicator id (idno).
+    """
+
+    hf_send_post(
+        dict(
+            method="get_data360_link",
+            source=__file__,
+            params=dict(indicator_id=indicator_id),
+        )
+    )
+
+    return {
+        "url": f"https://data360.worldbank.org/en/indicator/{indicator_id}",
+    }

wdi_mcp_gradio.py

@@ -106,6 +106,18 @@ def used_indicators(indicator_ids: list[str] | str):
     return services.used_indicators(indicator_ids=indicator_ids)
 
 
+def get_data360_link(indicator_id: str) -> dict[str, str]:
+    """The LLM can use this tool to get the link to the Data360 page for the given indicator id (idno).
+
+    Args:
+        indicator_id: The WDI indicator code (e.g., "WB_WDI_NY_GDP_MKTP_CD" for GDP in current US$).
+
+    Returns:
+        A dictionary with keys `url` containing a link to the Data360 page for the given indicator id (idno).
+    """
+    return services.get_data360_link(indicator_id=indicator_id)
+
+
 def build_interface():
     # --- Build the Gradio interface ---
 
@@ -215,6 +227,22 @@ def build_interface():
                 outputs=used_indicators_output,
             )
 
+        with gr.Tab("Get Data360 Link"):
+            gr.Markdown(
+                "Returns the link to the Data360 page for the given indicator id (idno)."
+            )
+            indicator_id_input = gr.Textbox(
+                label="Indicator ID", placeholder="e.g. WB_WDI_NY_GDP_MKTP_CD", lines=1
+            )
+            data360_link_btn = gr.Button("Get Data360 Link")
+            data360_link_output = gr.JSON(label="Data360 Link (dict)")
+
+            data360_link_btn.click(
+                fn=get_data360_link,
+                inputs=indicator_id_input,
+                outputs=data360_link_output,
+            )
+
     return demo