perf: USER_CONFIG保存KV前裁剪

TBXark · Aug 28, 2024 · 95b4545 · 95b4545
1 parent 8ae55be
commit 95b4545
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 12 deletions.
diff --git a/dist/buildinfo.json b/dist/buildinfo.json
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/timestamp b/dist/timestamp
diff --git a/src/telegram/command/system.ts b/src/telegram/command/system.ts
@@ -152,7 +152,7 @@ export class SetEnvCommandHandler implements CommandHandler {
             console.log('Update user config: ', key, context.USER_CONFIG[key]);
             await ENV.DATABASE.put(
                 context.SHARE_CONTEXT.configStoreKey,
-                JSON.stringify(context.USER_CONFIG),
+                JSON.stringify(ConfigMerger.trim(context.USER_CONFIG, ENV.LOCK_USER_CONFIG_KEYS)),
             );
             return sender.sendPlainText('Update user config success');
         } catch (e) {
@@ -187,7 +187,7 @@ export class SetEnvsCommandHandler implements CommandHandler {
             context.USER_CONFIG.DEFINE_KEYS = Array.from(new Set(context.USER_CONFIG.DEFINE_KEYS));
             await ENV.DATABASE.put(
                 context.SHARE_CONTEXT.configStoreKey,
-                JSON.stringify(context.USER_CONFIG),
+                JSON.stringify(ConfigMerger.trim(context.USER_CONFIG, ENV.LOCK_USER_CONFIG_KEYS)),
             );
             return sender.sendPlainText('Update user config success');
         } catch (e) {
@@ -210,7 +210,7 @@ export class DelEnvCommandHandler implements CommandHandler {
             context.USER_CONFIG.DEFINE_KEYS = context.USER_CONFIG.DEFINE_KEYS.filter(key => key !== subcommand);
             await ENV.DATABASE.put(
                 context.SHARE_CONTEXT.configStoreKey,
-                JSON.stringify(context.USER_CONFIG),
+                JSON.stringify(ConfigMerger.trim(context.USER_CONFIG, ENV.LOCK_USER_CONFIG_KEYS)),
             );
             return sender.sendPlainText('Delete user config success');
         } catch (e) {
@@ -292,7 +292,7 @@ export class SystemCommandHandler implements CommandHandler {
             context.USER_CONFIG.MISTRAL_API_KEY = '******';
             context.USER_CONFIG.COHERE_API_KEY = '******';
             context.USER_CONFIG.ANTHROPIC_API_KEY = '******';
-            const config = context.USER_CONFIG;
+            const config = ConfigMerger.trim(context.USER_CONFIG, ENV.LOCK_USER_CONFIG_KEYS);
             msg = `<pre>\n${msg}`;
             msg += `USER_CONFIG: ${JSON.stringify(config, null, 2)}\n`;
             msg += `CHAT_CONTEXT: ${JSON.stringify(sender.context || {}, null, 2)}\n`;