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

add video conclusion plugin

Browse Source
This commit is contained in:
Sheng Fan
2024-04-09 22:04:37 +08:00
parent c77c14c69a
commit 74ac8b612b
3 changed files with 151 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/api/langchain-tools/nodejs_tools.ts
View File

@@ -11,6 +11,7 @@ import { BilibiliVideoInfoTool } from "./bilibili_vid_info";
import { BilibiliVideoSearchTool } from "./bilibili_vid_search";
import { BilibiliMusicRecognitionTool } from "./bilibili_music_recognition";
import { RAGSearch } from "./rag_search";
import { BilibiliVideoConclusionTool } from "./bilibili_vid_conclusion";
export class NodeJSTool {
private apiKey: string | undefined;
@@ -57,6 +58,7 @@ export class NodeJSTool {
const bilibiliVideoInfoTool = new BilibiliVideoInfoTool();
const bilibiliVideoSearchTool = new BilibiliVideoSearchTool();
const bilibiliMusicRecognitionTool = new BilibiliMusicRecognitionTool();
const bilibiliVideoConclusionTool = new BilibiliVideoConclusionTool();
let tools = [
calculatorTool,
webBrowserTool,
@@ -68,6 +70,7 @@ export class NodeJSTool {
bilibiliVideoInfoTool,
bilibiliVideoSearchTool,
bilibiliMusicRecognitionTool,
bilibiliVideoConclusionTool,
];
if (!!process.env.ENABLE_RAG) {
tools.push(new RAGSearch(this.sessionId, this.model, this.ragEmbeddings));