From 8e16d39e7201272d102c650e711d50df8485eadf Mon Sep 17 00:00:00 2001
From: moe-berri <moe@berri.ai>
Date: Tue, 6 Oct 2026 14:44:12 -0700
Subject: [PATCH] fix(lens): refresh content using resolved trace references

---
 .../conversation/TraceConversation.integration.test.tsx      | 4 ++++
 .../src/components/lens/traces/detail/run/RunView.tsx        | 5 +++--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/lens/traces/detail/conversation/TraceConversation.integration.test.tsx b/ui/litellm-dashboard/src/components/lens/traces/detail/conversation/TraceConversation.integration.test.tsx
index 85df5f5255b..8154bd48a65 100644
--- a/ui/litellm-dashboard/src/components/lens/traces/detail/conversation/TraceConversation.integration.test.tsx
+++ b/ui/litellm-dashboard/src/components/lens/traces/detail/conversation/TraceConversation.integration.test.tsx
@@ -48,6 +48,10 @@ describe("TraceConversation", () => {
 
   it.each([false, true])("refreshes unchanged spans without hiding loaded content on failure (%s)", async (failed) => {
     const user = userEvent.setup();
+    vi.mocked(agentTraceCall).mockResolvedValue({
+      ...trace,
+      summary: { ...trace.summary, trace_ref: "resolved-reference" },
+    });
     renderWithProviders(
       <RoutedRunView traceId={trace.summary.trace_id} accessToken="test" onBack={vi.fn()} embedded />,
     );
diff --git a/ui/litellm-dashboard/src/components/lens/traces/detail/run/RunView.tsx b/ui/litellm-dashboard/src/components/lens/traces/detail/run/RunView.tsx
index f383fc46b3e..e14a23b5070 100644
--- a/ui/litellm-dashboard/src/components/lens/traces/detail/run/RunView.tsx
+++ b/ui/litellm-dashboard/src/components/lens/traces/detail/run/RunView.tsx
@@ -116,9 +116,10 @@ function LoadedRun({
   };
   const refreshRun = () =>
     readManually(async () => {
-      await traceQuery.refetch();
+      const refreshed = await traceQuery.refetch();
+      const contentRef = refreshed.data?.pages[0].summary.trace_ref ?? traceRef;
       await queryClient.invalidateQueries({
-        queryKey: ["agentTraceSpan", traceId, traceRef],
+        queryKey: ["agentTraceSpan", traceId, contentRef],
         predicate: (query) => query.queryKey.at(-1) === accessToken,
       });
     });