feat:私聊独有replyer

pull/1239/head
SengokuCola 2025-09-16 15:02:12 +08:00
parent 755ceba0a4
commit 750bdfe04d
6 changed files with 1091 additions and 9 deletions

View File

@ -32,9 +32,9 @@ from src.person_info.person_info import Person, is_person_known
from src.plugin_system.base.component_types import ActionInfo, EventType
from src.plugin_system.apis import llm_api
from src.chat.replyer.lpmm_prompt import init_lpmm_prompt
from src.chat.replyer.replyer_prompt import init_replyer_prompt
from src.chat.replyer.rewrite_prompt import init_rewrite_prompt
from src.chat.replyer.prompt.lpmm_prompt import init_lpmm_prompt
from src.chat.replyer.prompt.replyer_prompt import init_replyer_prompt
from src.chat.replyer.prompt.rewrite_prompt import init_rewrite_prompt
init_lpmm_prompt()
init_replyer_prompt()

File diff suppressed because it is too large Load Diff

View File

@ -52,4 +52,24 @@ def init_replyer_prompt():
{moderation_prompt}不要输出多余内容(包括前后缀冒号和引号括号表情包at或 @等 )
""",
"replyer_self_prompt",
)
Prompt(
"""{knowledge_prompt}{relation_info_block}{tool_info_block}{extra_info_block}
{expression_habits_block}
你正在和{sender_name}聊天这是你们之前聊的内容:
{time_block}
{dialogue_prompt}
{reply_target_block}
{identity}
你正在和{sender_name}聊天,现在请你读读之前的聊天记录然后给出日常且口语化的回复平淡一些
尽量简短一些{keywords_reaction_prompt}请注意把握聊天内容不要回复的太有条理可以有个性
{reply_style}
请注意不要输出多余内容(包括前后缀冒号和引号括号表情等)只输出回复内容
{moderation_prompt}不要输出多余内容(包括前后缀冒号和引号括号表情包at或 @等 )""",
"private_replyer_prompt",
)

View File

@ -3,20 +3,21 @@ from typing import Dict, Optional
from src.common.logger import get_logger
from src.chat.message_receive.chat_stream import ChatStream, get_chat_manager
from src.chat.replyer.default_generator import DefaultReplyer
from src.chat.replyer.private_generator import PrivateReplyer
logger = get_logger("ReplyerManager")
class ReplyerManager:
def __init__(self):
self._repliers: Dict[str, DefaultReplyer] = {}
self._repliers: Dict[str, DefaultReplyer | PrivateReplyer] = {}
def get_replyer(
self,
chat_stream: Optional[ChatStream] = None,
chat_id: Optional[str] = None,
request_type: str = "replyer",
) -> Optional[DefaultReplyer]:
) -> Optional[DefaultReplyer | PrivateReplyer]:
"""
获取或创建回复器实例
@ -46,10 +47,17 @@ class ReplyerManager:
return None
# model_configs 只在此时(初始化时)生效
replyer = DefaultReplyer(
chat_stream=target_stream,
request_type=request_type,
)
if target_stream.group_info:
replyer = DefaultReplyer(
chat_stream=target_stream,
request_type=request_type,
)
else:
replyer = PrivateReplyer(
chat_stream=target_stream,
request_type=request_type,
)
self._repliers[stream_id] = replyer
return replyer