feat(core): add restart command and middleware for ChatLuna service

packages/core/src/chains/rooms.ts

@@ -192,7 +192,6 @@ export async function getTemplateConversationRoom(
 
         ctx.scope.parent.scope.update(config, true)
 
-
         // throw new ChatLunaError(ChatLunaErrorCode.INIT_ROOM)
     }
 

packages/core/src/commands/chat.ts

@@ -92,4 +92,8 @@ export function apply(ctx: Context, config: Config, chain: ChatChain) {
             await chain.receiveCommand(session, 'wipe')
         }
     )
+
+    ctx.command('chatluna.restart').action(async ({ options, session }) => {
+        await chain.receiveCommand(session, 'restart')
+    })
 }

packages/core/src/locales/en-US.yml

@@ -255,6 +255,11 @@ commands:
                 incorrect_input: 'Incorrect input. Deletion cancelled.'
                 success: 'Data deleted. Restarting to apply changes.'
 
+        restart:
+            description: Restart the ChatLuna.
+            messages:
+                success: Successfully restarted ChatLuna.
+
         auth:
             description: ChatLuna authentication commands.
             list:

packages/core/src/locales/zh-CN.yml

@@ -255,6 +255,11 @@ commands:
                 incorrect_input: '你的输入不正确，已取消删除。'
                 success: '已删除相关数据，即将重启完成更改。'
 
+        restart:
+            description: 重启 ChatLuna 服务。
+            messages:
+                success: '已成功重启 ChatLuna 插件。'
+
         auth:
             description: ChatLuna 鉴权相关指令。
             list:

packages/core/src/middleware.ts

@@ -43,6 +43,7 @@ import { apply as render_message } from './middlewares/render_message'
 import { apply as request_model } from './middlewares/request_model'
 import { apply as resolve_model } from './middlewares/resolve_model'
 import { apply as resolve_room } from './middlewares/resolve_room'
+import { apply as restart } from './middlewares/restart'
 import { apply as rollback_chat } from './middlewares/rollback_chat'
 import { apply as room_info } from './middlewares/room_info'
 import { apply as room_permission } from './middlewares/room_permission'
@@ -111,6 +112,7 @@ export async function middleware(ctx: Context, config: Config) {
             request_model,
             resolve_model,
             resolve_room,
+            restart,
             rollback_chat,
             room_info,
             room_permission,

packages/core/src/middlewares/restart.ts

@@ -0,0 +1,25 @@
+import { Context } from 'koishi'
+import { Config } from '../config'
+import { ChainMiddlewareRunStatus, ChatChain } from '../chains/chain'
+
+export function apply(ctx: Context, config: Config, chain: ChatChain) {
+    chain
+        .middleware('restart', async (session, context) => {
+            const { command } = context
+
+            if (command !== 'restart') return ChainMiddlewareRunStatus.SKIPPED
+
+            ctx.scope.parent.scope.update(config, true)
+
+            context.message = session.text('.success')
+
+            return ChainMiddlewareRunStatus.STOP
+        })
+        .before('black_list')
+}
+
+declare module '../chains/chain' {
+    interface ChainMiddlewareName {
+        restart: never
+    }
+}

packages/core/src/middlewares/set_default_embeddings.ts

@@ -59,7 +59,6 @@ export function apply(ctx: Context, config: Config, chain: ChatChain) {
 
             ctx.scope.parent.scope.update(config, true)
 
-
             return ChainMiddlewareRunStatus.STOP
         })
         .after('lifecycle-handle_command')

packages/long-memory/src/plugin.ts

@@ -2,8 +2,7 @@ import { Context } from 'koishi'
 import { Config } from 'koishi-plugin-chatluna-long-memory'
 // import start
 import { apply as config } from './plugins/config'
-import { apply as memory } from './plugins/memory'
-// import end
+import { apply as memory } from './plugins/memory' // import end
 
 export async function plugins(ctx: Context, parent: Config) {
     type Plugin = (ctx: Context, config: Config) => PromiseLike<void> | void