feat(frontend/copilot): add per-turn work-done summary stats

autogpt_platform/backend/backend/copilot/stream_registry.py

@@ -733,7 +733,10 @@ async def mark_session_completed(
     # This is the SINGLE place that publishes StreamFinish — services and
     # the processor must NOT publish it themselves.
     try:
-        await publish_chunk(turn_id, StreamFinish())
+        await publish_chunk(
+            turn_id,
+            StreamFinish(),
+        )
     except Exception as e:
         logger.error(
             f"Failed to publish StreamFinish for session {session_id}: {e}. "

autogpt_platform/frontend/src/app/(platform)/copilot/components/ChatMessagesContainer/ChatMessagesContainer.tsx

@@ -10,6 +10,7 @@ import {
 } from "@/components/ai-elements/message";
 import { LoadingSpinner } from "@/components/atoms/LoadingSpinner/LoadingSpinner";
 import { FileUIPart, UIDataTypes, UIMessage, UITools } from "ai";
+import { TurnStatsBar } from "../JobStatsBar/TurnStatsBar";
 import { MessageAttachments } from "./components/MessageAttachments";
 import { MessagePartRenderer } from "./components/MessagePartRenderer";
 import { ThinkingIndicator } from "./components/ThinkingIndicator";
@@ -25,6 +26,24 @@ interface Props {
   headerSlot?: React.ReactNode;
 }
 
+/** Collect all messages belonging to a turn: the user message + every
+ *  assistant message up to (but not including) the next user message. */
+function getTurnMessages(
+  messages: UIMessage<unknown, UIDataTypes, UITools>[],
+  lastAssistantIndex: number,
+): UIMessage<unknown, UIDataTypes, UITools>[] {
+  let userIndex = lastAssistantIndex - 1;
+  while (userIndex >= 0 && messages[userIndex].role !== "user") {
+    userIndex--;
+  }
+  let endIndex = lastAssistantIndex + 1;
+  while (endIndex < messages.length && messages[endIndex].role !== "user") {
+    endIndex++;
+  }
+  const start = userIndex >= 0 ? userIndex : lastAssistantIndex;
+  return messages.slice(start, endIndex);
+}
+
 export function ChatMessagesContainer({
   messages,
   status,
@@ -34,21 +53,16 @@ export function ChatMessagesContainer({
 }: Props) {
   const lastMessage = messages[messages.length - 1];
 
-  // Determine if something is visibly "in-flight" in the last assistant message:
-  // - Text is actively streaming (last part is non-empty text)
-  // - A tool call is pending (state is input-streaming or input-available)
   const hasInflight = (() => {
     if (lastMessage?.role !== "assistant") return false;
     const parts = lastMessage.parts;
     if (parts.length === 0) return false;
 
     const lastPart = parts[parts.length - 1];
 
-    // Text is actively being written
     if (lastPart.type === "text" && lastPart.text.trim().length > 0)
       return true;
 
-    // A tool call is still pending (no output yet)
     if (
       lastPart.type.startsWith("tool-") &&
       "state" in lastPart &&
@@ -82,13 +96,16 @@ export function ChatMessagesContainer({
 
           const isAssistant = message.role === "assistant";
 
-          // Past assistant messages are always done; the last one
-          // is done only when the stream has finished.
           const isAssistantDone =
             isAssistant &&
             (!isLastAssistant ||
               (status !== "streaming" && status !== "submitted"));
 
+          const isLastAssistantInTurn =
+            isAssistant &&
+            (messageIndex === messages.length - 1 ||
+              messages[messageIndex + 1].role === "user");
+
           const fileParts = message.parts.filter(
             (p): p is FileUIPart => p.type === "file",
           );
@@ -110,6 +127,15 @@ export function ChatMessagesContainer({
                     partIndex={i}
                   />
                 ))}
+                {isLastAssistantInTurn &&
+                  !(
+                    isLastAssistant &&
+                    (status === "streaming" || status === "submitted")
+                  ) && (
+                    <TurnStatsBar
+                      turnMessages={getTurnMessages(messages, messageIndex)}
+                    />
+                  )}
                 {isLastAssistant && showThinking && (
                   <ThinkingIndicator active={showThinking} />
                 )}
@@ -120,23 +146,30 @@ export function ChatMessagesContainer({
                   isUser={message.role === "user"}
                 />
               )}
-              {isAssistantDone &&
+              {isLastAssistantInTurn &&
+                isAssistantDone &&
                 (() => {
-                  const textParts = message.parts.filter(
-                    (p): p is Extract<typeof p, { type: "text" }> =>
-                      p.type === "text",
-                  );
+                  const turnMsgs = getTurnMessages(messages, messageIndex);
+                  const allTextParts = turnMsgs
+                    .filter((m) => m.role === "assistant")
+                    .flatMap((m) =>
+                      m.parts.filter(
+                        (p): p is Extract<typeof p, { type: "text" }> =>
+                          p.type === "text",
+                      ),
+                    );
 
-                  // Hide actions when the message ended with an error or cancellation
-                  const lastTextPart = textParts[textParts.length - 1];
+                  const lastTextPart = allTextParts[allTextParts.length - 1];
                   if (lastTextPart) {
                     const { markerType } = parseSpecialMarkers(
                       lastTextPart.text,
                     );
                     if (markerType === "error") return null;
                   }
 
-                  const textContent = textParts.map((p) => p.text).join("\n");
+                  const textContent = allTextParts
+                    .map((p) => p.text)
+                    .join("\n");
                   return (
                     <MessageActions>
                       <CopyButton text={textContent} />

autogpt_platform/frontend/src/app/(platform)/copilot/components/JobStatsBar/TurnStatsBar.tsx

@@ -0,0 +1,29 @@
+import type { UIDataTypes, UIMessage, UITools } from "ai";
+import { getWorkDoneCounters } from "./useWorkDoneCounters";
+
+interface Props {
+  turnMessages: UIMessage<unknown, UIDataTypes, UITools>[];
+}
+
+export function TurnStatsBar({ turnMessages }: Props) {
+  const { counters } = getWorkDoneCounters(turnMessages);
+
+  if (counters.length === 0) return null;
+
+  return (
+    <div className="mt-2 flex items-center gap-1.5">
+      {counters.map(function renderCounter(counter, index) {
+        return (
+          <span key={counter.category} className="flex items-center gap-1">
+            {index > 0 && (
+              <span className="text-xs text-neutral-300">&middot;</span>
+            )}
+            <span className="text-[11px] tabular-nums text-neutral-500">
+              {counter.count} {counter.label}
+            </span>
+          </span>
+        );
+      })}
+    </div>
+  );
+}

autogpt_platform/frontend/src/app/(platform)/copilot/components/JobStatsBar/useWorkDoneCounters.ts

@@ -0,0 +1,74 @@
+import type { UIDataTypes, UIMessage, UITools } from "ai";
+
+const TOOL_TO_CATEGORY: Record<string, string> = {
+  find_agent: "search",
+  find_library_agent: "search",
+  run_agent: "agent run",
+  run_block: "block run",
+  create_agent: "agent created",
+  edit_agent: "agent edited",
+  schedule_agent: "agent scheduled",
+};
+
+const MAX_COUNTERS = 3;
+
+function pluralize(label: string, count: number): string {
+  if (count === 1) return label;
+
+  // "agent created" -> "agents created", "agent edited" -> "agents edited"
+  const nounVerbMatch = label.match(
+    /^(\w+)\s+(created|edited|scheduled|run)$/i,
+  );
+  if (nounVerbMatch) {
+    return pluralizeWord(nounVerbMatch[1]) + " " + nounVerbMatch[2];
+  }
+
+  return pluralizeWord(label);
+}
+
+function pluralizeWord(word: string): string {
+  if (word.endsWith("ch") || word.endsWith("sh") || word.endsWith("x"))
+    return word + "es";
+  return word + "s";
+}
+
+export interface WorkDoneCounter {
+  label: string;
+  count: number;
+  category: string;
+}
+
+export function getWorkDoneCounters(
+  messages: UIMessage<unknown, UIDataTypes, UITools>[],
+) {
+  const categoryCounts = new Map<string, number>();
+
+  for (const message of messages) {
+    if (message.role !== "assistant") continue;
+
+    for (const part of message.parts) {
+      if (!part.type.startsWith("tool-")) continue;
+
+      const toolName = part.type.replace("tool-", "");
+      const category = TOOL_TO_CATEGORY[toolName];
+      if (!category) continue;
+
+      categoryCounts.set(category, (categoryCounts.get(category) ?? 0) + 1);
+    }
+  }
+
+  const counters: WorkDoneCounter[] = Array.from(categoryCounts.entries())
+    .map(function toCounter([category, count]) {
+      return {
+        label: pluralize(category, count),
+        count,
+        category,
+      };
+    })
+    .sort(function byCountDesc(a, b) {
+      return b.count - a.count;
+    })
+    .slice(0, MAX_COUNTERS);
+
+  return { counters };
+}