Update agents.py

agents.py

@@ -398,6 +398,28 @@ def build_graph():
     return graph.compile()
 
 
+AGENT_LABELS = {
+    "search_agent": "🔍 Search agent",
+    "code_agent": "💻 Code agent",
+    "data_agent": "📊 Data agent",
+}
+
+
+def _format_attribution(results: dict) -> str:
+    """Build a deterministic 'which agent did what' footer from state.results.
+    Built in code rather than asked of the model, so attribution can't drift
+    from what actually ran — dict insertion order matches invocation order
+    since specialists write into results as they're called."""
+    if not results:
+        return ""
+    lines = ["", "---", "**Agents used:**"]
+    for name, summary in results.items():
+        label = AGENT_LABELS.get(name, name)
+        preview = summary if len(summary) <= 160 else summary[:157] + "..."
+        lines.append(f"- **{label}** — {preview}")
+    return "\n".join(lines)
+
+
 def run_task(app, task: str, session_state: dict) -> str:
     """Run one task through the graph. session_state is the gr.State dict
     used to persist conversation history across turns in the UI."""
@@ -413,6 +435,7 @@ def run_task(app, task: str, session_state: dict) -> str:
     # round-trips and is what actually prevents runaway loops.
     final_state = app.invoke(initial_state, config={"recursion_limit": 2 * MAX_TURNS + 5})
     answer = final_state["messages"][-1].content
+    answer += _format_attribution(final_state.get("results", {}))
 
     session_state.setdefault("history", []).append({"task": task, "answer": answer})
     return answer