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

…onstruct feat: agent make constructScratchPad async
lemonade-hq · Apr 20, 2023 · f78f2e9 · f78f2e9
2 parents 7b9b67d + 058f20a
commit f78f2e9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/langchain/src/agents/agent.ts b/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,

diff --git a/langchain/src/agents/chat/index.ts b/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}`;
     }

diff --git a/langchain/src/agents/chat_convo/index.ts b/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));