Skip to content

Commit dd11222

Browse files
MarkDaoustcopybara-github
authored andcommitted
chore: [Multimodal FileSearch] Move embedding_model to body
PiperOrigin-RevId: 910258011
1 parent fe5d69e commit dd11222

2 files changed

Lines changed: 8 additions & 1 deletion

File tree

google/genai/file_search_stores.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def _CreateFileSearchStoreConfig_to_mldev(
5151
if getv(from_object, ['embedding_model']) is not None:
5252
setv(
5353
parent_object,
54-
['_query', 'embeddingModel'],
54+
['embeddingModel'],
5555
t.t_model(api_client, getv(from_object, ['embedding_model'])),
5656
)
5757

google/genai/types.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15296,6 +15296,10 @@ class FileSearchStore(_common.BaseModel):
1529615296
description="""The size of raw bytes ingested into the FileSearchStore. This is the
1529715297
total size of all the documents in the FileSearchStore.""",
1529815298
)
15299+
embedding_model: Optional[str] = Field(
15300+
default=None,
15301+
description="""The embedding model used by the FileSearchStore.""",
15302+
)
1529915303

1530015304

1530115305
class FileSearchStoreDict(TypedDict, total=False):
@@ -15326,6 +15330,9 @@ class FileSearchStoreDict(TypedDict, total=False):
1532615330
"""The size of raw bytes ingested into the FileSearchStore. This is the
1532715331
total size of all the documents in the FileSearchStore."""
1532815332

15333+
embedding_model: Optional[str]
15334+
"""The embedding model used by the FileSearchStore."""
15335+
1532915336

1533015337
FileSearchStoreOrDict = Union[FileSearchStore, FileSearchStoreDict]
1533115338

0 commit comments

Comments
 (0)