feat: add agent cost reporting

2026-07-09 20:29:56 +08:00
parent fc5a5365d8
commit 716101eed0
9 changed files with 275 additions and 7 deletions
@@ -106,20 +106,23 @@ export class StreamingAgentCard {
    recordToolUseEnd({ runId: this.runId, ...params });
    this.scheduleFlush();
  }
-  async finish(fallbackText: string, options: { readonly interrupted?: boolean } = {}): Promise<void> {
+  async finish(fallbackText: string, options: { readonly interrupted?: boolean; readonly footerText?: string | undefined } = {}): Promise<void> {
    await this.flushChain;
    this.interrupted = options.interrupted === true;
+    const footerText = options.footerText ?? "";
+    const fallbackWithFooter = appendFooter(fallbackText, footerText);
    try {
      let updated = true;
      if (this.text.length > 0) {
+        this.text = appendFooter(this.text, footerText);
        updated = await this.flushCard("complete", this.text);
-      } else if (this.currentMessageId === null && fallbackText.length > 0) {
+      } else if (this.currentMessageId === null && fallbackWithFooter.length > 0) {
        // No streaming text was sent. If we never created a card, send one now.
-        this.text = fallbackText;
+        this.text = fallbackWithFooter;
        updated = await this.flushCard("complete", this.text);
      } else if (this.currentMessageId !== null) {
        // Patch the existing card with the final text.
-        updated = await this.flushCard("complete", fallbackText);
+        updated = await this.flushCard("complete", fallbackWithFooter);
      }
      if (!updated && this.interrupted) {
        await sendText(this.rt, this.chatId, "\u5DF2\u4E2D\u65AD\u5F53\u524D\u8FD0\u884C\u3002", this.sendOptions);
@@ -228,3 +231,9 @@ export class StreamingAgentCard {
    return "thinking";
  }
 }
+
+function appendFooter(text: string, footerText: string): string {
+  if (footerText === "") return text;
+  if (text === "") return footerText;
+  return `${text.trimEnd()}\n\n${footerText}`;
+}