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

support stable diffusion plugin (#49)

Browse Source
This commit is contained in:
Hk-Gosuto
2023-10-17 12:52:33 +08:00
committed by GitHub
parent 3454ea3602
commit 516f3f4cb1
14 changed files with 269 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -21,6 +21,7 @@ import { DynamicTool, Tool } from "langchain/tools";
import { DallEAPIWrapper } from "@/app/api/langchain-tools/dalle_image_generator";
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";
const serverConfig = getServerSideConfig();
@@ -228,10 +229,13 @@ async function handle(req: NextRequest) {
const webBrowserTool = new WebBrowser({ model, embeddings });
const calculatorTool = new Calculator();
const dallEAPITool = new DallEAPIWrapper(apiKey, baseUrl);
const stableDiffusionTool = new StableDiffusionWrapper();
if (useTools.includes("web-search")) tools.push(searchTool);
if (useTools.includes(webBrowserTool.name)) tools.push(webBrowserTool);
if (useTools.includes(calculatorTool.name)) tools.push(calculatorTool);
if (useTools.includes(dallEAPITool.name)) tools.push(dallEAPITool);
if (useTools.includes(stableDiffusionTool.name))
tools.push(stableDiffusionTool);
useTools.forEach((toolName) => {
if (toolName) {