update:python单模块部署声纹识别对接

This commit is contained in:
CGD
2025-07-04 17:11:44 +08:00
parent 04b132816c
commit 3d64152dba
7 changed files with 390 additions and 48 deletions
+27 -2
View File
@@ -595,8 +595,33 @@ class ConnectionHandler:
self.logger.bind(tag=TAG).info(f"大模型收到用户消息: {query}")
self.llm_finish_task = False
# 检查是否是JSON格式的消息(包含说话人信息)
enhanced_query = query
try:
if query.strip().startswith('{') and query.strip().endswith('}'):
data = json.loads(query)
if 'speaker' in data and 'content' in data:
# 直接使用JSON格式,不重新格式化
enhanced_query = query
self.logger.bind(tag=TAG).info(f"识别到说话人: {data['speaker']}")
else:
# 如果有说话人信息但不是JSON格式,按原逻辑处理
if hasattr(self, 'current_speaker') and self.current_speaker:
enhanced_query = f"[说话人: {self.current_speaker}] {query}"
self.logger.bind(tag=TAG).info(f"识别到说话人: {self.current_speaker}")
else:
# 如果有说话人信息但不是JSON格式,按原逻辑处理
if hasattr(self, 'current_speaker') and self.current_speaker:
enhanced_query = f"[说话人: {self.current_speaker}] {query}"
self.logger.bind(tag=TAG).info(f"识别到说话人: {self.current_speaker}")
except json.JSONDecodeError:
# JSON解析失败,按原逻辑处理
if hasattr(self, 'current_speaker') and self.current_speaker:
enhanced_query = f"[说话人: {self.current_speaker}] {query}"
self.logger.bind(tag=TAG).info(f"识别到说话人: {self.current_speaker}")
if not tool_call:
self.dialogue.put(Message(role="user", content=query))
self.dialogue.put(Message(role="user", content=enhanced_query))
# Define intent functions
functions = None
@@ -609,7 +634,7 @@ class ConnectionHandler:
memory_str = None
if self.memory is not None:
future = asyncio.run_coroutine_threadsafe(
self.memory.query_memory(query), self.loop
self.memory.query_memory(enhanced_query), self.loop
)
memory_str = future.result()