update: 更新了进入沉默模式的功能

This commit is contained in:
killua committed 2024-06-12 22:56:39 +08:00
1 parent 76c7adc1b5
commit 34f3fd62f5
4 files changed
+34

No files matched your search

+14
View File
@@ -29,6 +29,7 @@ LAST_FRAME =3
class XF_ASR(ASR):
def __init__(self):
super().__init__()
self.websocket = None
self.current_message = ""
self.audio = ""
@@ -58,6 +59,16 @@ class XF_ASR(ASR):
self.current_message += parse_xfasr_recv(json.loads(await self.websocket.recv()))
if self.current_message == "":
raise AsrResultNoneError()
if "进入沉默模式" in self.current_message:
self.is_slience = True
asyncio.create_task(self.websocket.close())
raise EnterSlienceMode()
if "退出沉默模式" in self.current_message:
self.is_slience = False
self.current_message = "已退出沉默模式"
if self.is_slience:
asyncio.create_task(self.websocket.close())
raise SlienceMode()
asyncio.create_task(self.websocket.close())
return [{"text":self.current_message, "audio":self.audio}]
@@ -145,6 +156,9 @@ class VOLCENGINE_LLM(LLM):
model = model,
messages=messages,
stream=True,
temperature=llm_info['temperature'],
top_p=llm_info['top_p'],
max_tokens=llm_info['max_tokens'],
stream_options={'include_usage': True}
)
for chunk in stream: