feat: add colored console output for agent logs

Author: bhouston

packages/agent/src/tools/agent/agentStart.ts

@@ -1,5 +1,6 @@
 import { z } from 'zod';
 import { zodToJsonSchema } from 'zod-to-json-schema';
+import chalk from 'chalk';
 
 import {
   getDefaultSystemPrompt,
@@ -15,6 +16,23 @@ import { AgentStatus, AgentState } from './AgentTracker.js';
 // For backward compatibility
 export const agentStates = new Map<string, AgentState>();
 
+// Generate a random color for an agent
+// Avoid colors that are too light or too similar to error/warning colors
+const getRandomAgentColor = () => {
+  // List of bright chalk colors that are visually distinct
+  const colors = [
+    chalk.cyan,
+    chalk.green,
+    chalk.blue,
+    chalk.magenta,
+    chalk.blueBright,
+    chalk.greenBright,
+    chalk.cyanBright,
+    chalk.magentaBright
+  ];
+  return colors[Math.floor(Math.random() * colors.length)];
+};
+
 const parameterSchema = z.object({
   description: z
     .string()
@@ -155,9 +173,13 @@ export const agentStartTool: Tool<Parameters, ReturnType> = {
     // This is wrapped in a try-catch to maintain backward compatibility with tests
     let subAgentLogger = context.logger;
     try {
+      // Generate a random color for this agent
+      const agentColor = getRandomAgentColor();
+      
       subAgentLogger = new Logger({
         name: 'agent',
         parent: context.logger,
+        color: agentColor, // Assign the random color to the agent
       });
       // Add the listener to the sub-agent logger as well
       subAgentLogger.listeners.push(logCaptureListener);

packages/agent/src/utils/logger.ts

@@ -13,6 +13,7 @@ export type LoggerProps = {
   logLevel?: LogLevel;
   parent?: Logger;
   customPrefix?: string;
+  color?: ChalkInstance;
 };
 
 export type LoggerListener = (
@@ -29,6 +30,7 @@ export class Logger {
   public readonly name: string;
   public readonly nesting: number;
   public readonly customPrefix?: string;
+  public readonly color?: ChalkInstance;
 
   readonly listeners: LoggerListener[] = [];
 
@@ -37,12 +39,15 @@ export class Logger {
     parent = undefined,
     logLevel = parent?.logLevel ?? LogLevel.info,
     customPrefix,
+    color,
   }: LoggerProps) {
     this.customPrefix = customPrefix;
     this.name = name;
     this.parent = parent;
     this.logLevel = logLevel;
     this.logLevelIndex = logLevel;
+    // Inherit color from parent if not provided and parent has a color
+    this.color = color ?? parent?.color;
 
     // Calculate indent level and offset based on parent chain
     this.nesting = 0;
@@ -108,16 +113,26 @@ export const consoleOutputLogger: LoggerListener = (
   lines: string[],
 ) => {
   const getColor = (level: LogLevel, _nesting: number = 0): ChalkInstance => {
+    // Always use red for errors and yellow for warnings regardless of agent color
+    if (level === LogLevel.error) {
+      return chalk.red;
+    }
+    if (level === LogLevel.warn) {
+      return chalk.yellow;
+    }
+    
+    // Use logger's color if available for log level
+    if (level === LogLevel.log && logger.color) {
+      return logger.color;
+    }
+    
+    // Default colors for different log levels
     switch (level) {
       case LogLevel.debug:
       case LogLevel.info:
         return chalk.white.dim;
       case LogLevel.log:
         return chalk.white;
-      case LogLevel.warn:
-        return chalk.yellow;
-      case LogLevel.error:
-        return chalk.red;
       default:
         throw new Error(`Unknown log level: ${level}`);
     }