update:增加私有唤醒答复文件 (#488)

This commit is contained in:
hrz
2025-03-23 22:40:31 +08:00
committed by GitHub
parent b1b99f1066
commit 4acb5f924a
@@ -31,7 +31,7 @@ async def checkWakeupWords(conn, text):
conn.tts_last_text_index = 0 conn.tts_last_text_index = 0
conn.llm_finish_task = True conn.llm_finish_task = True
file = getWakeupWordFile() file = getWakeupWordFile(WAKEUP_CONFIG["file_name"])
if file is None: if file is None:
asyncio.create_task(wakeupWordsResponse(conn)) asyncio.create_task(wakeupWordsResponse(conn))
return False return False
@@ -43,10 +43,14 @@ async def checkWakeupWords(conn, text):
return False return False
def getWakeupWordFile(): def getWakeupWordFile(file_name):
for file in os.listdir(WAKEUP_CONFIG["dir"]):
if "my_"+file_name in file:
return f"config/assets/{file}"
"""查找config/assets/目录下名称为wakeup_words的文件""" """查找config/assets/目录下名称为wakeup_words的文件"""
for file in os.listdir(WAKEUP_CONFIG["dir"]): for file in os.listdir(WAKEUP_CONFIG["dir"]):
if WAKEUP_CONFIG["file_name"] in file: if file_name in file:
return f"config/assets/{file}" return f"config/assets/{file}"
return None return None
@@ -60,9 +64,9 @@ async def wakeupWordsResponse(conn):
file_type = os.path.splitext(tts_file)[1] file_type = os.path.splitext(tts_file)[1]
if file_type: if file_type:
file_type = file_type.lstrip('.') file_type = file_type.lstrip('.')
old_file = getWakeupWordFile() old_file = getWakeupWordFile("my_" +WAKEUP_CONFIG["file_name"])
if old_file is not None: if old_file is not None:
os.remove(old_file) os.remove(old_file)
"""将文件挪到"wakeup_words.mp3""" """将文件挪到"wakeup_words.mp3"""
shutil.move(tts_file, WAKEUP_CONFIG["dir"] + WAKEUP_CONFIG["file_name"] + "." + file_type) shutil.move(tts_file, WAKEUP_CONFIG["dir"] + "my_" +WAKEUP_CONFIG["file_name"] + "." + file_type)
WAKEUP_CONFIG["create_time"] = time.time() WAKEUP_CONFIG["create_time"] = time.time()