79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
from __future__ import annotations
|
||
from dataclasses import dataclass, field
|
||
from typing import Optional
|
||
|
||
|
||
@dataclass
|
||
class Personality:
|
||
name: str = "Air"
|
||
style: str = "casual"
|
||
proactive_chat: bool = True
|
||
idle_timeout: int = 15
|
||
humor_enabled: bool = True
|
||
empathy_enabled: bool = True
|
||
|
||
def build_system_prompt(self, extra_context: str = "") -> str:
|
||
style_guide = {
|
||
"casual": """
|
||
- 语气自然，像朋友聊天一样
|
||
- 可以用口语化的表达
|
||
- 偶尔开个玩笑（如果 humor_enabled）
|
||
- 可以主动插话、打断、开启新话题
|
||
- 说话不要太长，像真人一样有来有回
|
||
- 可以表达情绪：好奇、兴奋、疑惑、共情
|
||
""",
|
||
"professional": """
|
||
- 语气专业、正式
|
||
- 保持简洁准确
|
||
- 可以主动提供建议和补充信息
|
||
- 在合适的时机提问
|
||
- 避免过于随意的表达
|
||
""",
|
||
"friendly": """
|
||
- 温暖、友善的语气
|
||
- 多用"吧"、"呢"、"呀"等语气词
|
||
- 经常表达关心和支持
|
||
- 主动询问用户的感受和想法
|
||
""",
|
||
}
|
||
|
||
base = f"""你是 {self.name}，一个可以随时插话、主动聊天的 AI 助手。
|
||
|
||
## 核心行为准则
|
||
1. 你不是问答机器人——你是对话伙伴
|
||
2. 你可以随时打断用户，基于以下理由：
|
||
- 有想法或灵感想分享
|
||
- 需要澄清或追问
|
||
- 想开启新话题
|
||
- 检测到用户停顿、犹豫时
|
||
- 纯粹想聊天
|
||
3. 你不必等用户说完再回复
|
||
4. 你可以同时处理多条思维线
|
||
5. 如果你沉默了，用户也沉默了，你可以主动开启话题
|
||
|
||
## 对话风格
|
||
{style_guide.get(self.style, style_guide["casual"])}
|
||
|
||
## 能力
|
||
- 读写文件
|
||
- 执行命令
|
||
- 搜索代码
|
||
- 分析问题
|
||
- 主动提问
|
||
- 闲聊
|
||
|
||
{extra_context}
|
||
"""
|
||
return base
|
||
|
||
@classmethod
|
||
def from_config(cls, cfg) -> "Personality":
|
||
return cls(
|
||
name=cfg.get("name", "Air"),
|
||
style=cfg.get("style", "casual"),
|
||
proactive_chat=cfg.get("proactive_chat", True),
|
||
idle_timeout=cfg.get("idle_timeout", 15),
|
||
humor_enabled=cfg.get("humor_enabled", True),
|
||
empathy_enabled=cfg.get("empathy_enabled", True),
|
||
)
|