xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-11-13 12:43:42 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

feat: add pdf browser plugin

Browse Source
This commit is contained in:
Hk-Gosuto
2023-11-23 13:32:16 +08:00
parent 787d65fc51
commit a8e19a81ff
5 changed files with 734 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/api/langchain/tool/agent/route.ts
View File

@@ -23,6 +23,7 @@ import { BaiduSearch } from "@/app/api/langchain-tools/baidu_search";
import { GoogleSearch } from "@/app/api/langchain-tools/google_search";
import { StableDiffusionWrapper } from "@/app/api/langchain-tools/stable_diffusion_image_generator";
import { ArxivAPIWrapper } from "@/app/api/langchain-tools/arxiv";
import { PDFBrowser } from "@/app/api/langchain-tools/pdf_browser";
const serverConfig = getServerSideConfig();
@@ -254,6 +255,9 @@ async function handle(req: NextRequest) {
tools.push(stableDiffusionTool);
if (useTools.includes(arxivAPITool.name)) tools.push(arxivAPITool);
const pdfBrowserTool = new PDFBrowser(model, embeddings);
tools.push(pdfBrowserTool);
useTools.forEach((toolName) => {
if (toolName) {
var tool = langchainTools[