update:创建智能体设置默认的插件

2026-07-26 00:53:54 +08:00 · 2025-06-12 16:52:56 +08:00
parent 535c088404
commit e943b07344
13 changed files with 171 additions and 249 deletions
@@ -445,14 +445,17 @@ class ConnectionHandler:
        if private_config.get("Intent", None) is not None:
            init_intent = True
            self.config["Intent"] = private_config["Intent"]
-            model_intent = private_config.get('selected_module', {}).get('Intent', {})
+            model_intent = private_config.get("selected_module", {}).get("Intent", {})
            self.config["selected_module"]["Intent"] = model_intent
            # 加载插件配置
-            if model_intent != 'Intent_nointent':
+            if model_intent != "Intent_nointent":
                plugin_from_server = private_config.get("plugins", {})
                for plugin, config_str in plugin_from_server.items():
                    plugin_from_server[plugin] = json.loads(config_str)
-                self.config['plugins'] = plugin_from_server
+                self.config["plugins"] = plugin_from_server
+                self.config["Intent"][self.config["selected_module"]["Intent"]][
+                    "functions"
+                ] = plugin_from_server.keys()
        if private_config.get("prompt", None) is not None:
            self.config["prompt"] = private_config["prompt"]
        if private_config.get("summaryMemory", None) is not None:
@@ -61,7 +61,6 @@ class FunctionHandler:
        self.function_registry.register_function("plugin_loader")
        self.function_registry.register_function("get_time")
        self.function_registry.register_function("get_lunar")
-        # self.function_registry.register_function("handle_speaker_volume_or_screen_brightness")

    def register_config_functions(self):
        """注册配置中的函数,可以不同客户端使用不同的配置"""
@@ -12,10 +12,6 @@ class MemoryProviderBase(ABC):

    def set_llm(self, llm):
        self.llm = llm
-        # 获取模型名称和类型信息
-        model_name = getattr(llm, "model_name", str(llm.__class__.__name__))
-        # 记录更详细的日志
-        logger.bind(tag=TAG).info(f"记忆总结设置LLM: {model_name}")

    @abstractmethod
    async def save_memory(self, msgs):
