GithubProxy/ChatGPT-Next-Web
1
0
Fork 1
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-05-24 14:40:22 +09:00
Code Issues Packages Projects Releases Wiki Activity

feat: agent支持自定义openaikey

Browse Source
This commit is contained in:
Hk-Gosuto 2023-08-18 13:40:04 +08:00
parent 20366fee35
commit 9a8b3f06cd
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,6 +19,7 @@ import { SerpAPI } from "langchain/tools";
import { Calculator } from "langchain/tools/calculator"; import { Calculator } from "langchain/tools/calculator";
import { DuckDuckGo } from "@/app/api/langchain-tools/duckduckgo_search"; import { DuckDuckGo } from "@/app/api/langchain-tools/duckduckgo_search";
import { HttpGetTool } from "@/app/api/langchain-tools/http_get"; import { HttpGetTool } from "@/app/api/langchain-tools/http_get";
import { ACCESS_CODE_PREFIX } from "@/app/constant";
const serverConfig = getServerSideConfig(); const serverConfig = getServerSideConfig();
@ -36,6 +37,7 @@ interface RequestBody {
frequency_penalty?: number; frequency_penalty?: number;
top_p?: number; top_p?: number;
baseUrl?: string; baseUrl?: string;
apiKey?: string;
maxIterations: number; maxIterations: number;
returnIntermediateSteps: boolean; returnIntermediateSteps: boolean;
} }
@ -67,6 +69,13 @@ async function handle(req: NextRequest) {
const transformStream = new TransformStream(); const transformStream = new TransformStream();
const writer = transformStream.writable.getWriter(); const writer = transformStream.writable.getWriter();
const reqBody: RequestBody = await req.json(); const reqBody: RequestBody = await req.json();
const authToken = req.headers.get("Authorization") ?? "";
const token = authToken.trim().replaceAll("Bearer ", "").trim();
const isOpenAiKey = !token.startsWith(ACCESS_CODE_PREFIX);
let apiKey = serverConfig.apiKey;
if (isOpenAiKey && token) {
apiKey = token;
}
const handler = BaseCallbackHandler.fromMethods({ const handler = BaseCallbackHandler.fromMethods({
async handleLLMNewToken(token: string) { async handleLLMNewToken(token: string) {
@ -209,7 +218,7 @@ async function handle(req: NextRequest) {
const llm = new ChatOpenAI( const llm = new ChatOpenAI(
{ {
modelName: reqBody.model, modelName: reqBody.model,
openAIApiKey: serverConfig.apiKey, openAIApiKey: apiKey,
temperature: reqBody.temperature, temperature: reqBody.temperature,
streaming: reqBody.stream, streaming: reqBody.stream,
topP: reqBody.top_p, topP: reqBody.top_p,