support bilibili-related plugins in edge deployments

This commit is contained in:
Sheng Fan 2024-04-10 19:53:06 +08:00
parent 45723956b5
commit d5531d0e9c
1 changed files with 12 additions and 0 deletions

View File

@ -6,6 +6,10 @@ import { Calculator } from "langchain/tools/calculator";
import { WebBrowser } from "langchain/tools/webbrowser"; import { WebBrowser } from "langchain/tools/webbrowser";
import { Embeddings } from "langchain/dist/embeddings/base.js"; import { Embeddings } from "langchain/dist/embeddings/base.js";
import { WolframAlphaTool } from "@/app/api/langchain-tools/wolframalpha"; import { WolframAlphaTool } from "@/app/api/langchain-tools/wolframalpha";
import { BilibiliVideoInfoTool } from "./bilibili_vid_info";
import { BilibiliVideoSearchTool } from "./bilibili_vid_search";
import { BilibiliVideoConclusionTool } from "./bilibili_vid_conclusion";
import { BilibiliMusicRecognitionTool } from "./bilibili_music_recognition";
export class EdgeTool { export class EdgeTool {
private apiKey: string | undefined; private apiKey: string | undefined;
@ -46,6 +50,10 @@ export class EdgeTool {
const stableDiffusionTool = new StableDiffusionWrapper(); const stableDiffusionTool = new StableDiffusionWrapper();
const arxivAPITool = new ArxivAPIWrapper(); const arxivAPITool = new ArxivAPIWrapper();
const wolframAlphaTool = new WolframAlphaTool(); const wolframAlphaTool = new WolframAlphaTool();
const bilibiliVideoInfoTool = new BilibiliVideoInfoTool();
const bilibiliVideoSearchTool = new BilibiliVideoSearchTool();
const bilibiliVideoConclusionTool = new BilibiliVideoConclusionTool();
const bilibiliMusicRecognitionTool = new BilibiliMusicRecognitionTool();
let tools = [ let tools = [
calculatorTool, calculatorTool,
webBrowserTool, webBrowserTool,
@ -53,6 +61,10 @@ export class EdgeTool {
stableDiffusionTool, stableDiffusionTool,
arxivAPITool, arxivAPITool,
wolframAlphaTool, wolframAlphaTool,
bilibiliVideoInfoTool,
bilibiliVideoSearchTool,
bilibiliMusicRecognitionTool,
bilibiliVideoConclusionTool,
]; ];
return tools; return tools;
} }