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 672022280f4..f9dc9be8d3b 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( , ); 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 4063cb8f426..339b0c3ea44 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, }); });