mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-13 03:59:12 +00:00
fix(lens): integrate resolved-reference refresh
This commit is contained in:
commit
8ea402ce71
2 changed files with 7 additions and 2 deletions
|
|
@ -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 />,
|
||||
);
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
});
|
||||
});
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue