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.
Files changed (319) hide show
  1. google/genai/__init__.py +1 -0
  2. google/genai/_interactions/__init__.py +117 -0
  3. google/genai/_interactions/_base_client.py +2019 -0
  4. google/genai/_interactions/_client.py +511 -0
  5. google/genai/_interactions/_compat.py +234 -0
  6. google/genai/_interactions/_constants.py +29 -0
  7. google/genai/_interactions/_exceptions.py +122 -0
  8. google/genai/_interactions/_files.py +139 -0
  9. google/genai/_interactions/_models.py +873 -0
  10. google/genai/_interactions/_qs.py +165 -0
  11. google/genai/_interactions/_resource.py +58 -0
  12. google/genai/_interactions/_response.py +847 -0
  13. google/genai/_interactions/_streaming.py +354 -0
  14. google/genai/_interactions/_types.py +276 -0
  15. google/genai/_interactions/_utils/__init__.py +79 -0
  16. google/genai/_interactions/_utils/_compat.py +61 -0
  17. google/genai/_interactions/_utils/_datetime_parse.py +151 -0
  18. google/genai/_interactions/_utils/_logs.py +40 -0
  19. google/genai/_interactions/_utils/_proxy.py +80 -0
  20. google/genai/_interactions/_utils/_reflection.py +57 -0
  21. google/genai/_interactions/_utils/_resources_proxy.py +39 -0
  22. google/genai/_interactions/_utils/_streams.py +27 -0
  23. google/genai/_interactions/_utils/_sync.py +73 -0
  24. google/genai/_interactions/_utils/_transform.py +472 -0
  25. google/genai/_interactions/_utils/_typing.py +172 -0
  26. google/genai/_interactions/_utils/_utils.py +437 -0
  27. google/genai/_interactions/_version.py +18 -0
  28. google/genai/_interactions/resources/__init__.py +34 -0
  29. google/genai/_interactions/resources/interactions.py +1350 -0
  30. google/genai/_interactions/types/__init__.py +107 -0
  31. google/genai/_interactions/types/allowed_tools.py +33 -0
  32. google/genai/_interactions/types/allowed_tools_param.py +35 -0
  33. google/genai/_interactions/types/annotation.py +42 -0
  34. google/genai/_interactions/types/annotation_param.py +42 -0
  35. google/genai/_interactions/types/audio_content.py +38 -0
  36. google/genai/_interactions/types/audio_content_param.py +45 -0
  37. google/genai/_interactions/types/audio_mime_type.py +25 -0
  38. google/genai/_interactions/types/audio_mime_type_param.py +27 -0
  39. google/genai/_interactions/types/code_execution_call_arguments.py +33 -0
  40. google/genai/_interactions/types/code_execution_call_arguments_param.py +32 -0
  41. google/genai/_interactions/types/code_execution_call_content.py +37 -0
  42. google/genai/_interactions/types/code_execution_call_content_param.py +37 -0
  43. google/genai/_interactions/types/code_execution_result_content.py +42 -0
  44. google/genai/_interactions/types/code_execution_result_content_param.py +41 -0
  45. google/genai/_interactions/types/content_delta.py +358 -0
  46. google/genai/_interactions/types/content_start.py +79 -0
  47. google/genai/_interactions/types/content_stop.py +35 -0
  48. google/genai/_interactions/types/deep_research_agent_config.py +33 -0
  49. google/genai/_interactions/types/deep_research_agent_config_param.py +32 -0
  50. google/genai/_interactions/types/document_content.py +36 -0
  51. google/genai/_interactions/types/document_content_param.py +43 -0
  52. google/genai/_interactions/types/dynamic_agent_config.py +44 -0
  53. google/genai/_interactions/types/dynamic_agent_config_param.py +33 -0
  54. google/genai/_interactions/types/error_event.py +46 -0
  55. google/genai/_interactions/types/file_search_result_content.py +46 -0
  56. google/genai/_interactions/types/file_search_result_content_param.py +46 -0
  57. google/genai/_interactions/types/function.py +38 -0
  58. google/genai/_interactions/types/function_call_content.py +39 -0
  59. google/genai/_interactions/types/function_call_content_param.py +39 -0
  60. google/genai/_interactions/types/function_param.py +37 -0
  61. google/genai/_interactions/types/function_result_content.py +52 -0
  62. google/genai/_interactions/types/function_result_content_param.py +54 -0
  63. google/genai/_interactions/types/generation_config.py +57 -0
  64. google/genai/_interactions/types/generation_config_param.py +59 -0
  65. google/genai/_interactions/types/google_search_call_arguments.py +29 -0
  66. google/genai/_interactions/types/google_search_call_arguments_param.py +31 -0
  67. google/genai/_interactions/types/google_search_call_content.py +37 -0
  68. google/genai/_interactions/types/google_search_call_content_param.py +37 -0
  69. google/genai/_interactions/types/google_search_result.py +35 -0
  70. google/genai/_interactions/types/google_search_result_content.py +43 -0
  71. google/genai/_interactions/types/google_search_result_content_param.py +44 -0
  72. google/genai/_interactions/types/google_search_result_param.py +35 -0
  73. google/genai/_interactions/types/image_content.py +41 -0
  74. google/genai/_interactions/types/image_content_param.py +48 -0
  75. google/genai/_interactions/types/image_mime_type.py +23 -0
  76. google/genai/_interactions/types/image_mime_type_param.py +25 -0
  77. google/genai/_interactions/types/interaction.py +165 -0
  78. google/genai/_interactions/types/interaction_create_params.py +212 -0
  79. google/genai/_interactions/types/interaction_event.py +37 -0
  80. google/genai/_interactions/types/interaction_get_params.py +46 -0
  81. google/genai/_interactions/types/interaction_sse_event.py +32 -0
  82. google/genai/_interactions/types/interaction_status_update.py +37 -0
  83. google/genai/_interactions/types/mcp_server_tool_call_content.py +42 -0
  84. google/genai/_interactions/types/mcp_server_tool_call_content_param.py +42 -0
  85. google/genai/_interactions/types/mcp_server_tool_result_content.py +52 -0
  86. google/genai/_interactions/types/mcp_server_tool_result_content_param.py +54 -0
  87. google/genai/_interactions/types/model.py +36 -0
  88. google/genai/_interactions/types/model_param.py +38 -0
  89. google/genai/_interactions/types/speech_config.py +35 -0
  90. google/genai/_interactions/types/speech_config_param.py +35 -0
  91. google/genai/_interactions/types/text_content.py +37 -0
  92. google/genai/_interactions/types/text_content_param.py +38 -0
  93. google/genai/_interactions/types/thinking_level.py +22 -0
  94. google/genai/_interactions/types/thought_content.py +41 -0
  95. google/genai/_interactions/types/thought_content_param.py +47 -0
  96. google/genai/_interactions/types/tool.py +100 -0
  97. google/genai/_interactions/types/tool_choice.py +26 -0
  98. google/genai/_interactions/types/tool_choice_config.py +28 -0
  99. google/genai/_interactions/types/tool_choice_config_param.py +29 -0
  100. google/genai/_interactions/types/tool_choice_param.py +28 -0
  101. google/genai/_interactions/types/tool_choice_type.py +22 -0
  102. google/genai/_interactions/types/tool_param.py +97 -0
  103. google/genai/_interactions/types/turn.py +76 -0
  104. google/genai/_interactions/types/turn_param.py +73 -0
  105. google/genai/_interactions/types/url_context_call_arguments.py +29 -0
  106. google/genai/_interactions/types/url_context_call_arguments_param.py +31 -0
  107. google/genai/_interactions/types/url_context_call_content.py +37 -0
  108. google/genai/_interactions/types/url_context_call_content_param.py +37 -0
  109. google/genai/_interactions/types/url_context_result.py +33 -0
  110. google/genai/_interactions/types/url_context_result_content.py +43 -0
  111. google/genai/_interactions/types/url_context_result_content_param.py +44 -0
  112. google/genai/_interactions/types/url_context_result_param.py +32 -0
  113. google/genai/_interactions/types/usage.py +106 -0
  114. google/genai/_interactions/types/usage_param.py +106 -0
  115. google/genai/_interactions/types/video_content.py +41 -0
  116. google/genai/_interactions/types/video_content_param.py +48 -0
  117. google/genai/_interactions/types/video_mime_type.py +36 -0
  118. google/genai/_interactions/types/video_mime_type_param.py +38 -0
  119. google/genai/_live_converters.py +31 -0
  120. google/genai/_tokens_converters.py +5 -0
  121. google/genai/batches.py +7 -0
  122. google/genai/client.py +223 -0
  123. google/genai/interactions.py +17 -0
  124. google/genai/live.py +4 -3
  125. google/genai/models.py +12 -0
  126. google/genai/tests/__init__.py +21 -0
  127. google/genai/tests/afc/__init__.py +21 -0
  128. google/genai/tests/afc/test_convert_if_exist_pydantic_model.py +309 -0
  129. google/genai/tests/afc/test_convert_number_values_for_function_call_args.py +63 -0
  130. google/genai/tests/afc/test_find_afc_incompatible_tool_indexes.py +240 -0
  131. google/genai/tests/afc/test_generate_content_stream_afc.py +530 -0
  132. google/genai/tests/afc/test_generate_content_stream_afc_thoughts.py +77 -0
  133. google/genai/tests/afc/test_get_function_map.py +176 -0
  134. google/genai/tests/afc/test_get_function_response_parts.py +277 -0
  135. google/genai/tests/afc/test_get_max_remote_calls_for_afc.py +130 -0
  136. google/genai/tests/afc/test_invoke_function_from_dict_args.py +241 -0
  137. google/genai/tests/afc/test_raise_error_for_afc_incompatible_config.py +159 -0
  138. google/genai/tests/afc/test_should_append_afc_history.py +53 -0
  139. google/genai/tests/afc/test_should_disable_afc.py +214 -0
  140. google/genai/tests/batches/__init__.py +17 -0
  141. google/genai/tests/batches/test_cancel.py +77 -0
  142. google/genai/tests/batches/test_create.py +78 -0
  143. google/genai/tests/batches/test_create_with_bigquery.py +113 -0
  144. google/genai/tests/batches/test_create_with_file.py +82 -0
  145. google/genai/tests/batches/test_create_with_gcs.py +125 -0
  146. google/genai/tests/batches/test_create_with_inlined_requests.py +255 -0
  147. google/genai/tests/batches/test_delete.py +86 -0
  148. google/genai/tests/batches/test_embedding.py +157 -0
  149. google/genai/tests/batches/test_get.py +78 -0
  150. google/genai/tests/batches/test_list.py +79 -0
  151. google/genai/tests/caches/__init__.py +17 -0
  152. google/genai/tests/caches/constants.py +29 -0
  153. google/genai/tests/caches/test_create.py +210 -0
  154. google/genai/tests/caches/test_create_custom_url.py +105 -0
  155. google/genai/tests/caches/test_delete.py +54 -0
  156. google/genai/tests/caches/test_delete_custom_url.py +52 -0
  157. google/genai/tests/caches/test_get.py +94 -0
  158. google/genai/tests/caches/test_get_custom_url.py +52 -0
  159. google/genai/tests/caches/test_list.py +68 -0
  160. google/genai/tests/caches/test_update.py +70 -0
  161. google/genai/tests/caches/test_update_custom_url.py +58 -0
  162. google/genai/tests/chats/__init__.py +1 -0
  163. google/genai/tests/chats/test_get_history.py +597 -0
  164. google/genai/tests/chats/test_send_message.py +844 -0
  165. google/genai/tests/chats/test_validate_response.py +90 -0
  166. google/genai/tests/client/__init__.py +17 -0
  167. google/genai/tests/client/test_async_stream.py +427 -0
  168. google/genai/tests/client/test_client_close.py +197 -0
  169. google/genai/tests/client/test_client_initialization.py +1687 -0
  170. google/genai/tests/client/test_client_requests.py +355 -0
  171. google/genai/tests/client/test_custom_client.py +77 -0
  172. google/genai/tests/client/test_http_options.py +178 -0
  173. google/genai/tests/client/test_replay_client_equality.py +168 -0
  174. google/genai/tests/client/test_retries.py +846 -0
  175. google/genai/tests/client/test_upload_errors.py +136 -0
  176. google/genai/tests/common/__init__.py +17 -0
  177. google/genai/tests/common/test_common.py +954 -0
  178. google/genai/tests/conftest.py +162 -0
  179. google/genai/tests/documents/__init__.py +17 -0
  180. google/genai/tests/documents/test_delete.py +51 -0
  181. google/genai/tests/documents/test_get.py +85 -0
  182. google/genai/tests/documents/test_list.py +72 -0
  183. google/genai/tests/errors/__init__.py +1 -0
  184. google/genai/tests/errors/test_api_error.py +417 -0
  185. google/genai/tests/file_search_stores/__init__.py +17 -0
  186. google/genai/tests/file_search_stores/test_create.py +66 -0
  187. google/genai/tests/file_search_stores/test_delete.py +64 -0
  188. google/genai/tests/file_search_stores/test_get.py +94 -0
  189. google/genai/tests/file_search_stores/test_import_file.py +112 -0
  190. google/genai/tests/file_search_stores/test_list.py +57 -0
  191. google/genai/tests/file_search_stores/test_upload_to_file_search_store.py +141 -0
  192. google/genai/tests/files/__init__.py +17 -0
  193. google/genai/tests/files/test_delete.py +46 -0
  194. google/genai/tests/files/test_download.py +85 -0
  195. google/genai/tests/files/test_get.py +46 -0
  196. google/genai/tests/files/test_list.py +72 -0
  197. google/genai/tests/files/test_upload.py +255 -0
  198. google/genai/tests/imports/test_no_optional_imports.py +28 -0
  199. google/genai/tests/interactions/__init__.py +0 -0
  200. google/genai/tests/interactions/test_integration.py +80 -0
  201. google/genai/tests/live/__init__.py +16 -0
  202. google/genai/tests/live/test_live.py +2177 -0
  203. google/genai/tests/live/test_live_music.py +362 -0
  204. google/genai/tests/live/test_live_response.py +163 -0
  205. google/genai/tests/live/test_send_client_content.py +147 -0
  206. google/genai/tests/live/test_send_realtime_input.py +268 -0
  207. google/genai/tests/live/test_send_tool_response.py +222 -0
  208. google/genai/tests/local_tokenizer/__init__.py +17 -0
  209. google/genai/tests/local_tokenizer/test_local_tokenizer.py +343 -0
  210. google/genai/tests/local_tokenizer/test_local_tokenizer_loader.py +235 -0
  211. google/genai/tests/mcp/__init__.py +17 -0
  212. google/genai/tests/mcp/test_has_mcp_tool_usage.py +89 -0
  213. google/genai/tests/mcp/test_mcp_to_gemini_tools.py +191 -0
  214. google/genai/tests/mcp/test_parse_config_for_mcp_sessions.py +201 -0
  215. google/genai/tests/mcp/test_parse_config_for_mcp_usage.py +130 -0
  216. google/genai/tests/mcp/test_set_mcp_usage_header.py +72 -0
  217. google/genai/tests/models/__init__.py +17 -0
  218. google/genai/tests/models/constants.py +8 -0
  219. google/genai/tests/models/test_compute_tokens.py +120 -0
  220. google/genai/tests/models/test_count_tokens.py +159 -0
  221. google/genai/tests/models/test_delete.py +107 -0
  222. google/genai/tests/models/test_edit_image.py +264 -0
  223. google/genai/tests/models/test_embed_content.py +94 -0
  224. google/genai/tests/models/test_function_call_streaming.py +442 -0
  225. google/genai/tests/models/test_generate_content.py +2502 -0
  226. google/genai/tests/models/test_generate_content_cached_content.py +132 -0
  227. google/genai/tests/models/test_generate_content_config_zero_value.py +103 -0
  228. google/genai/tests/models/test_generate_content_from_apikey.py +44 -0
  229. google/genai/tests/models/test_generate_content_http_options.py +40 -0
  230. google/genai/tests/models/test_generate_content_image_generation.py +143 -0
  231. google/genai/tests/models/test_generate_content_mcp.py +343 -0
  232. google/genai/tests/models/test_generate_content_media_resolution.py +97 -0
  233. google/genai/tests/models/test_generate_content_model.py +139 -0
  234. google/genai/tests/models/test_generate_content_part.py +821 -0
  235. google/genai/tests/models/test_generate_content_thought.py +76 -0
  236. google/genai/tests/models/test_generate_content_tools.py +1761 -0
  237. google/genai/tests/models/test_generate_images.py +191 -0
  238. google/genai/tests/models/test_generate_videos.py +759 -0
  239. google/genai/tests/models/test_get.py +104 -0
  240. google/genai/tests/models/test_list.py +233 -0
  241. google/genai/tests/models/test_recontext_image.py +189 -0
  242. google/genai/tests/models/test_segment_image.py +148 -0
  243. google/genai/tests/models/test_update.py +95 -0
  244. google/genai/tests/models/test_upscale_image.py +157 -0
  245. google/genai/tests/operations/__init__.py +17 -0
  246. google/genai/tests/operations/test_get.py +38 -0
  247. google/genai/tests/public_samples/__init__.py +17 -0
  248. google/genai/tests/public_samples/test_gemini_text_only.py +34 -0
  249. google/genai/tests/pytest_helper.py +229 -0
  250. google/genai/tests/shared/__init__.py +16 -0
  251. google/genai/tests/shared/batches/__init__.py +14 -0
  252. google/genai/tests/shared/batches/test_create_delete.py +57 -0
  253. google/genai/tests/shared/batches/test_create_get_cancel.py +56 -0
  254. google/genai/tests/shared/batches/test_list.py +40 -0
  255. google/genai/tests/shared/caches/__init__.py +14 -0
  256. google/genai/tests/shared/caches/test_create_get_delete.py +67 -0
  257. google/genai/tests/shared/caches/test_create_update_get.py +71 -0
  258. google/genai/tests/shared/caches/test_list.py +40 -0
  259. google/genai/tests/shared/chats/__init__.py +14 -0
  260. google/genai/tests/shared/chats/test_send_message.py +48 -0
  261. google/genai/tests/shared/chats/test_send_message_stream.py +50 -0
  262. google/genai/tests/shared/files/__init__.py +14 -0
  263. google/genai/tests/shared/files/test_list.py +41 -0
  264. google/genai/tests/shared/files/test_upload_get_delete.py +54 -0
  265. google/genai/tests/shared/models/__init__.py +14 -0
  266. google/genai/tests/shared/models/test_compute_tokens.py +41 -0
  267. google/genai/tests/shared/models/test_count_tokens.py +40 -0
  268. google/genai/tests/shared/models/test_edit_image.py +67 -0
  269. google/genai/tests/shared/models/test_embed.py +40 -0
  270. google/genai/tests/shared/models/test_generate_content.py +39 -0
  271. google/genai/tests/shared/models/test_generate_content_stream.py +54 -0
  272. google/genai/tests/shared/models/test_generate_images.py +40 -0
  273. google/genai/tests/shared/models/test_generate_videos.py +38 -0
  274. google/genai/tests/shared/models/test_list.py +37 -0
  275. google/genai/tests/shared/models/test_recontext_image.py +55 -0
  276. google/genai/tests/shared/models/test_segment_image.py +52 -0
  277. google/genai/tests/shared/models/test_upscale_image.py +52 -0
  278. google/genai/tests/shared/tunings/__init__.py +16 -0
  279. google/genai/tests/shared/tunings/test_create.py +46 -0
  280. google/genai/tests/shared/tunings/test_create_get_cancel.py +56 -0
  281. google/genai/tests/shared/tunings/test_list.py +39 -0
  282. google/genai/tests/tokens/__init__.py +16 -0
  283. google/genai/tests/tokens/test_create.py +154 -0
  284. google/genai/tests/transformers/__init__.py +17 -0
  285. google/genai/tests/transformers/test_blobs.py +71 -0
  286. google/genai/tests/transformers/test_bytes.py +15 -0
  287. google/genai/tests/transformers/test_duck_type.py +96 -0
  288. google/genai/tests/transformers/test_function_responses.py +72 -0
  289. google/genai/tests/transformers/test_schema.py +653 -0
  290. google/genai/tests/transformers/test_t_batch.py +286 -0
  291. google/genai/tests/transformers/test_t_content.py +160 -0
  292. google/genai/tests/transformers/test_t_contents.py +398 -0
  293. google/genai/tests/transformers/test_t_part.py +85 -0
  294. google/genai/tests/transformers/test_t_parts.py +87 -0
  295. google/genai/tests/transformers/test_t_tool.py +157 -0
  296. google/genai/tests/transformers/test_t_tools.py +195 -0
  297. google/genai/tests/tunings/__init__.py +16 -0
  298. google/genai/tests/tunings/test_cancel.py +39 -0
  299. google/genai/tests/tunings/test_end_to_end.py +106 -0
  300. google/genai/tests/tunings/test_get.py +67 -0
  301. google/genai/tests/tunings/test_list.py +75 -0
  302. google/genai/tests/tunings/test_tune.py +268 -0
  303. google/genai/tests/types/__init__.py +16 -0
  304. google/genai/tests/types/test_bytes_internal.py +271 -0
  305. google/genai/tests/types/test_bytes_type.py +152 -0
  306. google/genai/tests/types/test_future.py +101 -0
  307. google/genai/tests/types/test_optional_types.py +36 -0
  308. google/genai/tests/types/test_part_type.py +616 -0
  309. google/genai/tests/types/test_schema_from_json_schema.py +417 -0
  310. google/genai/tests/types/test_schema_json_schema.py +468 -0
  311. google/genai/tests/types/test_types.py +2903 -0
  312. google/genai/types.py +72 -0
  313. google/genai/version.py +1 -1
  314. {google_genai-1.54.0.dist-info → google_genai-1.55.0.dist-info}/METADATA +3 -1
  315. google_genai-1.55.0.dist-info/RECORD +345 -0
  316. google_genai-1.54.0.dist-info/RECORD +0 -41
  317. {google_genai-1.54.0.dist-info → google_genai-1.55.0.dist-info}/WHEEL +0 -0
  318. {google_genai-1.54.0.dist-info → google_genai-1.55.0.dist-info}/licenses/LICENSE +0 -0
  319. {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()