remove bind

Author: leehuwuj

templates/components/agents/typescript/financial_report/workflow/fin-report.ts

@@ -85,7 +85,7 @@ export class FinancialReportWorkflow extends Workflow<
         inputs: [StartEvent<AgentInput>],
         outputs: [InputEvent],
       },
-      this.prepareChatHistory.bind(this),
+      this.prepareChatHistory,
     );
 
     this.addStep(
@@ -99,38 +99,38 @@ export class FinancialReportWorkflow extends Workflow<
           StopEvent,
         ],
       },
-      this.handleLLMInput.bind(this),
+      this.handleLLMInput,
     );
 
     this.addStep(
       {
         inputs: [ResearchEvent],
         outputs: [AnalyzeEvent],
       },
-      this.handleResearch.bind(this),
+      this.handleResearch,
     );
 
     this.addStep(
       {
         inputs: [AnalyzeEvent],
         outputs: [InputEvent],
       },
-      this.handleAnalyze.bind(this),
+      this.handleAnalyze,
     );
 
     this.addStep(
       {
         inputs: [ReportGenerationEvent],
         outputs: [InputEvent],
       },
-      this.handleReportGeneration.bind(this),
+      this.handleReportGeneration,
     );
   }
 
-  private async prepareChatHistory(
+  prepareChatHistory = async (
     ctx: HandlerContext<null>,
     ev: StartEvent<AgentInput>,
-  ) {
+  ): Promise<InputEvent> => {
     const { message } = ev.data;
 
     if (this.systemPrompt) {
@@ -139,9 +139,18 @@ export class FinancialReportWorkflow extends Workflow<
     this.memory.put({ role: "user", content: message });
 
     return new InputEvent({ input: this.memory.getMessages() });
-  }
+  };
 
-  private async handleLLMInput(ctx: HandlerContext<null>, ev: InputEvent) {
+  handleLLMInput = async (
+    ctx: HandlerContext<null>,
+    ev: InputEvent,
+  ): Promise<
+    | InputEvent
+    | ResearchEvent
+    | AnalyzeEvent
+    | ReportGenerationEvent
+    | StopEvent
+  > => {
     const chatHistory = ev.data.input;
 
     const tools = [this.codeInterpreterTool, this.documentGeneratorTool];
@@ -190,9 +199,12 @@ export class FinancialReportWorkflow extends Workflow<
         }
         throw new Error(`Unknown tool: ${toolName}`);
     }
-  }
+  };
 
-  private async handleResearch(ctx: HandlerContext<null>, ev: ResearchEvent) {
+  handleResearch = async (
+    ctx: HandlerContext<null>,
+    ev: ResearchEvent,
+  ): Promise<AnalyzeEvent> => {
     ctx.sendEvent(
       new AgentRunEvent({
         agent: "Researcher",
@@ -219,12 +231,15 @@ export class FinancialReportWorkflow extends Workflow<
           "I have finished researching the data, please analyze the data.",
       },
     });
-  }
+  };
 
   /**
    * Analyze a research result or a tool call for code interpreter from the LLM
    */
-  private async handleAnalyze(ctx: HandlerContext<null>, ev: AnalyzeEvent) {
+  handleAnalyze = async (
+    ctx: HandlerContext<null>,
+    ev: AnalyzeEvent,
+  ): Promise<InputEvent> => {
     ctx.sendEvent(
       new AgentRunEvent({
         agent: "Analyst",
@@ -283,12 +298,12 @@ export class FinancialReportWorkflow extends Workflow<
     return new InputEvent({
       input: this.memory.getMessages(),
     });
-  }
+  };
 
-  private async handleReportGeneration(
+  handleReportGeneration = async (
     ctx: HandlerContext<null>,
     ev: ReportGenerationEvent,
-  ) {
+  ): Promise<InputEvent> => {
     const { toolCalls } = ev.data;
 
     const toolMsgs = await callTools({
@@ -301,5 +316,5 @@ export class FinancialReportWorkflow extends Workflow<
       this.memory.put(toolMsg);
     }
     return new InputEvent({ input: this.memory.getMessages() });
-  }
+  };
 }

templates/components/agents/typescript/form_filling/workflow/form-filling.ts

@@ -87,7 +87,7 @@ export class FormFillingWorkflow extends Workflow<
         inputs: [StartEvent<AgentInput>],
         outputs: [InputEvent],
       },
-      this.prepareChatHistory.bind(this),
+      this.prepareChatHistory,
     );
 
     this.addStep(
@@ -101,38 +101,38 @@ export class FormFillingWorkflow extends Workflow<
           StopEvent,
         ],
       },
