From af5d75b4635f225e59a1aaa504054591745d102b Mon Sep 17 00:00:00 2001 From: minecraft1024a Date: Sat, 1 Nov 2025 16:56:44 +0800 Subject: [PATCH] =?UTF-8?q?=E5=9B=9E=E5=A4=8D=E5=99=A8=E4=BD=A0=E4=B8=8D?= =?UTF-8?q?=E8=AE=B8=E4=BC=A0dict?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/chat/replyer/default_generator.py | 4 ++-- src/plugin_system/apis/generator_api.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/chat/replyer/default_generator.py b/src/chat/replyer/default_generator.py index 89016ac97..106295ca9 100644 --- a/src/chat/replyer/default_generator.py +++ b/src/chat/replyer/default_generator.py @@ -300,7 +300,7 @@ class DefaultReplyer: enable_tool: bool = True, from_plugin: bool = True, stream_id: str | None = None, - reply_message: dict[str, Any] | None = None, + reply_message: DatabaseMessages | None = None, ) -> tuple[bool, dict[str, Any] | None, str | None]: # sourcery skip: merge-nested-ifs """ @@ -334,7 +334,7 @@ class DefaultReplyer: extra_info=extra_info, available_actions=available_actions, enable_tool=enable_tool, - reply_message=DatabaseMessages(**reply_message) if isinstance(reply_message, dict) else reply_message, + reply_message=reply_message, ) if not prompt: diff --git a/src/plugin_system/apis/generator_api.py b/src/plugin_system/apis/generator_api.py index dc6a1e6d9..ef3e974bb 100644 --- a/src/plugin_system/apis/generator_api.py +++ b/src/plugin_system/apis/generator_api.py @@ -15,6 +15,7 @@ from rich.traceback import install from src.chat.message_receive.chat_stream import ChatStream from src.chat.utils.utils import process_llm_response +from src.common.data_models.database_data_model import DatabaseMessages from src.common.logger import get_logger from src.plugin_system.base.component_types import ActionInfo @@ -81,7 +82,7 @@ async def generate_reply( chat_id: str | None = None, action_data: dict[str, Any] | None = None, reply_to: str = "", - reply_message: dict[str, Any] | None = None, + reply_message: DatabaseMessages | None = None, extra_info: str = "", available_actions: dict[str, ActionInfo] | None = None, enable_tool: bool = False,