Merge 78c59f4f28 into e5b2a872d3

🚨 auto fix by pre-commit hooks
✨ feat(search): 为搜索功能默认启用 Gemini Google Search 工具
2026-10-11 06:39:59 +08:00 · 2025-12-12 09:12:39 +00:00 · 2025-12-12 09:12:35 +00:00 · 2025-12-12 17:12:18 +08:00 · 2025-12-12 17:12:18 +08:00
2 changed files with 5 additions and 2 deletions
@@ -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()],
    )
@@ -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,
Author	SHA1	Message	Date
Rumio	dc92d7abc1	Merge `78c59f4f28` into `e5b2a872d3`	2025-12-12 09:12:39 +00:00
pre-commit-ci[bot]	78c59f4f28	🚨 auto fix by pre-commit hooks	2025-12-12 09:12:35 +00:00
webjoin111	de5c77407d	✨ feat(search): 为搜索功能默认启用 Gemini Google Search 工具	2025-12-12 17:12:18 +08:00
webjoin111	30101cf1f0	✨ feat(llm): 支持结构化生成函数接收 UniMessage	2025-12-12 17:12:18 +08:00