Compare commits

..

1 Commits

Author SHA1 Message Date
pre-commit-ci[bot]
5bc5feca78 🚨 auto fix by pre-commit hooks 2025-12-09 03:23:03 +00:00
2 changed files with 2 additions and 5 deletions

View File

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

View File

@ -445,12 +445,11 @@ 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(
self, self,
message: str | UniMessage | LLMMessage | list[LLMContentPart] | None, message: str | LLMMessage | list[LLMContentPart] | None,
response_model: type[T], response_model: type[T],
*, *,
model: ModelName = None, model: ModelName = None,