diff --git a/src/actions/analytics/events.ts b/src/actions/analytics/events.ts index 4e756b88a8..08d3987117 100644 --- a/src/actions/analytics/events.ts +++ b/src/actions/analytics/events.ts @@ -258,7 +258,8 @@ export const getTasks = async ({ WITH first_messages AS ( SELECT taskId, - argMin(text, ts) as title + argMin(text, ts) as title, + argMin(mode, ts) as mode FROM messages WHERE orgId = {orgId: String} GROUP BY taskId @@ -268,6 +269,7 @@ export const getTasks = async ({ e.userId, e.apiProvider AS provider, e.modelId as model, + any(fm.mode) AS mode, MAX(CASE WHEN e.type = 'Task Completed' THEN 1 ELSE 0 END) AS completed, SUM(CASE WHEN e.type = 'LLM Completion' THEN COALESCE(e.inputTokens, 0) + COALESCE(e.outputTokens, 0) ELSE 0 END) AS tokens, SUM(CASE WHEN e.type = 'LLM Completion' THEN COALESCE(e.cost, 0) ELSE 0 END) AS cost, diff --git a/src/app/(authenticated)/usage/Messages.tsx b/src/app/(authenticated)/usage/Messages.tsx index 9143602b8e..3881377443 100644 --- a/src/app/(authenticated)/usage/Messages.tsx +++ b/src/app/(authenticated)/usage/Messages.tsx @@ -28,10 +28,17 @@ export const Messages = ({ messages }: MessagesProps) => { message.role === 'user' ? 'bg-primary/10' : 'bg-secondary/10', )} > -