GithubProxy/ChatGPT-Next-Web
1
0
Fork 1
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-05-26 07:30:18 +09:00
Code Issues Packages Projects Releases Wiki Activity

chore: update pdf browser plugin

Browse Source
This commit is contained in:
Hk-Gosuto 2024-02-12 23:10:27 +08:00
parent 8691e95f12
commit 5e43a4ff2d
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/api/langchain-tools/pdf_browser.ts
View File

@ -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";