google-genai 1.55.0__py3-none-any.whl → 1.56.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.
Files changed (241) hide show
  1. google/genai/_interactions/_base_client.py +8 -2
  2. google/genai/_interactions/resources/interactions.py +6 -6
  3. google/genai/_interactions/types/__init__.py +2 -0
  4. google/genai/_interactions/types/audio_content.py +0 -1
  5. google/genai/_interactions/types/audio_content_param.py +0 -1
  6. google/genai/_interactions/types/code_execution_call_content.py +0 -1
  7. google/genai/_interactions/types/code_execution_call_content_param.py +0 -1
  8. google/genai/_interactions/types/code_execution_result_content.py +0 -1
  9. google/genai/_interactions/types/code_execution_result_content_param.py +0 -1
  10. google/genai/_interactions/types/content_delta.py +7 -23
  11. google/genai/_interactions/types/deep_research_agent_config.py +0 -1
  12. google/genai/_interactions/types/deep_research_agent_config_param.py +0 -1
  13. google/genai/_interactions/types/document_content.py +3 -2
  14. google/genai/_interactions/types/document_content_param.py +3 -2
  15. google/genai/{tests/__init__.py → _interactions/types/document_mime_type.py} +5 -3
  16. google/genai/{tests/afc/__init__.py → _interactions/types/document_mime_type_param.py} +8 -4
  17. google/genai/_interactions/types/dynamic_agent_config.py +0 -1
  18. google/genai/_interactions/types/dynamic_agent_config_param.py +0 -1
  19. google/genai/_interactions/types/file_search_result_content.py +0 -1
  20. google/genai/_interactions/types/file_search_result_content_param.py +0 -1
  21. google/genai/_interactions/types/function_call_content.py +0 -1
  22. google/genai/_interactions/types/function_call_content_param.py +0 -1
  23. google/genai/_interactions/types/function_result_content.py +1 -2
  24. google/genai/_interactions/types/function_result_content_param.py +1 -2
  25. google/genai/_interactions/types/google_search_call_content.py +0 -1
  26. google/genai/_interactions/types/google_search_call_content_param.py +0 -1
  27. google/genai/_interactions/types/google_search_result_content.py +0 -1
  28. google/genai/_interactions/types/google_search_result_content_param.py +0 -1
  29. google/genai/_interactions/types/image_content.py +1 -2
  30. google/genai/_interactions/types/image_content_param.py +1 -2
  31. google/genai/_interactions/types/mcp_server_tool_call_content.py +0 -1
  32. google/genai/_interactions/types/mcp_server_tool_call_content_param.py +0 -1
  33. google/genai/_interactions/types/mcp_server_tool_result_content.py +1 -2
  34. google/genai/_interactions/types/mcp_server_tool_result_content_param.py +1 -2
  35. google/genai/_interactions/types/text_content.py +0 -1
  36. google/genai/_interactions/types/text_content_param.py +0 -1
  37. google/genai/_interactions/types/thinking_level.py +1 -1
  38. google/genai/_interactions/types/thought_content.py +0 -1
  39. google/genai/_interactions/types/thought_content_param.py +0 -1
  40. google/genai/_interactions/types/url_context_call_content.py +0 -1
  41. google/genai/_interactions/types/url_context_call_content_param.py +0 -1
  42. google/genai/_interactions/types/url_context_result_content.py +0 -1
  43. google/genai/_interactions/types/url_context_result_content_param.py +0 -1
  44. google/genai/_interactions/types/video_content.py +1 -2
  45. google/genai/_interactions/types/video_content_param.py +1 -2
  46. google/genai/_live_converters.py +2 -30
  47. google/genai/client.py +3 -1
  48. google/genai/models.py +2 -29
  49. google/genai/tunings.py +1 -27
  50. google/genai/types.py +20 -22
  51. google/genai/version.py +1 -1
  52. {google_genai-1.55.0.dist-info → google_genai-1.56.0.dist-info}/METADATA +224 -22
  53. google_genai-1.56.0.dist-info/RECORD +162 -0
  54. google/genai/tests/afc/test_convert_if_exist_pydantic_model.py +0 -309
  55. google/genai/tests/afc/test_convert_number_values_for_function_call_args.py +0 -63
  56. google/genai/tests/afc/test_find_afc_incompatible_tool_indexes.py +0 -240
  57. google/genai/tests/afc/test_generate_content_stream_afc.py +0 -530
  58. google/genai/tests/afc/test_generate_content_stream_afc_thoughts.py +0 -77
  59. google/genai/tests/afc/test_get_function_map.py +0 -176
  60. google/genai/tests/afc/test_get_function_response_parts.py +0 -277
  61. google/genai/tests/afc/test_get_max_remote_calls_for_afc.py +0 -130
  62. google/genai/tests/afc/test_invoke_function_from_dict_args.py +0 -241
  63. google/genai/tests/afc/test_raise_error_for_afc_incompatible_config.py +0 -159
  64. google/genai/tests/afc/test_should_append_afc_history.py +0 -53
  65. google/genai/tests/afc/test_should_disable_afc.py +0 -214
  66. google/genai/tests/batches/__init__.py +0 -17
  67. google/genai/tests/batches/test_cancel.py +0 -77
  68. google/genai/tests/batches/test_create.py +0 -78
  69. google/genai/tests/batches/test_create_with_bigquery.py +0 -113
  70. google/genai/tests/batches/test_create_with_file.py +0 -82
  71. google/genai/tests/batches/test_create_with_gcs.py +0 -125
  72. google/genai/tests/batches/test_create_with_inlined_requests.py +0 -255
  73. google/genai/tests/batches/test_delete.py +0 -86
  74. google/genai/tests/batches/test_embedding.py +0 -157
  75. google/genai/tests/batches/test_get.py +0 -78
  76. google/genai/tests/batches/test_list.py +0 -79
  77. google/genai/tests/caches/__init__.py +0 -17
  78. google/genai/tests/caches/constants.py +0 -29
  79. google/genai/tests/caches/test_create.py +0 -210
  80. google/genai/tests/caches/test_create_custom_url.py +0 -105
  81. google/genai/tests/caches/test_delete.py +0 -54
  82. google/genai/tests/caches/test_delete_custom_url.py +0 -52
  83. google/genai/tests/caches/test_get.py +0 -94
  84. google/genai/tests/caches/test_get_custom_url.py +0 -52
  85. google/genai/tests/caches/test_list.py +0 -68
  86. google/genai/tests/caches/test_update.py +0 -70
  87. google/genai/tests/caches/test_update_custom_url.py +0 -58
  88. google/genai/tests/chats/__init__.py +0 -1
  89. google/genai/tests/chats/test_get_history.py +0 -597
  90. google/genai/tests/chats/test_send_message.py +0 -844
  91. google/genai/tests/chats/test_validate_response.py +0 -90
  92. google/genai/tests/client/__init__.py +0 -17
  93. google/genai/tests/client/test_async_stream.py +0 -427
  94. google/genai/tests/client/test_client_close.py +0 -197
  95. google/genai/tests/client/test_client_initialization.py +0 -1687
  96. google/genai/tests/client/test_client_requests.py +0 -355
  97. google/genai/tests/client/test_custom_client.py +0 -77
  98. google/genai/tests/client/test_http_options.py +0 -178
  99. google/genai/tests/client/test_replay_client_equality.py +0 -168
  100. google/genai/tests/client/test_retries.py +0 -846
  101. google/genai/tests/client/test_upload_errors.py +0 -136
  102. google/genai/tests/common/__init__.py +0 -17
  103. google/genai/tests/common/test_common.py +0 -954
  104. google/genai/tests/conftest.py +0 -162
  105. google/genai/tests/documents/__init__.py +0 -17
  106. google/genai/tests/documents/test_delete.py +0 -51
  107. google/genai/tests/documents/test_get.py +0 -85
  108. google/genai/tests/documents/test_list.py +0 -72
  109. google/genai/tests/errors/__init__.py +0 -1
  110. google/genai/tests/errors/test_api_error.py +0 -417
  111. google/genai/tests/file_search_stores/__init__.py +0 -17
  112. google/genai/tests/file_search_stores/test_create.py +0 -66
  113. google/genai/tests/file_search_stores/test_delete.py +0 -64
  114. google/genai/tests/file_search_stores/test_get.py +0 -94
  115. google/genai/tests/file_search_stores/test_import_file.py +0 -112
  116. google/genai/tests/file_search_stores/test_list.py +0 -57
  117. google/genai/tests/file_search_stores/test_upload_to_file_search_store.py +0 -141
  118. google/genai/tests/files/__init__.py +0 -17
  119. google/genai/tests/files/test_delete.py +0 -46
  120. google/genai/tests/files/test_download.py +0 -85
  121. google/genai/tests/files/test_get.py +0 -46
  122. google/genai/tests/files/test_list.py +0 -72
  123. google/genai/tests/files/test_upload.py +0 -255
  124. google/genai/tests/imports/test_no_optional_imports.py +0 -28
  125. google/genai/tests/interactions/test_integration.py +0 -80
  126. google/genai/tests/live/__init__.py +0 -16
  127. google/genai/tests/live/test_live.py +0 -2177
  128. google/genai/tests/live/test_live_music.py +0 -362
  129. google/genai/tests/live/test_live_response.py +0 -163
  130. google/genai/tests/live/test_send_client_content.py +0 -147
  131. google/genai/tests/live/test_send_realtime_input.py +0 -268
  132. google/genai/tests/live/test_send_tool_response.py +0 -222
  133. google/genai/tests/local_tokenizer/__init__.py +0 -17
  134. google/genai/tests/local_tokenizer/test_local_tokenizer.py +0 -343
  135. google/genai/tests/local_tokenizer/test_local_tokenizer_loader.py +0 -235
  136. google/genai/tests/mcp/__init__.py +0 -17
  137. google/genai/tests/mcp/test_has_mcp_tool_usage.py +0 -89
  138. google/genai/tests/mcp/test_mcp_to_gemini_tools.py +0 -191
  139. google/genai/tests/mcp/test_parse_config_for_mcp_sessions.py +0 -201
  140. google/genai/tests/mcp/test_parse_config_for_mcp_usage.py +0 -130
  141. google/genai/tests/mcp/test_set_mcp_usage_header.py +0 -72
  142. google/genai/tests/models/__init__.py +0 -17
  143. google/genai/tests/models/constants.py +0 -8
  144. google/genai/tests/models/test_compute_tokens.py +0 -120
  145. google/genai/tests/models/test_count_tokens.py +0 -159
  146. google/genai/tests/models/test_delete.py +0 -107
  147. google/genai/tests/models/test_edit_image.py +0 -264
  148. google/genai/tests/models/test_embed_content.py +0 -94
  149. google/genai/tests/models/test_function_call_streaming.py +0 -442
  150. google/genai/tests/models/test_generate_content.py +0 -2502
  151. google/genai/tests/models/test_generate_content_cached_content.py +0 -132
  152. google/genai/tests/models/test_generate_content_config_zero_value.py +0 -103
  153. google/genai/tests/models/test_generate_content_from_apikey.py +0 -44
  154. google/genai/tests/models/test_generate_content_http_options.py +0 -40
  155. google/genai/tests/models/test_generate_content_image_generation.py +0 -143
  156. google/genai/tests/models/test_generate_content_mcp.py +0 -343
  157. google/genai/tests/models/test_generate_content_media_resolution.py +0 -97
  158. google/genai/tests/models/test_generate_content_model.py +0 -139
  159. google/genai/tests/models/test_generate_content_part.py +0 -821
  160. google/genai/tests/models/test_generate_content_thought.py +0 -76
  161. google/genai/tests/models/test_generate_content_tools.py +0 -1761
  162. google/genai/tests/models/test_generate_images.py +0 -191
  163. google/genai/tests/models/test_generate_videos.py +0 -759
  164. google/genai/tests/models/test_get.py +0 -104
  165. google/genai/tests/models/test_list.py +0 -233
  166. google/genai/tests/models/test_recontext_image.py +0 -189
  167. google/genai/tests/models/test_segment_image.py +0 -148
  168. google/genai/tests/models/test_update.py +0 -95
  169. google/genai/tests/models/test_upscale_image.py +0 -157
  170. google/genai/tests/operations/__init__.py +0 -17
  171. google/genai/tests/operations/test_get.py +0 -38
  172. google/genai/tests/public_samples/__init__.py +0 -17
  173. google/genai/tests/public_samples/test_gemini_text_only.py +0 -34
  174. google/genai/tests/pytest_helper.py +0 -229
  175. google/genai/tests/shared/__init__.py +0 -16
  176. google/genai/tests/shared/batches/__init__.py +0 -14
  177. google/genai/tests/shared/batches/test_create_delete.py +0 -57
  178. google/genai/tests/shared/batches/test_create_get_cancel.py +0 -56
  179. google/genai/tests/shared/batches/test_list.py +0 -40
  180. google/genai/tests/shared/caches/__init__.py +0 -14
  181. google/genai/tests/shared/caches/test_create_get_delete.py +0 -67
  182. google/genai/tests/shared/caches/test_create_update_get.py +0 -71
  183. google/genai/tests/shared/caches/test_list.py +0 -40
  184. google/genai/tests/shared/chats/__init__.py +0 -14
  185. google/genai/tests/shared/chats/test_send_message.py +0 -48
  186. google/genai/tests/shared/chats/test_send_message_stream.py +0 -50
  187. google/genai/tests/shared/files/__init__.py +0 -14
  188. google/genai/tests/shared/files/test_list.py +0 -41
  189. google/genai/tests/shared/files/test_upload_get_delete.py +0 -54
  190. google/genai/tests/shared/models/__init__.py +0 -14
  191. google/genai/tests/shared/models/test_compute_tokens.py +0 -41
  192. google/genai/tests/shared/models/test_count_tokens.py +0 -40
  193. google/genai/tests/shared/models/test_edit_image.py +0 -67
  194. google/genai/tests/shared/models/test_embed.py +0 -40
  195. google/genai/tests/shared/models/test_generate_content.py +0 -39
  196. google/genai/tests/shared/models/test_generate_content_stream.py +0 -54
  197. google/genai/tests/shared/models/test_generate_images.py +0 -40
  198. google/genai/tests/shared/models/test_generate_videos.py +0 -38
  199. google/genai/tests/shared/models/test_list.py +0 -37
  200. google/genai/tests/shared/models/test_recontext_image.py +0 -55
  201. google/genai/tests/shared/models/test_segment_image.py +0 -52
  202. google/genai/tests/shared/models/test_upscale_image.py +0 -52
  203. google/genai/tests/shared/tunings/__init__.py +0 -16
  204. google/genai/tests/shared/tunings/test_create.py +0 -46
  205. google/genai/tests/shared/tunings/test_create_get_cancel.py +0 -56
  206. google/genai/tests/shared/tunings/test_list.py +0 -39
  207. google/genai/tests/tokens/__init__.py +0 -16
  208. google/genai/tests/tokens/test_create.py +0 -154
  209. google/genai/tests/transformers/__init__.py +0 -17
  210. google/genai/tests/transformers/test_blobs.py +0 -71
  211. google/genai/tests/transformers/test_bytes.py +0 -15
  212. google/genai/tests/transformers/test_duck_type.py +0 -96
  213. google/genai/tests/transformers/test_function_responses.py +0 -72
  214. google/genai/tests/transformers/test_schema.py +0 -653
  215. google/genai/tests/transformers/test_t_batch.py +0 -286
  216. google/genai/tests/transformers/test_t_content.py +0 -160
  217. google/genai/tests/transformers/test_t_contents.py +0 -398
  218. google/genai/tests/transformers/test_t_part.py +0 -85
  219. google/genai/tests/transformers/test_t_parts.py +0 -87
  220. google/genai/tests/transformers/test_t_tool.py +0 -157
  221. google/genai/tests/transformers/test_t_tools.py +0 -195
  222. google/genai/tests/tunings/__init__.py +0 -16
  223. google/genai/tests/tunings/test_cancel.py +0 -39
  224. google/genai/tests/tunings/test_end_to_end.py +0 -106
  225. google/genai/tests/tunings/test_get.py +0 -67
  226. google/genai/tests/tunings/test_list.py +0 -75
  227. google/genai/tests/tunings/test_tune.py +0 -268
  228. google/genai/tests/types/__init__.py +0 -16
  229. google/genai/tests/types/test_bytes_internal.py +0 -271
  230. google/genai/tests/types/test_bytes_type.py +0 -152
  231. google/genai/tests/types/test_future.py +0 -101
  232. google/genai/tests/types/test_optional_types.py +0 -36
  233. google/genai/tests/types/test_part_type.py +0 -616
  234. google/genai/tests/types/test_schema_from_json_schema.py +0 -417
  235. google/genai/tests/types/test_schema_json_schema.py +0 -468
  236. google/genai/tests/types/test_types.py +0 -2903
  237. google_genai-1.55.0.dist-info/RECORD +0 -345
  238. /google/genai/{tests/interactions/__init__.py → _interactions/py.typed} +0 -0
  239. {google_genai-1.55.0.dist-info → google_genai-1.56.0.dist-info}/WHEEL +0 -0
  240. {google_genai-1.55.0.dist-info → google_genai-1.56.0.dist-info}/licenses/LICENSE +0 -0
  241. {google_genai-1.55.0.dist-info → google_genai-1.56.0.dist-info}/top_level.txt +0 -0
@@ -1,210 +0,0 @@
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
- )
@@ -1,105 +0,0 @@
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',)
@@ -1,54 +0,0 @@
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)
@@ -1,52 +0,0 @@
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
- )
@@ -1,94 +0,0 @@
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
@@ -1,52 +0,0 @@
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
- )
@@ -1,68 +0,0 @@
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()