mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-22 15:13:55 +08:00
2e182626b9
* function call功能完善，增加天气查询，支持插件式扩展 * 增加角色切换功能，通过切换system提示词，修改角色认知 * 增加插件管理系统，可以通过语音加载和卸载插件 * docs: 添加命令操作 (#329) * docs: 添加命令操作 * feat: 添加 docker-setup.sh 脚本以简化服务端部署 - 新增 docker-setup.sh 脚本，自动创建目录结构、下载语音识别模型和配置文件，并检查文件完整性。 - 更新 Deployment.md 文档，提供一键执行脚本的说明和使用示例。 * docs: 更新 Deployment.md，添加环境访问 GitHub 的注意事项 * refactor: 更新 docker-setup.sh 脚本以支持多操作系统下载命令 - 修改脚本以检测操作系统类型，并根据不同系统选择合适的下载命令（curl 或 wget）。 - 优化错误处理，确保在下载失败时提供清晰的提示信息。 - 更新 Deployment.md 文档，调整懒人脚本的使用说明，增加手动部署的步骤。 * Update docker-setup.sh * Update docker-setup.sh --------- Co-authored-by: 欣南科技 <huangrongzhuang@xin-nan.com> * update:优化插件加载的配置提示 * update:增加自动安装脚本的操作说明 * update:优化插件配置，去掉旧版本的时间设定 --------- Co-authored-by: 玄凤科技 <eric230308@gmail.com> Co-authored-by: TinsFox <fox@tinsfox.com> Co-authored-by: hrz <1710360675@qq.com>
89 lines
4.0 KiB
Python
89 lines
4.0 KiB
Python
import asyncio
|
|
import websockets
|
|
from config.logger import setup_logging
|
|
from core.connection import ConnectionHandler
|
|
from core.handle.musicHandler import MusicHandler
|
|
from core.utils.util import get_local_ip
|
|
from core.utils import asr, vad, llm, tts, memory, intent
|
|
|
|
TAG = __name__
|
|
|
|
|
|
class WebSocketServer:
|
|
def __init__(self, config: dict):
|
|
self.config = config
|
|
self.logger = setup_logging()
|
|
self._vad, self._asr, self._llm, self._tts, self._music, self._memory, self.intent = self._create_processing_instances()
|
|
self.active_connections = set() # 添加全局连接记录
|
|
|
|
def _create_processing_instances(self):
|
|
memory_cls_name = self.config["selected_module"].get("Memory", "nomem") # 默认使用nomem
|
|
has_memory_cfg = self.config.get("Memory") and memory_cls_name in self.config["Memory"]
|
|
memory_cfg = self.config["Memory"][memory_cls_name] if has_memory_cfg else {}
|
|
|
|
"""创建处理模块实例"""
|
|
return (
|
|
vad.create_instance(
|
|
self.config["selected_module"]["VAD"],
|
|
self.config["VAD"][self.config["selected_module"]["VAD"]]
|
|
),
|
|
asr.create_instance(
|
|
self.config["selected_module"]["ASR"]
|
|
if not 'type' in self.config["ASR"][self.config["selected_module"]["ASR"]]
|
|
else
|
|
self.config["ASR"][self.config["selected_module"]["ASR"]]["type"],
|
|
self.config["ASR"][self.config["selected_module"]["ASR"]],
|
|
self.config["delete_audio"]
|
|
),
|
|
llm.create_instance(
|
|
self.config["selected_module"]["LLM"]
|
|
if not 'type' in self.config["LLM"][self.config["selected_module"]["LLM"]]
|
|
else
|
|
self.config["LLM"][self.config["selected_module"]["LLM"]]['type'],
|
|
self.config["LLM"][self.config["selected_module"]["LLM"]],
|
|
),
|
|
tts.create_instance(
|
|
self.config["selected_module"]["TTS"]
|
|
if not 'type' in self.config["TTS"][self.config["selected_module"]["TTS"]]
|
|
else
|
|
self.config["TTS"][self.config["selected_module"]["TTS"]]["type"],
|
|
self.config["TTS"][self.config["selected_module"]["TTS"]],
|
|
self.config["delete_audio"]
|
|
),
|
|
MusicHandler(self.config),
|
|
memory.create_instance(memory_cls_name, memory_cfg),
|
|
intent.create_instance(
|
|
self.config["selected_module"]["Intent"]
|
|
if not 'type' in self.config["Intent"][self.config["selected_module"]["Intent"]]
|
|
else
|
|
self.config["Intent"][self.config["selected_module"]["Intent"]]["type"],
|
|
self.config["Intent"][self.config["selected_module"]["Intent"]]
|
|
),
|
|
)
|
|
|
|
async def start(self):
|
|
server_config = self.config["server"]
|
|
host = server_config["ip"]
|
|
port = server_config["port"]
|
|
selected_module = self.config.get("selected_module")
|
|
self.logger.bind(tag=TAG).info(f"selected_module values: {', '.join(selected_module.values())}")
|
|
|
|
self.logger.bind(tag=TAG).info("Server is running at ws://{}:{}", get_local_ip(), port)
|
|
self.logger.bind(tag=TAG).info("=======上面的地址是websocket协议地址，请勿用浏览器访问=======")
|
|
async with websockets.serve(
|
|
self._handle_connection,
|
|
host,
|
|
port
|
|
):
|
|
await asyncio.Future()
|
|
|
|
async def _handle_connection(self, websocket):
|
|
"""处理新连接，每次创建独立的ConnectionHandler"""
|
|
# 创建ConnectionHandler时传入当前server实例
|
|
handler = ConnectionHandler(self.config, self._vad, self._asr, self._llm, self._tts, self._music, self._memory, self.intent)
|
|
self.active_connections.add(handler)
|
|
try:
|
|
await handler.handle_connection(websocket)
|
|
finally:
|
|
self.active_connections.discard(handler)
|