Skip to content

Commit 8eba82d

Browse files
committed
use file_search_store_names for Google file search
Changed from file_names to file_search_store_names to match the Google SDK and maintain consistency with OpenAI's store-based approach.
1 parent c83f125 commit 8eba82d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pydantic_ai_slim/pydantic_ai/models/google.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,7 @@ def _get_tools(self, model_request_parameters: ModelRequestParameters) -> list[T
343343
elif isinstance(tool, CodeExecutionTool):
344344
tools.append(ToolDict(code_execution=ToolCodeExecutionDict()))
345345
elif isinstance(tool, FileSearchTool):
346-
tools.append(ToolDict(file_search={'file_names': tool.vector_store_ids})) # type: ignore[reportGeneralTypeIssues]
346+
tools.append(ToolDict(file_search={'file_search_store_names': tool.vector_store_ids})) # type: ignore[reportGeneralTypeIssues]
347347
elif isinstance(tool, ImageGenerationTool): # pragma: no branch
348348
if not self.profile.supports_image_output:
349349
raise UserError(

0 commit comments

Comments
 (0)