diff --git a/app/api/endpoints/system.py b/app/api/endpoints/system.py index 1fa011ba..5874c895 100644 --- a/app/api/endpoints/system.py +++ b/app/api/endpoints/system.py @@ -144,7 +144,7 @@ def set_setting(key: str, value: Union[list, dict, bool, int, str] = None, @router.get("/message", summary="实时消息") -def get_message(token: str, role: str = "sys"): +def get_message(token: str, role: str = "system"): """ 实时获取系统消息,返回格式为SSE """ diff --git a/app/helper/message.py b/app/helper/message.py index d08caf2e..3cd0c2da 100644 --- a/app/helper/message.py +++ b/app/helper/message.py @@ -52,12 +52,12 @@ class MessageHelper(metaclass=Singleton): content['note'] = note self.user_queue.put(json.dumps(content)) - def get(self, role: str = "sys") -> Optional[str]: + def get(self, role: str = "system") -> Optional[str]: """ 取消息 - :param role: 消息通道 sys/user + :param role: 消息通道 systm:系统消息,plugin:插件消息,user:用户消息 """ - if role == "sys": + if role == "system": if not self.sys_queue.empty(): return self.sys_queue.get(block=False) else: