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

feat: add wolframAlpha plugin

Browse Source
This commit is contained in:
Hk-Gosuto
2023-12-29 19:00:01 +08:00
parent 4551abdce8
commit 2a9b36f626
4 changed files with 40 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
app/api/langchain/tool/agent/agentapi.ts
View File

@@ -16,7 +16,7 @@ import { ACCESS_CODE_PREFIX, ServiceProvider } from "@/app/constant";
import * as langchainTools from "langchain/tools";
import { HttpGetTool } from "@/app/api/langchain-tools/http_get";
import { DuckDuckGo } from "@/app/api/langchain-tools/duckduckgo_search";
import { DynamicTool, Tool } from "langchain/tools";
import { DynamicTool, Tool, WolframAlphaTool } from "langchain/tools";
import { BaiduSearch } from "@/app/api/langchain-tools/baidu_search";
import { GoogleSearch } from "@/app/api/langchain-tools/google_search";
import { useAccessStore } from "@/app/store";
@@ -282,6 +282,16 @@ export class AgentApi {
var tool = langchainTools[
toolName as keyof typeof langchainTools
] as any;
if (
toolName === "wolfram_alpha" &&
process.env.WOLFRAM_ALPHA_APP_ID
) {
const tool = new WolframAlphaTool({
appid: process.env.WOLFRAM_ALPHA_APP_ID,
});
tools.push(tool);
return;
}
if (tool) {
tools.push(new tool());
}
@@ -387,7 +397,6 @@ export class AgentApi {
[handler],
);
console.log("returning response");
return new Response(this.transformStream.readable, {
headers: { "Content-Type": "text/event-stream" },
});