scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-13 03:59:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

fix(lens): retain refreshed actor conversation content

Browse source
This commit is contained in:
moe-berri 2026-10-06 14:34:30 -07:00
commit dbe7a90f47
3 changed files with 30 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
ui/litellm-dashboard/src/components/lens/traces/detail/conversation/TraceConversation.integration.test.tsx
View file

@ -46,6 +46,27 @@ describe("TraceConversation", () => {
.mockImplementation(async (_token, _trace, id) => (id === "root" ? rootDetail : { ...toolDetail, span_id: id }));
});
it.each([false, true])("refreshes unchanged spans without hiding loaded content on failure (%s)", async (failed) => {
const user = userEvent.setup();
renderWithProviders(
<RoutedRunView traceId={trace.summary.trace_id} accessToken="test" onBack={vi.fn()} embedded />,
);
await user.click(await screen.findByRole("tab", { name: "Conversation" }));
expect(await screen.findByText("The release is ready")).toBeVisible();
vi.mocked(agentTraceSpanCall).mockImplementation(async (_token, _trace, id) => {
if (failed) throw new Error("content refresh failed");
return id === "root" ? { ...rootDetail, output: "Updated final answer" } : toolDetail;
});
await user.click(screen.getByRole("button", { name: "Refresh run" }));
if (failed) {
expect(await screen.findAllByRole("button", { name: "Retry step" })).toHaveLength(2);
expect(screen.getByText("The release is ready")).toBeVisible();
} else {
expect(await screen.findByText("Updated final answer")).toBeVisible();
expect(screen.queryByText("The release is ready")).not.toBeInTheDocument();
}
});
it("switches to a readable transcript and opens the exact tool step from it", async () => {
const user = userEvent.setup();
renderWithProviders(

2
ui/litellm-dashboard/src/components/lens/traces/detail/conversation/useConversationDetails.ts
View file

@ -22,7 +22,7 @@ export function useConversationDetails(trace: Trace, accessToken: string) {
[traceId, traceRef, accessToken, traces],
);
const queries = useQueries({ queries: visible.map(spanQuery) });
const unresolvedIndex = queries.findIndex((query) => !query.isSuccess);
const unresolvedIndex = queries.findIndex((query) => query.data === undefined);
const loadedCount = unresolvedIndex < 0 ? queries.length : unresolvedIndex;
const details = new Map(
queries.slice(0, loadedCount).map((query, index) => [visible[index].span_id, query.data!] as const),

9
ui/litellm-dashboard/src/components/lens/traces/detail/run/RunView.tsx
View file

@ -114,7 +114,14 @@ function LoadedRun({
setManualRead(true);
void read().finally(() => setManualRead(false));
};
const refreshRun = () => readManually(() => traceQuery.refetch());
const refreshRun = () =>
readManually(async () => {
await traceQuery.refetch();
await queryClient.invalidateQueries({
queryKey: ["agentTraceSpan", traceId, traceRef],
predicate: (query) => query.queryKey.at(-1) === accessToken,
});
});
const toggleLive = () => {
if (live && !manualRead && !traceQuery.isFetchingNextPage) {
void queryClient.cancelQueries({ queryKey, exact: true });