allow accessing cost during steps

Author: bbohlender

docs/concepts/history.mdx

@@ -225,10 +225,6 @@ history.addEventListener('step-response', (event) => {
   console.log('Step completed:', event.historyId)
 })
 
-history.addEventListener('step-error', (event) => {
-  console.error('Step failed:', event.error)
-})
-
 history.addEventListener('data-reference-added', (event) => {
   console.log('Data added to history')
 })

packages/prai/src/console-logger.ts

@@ -150,17 +150,6 @@ export function consoleLogger(history: History, options?: { abort?: AbortSignal
     { signal: abortSignal },
   )
 
-  history.addEventListener(
-    'step-error',
-    (event) => {
-      logHeader(event.historyId, 'Step Error', Date.now(), colors.error)
-      console.error('  Error:')
-      console.log('  ' + colors.error(event.error))
-      logSeparator()
-    },
-    { signal: abortSignal },
-  )
-
   history.addEventListener(
     'subtask-start',
     (event) => {

packages/prai/src/history.ts

@@ -69,7 +69,6 @@ export type BuildReference = () => string
 export type EventMap = {
   'step-request': StepRequestEvent
   'step-response': StepResponseEvent
-  'step-error': StepErrorEvent
   'subtask-start': SubtaskStartEvent
   'data-reference-added': DataReferenceAddedEvent
   'subtask-response-referenced': SubtaskReponseReferencedEvent
@@ -87,12 +86,6 @@ export type StepResponseEvent = {
   message: Message
 }
 
-export type StepErrorEvent = {
-  type: 'step-error'
-  historyId: string
-  error: string
-}
-
 export type SubtaskStartEvent = {
   type: 'subtask-start-event'
   historyId: string
@@ -139,9 +132,6 @@ export class History {
   private cost: number | undefined = 0
 
   getCost() {
-    if (this.currentlyExecutingStepId != null) {
-      throw new Error('unable to compute the costs while still executing a step')
-    }
     return this.cost
   }
 
@@ -221,11 +211,6 @@ export class History {
     return stepId
   }
 
-  onStepError(error: any) {
-    const errorMessage = error instanceof Error ? error.message : String(error)
-    this.dispatchEvent('step-error', { type: 'step-error', historyId: this.id, error: errorMessage })
-  }
-
   /**
    * @deprecated used internally
    */

packages/redis/src/index.ts

@@ -77,16 +77,6 @@ export async function redisLogger(history: History, options: RedisLoggerOptions)
     { signal: abortSignal },
   )
 
-  history.addEventListener(
-    'step-error',
-    (event) => {
-      addStreamEntry('step-error', event.historyId, {
-        error: event.error,
-      })
-    },
-    { signal: abortSignal },
-  )
-
   history.addEventListener(
     'subtask-start',
     (event) => {