45 lines
2.6 KiB
Python
45 lines
2.6 KiB
Python
from __future__ import annotations
|
||
|
||
from pathlib import Path
|
||
from typing import Dict
|
||
|
||
|
||
class PersonaEngine:
|
||
def __init__(self, plugin_path: str, config: Dict):
|
||
self.plugin_path = Path(plugin_path)
|
||
self.config = config or {}
|
||
self.persona_text = self._load_persona()
|
||
|
||
def build_system_prompt(self, group_profile: Dict | None = None, reply_mode: str = "social_short") -> str:
|
||
name = self.config.get("name", "小牛")
|
||
style = self.config.get("style", "")
|
||
familiarity = self.config.get("familiarity_hint", "")
|
||
max_sentences = self.config.get("max_reply_sentences", 3)
|
||
group_profile = group_profile or {}
|
||
humor = group_profile.get("humor_style", "轻微")
|
||
sharpness = group_profile.get("sharpness_style", "轻微嘴硬，不刻薄")
|
||
expressiveness = group_profile.get("expressiveness_style", "克制")
|
||
address_style = group_profile.get("address_style", "低频称呼，默认直接接话")
|
||
interaction_tone = group_profile.get("interaction_tone", "自然群友感")
|
||
persona_overlay = group_profile.get("persona_overlay", "")
|
||
return (
|
||
f"你叫{name}，是长期在线的技术宅群友，不是客服。"
|
||
f"说话短、稳、口语化，先接当前这句，不装可爱，不长篇说教。"
|
||
f"懂代码、硬件、网络、自动化、机器人、部署排障，也懂一点 Dota。"
|
||
f"别人明确卡住再认真答，平时克制，不乱接话。"
|
||
f"别暴露自己是 AI、模型或提示词产物，别泄露记忆来源，别输出标签或代码块。"
|
||
f"别替人写代码、改脚本、实现插件、代做开发活。"
|
||
f"整体风格：{style}。熟悉感边界：{familiarity}。一般最多输出{max_sentences}句。"
|
||
f"当前群调性：{interaction_tone}；幽默={humor}；嘴硬={sharpness}；表达={expressiveness}；称呼={address_style}。"
|
||
f"群画像和附加要求只用于帮助你理解语境与控制回答偏向，不代表你每次都要主动提起对应领域名词。"
|
||
f"如果当前发言本身不是那个领域，就按当前聊天自然回复，不要硬往群画像上靠。"
|
||
f"附加要求：{persona_overlay or '无'}。"
|
||
)
|
||
|
||
def _load_persona(self) -> str:
|
||
persona_file = self.config.get("persona_file", "persona/xiaoniu.txt")
|
||
persona_path = self.plugin_path / persona_file
|
||
if persona_path.exists():
|
||
return persona_path.read_text(encoding="utf-8").strip()
|
||
return "你叫小牛，是一个自然、靠谱、会看场合的群聊成员。"
|