Merge 78c59f4f28 into e5b2a872d3

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)
@@ -182,7 +183,7 @@ async def embed_documents(
 
 
 async def generate_structured(
-    message: str | LLMMessage | list[LLMContentPart],
+    message: str | UniMessage | LLMMessage | list[LLMContentPart],
     response_model: type[T],
     *,
     model: ModelName = None,
@@ -402,4 +403,5 @@ async def search(
         model=model,
         instruction=instruction,
         config=final_config,
+        tools=[GeminiGoogleSearch()],
     )

zhenxun/services/llm/session.py

@@ -445,11 +445,12 @@ class AI:
             instruction=instruction,
             template_vars=template_vars,
             config=search_config,
+            tools=[GeminiGoogleSearch()],
         )
 
     async def generate_structured(
         self,
-        message: str | LLMMessage | list[LLMContentPart] | None,
+        message: str | UniMessage | LLMMessage | list[LLMContentPart] | None,
         response_model: type[T],
         *,
         model: ModelName = None,