From b4ac40683ca2562dd88b32a56e1ee99312f09457 Mon Sep 17 00:00:00 2001 From: Ivan Lopez Hernandez Date: Mon, 12 May 2025 16:16:45 -0700 Subject: [PATCH] feat:Added Async wrappers for the Files Module operations PiperOrigin-RevId: 757953630 --- google/genai/tests/files/test_delete.py | 6 +++++- google/genai/tests/files/test_get.py | 4 ++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/google/genai/tests/files/test_delete.py b/google/genai/tests/files/test_delete.py index a8f98761..602395c1 100644 --- a/google/genai/tests/files/test_delete.py +++ b/google/genai/tests/files/test_delete.py @@ -25,6 +25,10 @@ name='test_delete', parameters=types._DeleteFileParameters(name='files/1vx8znuf0yje'), exception_if_vertex='only supported in the Gemini Developer client', + skip_in_api_mode=( + 'The files have a TTL, they cannot be reliably retrieved for a long' + ' time.' + ), ), ] @@ -39,4 +43,4 @@ @pytest.mark.asyncio async def test_async(client): with pytest_helper.exception_if_vertex(client, ValueError): - file = await client.aio.files.get(name='files/vjvu9fwk2qj8') \ No newline at end of file + file = await client.aio.files.get(name='files/vjvu9fwk2qj8') diff --git a/google/genai/tests/files/test_get.py b/google/genai/tests/files/test_get.py index 97c7d364..80f3be94 100644 --- a/google/genai/tests/files/test_get.py +++ b/google/genai/tests/files/test_get.py @@ -25,6 +25,10 @@ name='test_get', parameters=types._GetFileParameters(name='files/vjvu9fwk2qj8'), exception_if_vertex='only supported in the Gemini Developer client', + skip_in_api_mode=( + 'The files have a TTL, they cannot be reliably retrieved for a long' + ' time.' + ), ), ]