使generator能够自定义是否切句或者生成错字

跨越了三份文件的参数传递((
This commit is contained in:
A0000Xz
2025-06-26 23:30:31 +08:00
parent 63eb678433
commit 514ccd6be3
3 changed files with 21 additions and 5 deletions

View File

@@ -73,6 +73,8 @@ async def generate_reply(
chat_stream=None,
action_data: Dict[str, Any] = None,
chat_id: str = None,
enable_splitter: bool=True,
enable_chinese_typo: bool=True
) -> Tuple[bool, List[Tuple[str, Any]]]:
"""生成回复
@@ -80,6 +82,8 @@ async def generate_reply(
chat_stream: 聊天流对象(优先)
action_data: 动作数据
chat_id: 聊天ID备用
enable_splitter: 是否启用消息分割器
enable_chinese_typo: 是否启用错字生成器
Returns:
Tuple[bool, List[Tuple[str, Any]]]: (是否成功, 回复集合)
@@ -96,6 +100,8 @@ async def generate_reply(
# 调用回复器生成回复
success, reply_set = await replyer.generate_reply_with_context(
reply_data=action_data or {},
enable_splitter=enable_splitter,
enable_chinese_typo=enable_chinese_typo
)
if success:
@@ -114,6 +120,8 @@ async def rewrite_reply(
chat_stream=None,
reply_data: Dict[str, Any] = None,
chat_id: str = None,
enable_splitter: bool=True,
enable_chinese_typo: bool=True
) -> Tuple[bool, List[Tuple[str, Any]]]:
"""重写回复
@@ -121,6 +129,8 @@ async def rewrite_reply(
chat_stream: 聊天流对象(优先)
reply_data: 回复数据
chat_id: 聊天ID备用
enable_splitter: 是否启用消息分割器
enable_chinese_typo: 是否启用错字生成器
Returns:
Tuple[bool, List[Tuple[str, Any]]]: (是否成功, 回复集合)
@@ -137,6 +147,8 @@ async def rewrite_reply(
# 调用回复器重写回复
success, reply_set = await replyer.rewrite_reply_with_context(
reply_data=reply_data or {},
enable_splitter=enable_splitter,
enable_chinese_typo=enable_chinese_typo
)
if success: