diff --git a/zhenxun/services/llm/api.py b/zhenxun/services/llm/api.py
index 644c6de0..524fac08 100644
--- a/zhenxun/services/llm/api.py
+++ b/zhenxun/services/llm/api.py
@@ -30,6 +30,7 @@ from .types import (
     ToolChoice,
 )
 from .types.exceptions import get_user_friendly_error_message
+from .types.models import GeminiGoogleSearch
 from .utils import create_multimodal_message
 
 T = TypeVar("T", bound=BaseModel)
@@ -302,7 +303,6 @@ async def _generate_image_from_message(
         messages = await normalize_to_llm_messages(message)
 
         async with await get_model_instance(model) as model_instance:
-
             response = await model_instance.generate_response(messages, config=config)
 
             if not response.images:
@@ -403,4 +403,5 @@ async def search(
         model=model,
         instruction=instruction,
         config=final_config,
+        tools=[GeminiGoogleSearch()],
     )
diff --git a/zhenxun/services/llm/session.py b/zhenxun/services/llm/session.py
index 5f2e646b..c1cdb678 100644
--- a/zhenxun/services/llm/session.py
+++ b/zhenxun/services/llm/session.py
@@ -445,6 +445,7 @@ class AI:
             instruction=instruction,
             template_vars=template_vars,
             config=search_config,
+            tools=[GeminiGoogleSearch()],
         )
 
     async def generate_structured(