add plugin: bili music recognition

This commit is contained in:
Sheng Fan
2024-04-06 23:14:53 +08:00
parent c7f1358709
commit 5743ceda59
3 changed files with 110 additions and 0 deletions

View File

@@ -9,6 +9,7 @@ import { WebBrowser } from "langchain/tools/webbrowser";
import { WolframAlphaTool } from "@/app/api/langchain-tools/wolframalpha";
import { BilibiliVideoInfoTool } from "./bilibili_vid_info";
import { BilibiliVideoSearchTool } from "./bilibili_vid_search";
import { BilibiliMusicRecognitionTool } from "./bilibili_music_recognition";
export class NodeJSTool {
private apiKey: string | undefined;
@@ -52,6 +53,7 @@ export class NodeJSTool {
const pdfBrowserTool = new PDFBrowser(this.model, this.embeddings);
const bilibiliVideoInfoTool = new BilibiliVideoInfoTool();
const bilibiliVideoSearchTool = new BilibiliVideoSearchTool();
const bilibiliMusicRecognitionTool = new BilibiliMusicRecognitionTool();
let tools = [
calculatorTool,
webBrowserTool,
@@ -62,6 +64,7 @@ export class NodeJSTool {
pdfBrowserTool,
bilibiliVideoInfoTool,
bilibiliVideoSearchTool,
bilibiliMusicRecognitionTool,
];
return tools;
}