add logs to component chains/agents

packages/components/src/utils.ts

@@ -3,9 +3,6 @@ import { load } from 'cheerio'
 import * as fs from 'fs'
 import * as path from 'path'
 import { JSDOM } from 'jsdom'
-import { BaseCallbackHandler } from 'langchain/callbacks'
-import { Server } from 'socket.io'
-import { ChainValues } from 'langchain/dist/schema'
 
 export const numberOrExpressionRegex = '^(\\d+\\.?\\d*|{{.*}})$' //return true if string consists only numbers OR expression {{}}
 export const notEmptyRegex = '(.|\\s)*\\S(.|\\s)*' //return true if string is not empty or blank
@@ -350,50 +347,9 @@ export const getEnvironmentVariable = (name: string): string | undefined => {
     }
 }
 
-/**
- * Custom chain handler class
+/*
+ * List of dependencies allowed to be import in vm2
  */
-export class CustomChainHandler extends BaseCallbackHandler {
-    name = 'custom_chain_handler'
-    isLLMStarted = false
-    socketIO: Server
-    socketIOClientId = ''
-    skipK = 0 // Skip streaming for first K numbers of handleLLMStart
-    returnSourceDocuments = false
-
-    constructor(socketIO: Server, socketIOClientId: string, skipK?: number, returnSourceDocuments?: boolean) {
-        super()
-        this.socketIO = socketIO
-        this.socketIOClientId = socketIOClientId
-        this.skipK = skipK ?? this.skipK
-        this.returnSourceDocuments = returnSourceDocuments ?? this.returnSourceDocuments
-    }
-
-    handleLLMStart() {
-        if (this.skipK > 0) this.skipK -= 1
-    }
-
-    handleLLMNewToken(token: string) {
-        if (this.skipK === 0) {
-            if (!this.isLLMStarted) {
-                this.isLLMStarted = true
-                this.socketIO.to(this.socketIOClientId).emit('start', token)
-            }
-            this.socketIO.to(this.socketIOClientId).emit('token', token)
-        }
-    }
-
-    handleLLMEnd() {
-        this.socketIO.to(this.socketIOClientId).emit('end')
-    }
-
-    handleChainEnd(outputs: ChainValues): void | Promise<void> {
-        if (this.returnSourceDocuments) {
-            this.socketIO.to(this.socketIOClientId).emit('sourceDocuments', outputs?.sourceDocuments)
-        }
-    }
-}
-
 export const availableDependencies = [
     '@dqbd/tiktoken',
     '@getzep/zep-js',