mirror of
https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git
synced 2025-10-01 23:56:39 +08:00
243 lines
7.1 KiB
TypeScript
243 lines
7.1 KiB
TypeScript
import md5 from "spark-md5";
|
||
import { DEFAULT_MODELS, DEFAULT_GA_ID } from "../constant";
|
||
|
||
declare global {
|
||
namespace NodeJS {
|
||
interface ProcessEnv {
|
||
PROXY_URL?: string; // docker only
|
||
|
||
OPENAI_API_KEY?: string;
|
||
CODE?: string;
|
||
|
||
BASE_URL?: string;
|
||
OPENAI_ORG_ID?: string; // openai only
|
||
|
||
// @ts-ignore
|
||
VERCEL?: string;
|
||
BUILD_MODE?: "standalone" | "export";
|
||
BUILD_APP?: string; // is building desktop app
|
||
|
||
HIDE_USER_API_KEY?: string; // disable user's api key input
|
||
DISABLE_GPT4?: string; // allow user to use gpt-4 or not
|
||
ENABLE_BALANCE_QUERY?: string; // allow user to query balance or not
|
||
DISABLE_FAST_LINK?: string; // disallow parse settings from url or not
|
||
CUSTOM_MODELS?: string; // to control custom models
|
||
DEFAULT_MODEL?: string; // to control default model in every new chat window
|
||
|
||
// stability only
|
||
STABILITY_URL?: string;
|
||
STABILITY_API_KEY?: string;
|
||
|
||
// azure only
|
||
AZURE_URL?: string; // https://{azure-url}/openai/deployments/{deploy-name}
|
||
AZURE_API_KEY?: string;
|
||
AZURE_API_VERSION?: string;
|
||
AZURE_VOICE_KEY?: string;
|
||
|
||
// google only
|
||
GOOGLE_API_KEY?: string;
|
||
GOOGLE_URL?: string;
|
||
|
||
// google tag manager
|
||
GTM_ID?: string;
|
||
|
||
// anthropic only
|
||
ANTHROPIC_URL?: string;
|
||
ANTHROPIC_API_KEY?: string;
|
||
ANTHROPIC_API_VERSION?: string;
|
||
|
||
// baidu only
|
||
BAIDU_URL?: string;
|
||
BAIDU_API_KEY?: string;
|
||
BAIDU_SECRET_KEY?: string;
|
||
|
||
// bytedance only
|
||
BYTEDANCE_URL?: string;
|
||
BYTEDANCE_API_KEY?: string;
|
||
|
||
// alibaba only
|
||
ALIBABA_URL?: string;
|
||
ALIBABA_API_KEY?: string;
|
||
|
||
// tencent only
|
||
TENCENT_URL?: string;
|
||
TENCENT_SECRET_KEY?: string;
|
||
TENCENT_SECRET_ID?: string;
|
||
|
||
// moonshot only
|
||
MOONSHOT_URL?: string;
|
||
MOONSHOT_API_KEY?: string;
|
||
|
||
// iflytek only
|
||
IFLYTEK_URL?: string;
|
||
IFLYTEK_API_KEY?: string;
|
||
IFLYTEK_API_SECRET?: string;
|
||
|
||
// custom template for preprocessing user input
|
||
DEFAULT_INPUT_TEMPLATE?: string;
|
||
}
|
||
}
|
||
}
|
||
|
||
const ACCESS_CODES = (function getAccessCodes(): Set<string> {
|
||
const code = process.env.CODE;
|
||
|
||
try {
|
||
const codes = (code?.split(",") ?? [])
|
||
.filter((v) => !!v)
|
||
.map((v) => md5.hash(v.trim()));
|
||
return new Set(codes);
|
||
} catch (e) {
|
||
return new Set();
|
||
}
|
||
})();
|
||
|
||
function getApiKey(keys?: string) {
|
||
const apiKeyEnvVar = keys ?? "";
|
||
const apiKeys = apiKeyEnvVar.split(",").map((v) => v.trim());
|
||
const randomIndex = Math.floor(Math.random() * apiKeys.length);
|
||
const apiKey = apiKeys[randomIndex];
|
||
if (apiKey) {
|
||
// console.log(
|
||
// `[Server Config] using ${randomIndex + 1} of ${
|
||
// apiKeys.length
|
||
// } api key - ${apiKey}`,
|
||
// );
|
||
}
|
||
|
||
return apiKey;
|
||
}
|
||
|
||
export const getServerSideConfig = () => {
|
||
if (typeof process === "undefined") {
|
||
throw Error(
|
||
"[Server Config] you are importing a nodejs-only module outside of nodejs",
|
||
);
|
||
}
|
||
|
||
const disableGPT4 = !!process.env.DISABLE_GPT4;
|
||
let customModels = process.env.CUSTOM_MODELS ?? "";
|
||
let defaultModel = process.env.DEFAULT_MODEL ?? "";
|
||
|
||
if (disableGPT4) {
|
||
if (customModels) customModels += ",";
|
||
customModels += DEFAULT_MODELS.filter(
|
||
(m) =>
|
||
(m.name.startsWith("gpt-4") || m.name.startsWith("chatgpt-4o")) &&
|
||
!m.name.startsWith("gpt-4o-mini"),
|
||
)
|
||
.map((m) => "-" + m.name)
|
||
.join(",");
|
||
if (
|
||
(defaultModel.startsWith("gpt-4") ||
|
||
defaultModel.startsWith("chatgpt-4o")) &&
|
||
!defaultModel.startsWith("gpt-4o-mini")
|
||
)
|
||
defaultModel = "";
|
||
}
|
||
|
||
const isStability = !!process.env.STABILITY_API_KEY;
|
||
|
||
const isAzure = !!process.env.AZURE_URL;
|
||
const isGoogle = !!process.env.GOOGLE_API_KEY;
|
||
const isAnthropic = !!process.env.ANTHROPIC_API_KEY;
|
||
const isTencent = !!process.env.TENCENT_API_KEY;
|
||
// 需要一个函数来判断请求中模型是否为微软的。
|
||
// 当前逻辑，gpt-4-32k模型为微软，别的不是
|
||
// const isAzure = !!process.env.AZURE_URL;
|
||
const hasAzure = !!process.env.AZURE_URL;
|
||
|
||
const isBaidu = !!process.env.BAIDU_API_KEY;
|
||
const isBytedance = !!process.env.BYTEDANCE_API_KEY;
|
||
const isAlibaba = !!process.env.ALIBABA_API_KEY;
|
||
const isMoonshot = !!process.env.MOONSHOT_API_KEY;
|
||
const isIflytek = !!process.env.IFLYTEK_API_KEY;
|
||
// const apiKeyEnvVar = process.env.OPENAI_API_KEY ?? "";
|
||
// const apiKeys = apiKeyEnvVar.split(",").map((v) => v.trim());
|
||
// const randomIndex = Math.floor(Math.random() * apiKeys.length);
|
||
// const apiKey = apiKeys[randomIndex];
|
||
// console.log(
|
||
// `[Server Config] using ${randomIndex + 1} of ${apiKeys.length} api key`,
|
||
// );
|
||
|
||
const allowedWebDevEndpoints = (
|
||
process.env.WHITE_WEBDEV_ENDPOINTS ?? ""
|
||
).split(",");
|
||
|
||
return {
|
||
baseUrl: process.env.BASE_URL,
|
||
apiKey: getApiKey(process.env.OPENAI_API_KEY),
|
||
openaiOrgId: process.env.OPENAI_ORG_ID,
|
||
|
||
isStability,
|
||
stabilityUrl: process.env.STABILITY_URL,
|
||
stabilityApiKey: getApiKey(process.env.STABILITY_API_KEY),
|
||
|
||
isAzure,
|
||
azureUrl: process.env.AZURE_URL,
|
||
azureApiKey: getApiKey(process.env.AZURE_API_KEY),
|
||
azureApiVersion: process.env.AZURE_API_VERSION,
|
||
|
||
azureVoiceKey: process.env.AZURE_VOICE_KEY ?? "",
|
||
|
||
isGoogle,
|
||
googleApiKey: getApiKey(process.env.GOOGLE_API_KEY),
|
||
googleUrl: process.env.GOOGLE_URL,
|
||
|
||
isAnthropic,
|
||
anthropicApiKey: getApiKey(process.env.ANTHROPIC_API_KEY),
|
||
anthropicApiVersion: process.env.ANTHROPIC_API_VERSION,
|
||
anthropicUrl: process.env.ANTHROPIC_URL,
|
||
|
||
isBaidu,
|
||
baiduUrl: process.env.BAIDU_URL,
|
||
baiduApiKey: getApiKey(process.env.BAIDU_API_KEY),
|
||
baiduSecretKey: process.env.BAIDU_SECRET_KEY,
|
||
|
||
isBytedance,
|
||
bytedanceApiKey: getApiKey(process.env.BYTEDANCE_API_KEY),
|
||
bytedanceUrl: process.env.BYTEDANCE_URL,
|
||
|
||
isAlibaba,
|
||
alibabaUrl: process.env.ALIBABA_URL,
|
||
alibabaApiKey: getApiKey(process.env.ALIBABA_API_KEY),
|
||
|
||
isTencent,
|
||
tencentUrl: process.env.TENCENT_URL,
|
||
tencentSecretKey: getApiKey(process.env.TENCENT_SECRET_KEY),
|
||
tencentSecretId: process.env.TENCENT_SECRET_ID,
|
||
|
||
isMoonshot,
|
||
moonshotUrl: process.env.MOONSHOT_URL,
|
||
moonshotApiKey: getApiKey(process.env.MOONSHOT_API_KEY),
|
||
|
||
isIflytek,
|
||
iflytekUrl: process.env.IFLYTEK_URL,
|
||
iflytekApiKey: process.env.IFLYTEK_API_KEY,
|
||
iflytekApiSecret: process.env.IFLYTEK_API_SECRET,
|
||
|
||
cloudflareAccountId: process.env.CLOUDFLARE_ACCOUNT_ID,
|
||
cloudflareKVNamespaceId: process.env.CLOUDFLARE_KV_NAMESPACE_ID,
|
||
cloudflareKVApiKey: getApiKey(process.env.CLOUDFLARE_KV_API_KEY),
|
||
cloudflareKVTTL: process.env.CLOUDFLARE_KV_TTL,
|
||
|
||
gtmId: process.env.GTM_ID,
|
||
gaId: process.env.GA_ID || DEFAULT_GA_ID,
|
||
|
||
needCode: ACCESS_CODES.size > 0,
|
||
code: process.env.CODE,
|
||
codes: ACCESS_CODES,
|
||
|
||
proxyUrl: process.env.PROXY_URL,
|
||
isVercel: !!process.env.VERCEL,
|
||
|
||
hideUserApiKey: !!process.env.HIDE_USER_API_KEY,
|
||
disableGPT4,
|
||
hideBalanceQuery: !process.env.ENABLE_BALANCE_QUERY,
|
||
disableFastLink: !!process.env.DISABLE_FAST_LINK,
|
||
customModels,
|
||
defaultModel,
|
||
allowedWebDevEndpoints,
|
||
};
|
||
};
|