Skip to content

Commit 419a7ce

Browse files
Accept partial file names (#386)
Adds prefix files/ if not present Change-Id: Iac5c4d0934620f2462cf15ae519474b0ce7908da
1 parent 23b81d7 commit 419a7ce

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

google/generativeai/files.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,16 +81,20 @@ def list_files(page_size=100) -> Iterable[file_types.File]:
8181
yield file_types.File(proto)
8282

8383

84-
def get_file(name) -> file_types.File:
84+
def get_file(name: str) -> file_types.File:
8585
"""Calls the API to retrieve a specified file using a supported file service."""
86+
if "/" not in name:
87+
name = f"files/{name}"
8688
client = get_default_file_client()
8789
return file_types.File(client.get_file(name=name))
8890

8991

90-
def delete_file(name):
92+
def delete_file(name: str | file_types.File | protos.File):
9193
"""Calls the API to permanently delete a specified file using a supported file service."""
9294
if isinstance(name, (file_types.File, protos.File)):
9395
name = name.name
96+
elif "/" not in name:
97+
name = f"files/{name}"
9498
request = protos.DeleteFileRequest(name=name)
9599
client = get_default_file_client()
96100
client.delete_file(request=request)

0 commit comments

Comments
 (0)