fix message

This commit is contained in:
jxxghp 2024-05-28 15:39:17 +08:00
parent cfd7208cda
commit 8eb89eec86
2 changed files with 4 additions and 4 deletions

View File

@ -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
"""

View File

@ -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: