From eebe31f69dd74d1fb2a3bc10e53d38beeb24d67a Mon Sep 17 00:00:00 2001
From: Soulter <37870767+Soulter@users.noreply.github.com>
Date: Fri, 23 Jan 2026 19:52:31 +0800
Subject: [PATCH] fix: update web_search_tavily handling for webchat platform
 (#4633)

* fix: update web_search_tavily handling for webchat platform

* chore: style consistent
---
 astrbot/core/astr_agent_hooks.py             | 4 +++-
 astrbot/core/pipeline/process_stage/utils.py | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/astrbot/core/astr_agent_hooks.py b/astrbot/core/astr_agent_hooks.py
index fa0bb02a9..717d4a3e1 100644
--- a/astrbot/core/astr_agent_hooks.py
+++ b/astrbot/core/astr_agent_hooks.py
@@ -56,8 +56,10 @@ class MainAgentHooks(BaseAgentRunHooks[AstrAgentContext]):
         )
 
         # special handle web_search_tavily
+        platform_name = run_context.context.event.get_platform_name()
         if (
-            tool.name == "web_search_tavily"
+            platform_name == "webchat"
+            and tool.name == "web_search_tavily"
             and len(run_context.messages) > 0
             and tool_result
             and len(tool_result.content)
diff --git a/astrbot/core/pipeline/process_stage/utils.py b/astrbot/core/pipeline/process_stage/utils.py
index 3526efdb0..57e8667af 100644
--- a/astrbot/core/pipeline/process_stage/utils.py
+++ b/astrbot/core/pipeline/process_stage/utils.py
@@ -41,6 +41,7 @@ TOOL_CALL_PROMPT = (
     "You MUST NOT return an empty response, especially after invoking a tool."
     "Before calling any tool, provide a brief explanatory message to the user stating the purpose of the tool call."
     "After the tool call is completed, you must briefly summarize the results returned by the tool for the user."
+    "Keep the role-play and style consistent throughout the conversation."
 )
 
 CHATUI_SPECIAL_DEFAULT_PERSONA_PROMPT = (