feat(search): 为搜索功能默认启用 Gemini Google Search 工具

This commit is contained in:
webjoin111 2025-12-12 17:07:00 +08:00
parent 30101cf1f0
commit de5c77407d
2 changed files with 3 additions and 1 deletions

View File

@ -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()],
)

View File

@ -445,6 +445,7 @@ class AI:
instruction=instruction,
template_vars=template_vars,
config=search_config,
tools=[GeminiGoogleSearch()],
)
async def generate_structured(