add finish message to print mode

Author: charleslien

common/src/types/print-mode.ts

@@ -38,8 +38,7 @@ export type PrintModeText = z.infer<typeof printModeTextSchema>
 
 export const printModeFinishSchema = z.object({
   type: z.literal('finish'),
-  agentId: z.string(),
-  model: z.string(),
+  agentId: z.string().optional(),
   totalCost: z.number(),
 })
 export type PrintModeFinish = z.infer<typeof printModeFinishSchema>

npm-app/src/cli.ts

@@ -27,6 +27,7 @@ import {
   yellow,
 } from 'picocolors'
 
+import { PrintModeFinish } from '@codebuff/common/types/print-mode'
 import {
   killAllBackgroundProcesses,
   sendKillSignalToAllBackgroundProcesses,
@@ -1201,6 +1202,17 @@ export class CLI {
       .flat()
       .reduce((sum, credits) => sum + credits, 0)
 
+    if (printModeIsEnabled()) {
+      const finishObj: PrintModeFinish = {
+        type: 'finish',
+        totalCost: totalCreditsUsedThisSession,
+      }
+      const agentId = CLI.getInstance().agent
+      if (agentId) {
+        finishObj.agentId = agentId
+      }
+      printModeLog(finishObj)
+    }
     let exitUsageMessage = `${pluralize(totalCreditsUsedThisSession, 'credit')} used this session`
     if (client.usageData.remainingBalance !== null) {
       exitUsageMessage += `, ${client.usageData.remainingBalance.toLocaleString()} credits left.`