From 6e8d60299cc62bd13129cd8fbfebc07bc16e2c3c Mon Sep 17 00:00:00 2001
From: "zhenguo.li" <zhenguo.li@bytedance.com>
Date: Thu, 18 Dec 2025 21:40:00 +0800
Subject: [PATCH] feat: add web search tool-specific ak/sk

---
 veadk/tools/builtin_tools/web_search.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/veadk/tools/builtin_tools/web_search.py b/veadk/tools/builtin_tools/web_search.py
index bd011d5a..56266097 100644
--- a/veadk/tools/builtin_tools/web_search.py
+++ b/veadk/tools/builtin_tools/web_search.py
@@ -38,7 +38,12 @@ def web_search(query: str, tool_context: ToolContext | None = None) -> list[str]
     """
     ak = None
     sk = None
-    if tool_context:
+    # First try to get tool-specific AK/SK
+    ak = os.getenv("TOOL_WEB_SEARCH_ACCESS_KEY")
+    sk = os.getenv("TOOL_WEB_SEARCH_SECRET_KEY")
+    if ak and sk:
+        logger.debug("Successfully get tool-specific AK/SK.")
+    elif tool_context:
         ak = tool_context.state.get("VOLCENGINE_ACCESS_KEY")
         sk = tool_context.state.get("VOLCENGINE_SECRET_KEY")
     session_token = ""