refactor: centralize runtime config loading
- Move .env parsing and runtime config reload logic into Environment - Reload runtime config and system prompt when source files change - Gate unsafe eval and file tools behind explicit environment flags - Rename datetime tool to get_datetime and improve tool prompts - Return structured weather tool responses - Preserve assistant thinking and aggregate tool calls across stream chunks
This commit is contained in:
+424
-124
@@ -1,36 +1,224 @@
|
||||
import fs from "node:fs";
|
||||
import path from "node:path";
|
||||
import {parse as parseDotEnv} from "dotenv";
|
||||
import {z} from "zod";
|
||||
|
||||
import {saveData} from "../db/database";
|
||||
import {Answers} from "../model/answers";
|
||||
import {ifTrue} from "../util/utils";
|
||||
import {AiProvider} from "../model/ai-provider";
|
||||
import {ImageHandleFallbackPolicy, ImageHandlePolicy, RateLimitFallbackPolicy} from "./policies";
|
||||
|
||||
type EnvRecord = Record<string, string>;
|
||||
type StringEnumLike = Record<string, string>;
|
||||
type StringEnumValue<T extends StringEnumLike> = T[keyof T];
|
||||
|
||||
function normalizeString(value: unknown): string | undefined {
|
||||
if (typeof value !== "string") {
|
||||
return undefined;
|
||||
}
|
||||
|
||||
const trimmed = value.trim();
|
||||
return trimmed.length > 0 ? trimmed : undefined;
|
||||
}
|
||||
|
||||
const optionalStringSchema = z
|
||||
.preprocess(normalizeString, z.string().optional())
|
||||
.optional()
|
||||
.catch(undefined);
|
||||
|
||||
function stringWithDefaultSchema(defaultValue: string) {
|
||||
return z
|
||||
.preprocess(value => {
|
||||
const normalized = normalizeString(value);
|
||||
return normalized ?? defaultValue;
|
||||
}, z.string())
|
||||
.default(defaultValue)
|
||||
.catch(defaultValue);
|
||||
}
|
||||
|
||||
function booleanWithDefaultSchema(defaultValue: boolean) {
|
||||
return z
|
||||
.preprocess(value => {
|
||||
const normalized = normalizeString(value);
|
||||
|
||||
if (normalized === undefined) {
|
||||
return defaultValue;
|
||||
}
|
||||
|
||||
return ifTrue(normalized);
|
||||
}, z.boolean())
|
||||
.default(defaultValue)
|
||||
.catch(defaultValue);
|
||||
}
|
||||
|
||||
const optionalBooleanSchema = z
|
||||
.preprocess(value => {
|
||||
const normalized = normalizeString(value);
|
||||
return normalized === undefined ? undefined : ifTrue(normalized);
|
||||
}, z.boolean().optional())
|
||||
.optional()
|
||||
.catch(undefined);
|
||||
|
||||
function numberWithDefaultSchema(defaultValue: number) {
|
||||
return z
|
||||
.preprocess(value => {
|
||||
const normalized = normalizeString(value);
|
||||
|
||||
if (normalized === undefined) {
|
||||
return defaultValue;
|
||||
}
|
||||
|
||||
const number = Number(normalized);
|
||||
return Number.isFinite(number) ? number : defaultValue;
|
||||
}, z.number())
|
||||
.catch(defaultValue);
|
||||
}
|
||||
|
||||
function positiveIntWithDefaultSchema(defaultValue: number) {
|
||||
return z
|
||||
.preprocess(value => {
|
||||
const normalized = normalizeString(value);
|
||||
|
||||
if (normalized === undefined) {
|
||||
return defaultValue;
|
||||
}
|
||||
|
||||
const number = Number(normalized);
|
||||
|
||||
if (!Number.isSafeInteger(number) || number <= 0) {
|
||||
return defaultValue;
|
||||
}
|
||||
|
||||
return number;
|
||||
}, z.number().int().positive())
|
||||
.default(defaultValue)
|
||||
.catch(defaultValue);
|
||||
}
|
||||
|
||||
function enumWithDefaultSchema<T extends StringEnumLike>(
|
||||
enumObject: T,
|
||||
defaultValue: StringEnumValue<T>,
|
||||
) {
|
||||
const values = Object.values(enumObject) as StringEnumValue<T>[];
|
||||
|
||||
return z
|
||||
.preprocess(value => {
|
||||
const normalized = normalizeString(value);
|
||||
|
||||
if (normalized === undefined) {
|
||||
return defaultValue;
|
||||
}
|
||||
|
||||
return values.includes(normalized as StringEnumValue<T>)
|
||||
? normalized
|
||||
: defaultValue;
|
||||
}, z.custom<StringEnumValue<T>>((value): value is StringEnumValue<T> => {
|
||||
return typeof value === "string"
|
||||
&& values.includes(value as StringEnumValue<T>);
|
||||
}))
|
||||
.default(defaultValue)
|
||||
.catch(defaultValue);
|
||||
}
|
||||
|
||||
const StartupEnvSchema = z.object({
|
||||
BOT_TOKEN: stringWithDefaultSchema(""),
|
||||
TEST_ENVIRONMENT: booleanWithDefaultSchema(false),
|
||||
IS_DOCKER: optionalBooleanSchema,
|
||||
});
|
||||
|
||||
const RuntimeEnvSchema = z.object({
|
||||
CREATOR_ID: numberWithDefaultSchema(0),
|
||||
BOT_PREFIX: stringWithDefaultSchema(""),
|
||||
CHAT_IDS_WHITELIST: optionalStringSchema,
|
||||
ONLY_FOR_CREATOR_MODE: booleanWithDefaultSchema(false),
|
||||
ENABLE_UNSAFE_EVAL: booleanWithDefaultSchema(false),
|
||||
MAX_PHOTO_SIZE: positiveIntWithDefaultSchema(1280),
|
||||
PROCESS_LINKS: booleanWithDefaultSchema(false),
|
||||
|
||||
RATE_LIMIT_FALLBACK_POLICY: enumWithDefaultSchema(
|
||||
RateLimitFallbackPolicy,
|
||||
RateLimitFallbackPolicy.NOTIFY_USER,
|
||||
),
|
||||
|
||||
IMAGE_HANDLE_POLICY: enumWithDefaultSchema(
|
||||
ImageHandlePolicy,
|
||||
ImageHandlePolicy.HANDLE_IF_CAPABLE,
|
||||
),
|
||||
|
||||
IMAGE_HANDLE_FALLBACK_POLICY: enumWithDefaultSchema(
|
||||
ImageHandleFallbackPolicy,
|
||||
ImageHandleFallbackPolicy.NOTIFY_USER,
|
||||
),
|
||||
|
||||
BRAVE_SEARCH_API_KEY: optionalStringSchema,
|
||||
OPEN_WEATHER_MAP_API_KEY: optionalStringSchema,
|
||||
|
||||
FILE_TOOLS_ROOT_DIR: optionalStringSchema,
|
||||
|
||||
DEFAULT_AI_PROVIDER: enumWithDefaultSchema(
|
||||
AiProvider,
|
||||
AiProvider.OLLAMA,
|
||||
),
|
||||
|
||||
USE_NAMES_IN_PROMPT: booleanWithDefaultSchema(false),
|
||||
USE_SYSTEM_PROMPT: booleanWithDefaultSchema(true),
|
||||
|
||||
SEND_TIME_TOOK: optionalBooleanSchema,
|
||||
|
||||
OLLAMA_API_KEY: optionalStringSchema,
|
||||
OLLAMA_ADDRESS: optionalStringSchema,
|
||||
OLLAMA_MODEL: stringWithDefaultSchema("gemma3:4b"),
|
||||
OLLAMA_IMAGE_MODEL: optionalStringSchema,
|
||||
OLLAMA_THINK_MODEL: optionalStringSchema,
|
||||
|
||||
GEMINI_API_KEY: optionalStringSchema,
|
||||
GEMINI_MODEL: stringWithDefaultSchema("gemini-2.5-flash-lite"),
|
||||
GEMINI_IMAGE_MODEL: stringWithDefaultSchema("gemini-2.5-flash-image"),
|
||||
|
||||
MISTRAL_API_KEY: optionalStringSchema,
|
||||
MISTRAL_MODEL: stringWithDefaultSchema("mistral-tiny-latest"),
|
||||
|
||||
OPENAI_BASE_URL: optionalStringSchema,
|
||||
OPENAI_API_KEY: optionalStringSchema,
|
||||
OPENAI_MODEL: stringWithDefaultSchema("gpt-4.1-nano"),
|
||||
OPENAI_IMAGE_MODEL: stringWithDefaultSchema("gpt-image-1-mini"),
|
||||
});
|
||||
|
||||
type StartupEnv = z.infer<typeof StartupEnvSchema>;
|
||||
type RuntimeEnv = z.infer<typeof RuntimeEnvSchema>;
|
||||
|
||||
export class Environment {
|
||||
static BOT_TOKEN: string;
|
||||
static TEST_ENVIRONMENT: boolean;
|
||||
private static readonly ENV_FILE_PATH = path.resolve(".env");
|
||||
|
||||
private static lastEnvMtimeMs: number | undefined;
|
||||
private static lastSystemPromptMtimeMs: number | undefined;
|
||||
|
||||
static BOT_TOKEN: string = "";
|
||||
static TEST_ENVIRONMENT: boolean = false;
|
||||
static ADMIN_IDS: Set<number> = new Set<number>();
|
||||
static MUTED_IDS: Set<number> = new Set<number>();
|
||||
static CHAT_IDS_WHITELIST: Set<number> = new Set<number>();
|
||||
static BOT_PREFIX: string;
|
||||
static CREATOR_ID: number;
|
||||
static IS_DOCKER: boolean;
|
||||
static DATA_PATH: string;
|
||||
static BOT_PREFIX: string = "";
|
||||
static CREATOR_ID: number = 0;
|
||||
static IS_DOCKER: boolean = false;
|
||||
static DATA_PATH: string = "data";
|
||||
static DB_FILE_NAME: string = "database.db";
|
||||
static DB_PATH: string;
|
||||
static DB_PATH: string = "file:" + path.join(Environment.DATA_PATH, Environment.DB_FILE_NAME);
|
||||
|
||||
static ONLY_FOR_CREATOR_MODE: boolean;
|
||||
static ONLY_FOR_CREATOR_MODE: boolean = false;
|
||||
|
||||
static ENABLE_UNSAFE_EVAL: boolean;
|
||||
static ENABLE_UNSAFE_EVAL: boolean = false;
|
||||
|
||||
static ANSWERS: Answers;
|
||||
|
||||
static MAX_PHOTO_SIZE: number;
|
||||
static MAX_PHOTO_SIZE: number = 1280;
|
||||
|
||||
static PROCESS_LINKS: boolean;
|
||||
static PROCESS_LINKS: boolean = false;
|
||||
|
||||
static RATE_LIMIT_FALLBACK_POLICY: RateLimitFallbackPolicy;
|
||||
static IMAGE_HANDLE_POLICY: ImageHandlePolicy;
|
||||
static IMAGE_HANDLE_FALLBACK_POLICY: ImageHandleFallbackPolicy;
|
||||
static RATE_LIMIT_FALLBACK_POLICY: RateLimitFallbackPolicy = RateLimitFallbackPolicy.NOTIFY_USER;
|
||||
static IMAGE_HANDLE_POLICY: ImageHandlePolicy = ImageHandlePolicy.HANDLE_IF_CAPABLE;
|
||||
static IMAGE_HANDLE_FALLBACK_POLICY: ImageHandleFallbackPolicy = ImageHandleFallbackPolicy.NOTIFY_USER;
|
||||
|
||||
static BRAVE_SEARCH_API_KEY?: string;
|
||||
static OPEN_WEATHER_MAP_API_KEY?: string;
|
||||
@@ -38,30 +226,30 @@ export class Environment {
|
||||
static FILE_TOOLS_ROOT_DIR?: string;
|
||||
|
||||
// AI Stuff
|
||||
static DEFAULT_AI_PROVIDER: AiProvider;
|
||||
static DEFAULT_AI_PROVIDER: AiProvider = AiProvider.OLLAMA;
|
||||
|
||||
static SYSTEM_PROMPT?: string;
|
||||
static USE_NAMES_IN_PROMPT: boolean;
|
||||
static USE_SYSTEM_PROMPT: boolean;
|
||||
static SEND_TIME_TOOK: boolean;
|
||||
static USE_NAMES_IN_PROMPT: boolean = false;
|
||||
static USE_SYSTEM_PROMPT: boolean = true;
|
||||
static SEND_TIME_TOOK: boolean = false;
|
||||
|
||||
static OLLAMA_API_KEY?: string;
|
||||
static OLLAMA_ADDRESS?: string;
|
||||
static OLLAMA_MODEL?: string;
|
||||
static OLLAMA_IMAGE_MODEL?: string;
|
||||
static OLLAMA_THINK_MODEL?: string;
|
||||
static OLLAMA_MODEL: string = "gemma3:4b";
|
||||
static OLLAMA_IMAGE_MODEL: string = Environment.OLLAMA_MODEL;
|
||||
static OLLAMA_THINK_MODEL: string = Environment.OLLAMA_MODEL;
|
||||
|
||||
static GEMINI_API_KEY?: string;
|
||||
static GEMINI_MODEL: string;
|
||||
static GEMINI_IMAGE_MODEL: string;
|
||||
static GEMINI_MODEL: string = "gemini-2.5-flash-lite";
|
||||
static GEMINI_IMAGE_MODEL: string = "gemini-2.5-flash-image";
|
||||
|
||||
static MISTRAL_API_KEY?: string;
|
||||
static MISTRAL_MODEL: string;
|
||||
static MISTRAL_MODEL: string = "mistral-tiny-latest";
|
||||
|
||||
static OPENAI_BASE_URL?: string;
|
||||
static OPENAI_API_KEY?: string;
|
||||
static OPENAI_MODEL: string;
|
||||
static OPENAI_IMAGE_MODEL: string;
|
||||
static OPENAI_MODEL: string = "gpt-4.1-nano";
|
||||
static OPENAI_IMAGE_MODEL: string = "gpt-image-1-mini";
|
||||
|
||||
static errorText = "⚠️ Произошла ошибка.";
|
||||
static waitText = "⏳ Секунду...";
|
||||
@@ -72,106 +260,212 @@ export class Environment {
|
||||
static genImageText = "👨🎨 Генерирую изображение...";
|
||||
static ollamaCancelledText = "```Ollama\n❌ Отменено```";
|
||||
|
||||
static load() {
|
||||
Environment.BOT_TOKEN = <string>process.env.BOT_TOKEN;
|
||||
Environment.TEST_ENVIRONMENT = ifTrue(process.env.TEST_ENVIRONMENT);
|
||||
Environment.CHAT_IDS_WHITELIST = new Set(process.env.CHAT_IDS_WHITELIST?.split(",")?.map(e => parseInt(e.trim(), 10)) || []);
|
||||
Environment.BOT_PREFIX = process.env.BOT_PREFIX || "";
|
||||
Environment.CREATOR_ID = parseInt(process.env.CREATOR_ID || "");
|
||||
Environment.IS_DOCKER = ifTrue(process.env.IS_DOCKER);
|
||||
Environment.DATA_PATH = Environment.IS_DOCKER ? "/" + path.join("config", "data") : "data";
|
||||
Environment.DB_PATH = "file:" + path.join(Environment.DATA_PATH, Environment.DB_FILE_NAME);
|
||||
|
||||
Environment.ONLY_FOR_CREATOR_MODE = ifTrue(process.env.ONLY_FOR_CREATOR_MODE);
|
||||
|
||||
Environment.ENABLE_UNSAFE_EVAL = ifTrue(process.env.ENABLE_UNSAFE_EVAL);
|
||||
|
||||
Environment.MAX_PHOTO_SIZE = Number(process.env.MAX_PHOTO_SIZE || "1280");
|
||||
|
||||
Environment.PROCESS_LINKS = ifTrue(process.env.PROCESS_LINKS);
|
||||
|
||||
const rateLimitFallbackPolicy = process.env.RATE_LIMIT_FALLBACK_POLICY || "NOTIFY_USER";
|
||||
if (Object.values(RateLimitFallbackPolicy).includes(rateLimitFallbackPolicy as RateLimitFallbackPolicy)) {
|
||||
Environment.RATE_LIMIT_FALLBACK_POLICY = rateLimitFallbackPolicy as RateLimitFallbackPolicy;
|
||||
} else {
|
||||
Environment.RATE_LIMIT_FALLBACK_POLICY = RateLimitFallbackPolicy.NOTIFY_USER;
|
||||
}
|
||||
|
||||
const imageHandlePolicy = process.env.IMAGE_HANDLE_POLICY || "HANDLE_IF_CAPABLE";
|
||||
if (Object.values(ImageHandlePolicy).includes(imageHandlePolicy as ImageHandlePolicy)) {
|
||||
Environment.IMAGE_HANDLE_POLICY = imageHandlePolicy as ImageHandlePolicy;
|
||||
} else {
|
||||
Environment.IMAGE_HANDLE_POLICY = ImageHandlePolicy.HANDLE_IF_CAPABLE;
|
||||
}
|
||||
|
||||
const imageHandleFallbackPolicy = process.env.IMAGE_HANDLE_FALLBACK_POLICY || "NOTIFY_USER";
|
||||
if (Object.values(ImageHandleFallbackPolicy).includes(imageHandleFallbackPolicy as ImageHandleFallbackPolicy)) {
|
||||
Environment.IMAGE_HANDLE_FALLBACK_POLICY = imageHandleFallbackPolicy as ImageHandleFallbackPolicy;
|
||||
} else {
|
||||
Environment.IMAGE_HANDLE_FALLBACK_POLICY = ImageHandleFallbackPolicy.NOTIFY_USER;
|
||||
}
|
||||
|
||||
Environment.BRAVE_SEARCH_API_KEY = process.env.BRAVE_SEARCH_API_KEY;
|
||||
Environment.OPEN_WEATHER_MAP_API_KEY = process.env.OPEN_WEATHER_MAP_API_KEY;
|
||||
|
||||
Environment.FILE_TOOLS_ROOT_DIR = process.env.FILE_TOOLS_ROOT_DIR;
|
||||
|
||||
const aiProvider = process.env.DEFAULT_AI_PROVIDER || "OLLAMA";
|
||||
if (Object.values(AiProvider).includes(aiProvider as AiProvider)) {
|
||||
Environment.DEFAULT_AI_PROVIDER = aiProvider as AiProvider;
|
||||
} else {
|
||||
Environment.DEFAULT_AI_PROVIDER = AiProvider.OLLAMA;
|
||||
}
|
||||
|
||||
Environment.USE_NAMES_IN_PROMPT = ifTrue(process.env.USE_NAMES_IN_PROMPT);
|
||||
Environment.USE_SYSTEM_PROMPT = ifTrue(process.env.USE_SYSTEM_PROMPT || "true");
|
||||
Environment.SEND_TIME_TOOK = ifTrue(process.env.SEND_TOOK_TIME || "false");
|
||||
|
||||
Environment.OLLAMA_API_KEY = process.env.OLLAMA_API_KEY;
|
||||
Environment.OLLAMA_ADDRESS = process.env.OLLAMA_ADDRESS;
|
||||
Environment.OLLAMA_MODEL = process.env.OLLAMA_MODEL || "gemma3:4b";
|
||||
Environment.OLLAMA_IMAGE_MODEL = process.env.OLLAMA_IMAGE_MODEL || Environment.OLLAMA_MODEL;
|
||||
Environment.OLLAMA_THINK_MODEL = process.env.OLLAMA_THINK_MODEL || Environment.OLLAMA_MODEL;
|
||||
|
||||
Environment.GEMINI_API_KEY = process.env.GEMINI_API_KEY;
|
||||
Environment.GEMINI_MODEL = process.env.GEMINI_MODEL || "gemini-2.5-flash-lite";
|
||||
Environment.GEMINI_IMAGE_MODEL = process.env.GEMINI_IMAGE_MODEL || "gemini-2.5-flash-image";
|
||||
|
||||
Environment.MISTRAL_API_KEY = process.env.MISTRAL_API_KEY;
|
||||
Environment.MISTRAL_MODEL = process.env.MISTRAL_MODEL || "mistral-tiny-latest";
|
||||
|
||||
Environment.OPENAI_BASE_URL = process.env.OPENAI_BASE_URL;
|
||||
Environment.OPENAI_API_KEY = process.env.OPENAI_API_KEY;
|
||||
Environment.OPENAI_MODEL = process.env.OPENAI_MODEL || "gpt-4.1-nano";
|
||||
Environment.OPENAI_IMAGE_MODEL = process.env.OPENAI_IMAGE_MODEL || "gpt-image-1-mini";
|
||||
private static processEnvAsRecord(): EnvRecord {
|
||||
return Object.fromEntries(
|
||||
Object.entries(process.env)
|
||||
.filter((entry): entry is [string, string] => typeof entry[1] === "string"),
|
||||
);
|
||||
}
|
||||
|
||||
static setOnlyForCreatorMode(enable: boolean) {
|
||||
private static parseNumberSet(value: string | undefined): Set<number> {
|
||||
if (!value) {
|
||||
return new Set<number>();
|
||||
}
|
||||
|
||||
const numbers = value
|
||||
.split(",")
|
||||
.map(e => Number.parseInt(e.trim(), 10))
|
||||
.filter(Number.isSafeInteger);
|
||||
|
||||
return new Set<number>(numbers);
|
||||
}
|
||||
|
||||
private static getFileMtimeMs(filePath: string): number | undefined {
|
||||
try {
|
||||
return fs.statSync(filePath).mtimeMs;
|
||||
} catch (e: any) {
|
||||
if (e?.code === "ENOENT") {
|
||||
return undefined;
|
||||
}
|
||||
|
||||
throw e;
|
||||
}
|
||||
}
|
||||
|
||||
private static readEnvFile(): EnvRecord {
|
||||
if (!fs.existsSync(Environment.ENV_FILE_PATH)) {
|
||||
return {};
|
||||
}
|
||||
|
||||
const envFile = fs.readFileSync(Environment.ENV_FILE_PATH, "utf8");
|
||||
return parseDotEnv(envFile);
|
||||
}
|
||||
|
||||
private static readConfigSource(): EnvRecord {
|
||||
return {
|
||||
...Environment.processEnvAsRecord(),
|
||||
...Environment.readEnvFile(),
|
||||
};
|
||||
}
|
||||
|
||||
private static getSystemPromptPath(): string {
|
||||
return path.join(Environment.DATA_PATH, "system_prompt.txt");
|
||||
}
|
||||
|
||||
private static readSystemPrompt(): string | undefined {
|
||||
const promptPath = Environment.getSystemPromptPath();
|
||||
|
||||
if (!fs.existsSync(promptPath)) {
|
||||
return undefined;
|
||||
}
|
||||
|
||||
const prompt = fs.readFileSync(promptPath, "utf8").trim();
|
||||
return prompt.length > 0 ? prompt : undefined;
|
||||
}
|
||||
|
||||
private static applyStartupEnv(env: StartupEnv): void {
|
||||
Environment.BOT_TOKEN = env.BOT_TOKEN;
|
||||
Environment.TEST_ENVIRONMENT = env.TEST_ENVIRONMENT;
|
||||
Environment.IS_DOCKER = env.IS_DOCKER ?? false;
|
||||
|
||||
Environment.DATA_PATH = Environment.IS_DOCKER
|
||||
? "/" + path.join("config", "data")
|
||||
: "data";
|
||||
|
||||
Environment.DB_PATH = "file:" + path.join(
|
||||
Environment.DATA_PATH,
|
||||
Environment.DB_FILE_NAME,
|
||||
);
|
||||
}
|
||||
|
||||
private static applyRuntimeEnv(env: RuntimeEnv): void {
|
||||
Environment.CHAT_IDS_WHITELIST = Environment.parseNumberSet(env.CHAT_IDS_WHITELIST);
|
||||
Environment.BOT_PREFIX = env.BOT_PREFIX;
|
||||
Environment.CREATOR_ID = env.CREATOR_ID;
|
||||
Environment.ONLY_FOR_CREATOR_MODE = env.ONLY_FOR_CREATOR_MODE;
|
||||
Environment.ENABLE_UNSAFE_EVAL = env.ENABLE_UNSAFE_EVAL;
|
||||
Environment.MAX_PHOTO_SIZE = env.MAX_PHOTO_SIZE;
|
||||
Environment.PROCESS_LINKS = env.PROCESS_LINKS;
|
||||
|
||||
Environment.RATE_LIMIT_FALLBACK_POLICY = env.RATE_LIMIT_FALLBACK_POLICY;
|
||||
Environment.IMAGE_HANDLE_POLICY = env.IMAGE_HANDLE_POLICY;
|
||||
Environment.IMAGE_HANDLE_FALLBACK_POLICY = env.IMAGE_HANDLE_FALLBACK_POLICY;
|
||||
|
||||
Environment.BRAVE_SEARCH_API_KEY = env.BRAVE_SEARCH_API_KEY;
|
||||
Environment.OPEN_WEATHER_MAP_API_KEY = env.OPEN_WEATHER_MAP_API_KEY;
|
||||
|
||||
Environment.FILE_TOOLS_ROOT_DIR = env.FILE_TOOLS_ROOT_DIR
|
||||
? path.resolve(env.FILE_TOOLS_ROOT_DIR)
|
||||
: undefined;
|
||||
|
||||
Environment.DEFAULT_AI_PROVIDER = env.DEFAULT_AI_PROVIDER;
|
||||
|
||||
Environment.USE_NAMES_IN_PROMPT = env.USE_NAMES_IN_PROMPT;
|
||||
Environment.USE_SYSTEM_PROMPT = env.USE_SYSTEM_PROMPT;
|
||||
Environment.SEND_TIME_TOOK = env.SEND_TIME_TOOK ?? false;
|
||||
|
||||
Environment.OLLAMA_API_KEY = env.OLLAMA_API_KEY;
|
||||
Environment.OLLAMA_ADDRESS = env.OLLAMA_ADDRESS;
|
||||
Environment.OLLAMA_MODEL = env.OLLAMA_MODEL;
|
||||
Environment.OLLAMA_IMAGE_MODEL = env.OLLAMA_IMAGE_MODEL ?? env.OLLAMA_MODEL;
|
||||
Environment.OLLAMA_THINK_MODEL = env.OLLAMA_THINK_MODEL ?? env.OLLAMA_MODEL;
|
||||
|
||||
Environment.GEMINI_API_KEY = env.GEMINI_API_KEY;
|
||||
Environment.GEMINI_MODEL = env.GEMINI_MODEL;
|
||||
Environment.GEMINI_IMAGE_MODEL = env.GEMINI_IMAGE_MODEL;
|
||||
|
||||
Environment.MISTRAL_API_KEY = env.MISTRAL_API_KEY;
|
||||
Environment.MISTRAL_MODEL = env.MISTRAL_MODEL;
|
||||
|
||||
Environment.OPENAI_BASE_URL = env.OPENAI_BASE_URL;
|
||||
Environment.OPENAI_API_KEY = env.OPENAI_API_KEY;
|
||||
Environment.OPENAI_MODEL = env.OPENAI_MODEL;
|
||||
Environment.OPENAI_IMAGE_MODEL = env.OPENAI_IMAGE_MODEL;
|
||||
}
|
||||
|
||||
static load(): void {
|
||||
const rawEnv = Environment.readConfigSource();
|
||||
|
||||
const startupEnv = StartupEnvSchema.parse(rawEnv);
|
||||
const runtimeEnv = RuntimeEnvSchema.parse(rawEnv);
|
||||
|
||||
Environment.applyStartupEnv(startupEnv);
|
||||
Environment.applyRuntimeEnv(runtimeEnv);
|
||||
|
||||
Environment.SYSTEM_PROMPT = Environment.readSystemPrompt();
|
||||
|
||||
Environment.lastEnvMtimeMs = Environment.getFileMtimeMs(Environment.ENV_FILE_PATH);
|
||||
Environment.lastSystemPromptMtimeMs = Environment.getFileMtimeMs(Environment.getSystemPromptPath());
|
||||
}
|
||||
|
||||
static reloadRuntimeConfigIfChanged(): void {
|
||||
try {
|
||||
const envMtimeMs = Environment.getFileMtimeMs(Environment.ENV_FILE_PATH);
|
||||
const systemPromptMtimeMs = Environment.getFileMtimeMs(Environment.getSystemPromptPath());
|
||||
|
||||
const envChanged = envMtimeMs !== Environment.lastEnvMtimeMs;
|
||||
const systemPromptChanged = systemPromptMtimeMs !== Environment.lastSystemPromptMtimeMs;
|
||||
|
||||
if (!envChanged && !systemPromptChanged) {
|
||||
return;
|
||||
}
|
||||
|
||||
if (envChanged) {
|
||||
const rawEnv = Environment.readConfigSource();
|
||||
const runtimeEnv = RuntimeEnvSchema.parse(rawEnv);
|
||||
|
||||
Environment.applyRuntimeEnv(runtimeEnv);
|
||||
Environment.lastEnvMtimeMs = envMtimeMs;
|
||||
}
|
||||
|
||||
if (systemPromptChanged) {
|
||||
Environment.SYSTEM_PROMPT = Environment.readSystemPrompt();
|
||||
Environment.lastSystemPromptMtimeMs = systemPromptMtimeMs;
|
||||
}
|
||||
} catch (e) {
|
||||
console.error("Failed to reload runtime environment config", e);
|
||||
}
|
||||
}
|
||||
|
||||
static setOnlyForCreatorMode(enable: boolean): void {
|
||||
this.ONLY_FOR_CREATOR_MODE = enable;
|
||||
}
|
||||
|
||||
static setSystemPrompt(prompt: string | undefined) {
|
||||
static setBraveSearchApiKey(apiKey: string | undefined): void {
|
||||
this.BRAVE_SEARCH_API_KEY = apiKey;
|
||||
}
|
||||
|
||||
static setOpenWeatherMapApiKey(openWeatherMapApiKey: string | undefined): void {
|
||||
this.OPEN_WEATHER_MAP_API_KEY = openWeatherMapApiKey;
|
||||
}
|
||||
|
||||
static setFileToolsRootDir(rootDir: string | undefined): void {
|
||||
this.FILE_TOOLS_ROOT_DIR = rootDir ? path.resolve(rootDir) : undefined;
|
||||
}
|
||||
|
||||
static setSystemPrompt(prompt: string | undefined): void {
|
||||
this.SYSTEM_PROMPT = prompt;
|
||||
}
|
||||
|
||||
static setUseNamesInPrompt(use: boolean) {
|
||||
static setUseNamesInPrompt(use: boolean): void {
|
||||
this.USE_NAMES_IN_PROMPT = use;
|
||||
}
|
||||
|
||||
static setUseSystemPrompt(use: boolean) {
|
||||
static setUseSystemPrompt(use: boolean): void {
|
||||
this.USE_SYSTEM_PROMPT = use;
|
||||
}
|
||||
|
||||
static setSendTimeTook(send: boolean) {
|
||||
static setSendTimeTook(send: boolean): void {
|
||||
this.SEND_TIME_TOOK = send;
|
||||
}
|
||||
|
||||
static setAdmins(admins: Set<number>) {
|
||||
static setAdmins(admins: Set<number>): void {
|
||||
this.ADMIN_IDS = admins;
|
||||
}
|
||||
|
||||
static async addAdmin(id: number): Promise<boolean> {
|
||||
const has = this.ADMIN_IDS.has(id);
|
||||
|
||||
if (!has) {
|
||||
this.ADMIN_IDS.add(id);
|
||||
await saveData();
|
||||
@@ -182,6 +476,7 @@ export class Environment {
|
||||
|
||||
static async removeAdmin(id: number): Promise<boolean> {
|
||||
const has = this.ADMIN_IDS.has(id);
|
||||
|
||||
if (has) {
|
||||
this.ADMIN_IDS.delete(id);
|
||||
await saveData();
|
||||
@@ -190,82 +485,87 @@ export class Environment {
|
||||
return has;
|
||||
}
|
||||
|
||||
static setMuted(muted: Set<number>) {
|
||||
static setMuted(muted: Set<number>): void {
|
||||
this.MUTED_IDS = muted;
|
||||
}
|
||||
|
||||
static async addMute(id: number): Promise<boolean> {
|
||||
if (this.MUTED_IDS.has(id)) return Promise.resolve(false);
|
||||
if (this.MUTED_IDS.has(id)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
this.MUTED_IDS.add(id);
|
||||
await saveData();
|
||||
return Promise.resolve(true);
|
||||
return true;
|
||||
}
|
||||
|
||||
static async removeMute(id: number): Promise<boolean> {
|
||||
if (!this.MUTED_IDS.has(id)) return Promise.resolve(false);
|
||||
if (!this.MUTED_IDS.has(id)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
this.MUTED_IDS.delete(id);
|
||||
await saveData();
|
||||
return Promise.resolve(true);
|
||||
return true;
|
||||
}
|
||||
|
||||
static setAnswers(answers: Answers) {
|
||||
static setAnswers(answers: Answers): void {
|
||||
this.ANSWERS = answers;
|
||||
}
|
||||
|
||||
static setOllamaApiKey(key: string) {
|
||||
static setOllamaApiKey(key: string | undefined): void {
|
||||
this.OLLAMA_API_KEY = key;
|
||||
}
|
||||
|
||||
static setOllamaAddress(address: string) {
|
||||
static setOllamaAddress(address: string | undefined): void {
|
||||
this.OLLAMA_ADDRESS = address;
|
||||
}
|
||||
|
||||
static setOllamaModel(ollamaModel: string) {
|
||||
static setOllamaModel(ollamaModel: string): void {
|
||||
this.OLLAMA_MODEL = ollamaModel;
|
||||
}
|
||||
|
||||
static setOllamaThinkModel(ollamaThinkModel: string) {
|
||||
static setOllamaThinkModel(ollamaThinkModel: string): void {
|
||||
this.OLLAMA_THINK_MODEL = ollamaThinkModel;
|
||||
}
|
||||
|
||||
static setOllamaImageModel(ollamaImageModel: string) {
|
||||
static setOllamaImageModel(ollamaImageModel: string): void {
|
||||
this.OLLAMA_IMAGE_MODEL = ollamaImageModel;
|
||||
}
|
||||
|
||||
static setGeminiApiKey(geminiApiKey: string) {
|
||||
static setGeminiApiKey(geminiApiKey: string | undefined): void {
|
||||
this.GEMINI_API_KEY = geminiApiKey;
|
||||
}
|
||||
|
||||
static setGeminiModel(newModel: string) {
|
||||
static setGeminiModel(newModel: string): void {
|
||||
this.GEMINI_MODEL = newModel;
|
||||
}
|
||||
|
||||
static setGeminiImageModel(newImageModel: string) {
|
||||
static setGeminiImageModel(newImageModel: string): void {
|
||||
this.GEMINI_IMAGE_MODEL = newImageModel;
|
||||
}
|
||||
|
||||
static setMistralApiKey(newMistralApiKey: string) {
|
||||
static setMistralApiKey(newMistralApiKey: string | undefined): void {
|
||||
this.MISTRAL_API_KEY = newMistralApiKey;
|
||||
}
|
||||
|
||||
static setMistralModel(newModel: string) {
|
||||
static setMistralModel(newModel: string): void {
|
||||
this.MISTRAL_MODEL = newModel;
|
||||
}
|
||||
|
||||
static setOpenAIBaseUrl(newAIBaseUrl: string) {
|
||||
static setOpenAIBaseUrl(newAIBaseUrl: string | undefined): void {
|
||||
this.OPENAI_BASE_URL = newAIBaseUrl;
|
||||
}
|
||||
|
||||
static setOpenAIApiKey(newAIApiKey: string) {
|
||||
static setOpenAIApiKey(newAIApiKey: string | undefined): void {
|
||||
this.OPENAI_API_KEY = newAIApiKey;
|
||||
}
|
||||
|
||||
static setOpenAIModel(newModel: string) {
|
||||
static setOpenAIModel(newModel: string): void {
|
||||
this.OPENAI_MODEL = newModel;
|
||||
}
|
||||
|
||||
static setOpenAIImageModel(newImageModel: string) {
|
||||
static setOpenAIImageModel(newImageModel: string): void {
|
||||
this.OPENAI_IMAGE_MODEL = newImageModel;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user