mirror of
https://github.com/xinnan-tech/xiaozhi-esp32-server.git
synced 2026-07-22 23:23:55 +08:00
huozaimengli
4b573fb4e2
为 ConnectionHandler 相关的函数参数添加类型注解，使用 TYPE_CHECKING 避免循环导入。主要修改包括： - 在 abortHandle、textHandle 等处理模块中为 conn 参数添加 ConnectionHandler 类型注解 - 在 websocket_server、connection 等核心模块中为方法参数添加类型注解 - 在 plugins_func 下的多个功能模块中为函数参数添加类型注解 - 在 providers 相关模块中为工具执行器和方法添加类型注解 - 统一代码格式，如将单引号字符串改为双引号 Fixes #2034
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
from plugins_func.register import register_function, ToolType, ActionResponse, Action
|
|
from config.logger import setup_logging
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from core.connection import ConnectionHandler
|
|
|
|
TAG = __name__
|
|
logger = setup_logging()
|
|
|
|
handle_exit_intent_function_desc = {
|
|
"type": "function",
|
|
"function": {
|
|
"name": "handle_exit_intent",
|
|
"description": "当用户想结束对话或需要退出系统时调用",
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"say_goodbye": {
|
|
"type": "string",
|
|
"description": "和用户友好结束对话的告别语",
|
|
}
|
|
},
|
|
"required": ["say_goodbye"],
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
@register_function(
|
|
"handle_exit_intent", handle_exit_intent_function_desc, ToolType.SYSTEM_CTL
|
|
)
|
|
def handle_exit_intent(conn: "ConnectionHandler", say_goodbye: str | None = None):
|
|
# 处理退出意图
|
|
try:
|
|
if say_goodbye is None:
|
|
say_goodbye = "再见，祝您生活愉快！"
|
|
conn.close_after_chat = True
|
|
logger.bind(tag=TAG).info(f"退出意图已处理:{say_goodbye}")
|
|
return ActionResponse(
|
|
action=Action.RESPONSE, result="退出意图已处理", response=say_goodbye
|
|
)
|
|
except Exception as e:
|
|
logger.bind(tag=TAG).error(f"处理退出意图错误: {e}")
|
|
return ActionResponse(
|
|
action=Action.NONE, result="退出意图处理失败", response=""
|
|
)
|