From 773fbbbc34d879df5bebcba08a076eabd7fd7492 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 d1a62f314..cf13a4cda 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 """ @@ -342,7 +342,7 @@ class DefaultReplyer: available_actions=available_actions, choosen_actions=choosen_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 20dcb4a85..54bbb3b43 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,