Kris
05abecc02b
该提交实现了支持企业微信、微信公众号、个人微信桥接三种模式的完整微信渠道适配器，包含以下核心模块： 1. 基础认证与配置相关：auth_adapter、config_reload、setup_contract等 2. 消息处理与格式转换：format、attachment_adapter、outbound_adapter等 3. 多模式客户端支持：wecom/mp子模块，包含加解密、消息收发能力 4. 辅助能力：限速器、防抖、会话绑定、事件映射、模板渲染等 5. 扩展能力：二维码登录、消息读取、特权用户、心跳监控等 实现了完整的微信生态对接能力，支持消息收发、事件处理、API调用限流、配置热重载等功能。
46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
|
|
PROMPT_PREFIXES = {
|
|
"wecom": "你是一个企业微信 AI 助手，通过企业微信与用户沟通。请保持专业、简洁，使用中文回复。",
|
|
"mp": "你是一个微信公众号 AI 助手，通过公众号消息与用户沟通。请保持友好、专业，使用中文回复。",
|
|
"personal": "你是一个微信 AI 助手，通过个人微信与用户沟通。请保持自然、亲切，使用中文回复。",
|
|
}
|
|
|
|
|
|
class WeChatAgentPromptAdapter:
|
|
def __init__(self):
|
|
self._custom_prefix: str | None = None
|
|
self._enabled: bool = True
|
|
|
|
def configure(self, config: dict[str, Any], mode: str = "wecom") -> None:
|
|
self._enabled = config.get("agent_prompt", {}).get("enabled", True)
|
|
self._custom_prefix = config.get("agent_prompt", {}).get("prefix")
|
|
if not self._custom_prefix:
|
|
self._custom_prefix = PROMPT_PREFIXES.get(mode, "")
|
|
|
|
def get_channel_prefix(self, config: dict[str, Any], mode: str = "wecom") -> str:
|
|
if not self._enabled:
|
|
return ""
|
|
if self._custom_prefix:
|
|
return self._custom_prefix
|
|
return PROMPT_PREFIXES.get(mode, "")
|
|
|
|
@staticmethod
|
|
def inject_prompt(base_prompt: str, channel_prefix: str) -> str:
|
|
if not channel_prefix:
|
|
return base_prompt
|
|
return f"{channel_prefix}\n\n{base_prompt}"
|
|
|
|
@staticmethod
|
|
def strip_channel_prefix(prompt: str, mode: str = "wecom") -> str:
|
|
prefix = PROMPT_PREFIXES.get(mode, "")
|
|
if prefix and prompt.startswith(prefix):
|
|
return prompt[len(prefix) :].strip()
|
|
return prompt
|
|
|
|
def is_enabled(self) -> bool:
|
|
return self._enabled
|