@@ -1,157 +0,0 @@
-from config.logger import setup_logging
-from plugins_func.register import register_function, ToolType, ActionResponse, Action
-from core.handle.iotHandle import get_iot_status, send_iot_conn
-import asyncio
-
-TAG = __name__
-logger = setup_logging()
-
-
-async def _get_device_status(conn, device_name, device_type, property_name):
-    """获取设备状态"""
-    status = await get_iot_status(conn, device_type, property_name)
-    if status is None:
-        raise Exception(f"你的设备不支持{device_name}控制")
-    return status
-
-
-async def _set_device_property(
-    conn,
-    device_name,
-    device_type,
-    method_name,
-    property_name,
-    new_value=None,
-    action=None,
-    step=10,
-):
-    """设置设备属性"""
-    current_value = await _get_device_status(
-        conn, device_name, device_type, property_name
-    )
-
-    if action == "raise":
-        current_value += step
-    elif action == "lower":
-        current_value -= step
-    elif action == "set":
-        if new_value is None:
-            raise Exception(f"缺少{property_name}参数")
-        current_value = new_value
-
-    # 限制属性范围在0到100之间
-    current_value = max(0, min(100, current_value))
-
-    await send_iot_conn(conn, device_type, method_name, {property_name: current_value})
-    return current_value
-
-
-def _handle_device_action(conn, func, success_message, error_message, *args, **kwargs):
-    """处理设备操作的通用函数"""
-    future = asyncio.run_coroutine_threadsafe(func(conn, *args, **kwargs), conn.loop)
-    try:
-        result = future.result()
-        logger.bind(tag=TAG).info(f"{success_message}: {result}")
-        response = f"{success_message}{result}"
-        return ActionResponse(action=Action.RESPONSE, result=result, response=response)
-    except Exception as e:
-        logger.bind(tag=TAG).error(f"{error_message}: {e}")
-        response = f"{error_message}: {e}"
-        return ActionResponse(action=Action.RESPONSE, result=None, response=response)
-
-
-# 设备控制
-handle_device_function_desc = {
-    "type": "function",
-    "function": {
-        "name": "handle_speaker_volume_or_screen_brightness",
-        "description": (
-            "用户想要获取或者设置设备的音量/亮度大小，或者用户觉得声音/亮度过高或过低，或者用户想提高或降低音量/亮度。\n"
-            "**严格限制**：仅当用户明确操作 **Speaker（音量）或Screen（亮度）** 时才能调用此函数！\n"
-            "对于其他设备（如AC、Battery、Switch等），请不要调用此函数，而是继续正常的对话。\n\n"
-            "示例：\n"
-            "- 用户说『现在亮度多少』 → 调用函数：device_type: Screen, action: get\n"
-            "- 用户说『设置音量为50』 → 调用函数：device_type: Speaker, action: set, value: 50\n"
-            "- 用户说『亮度太高了』 → 调用函数：device_type: Screen, action: lower\n"
-            "- 用户说『调大音量』 → 调用函数：device_type: Speaker, action: raise\n\n"
-            "**拒绝调用示例**（应继续对话而非调用本函数）：\n"
-            "- 用户说『空调调低一度』 → 不调用（设备类型为AC）\n"
-            "- 用户说『开关灯』 → 不调用（设备类型为Switch）\n"
-            "- 用户说『电量多少』 → 不调用（设备类型为Battery）\n"
-        ),
-        "parameters": {
-            "type": "object",
-            "properties": {
-                "device_type": {
-                    "type": "string",
-                    "description": "设备类型，**严格限定为Speaker（音量）或Screen（亮度）**，其他设备类型禁止调用此函数",
-                    "enum": ["Speaker", "Screen"],
-                },
-                "action": {
-                    "type": "string",
-                    "description": "动作名称，可选值：get(获取),set(设置),raise(提高),lower(降低)",
-                },
-                "value": {
-                    "type": "integer",
-                    "description": "值大小，可选值：0-100之间的整数",
-                },
-            },
-            "required": ["device_type", "action"],
-        },
-    },
-}
-
-
-@register_function(
-    "handle_speaker_volume_or_screen_brightness",
-    handle_device_function_desc,
-    ToolType.IOT_CTL,
-)
-def handle_speaker_volume_or_screen_brightness(
-    conn, device_type: str, action: str, value: int = None
-):
-    # 检查value是否为中文值
-    if (
-        value is not None
-        and isinstance(value, str)
-        and any("\u4e00" <= char <= "\u9fff" for char in str(value))
-    ):
-        raise Exception(
-            f"请直接告诉我要将{'音量' if device_type=='Speaker' else '亮度'}调整成多少"
-        )
-
-    if device_type == "Speaker":
-        method_name, property_name, device_name = "SetVolume", "volume", "音量"
-    elif device_type == "Screen":
-        method_name, property_name, device_name = "SetBrightness", "brightness", "亮度"
-    else:
-        raise Exception(f"未识别的设备类型: {device_type}")
-
-    if action not in ["get", "set", "raise", "lower"]:
-        raise Exception(f"未识别的动作名称: {action}")
-
-    if action == "get":
-        # get
-        return _handle_device_action(
-            conn,
-            _get_device_status,
-            f"当前{device_name}",
-            f"获取{device_name}失败",
-            device_name=device_name,
-            device_type=device_type,
-            property_name=property_name,
-        )
-    else:
-        # set, raise, lower
-        return _handle_device_action(
-            conn,
-            _set_device_property,
-            f"{device_name}已调整到",
-            f"{device_name}调整失败",
-            device_name=device_name,
-            device_type=device_type,
-            method_name=method_name,
-            property_name=property_name,
-            new_value=value,
-            action=action,
-        )