GithubProxy/ChatGPT-Next-Web
1
0
Fork 1
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-05-29 00:50:22 +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 { BaseLanguageModel } from "langchain/dist/base_language";
import { formatDocumentsAsString } from "langchain/util/document"; import { formatDocumentsAsString } from "langchain/util/document";
import { Embeddings } from "langchain/dist/embeddings/base.js"; 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] => { export const parseInputs = (inputs: string): [string, string] => {
const [baseUrl, task] = inputs.split(",").map((input) => { const [baseUrl, task] = inputs.split(",").map((input) => {
@ -131,7 +133,10 @@ export class PDFBrowser extends Tool {
doSummary ? "a summary" : task doSummary ? "a summary" : task
} from the above text.`; } 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"; name = "pdf-browser";