mirror of https://github.com/Mai-with-u/MaiBot.git
feat:私聊独有replyer
parent
755ceba0a4
commit
750bdfe04d
|
|
@ -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.base.component_types import ActionInfo, EventType
|
||||||
from src.plugin_system.apis import llm_api
|
from src.plugin_system.apis import llm_api
|
||||||
|
|
||||||
from src.chat.replyer.lpmm_prompt import init_lpmm_prompt
|
from src.chat.replyer.prompt.lpmm_prompt import init_lpmm_prompt
|
||||||
from src.chat.replyer.replyer_prompt import init_replyer_prompt
|
from src.chat.replyer.prompt.replyer_prompt import init_replyer_prompt
|
||||||
from src.chat.replyer.rewrite_prompt import init_rewrite_prompt
|
from src.chat.replyer.prompt.rewrite_prompt import init_rewrite_prompt
|
||||||
|
|
||||||
init_lpmm_prompt()
|
init_lpmm_prompt()
|
||||||
init_replyer_prompt()
|
init_replyer_prompt()
|
||||||
|
|
|
||||||
File diff suppressed because it is too large
Load Diff
|
|
@ -53,3 +53,23 @@ def init_replyer_prompt():
|
||||||
""",
|
""",
|
||||||
"replyer_self_prompt",
|
"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",
|
||||||
|
)
|
||||||
|
|
@ -3,20 +3,21 @@ from typing import Dict, Optional
|
||||||
from src.common.logger import get_logger
|
from src.common.logger import get_logger
|
||||||
from src.chat.message_receive.chat_stream import ChatStream, get_chat_manager
|
from src.chat.message_receive.chat_stream import ChatStream, get_chat_manager
|
||||||
from src.chat.replyer.default_generator import DefaultReplyer
|
from src.chat.replyer.default_generator import DefaultReplyer
|
||||||
|
from src.chat.replyer.private_generator import PrivateReplyer
|
||||||
|
|
||||||
logger = get_logger("ReplyerManager")
|
logger = get_logger("ReplyerManager")
|
||||||
|
|
||||||
|
|
||||||
class ReplyerManager:
|
class ReplyerManager:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self._repliers: Dict[str, DefaultReplyer] = {}
|
self._repliers: Dict[str, DefaultReplyer | PrivateReplyer] = {}
|
||||||
|
|
||||||
def get_replyer(
|
def get_replyer(
|
||||||
self,
|
self,
|
||||||
chat_stream: Optional[ChatStream] = None,
|
chat_stream: Optional[ChatStream] = None,
|
||||||
chat_id: Optional[str] = None,
|
chat_id: Optional[str] = None,
|
||||||
request_type: str = "replyer",
|
request_type: str = "replyer",
|
||||||
) -> Optional[DefaultReplyer]:
|
) -> Optional[DefaultReplyer | PrivateReplyer]:
|
||||||
"""
|
"""
|
||||||
获取或创建回复器实例。
|
获取或创建回复器实例。
|
||||||
|
|
||||||
|
|
@ -46,10 +47,17 @@ class ReplyerManager:
|
||||||
return None
|
return None
|
||||||
|
|
||||||
# model_configs 只在此时(初始化时)生效
|
# model_configs 只在此时(初始化时)生效
|
||||||
replyer = DefaultReplyer(
|
if target_stream.group_info:
|
||||||
chat_stream=target_stream,
|
replyer = DefaultReplyer(
|
||||||
request_type=request_type,
|
chat_stream=target_stream,
|
||||||
)
|
request_type=request_type,
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
replyer = PrivateReplyer(
|
||||||
|
chat_stream=target_stream,
|
||||||
|
request_type=request_type,
|
||||||
|
)
|
||||||
|
|
||||||
self._repliers[stream_id] = replyer
|
self._repliers[stream_id] = replyer
|
||||||
return replyer
|
return replyer
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue