From 5e43a4ff2db320fdadf1281100884326a5fd10e0 Mon Sep 17 00:00:00 2001 From: Hk-Gosuto Date: Mon, 12 Feb 2024 23:10:27 +0800 Subject: [PATCH] chore: update pdf browser plugin --- app/api/langchain-tools/pdf_browser.ts | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/app/api/langchain-tools/pdf_browser.ts b/app/api/langchain-tools/pdf_browser.ts index 2a607f5ed..4fb6d273b 100644 --- a/app/api/langchain-tools/pdf_browser.ts +++ b/app/api/langchain-tools/pdf_browser.ts @@ -11,6 +11,8 @@ import { CallbackManagerForToolRun } from "@langchain/core/callbacks/manager"; import { BaseLanguageModel } from "langchain/dist/base_language"; import { formatDocumentsAsString } from "langchain/util/document"; import { Embeddings } from "langchain/dist/embeddings/base.js"; +import { RunnableSequence } from "@langchain/core/runnables"; +import { StringOutputParser } from "@langchain/core/output_parsers"; export const parseInputs = (inputs: string): [string, string] => { const [baseUrl, task] = inputs.split(",").map((input) => { @@ -131,7 +133,10 @@ export class PDFBrowser extends Tool { doSummary ? "a summary" : task } from the above text.`; - return this.model.predict(input, undefined, runManager?.getChild()); + console.log("[pdf-browser]", input); + + const chain = RunnableSequence.from([this.model, new StringOutputParser()]); + return chain.invoke(input, runManager?.getChild()); } name = "pdf-browser";