From b692674c5237610c2baf6d6a5ddbb532fd43f8fe Mon Sep 17 00:00:00 2001
From: Soulter <37870767+Soulter@users.noreply.github.com>
Date: Tue, 14 Apr 2026 18:43:52 +0800
Subject: [PATCH] Update astrbot/core/astr_agent_hooks.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
---
 astrbot/core/astr_agent_hooks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/astrbot/core/astr_agent_hooks.py b/astrbot/core/astr_agent_hooks.py
index 551df5989..b45fdb2bd 100644
--- a/astrbot/core/astr_agent_hooks.py
+++ b/astrbot/core/astr_agent_hooks.py
@@ -21,7 +21,7 @@ class MainAgentHooks(BaseAgentRunHooks[AstrAgentContext]):
             run_context,
         )
 
-    async def on_agent_done(self, run_context, llm_response) -> None:
+    async def on_agent_done(self, run_context: ContextWrapper[AstrAgentContext], llm_response: LLMResponse) -> None:
         # 执行事件钩子
         if llm_response and llm_response.reasoning_content:
             # we will use this in result_decorate stage to inject reasoning content to chain