mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-28 22:03:52 +08:00
refactor: 优化超时配置,响应社区反馈
- 工具调用超时改为可配置项 tool_call_timeout,默认30秒 - OpenAI 超时恢复可配置机制,支持细粒度配置和单一值 - TTS 超时保护统一添加到所有流式TTS实现 - 将 ActionResponse 导入移到文件顶部 - 修复超时配置边界情况处理
This commit is contained in:
@@ -17,12 +17,22 @@ class LLMProvider(LLMProviderBase):
|
||||
self.base_url = config.get("base_url")
|
||||
else:
|
||||
self.base_url = config.get("url")
|
||||
custom_timeout = httpx.Timeout(
|
||||
pool=2.0,
|
||||
connect=3.0,
|
||||
write=5.0,
|
||||
read=10.0
|
||||
)
|
||||
|
||||
timeout_config = config.get("timeout")
|
||||
if isinstance(timeout_config, dict):
|
||||
# 细粒度超时配置
|
||||
custom_timeout = httpx.Timeout(
|
||||
pool=timeout_config.get("pool", 2.0),
|
||||
connect=timeout_config.get("connect", 3.0),
|
||||
write=timeout_config.get("write", 5.0),
|
||||
read=timeout_config.get("read", 60.0)
|
||||
)
|
||||
elif isinstance(timeout_config, (int, float)) and timeout_config > 0:
|
||||
# 兼容旧的单一超时配置(整数或浮点数)
|
||||
custom_timeout = httpx.Timeout(timeout_config)
|
||||
else:
|
||||
# 未配置或配置无效,使用默认值
|
||||
custom_timeout = httpx.Timeout(300)
|
||||
|
||||
param_defaults = {
|
||||
"max_tokens": int,
|
||||
|
||||
@@ -127,7 +127,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.start_session(self.conn.sentence_id),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result()
|
||||
future.result(timeout=self.tts_timeout)
|
||||
self.before_stop_play_files.clear()
|
||||
logger.bind(tag=TAG).info("TTS会话启动成功")
|
||||
except Exception as e:
|
||||
@@ -144,7 +144,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.text_to_speak(message.content_detail, None),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result()
|
||||
future.result(timeout=self.tts_timeout)
|
||||
logger.bind(tag=TAG).debug("TTS文本发送成功")
|
||||
except Exception as e:
|
||||
logger.bind(tag=TAG).error(f"发送TTS文本失败: {str(e)}")
|
||||
@@ -165,7 +165,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.finish_session(self.conn.sentence_id),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result()
|
||||
future.result(timeout=self.tts_timeout)
|
||||
except Exception as e:
|
||||
logger.bind(tag=TAG).error(f"结束TTS会话失败: {str(e)}")
|
||||
continue
|
||||
|
||||
@@ -232,7 +232,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.start_session(self.task_id),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result()
|
||||
future.result(timeout=self.tts_timeout)
|
||||
self.before_stop_play_files.clear()
|
||||
logger.bind(tag=TAG).debug("TTS会话启动成功")
|
||||
|
||||
@@ -250,7 +250,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.text_to_speak(message.content_detail, None),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result()
|
||||
future.result(timeout=self.tts_timeout)
|
||||
logger.bind(tag=TAG).debug("TTS文本发送成功")
|
||||
except Exception as e:
|
||||
logger.bind(tag=TAG).error(f"发送TTS文本失败: {str(e)}")
|
||||
@@ -270,7 +270,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.finish_session(self.task_id),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result()
|
||||
future.result(timeout=self.tts_timeout)
|
||||
except Exception as e:
|
||||
logger.bind(tag=TAG).error(f"结束TTS会话失败: {str(e)}")
|
||||
continue
|
||||
|
||||
@@ -36,6 +36,7 @@ class TTSProviderBase(ABC):
|
||||
self.delete_audio_file = delete_audio_file
|
||||
self.audio_file_type = "wav"
|
||||
self.output_file = config.get("output_dir", "tmp/")
|
||||
self.tts_timeout = int(config.get("tts_timeout", 15))
|
||||
self.tts_text_queue = queue.Queue()
|
||||
self.tts_audio_queue = queue.Queue()
|
||||
self.tts_audio_first_sentence = True
|
||||
@@ -357,7 +358,7 @@ class TTSProviderBase(ABC):
|
||||
sendAudioMessage(self.conn, sentence_type, audio_datas, text),
|
||||
self.conn.loop,
|
||||
)
|
||||
future.result(timeout=15)
|
||||
future.result(timeout=self.tts_timeout)
|
||||
|
||||
# 记录输出和报告
|
||||
if self.conn.max_output_size > 0 and text:
|
||||
|
||||
@@ -149,7 +149,6 @@ class TTSProvider(TTSProviderBase):
|
||||
self.cluster = config.get("cluster")
|
||||
self.resource_id = config.get("resource_id")
|
||||
self.activate_session = False
|
||||
self.timeout = 30
|
||||
if config.get("private_voice"):
|
||||
self.voice = config.get("private_voice")
|
||||
else:
|
||||
@@ -308,7 +307,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.start_session(self.conn.sentence_id),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result(self.timeout)
|
||||
future.result(timeout=self.tts_timeout)
|
||||
self.before_stop_play_files.clear()
|
||||
logger.bind(tag=TAG).debug("TTS会话启动成功")
|
||||
except Exception as e:
|
||||
@@ -325,7 +324,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.text_to_speak(message.content_detail, None),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result(self.timeout)
|
||||
future.result(timeout=self.tts_timeout)
|
||||
logger.bind(tag=TAG).debug("TTS文本发送成功")
|
||||
except Exception as e:
|
||||
logger.bind(tag=TAG).error(f"发送TTS文本失败: {str(e)}")
|
||||
@@ -345,7 +344,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.finish_session(self.conn.sentence_id),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result(self.timeout)
|
||||
future.result(timeout=self.tts_timeout)
|
||||
except Exception as e:
|
||||
logger.bind(tag=TAG).error(f"结束TTS会话失败: {str(e)}")
|
||||
continue
|
||||
|
||||
@@ -178,7 +178,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.start_session(self.conn.sentence_id),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result()
|
||||
future.result(timeout=self.tts_timeout)
|
||||
self.before_stop_play_files.clear()
|
||||
logger.bind(tag=TAG).info("TTS会话启动成功")
|
||||
|
||||
@@ -197,7 +197,7 @@ class TTSProvider(TTSProviderBase):
|
||||
self.text_to_speak(message.content_detail, None),
|
||||
loop=self.conn.loop,
|
||||
)
|
||||
future.result()
|
||||
future.result(timeout=self.tts_timeout)
|
||||
logger.bind(tag=TAG).debug("TTS文本发送成功")
|
||||
except Exception as e:
|
||||
logger.bind(tag=TAG).error(f"发送TTS文本失败: {str(e)}")
|
||||
|
||||
Reference in New Issue
Block a user