Merge pull request langchain-ai#902 from jacobrosenthal/jacob/async-construct

feat: agent make constructScratchPad async

Author: nfcampos

langchain/src/agents/agent.ts

@@ -262,7 +262,9 @@ export abstract class Agent extends BaseSingleActionAgent {
   /**
    * Construct a scratchpad to let the agent continue its thought process
    */
-  constructScratchPad(steps: AgentStep[]): string | BaseChatMessage[] {
+  async constructScratchPad(
+    steps: AgentStep[]
+  ): Promise<string | BaseChatMessage[]> {
     return steps.reduce(
       (thoughts, { action, observation }) =>
         thoughts +
@@ -280,7 +282,7 @@ export abstract class Agent extends BaseSingleActionAgent {
     inputs: ChainValues,
     suffix?: string
   ): Promise<AgentAction | AgentFinish> {
-    const thoughts = this.constructScratchPad(steps);
+    const thoughts = await this.constructScratchPad(steps);
     const newInputs: ChainValues = {
       ...inputs,
       agent_scratchpad: suffix ? `${thoughts}${suffix}` : thoughts,

langchain/src/agents/chat/index.ts

@@ -65,8 +65,8 @@ export class ChatAgent extends Agent {
     return new ChatAgentOutputParser();
   }
 
-  constructScratchPad(steps: AgentStep[]): string {
-    const agentScratchpad = super.constructScratchPad(steps);
+  async constructScratchPad(steps: AgentStep[]): Promise<string> {
+    const agentScratchpad = await super.constructScratchPad(steps);
     if (agentScratchpad) {
       return `This was your previous work (but I haven't seen any of it! I only see what you return as final answer):\n${agentScratchpad}`;
     }

langchain/src/agents/chat_convo/index.ts

@@ -75,7 +75,7 @@ export class ChatConversationalAgent extends Agent {
     }
   }
 
-  constructScratchPad(steps: AgentStep[]): BaseChatMessage[] {
+  async constructScratchPad(steps: AgentStep[]): Promise<BaseChatMessage[]> {
     const thoughts: BaseChatMessage[] = [];
     for (const step of steps) {
       thoughts.push(new AIChatMessage(step.action.log));