fix: bring back output parser

langchain-ai · Oct 16, 2023 · dc4320d · dc4320d
1 parent 9546e50
commit dc4320d
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 16 deletions.
diff --git a/langchain/src/agents/agent.ts b/langchain/src/agents/agent.ts
@@ -19,6 +19,7 @@ import {
   StoppingMethod,
 } from "./types.js";
 import { Runnable } from "../schema/runnable/base.js";
+import { StringOutputParser } from "../schema/output_parser.js";
 
 /**
  * Record type for arguments passed to output parsers.
@@ -416,15 +417,11 @@ export abstract class Agent<
     }
 
     const output = await this.runnable.invoke(newInputs, callbackManager);
-    console.log({
-      newInputs,
-      output,
-    });
-    return output;
-    // if (!this.outputParser) {
-    //   throw new Error("Output parser not set");
-    // }
-    // return this.outputParser.parse(output, callbackManager);
+
+    if (!this.outputParser) {
+      throw new Error("Output parser not set");
+    }
+    return this.outputParser.parse(output, callbackManager);
   }
 
   /**

diff --git a/langchain/src/agents/tests/agent.int.test.ts b/langchain/src/agents/tests/agent.int.test.ts
@@ -35,7 +35,12 @@ test("Run agent from hub", async () => {
     input:
       "Who is Olivia Wilde's boyfriend? What is his current age raised to the 0.23 power?",
   });
-  console.log(res);
+  console.log(
+    {
+      res,
+    },
+    "Run agent from hub response"
+  );
 });
 
 test("Pass runnable to agent executor", async () => {
@@ -75,7 +80,12 @@ test("Pass runnable to agent executor", async () => {
     input:
       "Who is Olivia Wilde's boyfriend? What is his current age raised to the 0.23 power?",
   });
-  console.log(res);
+  console.log(
+    {
+      res,
+    },
+    "Pass runnable to agent executor"
+  );
 });
 
 test("Run agent locally", async () => {
@@ -98,8 +108,12 @@ test("Run agent locally", async () => {
   console.log(`Executing with input "${input}"...`);
 
   const result = await executor.call({ input });
-
-  console.log(`Got output ${result.output}`);
+  console.log(
+    {
+      result,
+    },
+    "Run agent locally"
+  );
 });
 
 test("Run agent with an abort signal", async () => {
@@ -184,9 +198,12 @@ test("Run tool web-browser", async () => {
   console.log(`Executing with input "${input}"...`);
 
   const result = await executor.call({ input });
-
-  console.log(`Got output ${result.output}`);
-
+  console.log(
+    {
+      result,
+    },
+    "Run tool web-browser"
+  );
   expect(result.intermediateSteps.length).toBeGreaterThanOrEqual(1);
   expect(result.intermediateSteps[0].action.tool).toEqual("web-browser");
   expect(result.output).not.toEqual("");