feat: implement pretty logger

packages/framework/package.json

@@ -21,6 +21,7 @@
   },
   "type": "module",
   "dependencies": {
+    "chalk": "^5.3.0",
     "dedent": "^1.5.3",
     "openai": "^4.76.0",
     "zod": "^3.23.8"

packages/framework/src/supervisor/iterate.ts

@@ -10,7 +10,7 @@ import { selectAgent } from './selectAgent.js'
  */
 export async function iterate(workflow: Workflow, state: WorkflowState): Promise<WorkflowState> {
   const nextState = await nextTick(workflow, state)
-  workflow.snapshot(nextState)
+  workflow.snapshot({ prevState: state, nextState })
   return nextState
 }
 

packages/framework/src/supervisor/runTools.ts

@@ -7,7 +7,7 @@ import { Message } from '../types.js'
 /**
  * Asserts that given message requests tool calls
  */
-function isToolCallRequest(message?: Message): message is ParsedChatCompletionMessage<any> {
+export function isToolCallRequest(message?: Message): message is ParsedChatCompletionMessage<any> {
   return message ? 'tool_calls' in message : false
 }
 

packages/framework/src/telemetry.ts

@@ -1,9 +1,77 @@
+import chalk from 'chalk'
+
+import { isToolCallRequest } from './supervisor/runTools.js'
 import { WorkflowState } from './workflow.js'
 
-export type Telemetry = (state: WorkflowState) => void
+export type Telemetry = ({
+  prevState,
+  nextState,
+}: {
+  prevState: WorkflowState
+  nextState: WorkflowState
+}) => void
 
 export const noop: Telemetry = () => {}
 
-export const logger: Telemetry = (state) => {
-  console.log(state)
+export const logger: Telemetry = ({ prevState, nextState }) => {
+  const { status: prevStatus } = prevState
+  const { status: nextStatus } = nextState
+
+  const logMessage = (emoji: string, message: string, details: string = '') => {
+    console.log(`${emoji} ${chalk.bold(message)}${details ? `\n${chalk.gray(details)}` : ''}`)
+  }
+
+  if (prevStatus !== nextStatus) {
+    switch (nextStatus) {
+      case 'pending':
+        logMessage(
+          '🕒',
+          'Distributing the task...',
+          `Request: ${nextState.agentRequest[0].content}`
+        )
+        break
+      case 'assigned':
+        switch (nextState.agentStatus) {
+          case 'idle':
+            logMessage('🛠️', `Agent "${nextState.agent}" is beginning to work on the task`)
+            break
+          case 'step':
+            logMessage(
+              '🔄',
+              `Agent "${nextState.agent}" is proceeding to the next step`,
+              `Step details: ${nextState.agentRequest[0].content}`
+            )
+            break
+          case 'tool': {
+            const lastMessage = nextState.agentRequest[nextState.agentRequest.length - 1]
+            if (!isToolCallRequest(lastMessage)) {
+              return
+            }
+            const toolNames = lastMessage.tool_calls.map((tool) => tool.function.name)
+            logMessage(
+              '🔧',
+              `Agent "${nextState.agent}" is calling tools`,
+              `Tools: ${toolNames.join(', ')}`
+            )
+            break
+          }
+        }
+        break
+      case 'idle':
+        logMessage('✅', 'Moving to next task', `Iterations: ${nextState.messages.length}`)
+        break
+      case 'finished':
+        logMessage(
+          '🎉',
+          'Workflow finished successfully!',
+          `Total messages: ${nextState.messages.length}`
+        )
+        break
+      case 'failed':
+        logMessage('❌', 'Workflow failed', `Check logs for more details`)
+        break
+      default:
+        logMessage('ℹ️', 'State changed', `New status: ${nextStatus}`)
+    }
+  }
 }