Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-14 21:52:56 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
zhenxun/services/llm/api.py
View File

@ -30,6 +30,7 @@ from .types import (
ToolChoice, ToolChoice,
) )
from .types.exceptions import get_user_friendly_error_message from .types.exceptions import get_user_friendly_error_message
from .types.models import GeminiGoogleSearch
from .utils import create_multimodal_message from .utils import create_multimodal_message
T = TypeVar("T", bound=BaseModel) T = TypeVar("T", bound=BaseModel)
@ -302,7 +303,6 @@ async def _generate_image_from_message(
messages = await normalize_to_llm_messages(message) messages = await normalize_to_llm_messages(message)
async with await get_model_instance(model) as model_instance: async with await get_model_instance(model) as model_instance:
response = await model_instance.generate_response(messages, config=config) response = await model_instance.generate_response(messages, config=config)
if not response.images: if not response.images:
@ -403,4 +403,5 @@ async def search(
model=model, model=model,
instruction=instruction, instruction=instruction,
config=final_config, config=final_config,
tools=[GeminiGoogleSearch()],
) )

1
zhenxun/services/llm/session.py
View File

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