google-genai 1.54.0__py3-none-any.whl → 1.55.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- google/genai/__init__.py +1 -0
- google/genai/_interactions/__init__.py +117 -0
- google/genai/_interactions/_base_client.py +2019 -0
- google/genai/_interactions/_client.py +511 -0
- google/genai/_interactions/_compat.py +234 -0
- google/genai/_interactions/_constants.py +29 -0
- google/genai/_interactions/_exceptions.py +122 -0
- google/genai/_interactions/_files.py +139 -0
- google/genai/_interactions/_models.py +873 -0
- google/genai/_interactions/_qs.py +165 -0
- google/genai/_interactions/_resource.py +58 -0
- google/genai/_interactions/_response.py +847 -0
- google/genai/_interactions/_streaming.py +354 -0
- google/genai/_interactions/_types.py +276 -0
- google/genai/_interactions/_utils/__init__.py +79 -0
- google/genai/_interactions/_utils/_compat.py +61 -0
- google/genai/_interactions/_utils/_datetime_parse.py +151 -0
- google/genai/_interactions/_utils/_logs.py +40 -0
- google/genai/_interactions/_utils/_proxy.py +80 -0
- google/genai/_interactions/_utils/_reflection.py +57 -0
- google/genai/_interactions/_utils/_resources_proxy.py +39 -0
- google/genai/_interactions/_utils/_streams.py +27 -0
- google/genai/_interactions/_utils/_sync.py +73 -0
- google/genai/_interactions/_utils/_transform.py +472 -0
- google/genai/_interactions/_utils/_typing.py +172 -0
- google/genai/_interactions/_utils/_utils.py +437 -0
- google/genai/_interactions/_version.py +18 -0
- google/genai/_interactions/resources/__init__.py +34 -0
- google/genai/_interactions/resources/interactions.py +1350 -0
- google/genai/_interactions/types/__init__.py +107 -0
- google/genai/_interactions/types/allowed_tools.py +33 -0
- google/genai/_interactions/types/allowed_tools_param.py +35 -0
- google/genai/_interactions/types/annotation.py +42 -0
- google/genai/_interactions/types/annotation_param.py +42 -0
- google/genai/_interactions/types/audio_content.py +38 -0
- google/genai/_interactions/types/audio_content_param.py +45 -0
- google/genai/_interactions/types/audio_mime_type.py +25 -0
- google/genai/_interactions/types/audio_mime_type_param.py +27 -0
- google/genai/_interactions/types/code_execution_call_arguments.py +33 -0
- google/genai/_interactions/types/code_execution_call_arguments_param.py +32 -0
- google/genai/_interactions/types/code_execution_call_content.py +37 -0
- google/genai/_interactions/types/code_execution_call_content_param.py +37 -0
- google/genai/_interactions/types/code_execution_result_content.py +42 -0
- google/genai/_interactions/types/code_execution_result_content_param.py +41 -0
- google/genai/_interactions/types/content_delta.py +358 -0
- google/genai/_interactions/types/content_start.py +79 -0
- google/genai/_interactions/types/content_stop.py +35 -0
- google/genai/_interactions/types/deep_research_agent_config.py +33 -0
- google/genai/_interactions/types/deep_research_agent_config_param.py +32 -0
- google/genai/_interactions/types/document_content.py +36 -0
- google/genai/_interactions/types/document_content_param.py +43 -0
- google/genai/_interactions/types/dynamic_agent_config.py +44 -0
- google/genai/_interactions/types/dynamic_agent_config_param.py +33 -0
- google/genai/_interactions/types/error_event.py +46 -0
- google/genai/_interactions/types/file_search_result_content.py +46 -0
- google/genai/_interactions/types/file_search_result_content_param.py +46 -0
- google/genai/_interactions/types/function.py +38 -0
- google/genai/_interactions/types/function_call_content.py +39 -0
- google/genai/_interactions/types/function_call_content_param.py +39 -0
- google/genai/_interactions/types/function_param.py +37 -0
- google/genai/_interactions/types/function_result_content.py +52 -0
- google/genai/_interactions/types/function_result_content_param.py +54 -0
- google/genai/_interactions/types/generation_config.py +57 -0
- google/genai/_interactions/types/generation_config_param.py +59 -0
- google/genai/_interactions/types/google_search_call_arguments.py +29 -0
- google/genai/_interactions/types/google_search_call_arguments_param.py +31 -0
- google/genai/_interactions/types/google_search_call_content.py +37 -0
- google/genai/_interactions/types/google_search_call_content_param.py +37 -0
- google/genai/_interactions/types/google_search_result.py +35 -0
- google/genai/_interactions/types/google_search_result_content.py +43 -0
- google/genai/_interactions/types/google_search_result_content_param.py +44 -0
- google/genai/_interactions/types/google_search_result_param.py +35 -0
- google/genai/_interactions/types/image_content.py +41 -0
- google/genai/_interactions/types/image_content_param.py +48 -0
- google/genai/_interactions/types/image_mime_type.py +23 -0
- google/genai/_interactions/types/image_mime_type_param.py +25 -0
- google/genai/_interactions/types/interaction.py +165 -0
- google/genai/_interactions/types/interaction_create_params.py +212 -0
- google/genai/_interactions/types/interaction_event.py +37 -0
- google/genai/_interactions/types/interaction_get_params.py +46 -0
- google/genai/_interactions/types/interaction_sse_event.py +32 -0
- google/genai/_interactions/types/interaction_status_update.py +37 -0
- google/genai/_interactions/types/mcp_server_tool_call_content.py +42 -0
- google/genai/_interactions/types/mcp_server_tool_call_content_param.py +42 -0
- google/genai/_interactions/types/mcp_server_tool_result_content.py +52 -0
- google/genai/_interactions/types/mcp_server_tool_result_content_param.py +54 -0
- google/genai/_interactions/types/model.py +36 -0
- google/genai/_interactions/types/model_param.py +38 -0
- google/genai/_interactions/types/speech_config.py +35 -0
- google/genai/_interactions/types/speech_config_param.py +35 -0
- google/genai/_interactions/types/text_content.py +37 -0
- google/genai/_interactions/types/text_content_param.py +38 -0
- google/genai/_interactions/types/thinking_level.py +22 -0
- google/genai/_interactions/types/thought_content.py +41 -0
- google/genai/_interactions/types/thought_content_param.py +47 -0
- google/genai/_interactions/types/tool.py +100 -0
- google/genai/_interactions/types/tool_choice.py +26 -0
- google/genai/_interactions/types/tool_choice_config.py +28 -0
- google/genai/_interactions/types/tool_choice_config_param.py +29 -0
- google/genai/_interactions/types/tool_choice_param.py +28 -0
- google/genai/_interactions/types/tool_choice_type.py +22 -0
- google/genai/_interactions/types/tool_param.py +97 -0
- google/genai/_interactions/types/turn.py +76 -0
- google/genai/_interactions/types/turn_param.py +73 -0
- google/genai/_interactions/types/url_context_call_arguments.py +29 -0
- google/genai/_interactions/types/url_context_call_arguments_param.py +31 -0
- google/genai/_interactions/types/url_context_call_content.py +37 -0
- google/genai/_interactions/types/url_context_call_content_param.py +37 -0
- google/genai/_interactions/types/url_context_result.py +33 -0
- google/genai/_interactions/types/url_context_result_content.py +43 -0
- google/genai/_interactions/types/url_context_result_content_param.py +44 -0
- google/genai/_interactions/types/url_context_result_param.py +32 -0
- google/genai/_interactions/types/usage.py +106 -0
- google/genai/_interactions/types/usage_param.py +106 -0
- google/genai/_interactions/types/video_content.py +41 -0
- google/genai/_interactions/types/video_content_param.py +48 -0
- google/genai/_interactions/types/video_mime_type.py +36 -0
- google/genai/_interactions/types/video_mime_type_param.py +38 -0
- google/genai/_live_converters.py +31 -0
- google/genai/_tokens_converters.py +5 -0
- google/genai/batches.py +7 -0
- google/genai/client.py +223 -0
- google/genai/interactions.py +17 -0
- google/genai/live.py +4 -3
- google/genai/models.py +12 -0
- google/genai/tests/__init__.py +21 -0
- google/genai/tests/afc/__init__.py +21 -0
- google/genai/tests/afc/test_convert_if_exist_pydantic_model.py +309 -0
- google/genai/tests/afc/test_convert_number_values_for_function_call_args.py +63 -0
- google/genai/tests/afc/test_find_afc_incompatible_tool_indexes.py +240 -0
- google/genai/tests/afc/test_generate_content_stream_afc.py +530 -0
- google/genai/tests/afc/test_generate_content_stream_afc_thoughts.py +77 -0
- google/genai/tests/afc/test_get_function_map.py +176 -0
- google/genai/tests/afc/test_get_function_response_parts.py +277 -0
- google/genai/tests/afc/test_get_max_remote_calls_for_afc.py +130 -0
- google/genai/tests/afc/test_invoke_function_from_dict_args.py +241 -0
- google/genai/tests/afc/test_raise_error_for_afc_incompatible_config.py +159 -0
- google/genai/tests/afc/test_should_append_afc_history.py +53 -0
- google/genai/tests/afc/test_should_disable_afc.py +214 -0
- google/genai/tests/batches/__init__.py +17 -0
- google/genai/tests/batches/test_cancel.py +77 -0
- google/genai/tests/batches/test_create.py +78 -0
- google/genai/tests/batches/test_create_with_bigquery.py +113 -0
- google/genai/tests/batches/test_create_with_file.py +82 -0
- google/genai/tests/batches/test_create_with_gcs.py +125 -0
- google/genai/tests/batches/test_create_with_inlined_requests.py +255 -0
- google/genai/tests/batches/test_delete.py +86 -0
- google/genai/tests/batches/test_embedding.py +157 -0
- google/genai/tests/batches/test_get.py +78 -0
- google/genai/tests/batches/test_list.py +79 -0
- google/genai/tests/caches/__init__.py +17 -0
- google/genai/tests/caches/constants.py +29 -0
- google/genai/tests/caches/test_create.py +210 -0
- google/genai/tests/caches/test_create_custom_url.py +105 -0
- google/genai/tests/caches/test_delete.py +54 -0
- google/genai/tests/caches/test_delete_custom_url.py +52 -0
- google/genai/tests/caches/test_get.py +94 -0
- google/genai/tests/caches/test_get_custom_url.py +52 -0
- google/genai/tests/caches/test_list.py +68 -0
- google/genai/tests/caches/test_update.py +70 -0
- google/genai/tests/caches/test_update_custom_url.py +58 -0
- google/genai/tests/chats/__init__.py +1 -0
- google/genai/tests/chats/test_get_history.py +597 -0
- google/genai/tests/chats/test_send_message.py +844 -0
- google/genai/tests/chats/test_validate_response.py +90 -0
- google/genai/tests/client/__init__.py +17 -0
- google/genai/tests/client/test_async_stream.py +427 -0
- google/genai/tests/client/test_client_close.py +197 -0
- google/genai/tests/client/test_client_initialization.py +1687 -0
- google/genai/tests/client/test_client_requests.py +355 -0
- google/genai/tests/client/test_custom_client.py +77 -0
- google/genai/tests/client/test_http_options.py +178 -0
- google/genai/tests/client/test_replay_client_equality.py +168 -0
- google/genai/tests/client/test_retries.py +846 -0
- google/genai/tests/client/test_upload_errors.py +136 -0
- google/genai/tests/common/__init__.py +17 -0
- google/genai/tests/common/test_common.py +954 -0
- google/genai/tests/conftest.py +162 -0
- google/genai/tests/documents/__init__.py +17 -0
- google/genai/tests/documents/test_delete.py +51 -0
- google/genai/tests/documents/test_get.py +85 -0
- google/genai/tests/documents/test_list.py +72 -0
- google/genai/tests/errors/__init__.py +1 -0
- google/genai/tests/errors/test_api_error.py +417 -0
- google/genai/tests/file_search_stores/__init__.py +17 -0
- google/genai/tests/file_search_stores/test_create.py +66 -0
- google/genai/tests/file_search_stores/test_delete.py +64 -0
- google/genai/tests/file_search_stores/test_get.py +94 -0
- google/genai/tests/file_search_stores/test_import_file.py +112 -0
- google/genai/tests/file_search_stores/test_list.py +57 -0
- google/genai/tests/file_search_stores/test_upload_to_file_search_store.py +141 -0
- google/genai/tests/files/__init__.py +17 -0
- google/genai/tests/files/test_delete.py +46 -0
- google/genai/tests/files/test_download.py +85 -0
- google/genai/tests/files/test_get.py +46 -0
- google/genai/tests/files/test_list.py +72 -0
- google/genai/tests/files/test_upload.py +255 -0
- google/genai/tests/imports/test_no_optional_imports.py +28 -0
- google/genai/tests/interactions/__init__.py +0 -0
- google/genai/tests/interactions/test_integration.py +80 -0
- google/genai/tests/live/__init__.py +16 -0
- google/genai/tests/live/test_live.py +2177 -0
- google/genai/tests/live/test_live_music.py +362 -0
- google/genai/tests/live/test_live_response.py +163 -0
- google/genai/tests/live/test_send_client_content.py +147 -0
- google/genai/tests/live/test_send_realtime_input.py +268 -0
- google/genai/tests/live/test_send_tool_response.py +222 -0
- google/genai/tests/local_tokenizer/__init__.py +17 -0
- google/genai/tests/local_tokenizer/test_local_tokenizer.py +343 -0
- google/genai/tests/local_tokenizer/test_local_tokenizer_loader.py +235 -0
- google/genai/tests/mcp/__init__.py +17 -0
- google/genai/tests/mcp/test_has_mcp_tool_usage.py +89 -0
- google/genai/tests/mcp/test_mcp_to_gemini_tools.py +191 -0
- google/genai/tests/mcp/test_parse_config_for_mcp_sessions.py +201 -0
- google/genai/tests/mcp/test_parse_config_for_mcp_usage.py +130 -0
- google/genai/tests/mcp/test_set_mcp_usage_header.py +72 -0
- google/genai/tests/models/__init__.py +17 -0
- google/genai/tests/models/constants.py +8 -0
- google/genai/tests/models/test_compute_tokens.py +120 -0
- google/genai/tests/models/test_count_tokens.py +159 -0
- google/genai/tests/models/test_delete.py +107 -0
- google/genai/tests/models/test_edit_image.py +264 -0
- google/genai/tests/models/test_embed_content.py +94 -0
- google/genai/tests/models/test_function_call_streaming.py +442 -0
- google/genai/tests/models/test_generate_content.py +2502 -0
- google/genai/tests/models/test_generate_content_cached_content.py +132 -0
- google/genai/tests/models/test_generate_content_config_zero_value.py +103 -0
- google/genai/tests/models/test_generate_content_from_apikey.py +44 -0
- google/genai/tests/models/test_generate_content_http_options.py +40 -0
- google/genai/tests/models/test_generate_content_image_generation.py +143 -0
- google/genai/tests/models/test_generate_content_mcp.py +343 -0
- google/genai/tests/models/test_generate_content_media_resolution.py +97 -0
- google/genai/tests/models/test_generate_content_model.py +139 -0
- google/genai/tests/models/test_generate_content_part.py +821 -0
- google/genai/tests/models/test_generate_content_thought.py +76 -0
- google/genai/tests/models/test_generate_content_tools.py +1761 -0
- google/genai/tests/models/test_generate_images.py +191 -0
- google/genai/tests/models/test_generate_videos.py +759 -0
- google/genai/tests/models/test_get.py +104 -0
- google/genai/tests/models/test_list.py +233 -0
- google/genai/tests/models/test_recontext_image.py +189 -0
- google/genai/tests/models/test_segment_image.py +148 -0
- google/genai/tests/models/test_update.py +95 -0
- google/genai/tests/models/test_upscale_image.py +157 -0
- google/genai/tests/operations/__init__.py +17 -0
- google/genai/tests/operations/test_get.py +38 -0
- google/genai/tests/public_samples/__init__.py +17 -0
- google/genai/tests/public_samples/test_gemini_text_only.py +34 -0
- google/genai/tests/pytest_helper.py +229 -0
- google/genai/tests/shared/__init__.py +16 -0
- google/genai/tests/shared/batches/__init__.py +14 -0
- google/genai/tests/shared/batches/test_create_delete.py +57 -0
- google/genai/tests/shared/batches/test_create_get_cancel.py +56 -0
- google/genai/tests/shared/batches/test_list.py +40 -0
- google/genai/tests/shared/caches/__init__.py +14 -0
- google/genai/tests/shared/caches/test_create_get_delete.py +67 -0
- google/genai/tests/shared/caches/test_create_update_get.py +71 -0
- google/genai/tests/shared/caches/test_list.py +40 -0
- google/genai/tests/shared/chats/__init__.py +14 -0
- google/genai/tests/shared/chats/test_send_message.py +48 -0
- google/genai/tests/shared/chats/test_send_message_stream.py +50 -0
- google/genai/tests/shared/files/__init__.py +14 -0
- google/genai/tests/shared/files/test_list.py +41 -0
- google/genai/tests/shared/files/test_upload_get_delete.py +54 -0
- google/genai/tests/shared/models/__init__.py +14 -0
- google/genai/tests/shared/models/test_compute_tokens.py +41 -0
- google/genai/tests/shared/models/test_count_tokens.py +40 -0
- google/genai/tests/shared/models/test_edit_image.py +67 -0
- google/genai/tests/shared/models/test_embed.py +40 -0
- google/genai/tests/shared/models/test_generate_content.py +39 -0
- google/genai/tests/shared/models/test_generate_content_stream.py +54 -0
- google/genai/tests/shared/models/test_generate_images.py +40 -0
- google/genai/tests/shared/models/test_generate_videos.py +38 -0
- google/genai/tests/shared/models/test_list.py +37 -0
- google/genai/tests/shared/models/test_recontext_image.py +55 -0
- google/genai/tests/shared/models/test_segment_image.py +52 -0
- google/genai/tests/shared/models/test_upscale_image.py +52 -0
- google/genai/tests/shared/tunings/__init__.py +16 -0
- google/genai/tests/shared/tunings/test_create.py +46 -0
- google/genai/tests/shared/tunings/test_create_get_cancel.py +56 -0
- google/genai/tests/shared/tunings/test_list.py +39 -0
- google/genai/tests/tokens/__init__.py +16 -0
- google/genai/tests/tokens/test_create.py +154 -0
- google/genai/tests/transformers/__init__.py +17 -0
- google/genai/tests/transformers/test_blobs.py +71 -0
- google/genai/tests/transformers/test_bytes.py +15 -0
- google/genai/tests/transformers/test_duck_type.py +96 -0
- google/genai/tests/transformers/test_function_responses.py +72 -0
- google/genai/tests/transformers/test_schema.py +653 -0
- google/genai/tests/transformers/test_t_batch.py +286 -0
- google/genai/tests/transformers/test_t_content.py +160 -0
- google/genai/tests/transformers/test_t_contents.py +398 -0
- google/genai/tests/transformers/test_t_part.py +85 -0
- google/genai/tests/transformers/test_t_parts.py +87 -0
- google/genai/tests/transformers/test_t_tool.py +157 -0
- google/genai/tests/transformers/test_t_tools.py +195 -0
- google/genai/tests/tunings/__init__.py +16 -0
- google/genai/tests/tunings/test_cancel.py +39 -0
- google/genai/tests/tunings/test_end_to_end.py +106 -0
- google/genai/tests/tunings/test_get.py +67 -0
- google/genai/tests/tunings/test_list.py +75 -0
- google/genai/tests/tunings/test_tune.py +268 -0
- google/genai/tests/types/__init__.py +16 -0
- google/genai/tests/types/test_bytes_internal.py +271 -0
- google/genai/tests/types/test_bytes_type.py +152 -0
- google/genai/tests/types/test_future.py +101 -0
- google/genai/tests/types/test_optional_types.py +36 -0
- google/genai/tests/types/test_part_type.py +616 -0
- google/genai/tests/types/test_schema_from_json_schema.py +417 -0
- google/genai/tests/types/test_schema_json_schema.py +468 -0
- google/genai/tests/types/test_types.py +2903 -0
- google/genai/types.py +72 -0
- google/genai/version.py +1 -1
- {google_genai-1.54.0.dist-info → google_genai-1.55.0.dist-info}/METADATA +3 -1
- google_genai-1.55.0.dist-info/RECORD +345 -0
- google_genai-1.54.0.dist-info/RECORD +0 -41
- {google_genai-1.54.0.dist-info → google_genai-1.55.0.dist-info}/WHEEL +0 -0
- {google_genai-1.54.0.dist-info → google_genai-1.55.0.dist-info}/licenses/LICENSE +0 -0
- {google_genai-1.54.0.dist-info → google_genai-1.55.0.dist-info}/top_level.txt +0 -0
|
@@ -0,0 +1,210 @@
|
|
|
1
|
+
# Copyright 2025 Google LLC
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
#
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
from copy import deepcopy
|
|
18
|
+
import datetime
|
|
19
|
+
import pytest
|
|
20
|
+
import sys
|
|
21
|
+
from ... import types
|
|
22
|
+
from .. import pytest_helper
|
|
23
|
+
from ... import _transformers as t
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI = types._CreateCachedContentParameters(
|
|
27
|
+
model='gemini-2.5-flash',
|
|
28
|
+
config={
|
|
29
|
+
'contents': [
|
|
30
|
+
types.Content(
|
|
31
|
+
role='user',
|
|
32
|
+
parts=[
|
|
33
|
+
types.Part(
|
|
34
|
+
fileData=types.FileData(
|
|
35
|
+
fileUri='gs://cloud-samples-data/generative-ai/pdf/2312.11805v3.pdf',
|
|
36
|
+
mimeType='application/pdf',
|
|
37
|
+
)
|
|
38
|
+
),
|
|
39
|
+
types.Part(
|
|
40
|
+
fileData=types.FileData(
|
|
41
|
+
fileUri='gs://cloud-samples-data/generative-ai/pdf/2403.05530.pdf',
|
|
42
|
+
mimeType='application/pdf',
|
|
43
|
+
)
|
|
44
|
+
),
|
|
45
|
+
],
|
|
46
|
+
)
|
|
47
|
+
],
|
|
48
|
+
'system_instruction': t.t_content('What is the sum of the two pdfs?'),
|
|
49
|
+
'display_name': 'test cache',
|
|
50
|
+
'ttl': '86400s',
|
|
51
|
+
},
|
|
52
|
+
)
|
|
53
|
+
|
|
54
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE = types._CreateCachedContentParameters(
|
|
55
|
+
model='gemini-2.5-flash',
|
|
56
|
+
config={
|
|
57
|
+
'contents': [
|
|
58
|
+
types.Content(
|
|
59
|
+
role='user',
|
|
60
|
+
parts=[
|
|
61
|
+
types.Part(
|
|
62
|
+
fileData=types.FileData(
|
|
63
|
+
mimeType='application/pdf',
|
|
64
|
+
fileUri='https://generativelanguage.googleapis.com/v1beta/files/v200dhvn15h7',
|
|
65
|
+
)
|
|
66
|
+
)
|
|
67
|
+
],
|
|
68
|
+
)
|
|
69
|
+
],
|
|
70
|
+
'system_instruction': t.t_content('What is the sum of the two pdfs?'),
|
|
71
|
+
'display_name': 'test cache',
|
|
72
|
+
'ttl': '86400s',
|
|
73
|
+
},
|
|
74
|
+
)
|
|
75
|
+
|
|
76
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_PARTIAL_MODEL_1 = deepcopy(
|
|
77
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI
|
|
78
|
+
)
|
|
79
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_PARTIAL_MODEL_1.model = (
|
|
80
|
+
'models/gemini-2.5-flash'
|
|
81
|
+
)
|
|
82
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_PARTIAL_MODEL_2 = deepcopy(
|
|
83
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI
|
|
84
|
+
)
|
|
85
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_PARTIAL_MODEL_2.model = (
|
|
86
|
+
'publishers/google/models/gemini-2.5-flash'
|
|
87
|
+
)
|
|
88
|
+
|
|
89
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE_PARTIAL_MODEL_1 = deepcopy(
|
|
90
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE
|
|
91
|
+
)
|
|
92
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE_PARTIAL_MODEL_1.model = (
|
|
93
|
+
'models/gemini-2.5-flash'
|
|
94
|
+
)
|
|
95
|
+
|
|
96
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_CMEK = deepcopy(
|
|
97
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI
|
|
98
|
+
)
|
|
99
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_CMEK.config.kms_key_name = (
|
|
100
|
+
'projects/test-project/locations/us-central1/keyRings/test-keyring/cryptoKeys/test-key'
|
|
101
|
+
)
|
|
102
|
+
|
|
103
|
+
if sys.version_info >= (3, 11):
|
|
104
|
+
_EXPIRE_TIME = datetime.datetime.fromisoformat('2025-12-20T00:00:00Z')
|
|
105
|
+
else:
|
|
106
|
+
_EXPIRE_TIME = datetime.datetime.fromisoformat('2025-12-20T00:00:00+00:00')
|
|
107
|
+
|
|
108
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_EXPIRE_TIME = deepcopy(
|
|
109
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI
|
|
110
|
+
)
|
|
111
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_EXPIRE_TIME.config.ttl = None
|
|
112
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_EXPIRE_TIME.config.expire_time = (
|
|
113
|
+
_EXPIRE_TIME
|
|
114
|
+
)
|
|
115
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_EXPIRE_TIME.config.display_name = (
|
|
116
|
+
'test cache'
|
|
117
|
+
)
|
|
118
|
+
|
|
119
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE_EXPIRE_TIME = deepcopy(
|
|
120
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE
|
|
121
|
+
)
|
|
122
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE_EXPIRE_TIME.config.ttl = None
|
|
123
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE_EXPIRE_TIME.config.expire_time = (
|
|
124
|
+
_EXPIRE_TIME
|
|
125
|
+
)
|
|
126
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE_EXPIRE_TIME.config.display_name = (
|
|
127
|
+
'test cache'
|
|
128
|
+
)
|
|
129
|
+
|
|
130
|
+
|
|
131
|
+
# Replay mode is not supported for caches tests due to the error message
|
|
132
|
+
# inconsistency in api and replay mode.
|
|
133
|
+
# To run api mode tests, use the following steps:
|
|
134
|
+
# 1. First create the resource.
|
|
135
|
+
# sh run_tests.sh pytest -s tests/caches/test_create.py --mode=api
|
|
136
|
+
# 1.1 If mldev test_create fails, update the uploaded file using this colab
|
|
137
|
+
# https://colab.sandbox.google.com/drive/1Fv6KGSs0cg6tlpcUHdsclHussXMEGOXk#scrollTo=RSKmFPx00MVL.
|
|
138
|
+
# 2. Find the resource name in debugging print and change the resource name constants.py.
|
|
139
|
+
# 3. Run and record get and update tests.
|
|
140
|
+
# sh run_tests.sh pytest -s tests/caches/test_get.py --mode=api && sh run_tests.sh pytest -s tests/caches/test_update.py --mode=api && sh run_tests.sh pytest -s tests/caches/test_delete.py --mode=api
|
|
141
|
+
test_table: list[pytest_helper.TestTableItem] = [
|
|
142
|
+
pytest_helper.TestTableItem(
|
|
143
|
+
name='test_caches_create_with_gcs_uri',
|
|
144
|
+
exception_if_mldev='INVALID_ARGUMENT',
|
|
145
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI,
|
|
146
|
+
),
|
|
147
|
+
pytest_helper.TestTableItem(
|
|
148
|
+
name='test_caches_create_with_gcs_uri_cmek',
|
|
149
|
+
exception_if_mldev='not supported',
|
|
150
|
+
exception_if_vertex='INVALID_ARGUMENT', # The key is invalid.
|
|
151
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_CMEK,
|
|
152
|
+
),
|
|
153
|
+
pytest_helper.TestTableItem(
|
|
154
|
+
name='test_caches_create_with_gcs_uri_expire_time',
|
|
155
|
+
exception_if_mldev='INVALID_ARGUMENT',
|
|
156
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_EXPIRE_TIME,
|
|
157
|
+
),
|
|
158
|
+
pytest_helper.TestTableItem(
|
|
159
|
+
name='test_caches_create_model_partial_path_1',
|
|
160
|
+
exception_if_mldev='INVALID_ARGUMENT',
|
|
161
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_PARTIAL_MODEL_1,
|
|
162
|
+
),
|
|
163
|
+
pytest_helper.TestTableItem(
|
|
164
|
+
name='test_caches_create_model_partial_path_2',
|
|
165
|
+
exception_if_mldev='404',
|
|
166
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI_PARTIAL_MODEL_2,
|
|
167
|
+
),
|
|
168
|
+
pytest_helper.TestTableItem(
|
|
169
|
+
name='test_caches_create_with_googleai_file',
|
|
170
|
+
exception_if_vertex='Internal',
|
|
171
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE,
|
|
172
|
+
skip_in_api_mode='Create is not reproducible in the API mode.',
|
|
173
|
+
),
|
|
174
|
+
pytest_helper.TestTableItem(
|
|
175
|
+
name='test_caches_create_with_googleai_file_expire_time',
|
|
176
|
+
exception_if_vertex='Internal',
|
|
177
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE_EXPIRE_TIME,
|
|
178
|
+
skip_in_api_mode='Create is not reproducible in the API mode.',
|
|
179
|
+
),
|
|
180
|
+
pytest_helper.TestTableItem(
|
|
181
|
+
name='test_caches_create_with_googleai_file_model_partial_path_1',
|
|
182
|
+
exception_if_vertex='Internal',
|
|
183
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE_PARTIAL_MODEL_1,
|
|
184
|
+
skip_in_api_mode='Create is not reproducible in the API mode.',
|
|
185
|
+
),
|
|
186
|
+
]
|
|
187
|
+
pytestmark = [
|
|
188
|
+
pytest_helper.setup(
|
|
189
|
+
file=__file__,
|
|
190
|
+
globals_for_file=globals(),
|
|
191
|
+
test_method='caches.create',
|
|
192
|
+
test_table=test_table,
|
|
193
|
+
),
|
|
194
|
+
]
|
|
195
|
+
pytest_plugins = ('pytest_asyncio',)
|
|
196
|
+
|
|
197
|
+
|
|
198
|
+
@pytest.mark.asyncio
|
|
199
|
+
async def test_async_googleai_file_create(client):
|
|
200
|
+
if client._api_client.vertexai:
|
|
201
|
+
with pytest.raises(Exception):
|
|
202
|
+
await client.aio.caches.create(
|
|
203
|
+
model=_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE.model,
|
|
204
|
+
config=_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE.config,
|
|
205
|
+
)
|
|
206
|
+
else:
|
|
207
|
+
await client.aio.caches.create(
|
|
208
|
+
model=_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE.model,
|
|
209
|
+
config=_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE.config,
|
|
210
|
+
)
|
|
@@ -0,0 +1,105 @@
|
|
|
1
|
+
# Copyright 2025 Google LLC
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
#
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
import copy
|
|
18
|
+
import pytest
|
|
19
|
+
from ... import types
|
|
20
|
+
from .. import pytest_helper
|
|
21
|
+
from . import constants
|
|
22
|
+
from ... import _transformers as t
|
|
23
|
+
|
|
24
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI = types._CreateCachedContentParameters(
|
|
25
|
+
model='gemini-2.5-flash',
|
|
26
|
+
config={
|
|
27
|
+
'contents': [
|
|
28
|
+
types.Content(
|
|
29
|
+
role='user',
|
|
30
|
+
parts=[
|
|
31
|
+
types.Part(
|
|
32
|
+
fileData=types.FileData(
|
|
33
|
+
fileUri='gs://cloud-samples-data/generative-ai/pdf/2312.11805v3.pdf',
|
|
34
|
+
mimeType='application/pdf',
|
|
35
|
+
)
|
|
36
|
+
),
|
|
37
|
+
types.Part(
|
|
38
|
+
fileData=types.FileData(
|
|
39
|
+
fileUri='gs://cloud-samples-data/generative-ai/pdf/2403.05530.pdf',
|
|
40
|
+
mimeType='application/pdf',
|
|
41
|
+
)
|
|
42
|
+
),
|
|
43
|
+
],
|
|
44
|
+
)
|
|
45
|
+
],
|
|
46
|
+
'system_instruction': t.t_content('What is the sum of the two pdfs?'),
|
|
47
|
+
'display_name': 'test cache',
|
|
48
|
+
'ttl': '86400s',
|
|
49
|
+
'http_options': constants.VERTEX_HTTP_OPTIONS,
|
|
50
|
+
},
|
|
51
|
+
)
|
|
52
|
+
|
|
53
|
+
_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE = types._CreateCachedContentParameters(
|
|
54
|
+
model='gemini-2.5-flash',
|
|
55
|
+
config={
|
|
56
|
+
'contents': [
|
|
57
|
+
types.Content(
|
|
58
|
+
role='user',
|
|
59
|
+
parts=[
|
|
60
|
+
types.Part(
|
|
61
|
+
fileData=types.FileData(
|
|
62
|
+
mimeType='application/pdf',
|
|
63
|
+
fileUri='https://generativelanguage.googleapis.com/v1beta/files/v200dhvn15h7',
|
|
64
|
+
)
|
|
65
|
+
)
|
|
66
|
+
],
|
|
67
|
+
)
|
|
68
|
+
],
|
|
69
|
+
'system_instruction': t.t_content('What is the sum of the two pdfs?'),
|
|
70
|
+
'display_name': 'test cache',
|
|
71
|
+
'ttl': '86400s',
|
|
72
|
+
'http_options': constants.MLDEV_HTTP_OPTIONS,
|
|
73
|
+
},
|
|
74
|
+
)
|
|
75
|
+
|
|
76
|
+
# Replay mode is not supported for caches tests due to the error message
|
|
77
|
+
# inconsistency in api and replay mode.
|
|
78
|
+
# To run api mode tests, use the following steps:
|
|
79
|
+
# 1. First create the resource.
|
|
80
|
+
# sh run_tests.sh pytest -s tests/caches/test_create.py --mode=api
|
|
81
|
+
# 1.1 If mldev test_create fails, update the uploaded file using this colab
|
|
82
|
+
# https://colab.sandbox.google.com/drive/1Fv6KGSs0cg6tlpcUHdsclHussXMEGOXk#scrollTo=RSKmFPx00MVL.
|
|
83
|
+
# 2. Find the resource name in debugging print and change the resource name constants.py.
|
|
84
|
+
# 3. Run and record get and update tests.
|
|
85
|
+
# sh run_tests.sh pytest -s tests/caches/test_get.py --mode=api && sh run_tests.sh pytest -s tests/caches/test_update.py --mode=api && sh run_tests.sh pytest -s tests/caches/test_delete.py --mode=api
|
|
86
|
+
test_table: list[pytest_helper.TestTableItem] = [
|
|
87
|
+
pytest_helper.TestTableItem(
|
|
88
|
+
name='test_caches_create_with_gcs_uri',
|
|
89
|
+
exception_if_mldev='404',
|
|
90
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GCS_URI,
|
|
91
|
+
),
|
|
92
|
+
pytest_helper.TestTableItem(
|
|
93
|
+
name='test_caches_create_with_googleai_file',
|
|
94
|
+
exception_if_vertex='404',
|
|
95
|
+
parameters=_CREATE_CACHED_CONTENT_PARAMETERS_GOOGLEAI_FILE,
|
|
96
|
+
skip_in_api_mode='Create is not reproducible in the API mode.',
|
|
97
|
+
),
|
|
98
|
+
]
|
|
99
|
+
pytestmark = pytest_helper.setup(
|
|
100
|
+
file=__file__,
|
|
101
|
+
globals_for_file=globals(),
|
|
102
|
+
test_method='caches.create',
|
|
103
|
+
test_table=test_table,
|
|
104
|
+
)
|
|
105
|
+
pytest_plugins = ('pytest_asyncio',)
|
|
@@ -0,0 +1,54 @@
|
|
|
1
|
+
# Copyright 2025 Google LLC
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
#
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
import pytest
|
|
18
|
+
from ... import types
|
|
19
|
+
from .. import pytest_helper
|
|
20
|
+
from . import constants
|
|
21
|
+
|
|
22
|
+
|
|
23
|
+
test_table: list[pytest_helper.TestTableItem] = [
|
|
24
|
+
pytest_helper.TestTableItem(
|
|
25
|
+
skip_in_api_mode='Delete is not reproducible in the API mode.',
|
|
26
|
+
name='test_caches_delete_with_vertex_cache_name',
|
|
27
|
+
exception_if_mldev='PERMISSION_DENIED',
|
|
28
|
+
parameters=types._DeleteCachedContentParameters(
|
|
29
|
+
name=constants.CACHED_CONTENT_NAME_VERTEX,
|
|
30
|
+
),
|
|
31
|
+
),
|
|
32
|
+
pytest_helper.TestTableItem(
|
|
33
|
+
skip_in_api_mode='Delete is not reproducible in the API mode.',
|
|
34
|
+
name='test_caches_delete_with_mldev_cache_name',
|
|
35
|
+
exception_if_vertex='NOT_FOUND',
|
|
36
|
+
parameters=types._DeleteCachedContentParameters(
|
|
37
|
+
name=constants.CACHED_CONTENT_NAME_MLDEV,
|
|
38
|
+
),
|
|
39
|
+
),
|
|
40
|
+
]
|
|
41
|
+
pytestmark = pytest_helper.setup(
|
|
42
|
+
file=__file__,
|
|
43
|
+
globals_for_file=globals(),
|
|
44
|
+
test_method='caches.delete',
|
|
45
|
+
test_table=test_table,
|
|
46
|
+
)
|
|
47
|
+
|
|
48
|
+
|
|
49
|
+
@pytest.mark.asyncio
|
|
50
|
+
async def test_async_delete(client):
|
|
51
|
+
if client._api_client.vertexai:
|
|
52
|
+
await client.aio.caches.delete(name=constants.CACHED_CONTENT_NAME_VERTEX)
|
|
53
|
+
else:
|
|
54
|
+
await client.aio.caches.delete(name=constants.CACHED_CONTENT_NAME_MLDEV)
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
# Copyright 2025 Google LLC
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
#
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
import pytest
|
|
18
|
+
from ... import types
|
|
19
|
+
from .. import pytest_helper
|
|
20
|
+
from . import constants
|
|
21
|
+
|
|
22
|
+
|
|
23
|
+
test_table: list[pytest_helper.TestTableItem] = [
|
|
24
|
+
pytest_helper.TestTableItem(
|
|
25
|
+
skip_in_api_mode='Delete is not reproducible in the API mode.',
|
|
26
|
+
name='test_caches_delete_with_vertex_cache_name',
|
|
27
|
+
exception_if_mldev='404',
|
|
28
|
+
parameters=types._DeleteCachedContentParameters(
|
|
29
|
+
name=constants.CACHED_CONTENT_NAME_VERTEX,
|
|
30
|
+
config={
|
|
31
|
+
'http_options': constants.VERTEX_HTTP_OPTIONS,
|
|
32
|
+
},
|
|
33
|
+
),
|
|
34
|
+
),
|
|
35
|
+
pytest_helper.TestTableItem(
|
|
36
|
+
skip_in_api_mode='Delete is not reproducible in the API mode.',
|
|
37
|
+
name='test_caches_delete_with_mldev_cache_name',
|
|
38
|
+
exception_if_vertex='404',
|
|
39
|
+
parameters=types._DeleteCachedContentParameters(
|
|
40
|
+
name=constants.CACHED_CONTENT_NAME_MLDEV,
|
|
41
|
+
config={
|
|
42
|
+
'http_options': constants.MLDEV_HTTP_OPTIONS,
|
|
43
|
+
},
|
|
44
|
+
),
|
|
45
|
+
),
|
|
46
|
+
]
|
|
47
|
+
pytestmark = pytest_helper.setup(
|
|
48
|
+
file=__file__,
|
|
49
|
+
globals_for_file=globals(),
|
|
50
|
+
test_method='caches.delete',
|
|
51
|
+
test_table=test_table,
|
|
52
|
+
)
|
|
@@ -0,0 +1,94 @@
|
|
|
1
|
+
# Copyright 2025 Google LLC
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
#
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
import pytest
|
|
18
|
+
from ... import types
|
|
19
|
+
from .. import pytest_helper
|
|
20
|
+
from . import constants
|
|
21
|
+
|
|
22
|
+
|
|
23
|
+
test_table: list[pytest_helper.TestTableItem] = [
|
|
24
|
+
pytest_helper.TestTableItem(
|
|
25
|
+
skip_in_api_mode='Get is not reproducible in the API mode.',
|
|
26
|
+
name='test_caches_get_with_vertex_cache_name',
|
|
27
|
+
exception_if_mldev='PERMISSION_DENIED',
|
|
28
|
+
parameters=types._GetCachedContentParameters(
|
|
29
|
+
name=constants.CACHED_CONTENT_NAME_VERTEX,
|
|
30
|
+
),
|
|
31
|
+
),
|
|
32
|
+
pytest_helper.TestTableItem(
|
|
33
|
+
skip_in_api_mode='Get is not reproducible in the API mode.',
|
|
34
|
+
name='test_caches_get_with_mldev_cache_name',
|
|
35
|
+
exception_if_vertex='NOT_FOUND',
|
|
36
|
+
parameters=types._GetCachedContentParameters(
|
|
37
|
+
name=constants.CACHED_CONTENT_NAME_MLDEV,
|
|
38
|
+
),
|
|
39
|
+
),
|
|
40
|
+
pytest_helper.TestTableItem(
|
|
41
|
+
skip_in_api_mode='Get is not reproducible in the API mode.',
|
|
42
|
+
name='test_caches_get_with_mldev_cache_name_partial_1',
|
|
43
|
+
exception_if_vertex='NOT_FOUND',
|
|
44
|
+
parameters=types._GetCachedContentParameters(
|
|
45
|
+
name='cachedContents/o239k1gxzz0juy9wqstndhncr85krehehf551hqh'
|
|
46
|
+
),
|
|
47
|
+
),
|
|
48
|
+
]
|
|
49
|
+
pytestmark = pytest_helper.setup(
|
|
50
|
+
file=__file__,
|
|
51
|
+
globals_for_file=globals(),
|
|
52
|
+
test_method='caches.get',
|
|
53
|
+
test_table=test_table,
|
|
54
|
+
)
|
|
55
|
+
|
|
56
|
+
|
|
57
|
+
@pytest.mark.asyncio
|
|
58
|
+
async def test_async_get(client):
|
|
59
|
+
if client._api_client.vertexai:
|
|
60
|
+
response = await client.aio.caches.get(
|
|
61
|
+
name=constants.CACHED_CONTENT_NAME_VERTEX
|
|
62
|
+
)
|
|
63
|
+
assert response
|
|
64
|
+
else:
|
|
65
|
+
await client.aio.caches.get(name=constants.CACHED_CONTENT_NAME_MLDEV)
|
|
66
|
+
|
|
67
|
+
|
|
68
|
+
def test_different_cache_name_formats(client):
|
|
69
|
+
if client._api_client.vertexai:
|
|
70
|
+
response1 = client.caches.get(
|
|
71
|
+
name='projects/964831358985/locations/us-central1/cachedContents/2164089915711684608'
|
|
72
|
+
)
|
|
73
|
+
assert response1
|
|
74
|
+
response2 = client.caches.get(
|
|
75
|
+
name='locations/us-central1/cachedContents/2164089915711684608'
|
|
76
|
+
)
|
|
77
|
+
assert response2
|
|
78
|
+
response3 = client.caches.get(
|
|
79
|
+
name='cachedContents/2164089915711684608'
|
|
80
|
+
)
|
|
81
|
+
assert response3
|
|
82
|
+
response4 = client.caches.get(
|
|
83
|
+
name='2164089915711684608'
|
|
84
|
+
)
|
|
85
|
+
assert response4
|
|
86
|
+
else:
|
|
87
|
+
response1 = client.caches.get(
|
|
88
|
+
name='cachedContents/o239k1gxzz0juy9wqstndhncr85krehehf551hqh'
|
|
89
|
+
)
|
|
90
|
+
assert response1
|
|
91
|
+
response2 = client.caches.get(
|
|
92
|
+
name='o239k1gxzz0juy9wqstndhncr85krehehf551hqh'
|
|
93
|
+
)
|
|
94
|
+
assert response2
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
# Copyright 2025 Google LLC
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
#
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
import pytest
|
|
18
|
+
from ... import types
|
|
19
|
+
from .. import pytest_helper
|
|
20
|
+
from . import constants
|
|
21
|
+
|
|
22
|
+
|
|
23
|
+
test_table: list[pytest_helper.TestTableItem] = [
|
|
24
|
+
pytest_helper.TestTableItem(
|
|
25
|
+
skip_in_api_mode='Get is not reproducible in the API mode.',
|
|
26
|
+
name='test_caches_get_with_vertex_cache_name',
|
|
27
|
+
exception_if_mldev='404',
|
|
28
|
+
parameters=types._GetCachedContentParameters(
|
|
29
|
+
name=constants.CACHED_CONTENT_NAME_VERTEX,
|
|
30
|
+
config={
|
|
31
|
+
'http_options': constants.VERTEX_HTTP_OPTIONS,
|
|
32
|
+
},
|
|
33
|
+
),
|
|
34
|
+
),
|
|
35
|
+
pytest_helper.TestTableItem(
|
|
36
|
+
skip_in_api_mode='Get is not reproducible in the API mode.',
|
|
37
|
+
name='test_caches_get_with_mldev_cache_name',
|
|
38
|
+
exception_if_vertex='404',
|
|
39
|
+
parameters=types._GetCachedContentParameters(
|
|
40
|
+
name=constants.CACHED_CONTENT_NAME_MLDEV,
|
|
41
|
+
config={
|
|
42
|
+
'http_options': constants.MLDEV_HTTP_OPTIONS,
|
|
43
|
+
},
|
|
44
|
+
),
|
|
45
|
+
),
|
|
46
|
+
]
|
|
47
|
+
pytestmark = pytest_helper.setup(
|
|
48
|
+
file=__file__,
|
|
49
|
+
globals_for_file=globals(),
|
|
50
|
+
test_method='caches.get',
|
|
51
|
+
test_table=test_table,
|
|
52
|
+
)
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
# Copyright 2025 Google LLC
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
#
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
"""Test caches list method."""
|
|
18
|
+
|
|
19
|
+
import pytest
|
|
20
|
+
from ... import types
|
|
21
|
+
from .. import pytest_helper
|
|
22
|
+
|
|
23
|
+
|
|
24
|
+
test_table: list[pytest_helper.TestTableItem] = [
|
|
25
|
+
pytest_helper.TestTableItem(
|
|
26
|
+
skip_in_api_mode='List is not reproducible in the API mode.',
|
|
27
|
+
name='test_caches_list',
|
|
28
|
+
parameters=types._ListCachedContentsParameters(config={'page_size': 3}),
|
|
29
|
+
),
|
|
30
|
+
]
|
|
31
|
+
pytestmark = [
|
|
32
|
+
pytest_helper.setup(
|
|
33
|
+
file=__file__,
|
|
34
|
+
globals_for_file=globals(),
|
|
35
|
+
test_method='caches.list',
|
|
36
|
+
test_table=test_table,
|
|
37
|
+
)
|
|
38
|
+
]
|
|
39
|
+
|
|
40
|
+
|
|
41
|
+
def test_pager(client):
|
|
42
|
+
cached_contents = client.caches.list(config={'page_size': 2})
|
|
43
|
+
assert 'content-type' in cached_contents.sdk_http_response.headers
|
|
44
|
+
assert cached_contents.name == 'cached_contents'
|
|
45
|
+
assert cached_contents.page_size == 2
|
|
46
|
+
assert len(cached_contents) <= 2
|
|
47
|
+
|
|
48
|
+
# Iterate through all the pages. Then next_page() should raise an exception.
|
|
49
|
+
for _ in cached_contents:
|
|
50
|
+
pass
|
|
51
|
+
with pytest.raises(IndexError, match='No more pages to fetch.'):
|
|
52
|
+
cached_contents.next_page()
|
|
53
|
+
|
|
54
|
+
|
|
55
|
+
@pytest.mark.asyncio
|
|
56
|
+
async def test_async_pager(client):
|
|
57
|
+
cached_contents = await client.aio.caches.list(config={'page_size': 2})
|
|
58
|
+
|
|
59
|
+
assert 'Content-Type' in cached_contents.sdk_http_response.headers
|
|
60
|
+
assert cached_contents.name == 'cached_contents'
|
|
61
|
+
assert cached_contents.page_size == 2
|
|
62
|
+
assert len(cached_contents) <= 2
|
|
63
|
+
|
|
64
|
+
# Iterate through all the pages. Then next_page() should raise an exception.
|
|
65
|
+
async for _ in cached_contents:
|
|
66
|
+
pass
|
|
67
|
+
with pytest.raises(IndexError, match='No more pages to fetch.'):
|
|
68
|
+
await cached_contents.next_page()
|