diff --git a/internal/workbenchruntime/service.go b/internal/workbenchruntime/service.go index 06c2542e..892a8ca8 100644 --- a/internal/workbenchruntime/service.go +++ b/internal/workbenchruntime/service.go @@ -1247,8 +1247,8 @@ func (s *Server) querySummaryFacts(ctx context.Context, sessions []map[string]an if len(sessionIDs) > 0 { resultCount++ go func() { - messageFacts, err := s.queryFacts(ctx, factQuery{SessionIDs: sessionIDs, Families: []string{"messages"}, MessagesOrder: "updated_asc"}) - results <- summaryFactsResult{kind: "messageFacts", facts: messageFacts, err: err} + messageSummaries, err := s.queryMessageSummaries(ctx, sessionIDs) + results <- summaryFactsResult{kind: "messageSummaries", facts: map[string][]any{"messageSummaries": messageSummaries}, err: err} }() } if len(traceIDs) > 0 { @@ -1264,7 +1264,7 @@ func (s *Server) querySummaryFacts(ctx context.Context, sessions []map[string]an return nil, result.err } switch result.kind { - case "messageFacts": + case "messageSummaries": facts = mergeFacts(facts, result.facts) case "traceFacts": facts = mergeFacts(facts, result.facts)