-      this.handleLLMInput.bind(this),
+      this.handleLLMInput,
     );
 
     this.addStep(
       {
         inputs: [ExtractMissingCellsEvent],
         outputs: [InputEvent],
       },
-      this.handleExtractMissingCells.bind(this),
+      this.handleExtractMissingCells,
     );
 
     this.addStep(
       {
         inputs: [FindAnswersEvent],
         outputs: [InputEvent],
       },
-      this.handleFindAnswers.bind(this),
+      this.handleFindAnswers,
     );
 
     this.addStep(
       {
         inputs: [FillMissingCellsEvent],
         outputs: [InputEvent],
       },
-      this.handleFillMissingCells.bind(this),
+      this.handleFillMissingCells,
     );
   }
 
-  private async prepareChatHistory(
+  prepareChatHistory = async (
     ctx: HandlerContext<null>,
     ev: StartEvent<AgentInput>,
-  ) {
+  ): Promise<InputEvent> => {
     const { message } = ev.data;
 
     if (this.systemPrompt) {
@@ -141,9 +141,18 @@ export class FormFillingWorkflow extends Workflow<
     this.memory.put({ role: "user", content: message });
 
     return new InputEvent({ input: this.memory.getMessages() });
-  }
+  };
 
-  private async handleLLMInput(ctx: HandlerContext<null>, ev: InputEvent) {
+  handleLLMInput = async (
+    ctx: HandlerContext<null>,
+    ev: InputEvent,
+  ): Promise<
+    | InputEvent
+    | ExtractMissingCellsEvent
+    | FindAnswersEvent
+    | FillMissingCellsEvent
+    | StopEvent
+  > => {
     const chatHistory = ev.data.input;
 
     const tools = [this.extractorTool, this.fillMissingCellsTool];
@@ -192,12 +201,12 @@ export class FormFillingWorkflow extends Workflow<
         }
         throw new Error(`Unknown tool: ${toolName}`);
     }
-  }
+  };
 
-  private async handleExtractMissingCells(
+  handleExtractMissingCells = async (
     ctx: HandlerContext<null>,
     ev: ExtractMissingCellsEvent,
-  ) {
+  ): Promise<InputEvent> => {
     ctx.sendEvent(
       new AgentRunEvent({
         agent: "CSVExtractor",
@@ -216,12 +225,12 @@ export class FormFillingWorkflow extends Workflow<
       this.memory.put(toolMsg);
     }
     return new InputEvent({ input: this.memory.getMessages() });
-  }
+  };
 
-  private async handleFindAnswers(
+  handleFindAnswers = async (
     ctx: HandlerContext<null>,
     ev: FindAnswersEvent,
-  ) {
+  ): Promise<InputEvent> => {
     const { toolCalls } = ev.data;
     if (!this.queryEngineTools) {
       throw new Error("Query engine tool is not available");
@@ -244,12 +253,12 @@ export class FormFillingWorkflow extends Workflow<
       this.memory.put(toolMsg);
     }
     return new InputEvent({ input: this.memory.getMessages() });
-  }
+  };
 
-  private async handleFillMissingCells(
+  handleFillMissingCells = async (
     ctx: HandlerContext<null>,
     ev: FillMissingCellsEvent,
-  ) {
+  ): Promise<InputEvent> => {
     const { toolCalls } = ev.data;
 
     const toolMsgs = await callTools({
@@ -262,5 +271,5 @@ export class FormFillingWorkflow extends Workflow<
       this.memory.put(toolMsg);
     }
     return new InputEvent({ input: this.memory.getMessages() });
-  }
+  };
 }