diff --git a/google/cloud/aiplatform_v1/types/tool.py b/google/cloud/aiplatform_v1/types/tool.py index 66fe057181..2593cfe16d 100644 --- a/google/cloud/aiplatform_v1/types/tool.py +++ b/google/cloud/aiplatform_v1/types/tool.py @@ -102,6 +102,11 @@ class Tool(proto.Message): user queries. The search results will be grounded on Parallel.ai and presented to the model for response generation + exa_ai_search (google.cloud.aiplatform_v1.types.Tool.ExaAiSearch): + Optional. Uses Exa.ai to search for + information to answer user queries. The search + results will be grounded on Exa.ai and presented + to the model for response generation code_execution (google.cloud.aiplatform_v1.types.Tool.CodeExecution): Optional. CodeExecution tool type. Enables the model to execute code as part of @@ -222,6 +227,28 @@ class ParallelAiSearch(proto.Message): message=struct_pb2.Struct, ) + class ExaAiSearch(proto.Message): + r"""ExaAiSearch tool type. + A tool that uses the Exa.ai search engine for grounding. + + Attributes: + api_key (str): + Required. The API key for ExaAiSearch. + custom_configs (google.protobuf.struct_pb2.Struct): + Optional. This field can be used to pass any + parameter from the Exa.ai Search API. + """ + + api_key: str = proto.Field( + proto.STRING, + number=1, + ) + custom_configs: struct_pb2.Struct = proto.Field( + proto.MESSAGE, + number=2, + message=struct_pb2.Struct, + ) + class CodeExecution(proto.Message): r"""Tool that executes code generated by the model, and automatically returns the result to the model. @@ -308,6 +335,11 @@ class Environment(proto.Enum): number=13, message=ParallelAiSearch, ) + exa_ai_search: ExaAiSearch = proto.Field( + proto.MESSAGE, + number=15, + message=ExaAiSearch, + ) code_execution: CodeExecution = proto.Field( proto.MESSAGE, number=4, diff --git a/google/cloud/aiplatform_v1beta1/types/batch_prediction_job.py b/google/cloud/aiplatform_v1beta1/types/batch_prediction_job.py index 928793451f..e50e7a7b31 100644 --- a/google/cloud/aiplatform_v1beta1/types/batch_prediction_job.py +++ b/google/cloud/aiplatform_v1beta1/types/batch_prediction_job.py @@ -90,6 +90,12 @@ class BatchPredictionJob(proto.Message): prediction without requiring uploading to model registry. Exactly one of model and unmanaged_container_model must be set. + endpoint (str): + For Bring-Your-Own-Endpoint (BYOE), the name of the Endpoint + resource that produces the predictions via this job, must + share the same ancestor Location. Exactly one of model, + unmanaged_container_model, or endpoint must be set. Example: + ``projects/193595526740/locations/us-central1/endpoints/4203439000301600768`` input_config (google.cloud.aiplatform_v1beta1.types.BatchPredictionJob.InputConfig): Required. Input configuration of the instances on which predictions are performed. The schema of any single instance @@ -637,6 +643,10 @@ class OutputInfo(proto.Message): message=gca_unmanaged_container_model.UnmanagedContainerModel, ) ) + endpoint: str = proto.Field( + proto.STRING, + number=41, + ) input_config: InputConfig = proto.Field( proto.MESSAGE, number=4, diff --git a/google/cloud/aiplatform_v1beta1/types/tool.py b/google/cloud/aiplatform_v1beta1/types/tool.py index ae66e88303..9134a723cf 100644 --- a/google/cloud/aiplatform_v1beta1/types/tool.py +++ b/google/cloud/aiplatform_v1beta1/types/tool.py @@ -103,6 +103,11 @@ class Tool(proto.Message): user queries. The search results will be grounded on Parallel.ai and presented to the model for response generation + exa_ai_search (google.cloud.aiplatform_v1beta1.types.Tool.ExaAiSearch): + Optional. Uses Exa.ai to search for + information to answer user queries. The search + results will be grounded on Exa.ai and presented + to the model for response generation code_execution (google.cloud.aiplatform_v1beta1.types.Tool.CodeExecution): Optional. CodeExecution tool type. Enables the model to execute code as part of @@ -223,6 +228,28 @@ class ParallelAiSearch(proto.Message): message=struct_pb2.Struct, ) + class ExaAiSearch(proto.Message): + r"""ExaAiSearch tool type. + A tool that uses the Exa.ai search engine for grounding. + + Attributes: + api_key (str): + Required. The API key for ExaAiSearch. + custom_configs (google.protobuf.struct_pb2.Struct): + Optional. This field can be used to pass any + parameter from the Exa.ai Search API. + """ + + api_key: str = proto.Field( + proto.STRING, + number=1, + ) + custom_configs: struct_pb2.Struct = proto.Field( + proto.MESSAGE, + number=2, + message=struct_pb2.Struct, + ) + class CodeExecution(proto.Message): r"""Tool that executes code generated by the model, and automatically returns the result to the model. @@ -309,6 +336,11 @@ class Environment(proto.Enum): number=13, message=ParallelAiSearch, ) + exa_ai_search: ExaAiSearch = proto.Field( + proto.MESSAGE, + number=15, + message=ExaAiSearch, + ) code_execution: CodeExecution = proto.Field( proto.MESSAGE, number=4, diff --git a/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1.json b/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1.json index 6312bfc507..53753f28da 100644 --- a/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1.json +++ b/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1.json @@ -8,7 +8,7 @@ ], "language": "PYTHON", "name": "google-cloud-aiplatform", - "version": "1.163.0" + "version": "0.0.0" }, "snippets": [ { diff --git a/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1beta1.json b/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1beta1.json index a5e2602e2f..bb8afa1c8f 100644 --- a/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1beta1.json +++ b/samples/generated_samples/snippet_metadata_google.cloud.aiplatform.v1beta1.json @@ -8,7 +8,7 @@ ], "language": "PYTHON", "name": "google-cloud-aiplatform", - "version": "1.163.0" + "version": "0.0.0" }, "snippets": [ { diff --git a/tests/unit/gapic/aiplatform_v1/test_gen_ai_cache_service.py b/tests/unit/gapic/aiplatform_v1/test_gen_ai_cache_service.py index 3a0cc83fa5..99b0da69a5 100644 --- a/tests/unit/gapic/aiplatform_v1/test_gen_ai_cache_service.py +++ b/tests/unit/gapic/aiplatform_v1/test_gen_ai_cache_service.py @@ -4077,7 +4077,7 @@ def test_create_cached_content_rest_call_success(request_type): # send a request that will satisfy transcoding request_init = {'parent': 'projects/sample1/locations/sample2'} - request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'thought': True, 'thought_signature': b'thought_signature_blob', 'video_metadata': {'start_offset': {}, 'end_offset': {}, 'fps': 0.329}, 'media_resolution': {'level': 1}}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} + request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'thought': True, 'thought_signature': b'thought_signature_blob', 'video_metadata': {'start_offset': {}, 'end_offset': {}, 'fps': 0.329}, 'media_resolution': {'level': 1}}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'exa_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 @@ -4361,7 +4361,7 @@ def test_update_cached_content_rest_call_success(request_type): # send a request that will satisfy transcoding request_init = {'cached_content': {'name': 'projects/sample1/locations/sample2/cachedContents/sample3'}} - request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'projects/sample1/locations/sample2/cachedContents/sample3', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'thought': True, 'thought_signature': b'thought_signature_blob', 'video_metadata': {'start_offset': {}, 'end_offset': {}, 'fps': 0.329}, 'media_resolution': {'level': 1}}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} + request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'projects/sample1/locations/sample2/cachedContents/sample3', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'thought': True, 'thought_signature': b'thought_signature_blob', 'video_metadata': {'start_offset': {}, 'end_offset': {}, 'fps': 0.329}, 'media_resolution': {'level': 1}}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'exa_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 @@ -5387,7 +5387,7 @@ async def test_create_cached_content_rest_asyncio_call_success(request_type): # send a request that will satisfy transcoding request_init = {'parent': 'projects/sample1/locations/sample2'} - request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'thought': True, 'thought_signature': b'thought_signature_blob', 'video_metadata': {'start_offset': {}, 'end_offset': {}, 'fps': 0.329}, 'media_resolution': {'level': 1}}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} + request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'thought': True, 'thought_signature': b'thought_signature_blob', 'video_metadata': {'start_offset': {}, 'end_offset': {}, 'fps': 0.329}, 'media_resolution': {'level': 1}}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'exa_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 @@ -5685,7 +5685,7 @@ async def test_update_cached_content_rest_asyncio_call_success(request_type): # send a request that will satisfy transcoding request_init = {'cached_content': {'name': 'projects/sample1/locations/sample2/cachedContents/sample3'}} - request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'projects/sample1/locations/sample2/cachedContents/sample3', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'thought': True, 'thought_signature': b'thought_signature_blob', 'video_metadata': {'start_offset': {}, 'end_offset': {}, 'fps': 0.329}, 'media_resolution': {'level': 1}}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} + request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'projects/sample1/locations/sample2/cachedContents/sample3', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'thought': True, 'thought_signature': b'thought_signature_blob', 'video_metadata': {'start_offset': {}, 'end_offset': {}, 'fps': 0.329}, 'media_resolution': {'level': 1}}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'exa_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 diff --git a/tests/unit/gapic/aiplatform_v1beta1/test_gen_ai_cache_service.py b/tests/unit/gapic/aiplatform_v1beta1/test_gen_ai_cache_service.py index 60879167f0..fdb88e61a2 100644 --- a/tests/unit/gapic/aiplatform_v1beta1/test_gen_ai_cache_service.py +++ b/tests/unit/gapic/aiplatform_v1beta1/test_gen_ai_cache_service.py @@ -4077,7 +4077,7 @@ def test_create_cached_content_rest_call_success(request_type): # send a request that will satisfy transcoding request_init = {'parent': 'projects/sample1/locations/sample2'} - request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'id': 'id_value', 'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'id': 'id_value', 'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'video_metadata': {'start_offset': {}, 'end_offset': {}}, 'thought': True, 'thought_signature': b'thought_signature_blob'}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_corpora': ['rag_corpora_value1', 'rag_corpora_value2'], 'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'hybrid_search': {'alpha': 0.518}, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}, 'store_context': True}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} + request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'id': 'id_value', 'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'id': 'id_value', 'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'video_metadata': {'start_offset': {}, 'end_offset': {}}, 'thought': True, 'thought_signature': b'thought_signature_blob'}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_corpora': ['rag_corpora_value1', 'rag_corpora_value2'], 'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'hybrid_search': {'alpha': 0.518}, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}, 'store_context': True}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'exa_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 @@ -4361,7 +4361,7 @@ def test_update_cached_content_rest_call_success(request_type): # send a request that will satisfy transcoding request_init = {'cached_content': {'name': 'projects/sample1/locations/sample2/cachedContents/sample3'}} - request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'projects/sample1/locations/sample2/cachedContents/sample3', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'id': 'id_value', 'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'id': 'id_value', 'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'video_metadata': {'start_offset': {}, 'end_offset': {}}, 'thought': True, 'thought_signature': b'thought_signature_blob'}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_corpora': ['rag_corpora_value1', 'rag_corpora_value2'], 'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'hybrid_search': {'alpha': 0.518}, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}, 'store_context': True}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} + request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'projects/sample1/locations/sample2/cachedContents/sample3', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'id': 'id_value', 'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'id': 'id_value', 'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'video_metadata': {'start_offset': {}, 'end_offset': {}}, 'thought': True, 'thought_signature': b'thought_signature_blob'}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_corpora': ['rag_corpora_value1', 'rag_corpora_value2'], 'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'hybrid_search': {'alpha': 0.518}, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}, 'store_context': True}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'exa_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 @@ -5387,7 +5387,7 @@ async def test_create_cached_content_rest_asyncio_call_success(request_type): # send a request that will satisfy transcoding request_init = {'parent': 'projects/sample1/locations/sample2'} - request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'id': 'id_value', 'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'id': 'id_value', 'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'video_metadata': {'start_offset': {}, 'end_offset': {}}, 'thought': True, 'thought_signature': b'thought_signature_blob'}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_corpora': ['rag_corpora_value1', 'rag_corpora_value2'], 'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'hybrid_search': {'alpha': 0.518}, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}, 'store_context': True}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} + request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'id': 'id_value', 'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'id': 'id_value', 'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'video_metadata': {'start_offset': {}, 'end_offset': {}}, 'thought': True, 'thought_signature': b'thought_signature_blob'}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_corpora': ['rag_corpora_value1', 'rag_corpora_value2'], 'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'hybrid_search': {'alpha': 0.518}, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}, 'store_context': True}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'exa_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 @@ -5685,7 +5685,7 @@ async def test_update_cached_content_rest_asyncio_call_success(request_type): # send a request that will satisfy transcoding request_init = {'cached_content': {'name': 'projects/sample1/locations/sample2/cachedContents/sample3'}} - request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'projects/sample1/locations/sample2/cachedContents/sample3', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'id': 'id_value', 'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'id': 'id_value', 'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'video_metadata': {'start_offset': {}, 'end_offset': {}}, 'thought': True, 'thought_signature': b'thought_signature_blob'}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_corpora': ['rag_corpora_value1', 'rag_corpora_value2'], 'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'hybrid_search': {'alpha': 0.518}, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}, 'store_context': True}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} + request_init["cached_content"] = {'expire_time': {'seconds': 751, 'nanos': 543}, 'ttl': {'seconds': 751, 'nanos': 543}, 'name': 'projects/sample1/locations/sample2/cachedContents/sample3', 'display_name': 'display_name_value', 'model': 'model_value', 'system_instruction': {'role': 'role_value', 'parts': [{'text': 'text_value', 'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value'}, 'function_call': {'id': 'id_value', 'name': 'name_value', 'args': {'fields': {}}, 'partial_args': [{'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'json_path': 'json_path_value', 'will_continue': True}], 'will_continue': True}, 'function_response': {'id': 'id_value', 'name': 'name_value', 'response': {}, 'parts': [{'inline_data': {'mime_type': 'mime_type_value', 'data': b'data_blob', 'display_name': 'display_name_value'}, 'file_data': {'mime_type': 'mime_type_value', 'file_uri': 'file_uri_value', 'display_name': 'display_name_value'}}]}, 'executable_code': {'language': 1, 'code': 'code_value'}, 'code_execution_result': {'outcome': 1, 'output': 'output_value'}, 'video_metadata': {'start_offset': {}, 'end_offset': {}}, 'thought': True, 'thought_signature': b'thought_signature_blob'}]}, 'contents': {}, 'tools': [{'function_declarations': [{'name': 'name_value', 'description': 'description_value', 'parameters': {'type_': 1, 'format_': 'format__value', 'title': 'title_value', 'description': 'description_value', 'nullable': True, 'default': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {}, 'list_value': {'values': {}}}, 'items': {}, 'min_items': 965, 'max_items': 967, 'enum': ['enum_value1', 'enum_value2'], 'properties': {}, 'property_ordering': ['property_ordering_value1', 'property_ordering_value2'], 'required': ['required_value1', 'required_value2'], 'min_properties': 1520, 'max_properties': 1522, 'minimum': 0.764, 'maximum': 0.766, 'min_length': 1061, 'max_length': 1063, 'pattern': 'pattern_value', 'example': {}, 'any_of': {}, 'additional_properties': {}, 'ref': 'ref_value', 'defs': {}}, 'parameters_json_schema': {}, 'response': {}, 'response_json_schema': {}}], 'retrieval': {'vertex_ai_search': {'datastore': 'datastore_value', 'engine': 'engine_value', 'max_results': 1207, 'filter': 'filter_value', 'data_store_specs': [{'data_store': 'data_store_value', 'filter': 'filter_value'}]}, 'vertex_rag_store': {'rag_corpora': ['rag_corpora_value1', 'rag_corpora_value2'], 'rag_resources': [{'rag_corpus': 'rag_corpus_value', 'rag_file_ids': ['rag_file_ids_value1', 'rag_file_ids_value2']}], 'similarity_top_k': 1731, 'vector_distance_threshold': 0.2665, 'rag_retrieval_config': {'top_k': 541, 'hybrid_search': {'alpha': 0.518}, 'filter': {'vector_distance_threshold': 0.2665, 'vector_similarity_threshold': 0.2917, 'metadata_filter': 'metadata_filter_value'}, 'ranking': {'rank_service': {'model_name': 'model_name_value'}, 'llm_ranker': {'model_name': 'model_name_value'}}}, 'store_context': True}, 'disable_attribution': True}, 'google_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'google_search_retrieval': {'dynamic_retrieval_config': {'mode': 1, 'dynamic_threshold': 0.1809}}, 'google_maps': {'enable_widget': True}, 'enterprise_web_search': {'exclude_domains': ['exclude_domains_value1', 'exclude_domains_value2'], 'blocking_confidence': 30}, 'parallel_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'exa_ai_search': {'api_key': 'api_key_value', 'custom_configs': {}}, 'code_execution': {}, 'url_context': {}, 'computer_use': {'environment': 1, 'excluded_predefined_functions': ['excluded_predefined_functions_value1', 'excluded_predefined_functions_value2']}}], 'tool_config': {'function_calling_config': {'mode': 1, 'allowed_function_names': ['allowed_function_names_value1', 'allowed_function_names_value2'], 'stream_function_call_arguments': True}, 'retrieval_config': {'lat_lng': {'latitude': 0.86, 'longitude': 0.971}, 'language_code': 'language_code_value'}}, 'create_time': {}, 'update_time': {}, 'usage_metadata': {'total_token_count': 1836, 'text_count': 1101, 'image_count': 1163, 'video_duration_seconds': 2346, 'audio_duration_seconds': 2341}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 diff --git a/tests/unit/gapic/aiplatform_v1beta1/test_job_service.py b/tests/unit/gapic/aiplatform_v1beta1/test_job_service.py index 953dc2bf0d..b60da331dd 100644 --- a/tests/unit/gapic/aiplatform_v1beta1/test_job_service.py +++ b/tests/unit/gapic/aiplatform_v1beta1/test_job_service.py @@ -9106,6 +9106,7 @@ def test_create_batch_prediction_job(request_type, transport: str = 'grpc'): display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -9127,6 +9128,7 @@ def test_create_batch_prediction_job(request_type, transport: str = 'grpc'): assert response.display_name == 'display_name_value' assert response.model == 'model_value' assert response.model_version_id == 'model_version_id_value' + assert response.endpoint == 'endpoint_value' assert response.service_account == 'service_account_value' assert response.generate_explanation is True assert response.state == job_state.JobState.JOB_STATE_QUEUED @@ -9254,6 +9256,7 @@ async def test_create_batch_prediction_job_async(request_type, transport: str = display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -9275,6 +9278,7 @@ async def test_create_batch_prediction_job_async(request_type, transport: str = assert response.display_name == 'display_name_value' assert response.model == 'model_value' assert response.model_version_id == 'model_version_id_value' + assert response.endpoint == 'endpoint_value' assert response.service_account == 'service_account_value' assert response.generate_explanation is True assert response.state == job_state.JobState.JOB_STATE_QUEUED @@ -9461,6 +9465,7 @@ def test_get_batch_prediction_job(request_type, transport: str = 'grpc'): display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -9482,6 +9487,7 @@ def test_get_batch_prediction_job(request_type, transport: str = 'grpc'): assert response.display_name == 'display_name_value' assert response.model == 'model_value' assert response.model_version_id == 'model_version_id_value' + assert response.endpoint == 'endpoint_value' assert response.service_account == 'service_account_value' assert response.generate_explanation is True assert response.state == job_state.JobState.JOB_STATE_QUEUED @@ -9609,6 +9615,7 @@ async def test_get_batch_prediction_job_async(request_type, transport: str = 'gr display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -9630,6 +9637,7 @@ async def test_get_batch_prediction_job_async(request_type, transport: str = 'gr assert response.display_name == 'display_name_value' assert response.model == 'model_value' assert response.model_version_id == 'model_version_id_value' + assert response.endpoint == 'endpoint_value' assert response.service_account == 'service_account_value' assert response.generate_explanation is True assert response.state == job_state.JobState.JOB_STATE_QUEUED @@ -21786,6 +21794,7 @@ async def test_create_batch_prediction_job_empty_call_grpc_asyncio(): display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -21821,6 +21830,7 @@ async def test_get_batch_prediction_job_empty_call_grpc_asyncio(): display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -24778,7 +24788,7 @@ def test_create_batch_prediction_job_rest_call_success(request_type): # send a request that will satisfy transcoding request_init = {'parent': 'projects/sample1/locations/sample2'} - request_init["batch_prediction_job"] = {'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'model_version_id': 'model_version_id_value', 'unmanaged_container_model': {'artifact_uri': 'artifact_uri_value', 'predict_schemata': {'instance_schema_uri': 'instance_schema_uri_value', 'parameters_schema_uri': 'parameters_schema_uri_value', 'prediction_schema_uri': 'prediction_schema_uri_value'}, 'container_spec': {'image_uri': 'image_uri_value', 'command': ['command_value1', 'command_value2'], 'args': ['args_value1', 'args_value2'], 'env': [{'name': 'name_value', 'value': 'value_value'}], 'ports': [{'container_port': 1511}], 'predict_route': 'predict_route_value', 'health_route': 'health_route_value', 'invoke_route_prefix': 'invoke_route_prefix_value', 'grpc_ports': {}, 'deployment_timeout': {'seconds': 751, 'nanos': 543}, 'shared_memory_size_mb': 2231, 'startup_probe': {'exec_': {'command': ['command_value1', 'command_value2']}, 'http_get': {'path': 'path_value', 'port': 453, 'host': 'host_value', 'scheme': 'scheme_value', 'http_headers': [{'name': 'name_value', 'value': 'value_value'}]}, 'grpc': {'port': 453, 'service': 'service_value'}, 'tcp_socket': {'port': 453, 'host': 'host_value'}, 'period_seconds': 1489, 'timeout_seconds': 1621, 'failure_threshold': 1812, 'success_threshold': 1829, 'initial_delay_seconds': 2214}, 'health_probe': {}, 'liveness_probe': {}}}, 'input_config': {'gcs_source': {'uris': ['uris_value1', 'uris_value2']}, 'bigquery_source': {'input_uri': 'input_uri_value'}, 'vertex_multimodal_dataset_source': {'dataset_name': 'dataset_name_value'}, 'instances_format': 'instances_format_value'}, 'instance_config': {'instance_type': 'instance_type_value', 'key_field': 'key_field_value', 'included_fields': ['included_fields_value1', 'included_fields_value2'], 'excluded_fields': ['excluded_fields_value1', 'excluded_fields_value2']}, 'model_parameters': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {'fields': {}}, 'list_value': {'values': {}}}, 'output_config': {'gcs_destination': {'output_uri_prefix': 'output_uri_prefix_value'}, 'bigquery_destination': {'output_uri': 'output_uri_value'}, 'vertex_multimodal_dataset_destination': {'bigquery_destination': {}, 'display_name': 'display_name_value'}, 'predictions_format': 'predictions_format_value'}, 'dedicated_resources': {'machine_spec': {'machine_type': 'machine_type_value', 'accelerator_type': 1, 'accelerator_count': 1805, 'gpu_partition_size': 'gpu_partition_size_value', 'tpu_topology': 'tpu_topology_value', 'multihost_gpu_node_count': 2593, 'reservation_affinity': {'reservation_affinity_type': 1, 'key': 'key_value', 'values': ['values_value1', 'values_value2']}, 'min_gpu_driver_version': 'min_gpu_driver_version_value'}, 'starting_replica_count': 2355, 'max_replica_count': 1805, 'flex_start': {'max_runtime_duration': {}}, 'spot': True}, 'service_account': 'service_account_value', 'manual_batch_tuning_parameters': {'batch_size': 1052}, 'generate_explanation': True, 'explanation_spec': {'parameters': {'sampled_shapley_attribution': {'path_count': 1077}, 'integrated_gradients_attribution': {'step_count': 1092, 'smooth_grad_config': {'noise_sigma': 0.11660000000000001, 'feature_noise_sigma': {'noise_sigma': [{'name': 'name_value', 'sigma': 0.529}]}, 'noisy_sample_count': 1947}, 'blur_baseline_config': {'max_blur_sigma': 0.1482}}, 'xrai_attribution': {'step_count': 1092, 'smooth_grad_config': {}, 'blur_baseline_config': {}}, 'examples': {'example_gcs_source': {'data_format': 1, 'gcs_source': {}}, 'nearest_neighbor_search_config': {}, 'presets': {'query': 1, 'modality': 1}, 'gcs_source': {}, 'neighbor_count': 1494}, 'top_k': 541, 'output_indices': {}}, 'metadata': {'inputs': {}, 'outputs': {}, 'feature_attributions_schema_uri': 'feature_attributions_schema_uri_value', 'latent_space_source': 'latent_space_source_value'}}, 'output_info': {'gcs_output_directory': 'gcs_output_directory_value', 'bigquery_output_dataset': 'bigquery_output_dataset_value', 'vertex_multimodal_dataset_name': 'vertex_multimodal_dataset_name_value', 'bigquery_output_table': 'bigquery_output_table_value'}, 'state': 1, 'error': {'code': 411, 'message': 'message_value', 'details': [{'type_url': 'type.googleapis.com/google.protobuf.Duration', 'value': b'\x08\x0c\x10\xdb\x07'}]}, 'partial_failures': {}, 'resources_consumed': {'replica_hours': 0.13920000000000002}, 'completion_stats': {'successful_count': 1736, 'failed_count': 1261, 'incomplete_count': 1720, 'successful_forecast_point_count': 3335}, 'create_time': {'seconds': 751, 'nanos': 543}, 'start_time': {}, 'end_time': {}, 'update_time': {}, 'labels': {}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}, 'model_monitoring_config': {'objective_configs': [{'training_dataset': {'dataset': 'dataset_value', 'gcs_source': {}, 'bigquery_source': {}, 'data_format': 'data_format_value', 'target_field': 'target_field_value', 'logging_sampling_strategy': {'random_sample_config': {'sample_rate': 0.1165}}}, 'training_prediction_skew_detection_config': {'skew_thresholds': {}, 'attribution_score_skew_thresholds': {}, 'default_skew_threshold': {'value': 0.541}}, 'prediction_drift_detection_config': {'drift_thresholds': {}, 'attribution_score_drift_thresholds': {}, 'default_drift_threshold': {}}, 'explanation_config': {'enable_feature_attributes': True, 'explanation_baseline': {'gcs': {}, 'bigquery': {}, 'prediction_format': 2}}}], 'alert_config': {'email_alert_config': {'user_emails': ['user_emails_value1', 'user_emails_value2']}, 'enable_logging': True, 'notification_channels': ['notification_channels_value1', 'notification_channels_value2']}, 'analysis_instance_schema_uri': 'analysis_instance_schema_uri_value', 'stats_anomalies_base_directory': {}}, 'model_monitoring_stats_anomalies': [{'objective': 1, 'deployed_model_id': 'deployed_model_id_value', 'anomaly_count': 1401, 'feature_stats': [{'feature_display_name': 'feature_display_name_value', 'threshold': {}, 'training_stats': {'score': 0.54, 'stats_uri': 'stats_uri_value', 'anomaly_uri': 'anomaly_uri_value', 'distribution_deviation': 0.23700000000000002, 'anomaly_detection_threshold': 0.28750000000000003, 'start_time': {}, 'end_time': {}}, 'prediction_stats': {}}]}], 'model_monitoring_status': {}, 'disable_container_logging': True, 'satisfies_pzs': True, 'satisfies_pzi': True} + request_init["batch_prediction_job"] = {'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'model_version_id': 'model_version_id_value', 'unmanaged_container_model': {'artifact_uri': 'artifact_uri_value', 'predict_schemata': {'instance_schema_uri': 'instance_schema_uri_value', 'parameters_schema_uri': 'parameters_schema_uri_value', 'prediction_schema_uri': 'prediction_schema_uri_value'}, 'container_spec': {'image_uri': 'image_uri_value', 'command': ['command_value1', 'command_value2'], 'args': ['args_value1', 'args_value2'], 'env': [{'name': 'name_value', 'value': 'value_value'}], 'ports': [{'container_port': 1511}], 'predict_route': 'predict_route_value', 'health_route': 'health_route_value', 'invoke_route_prefix': 'invoke_route_prefix_value', 'grpc_ports': {}, 'deployment_timeout': {'seconds': 751, 'nanos': 543}, 'shared_memory_size_mb': 2231, 'startup_probe': {'exec_': {'command': ['command_value1', 'command_value2']}, 'http_get': {'path': 'path_value', 'port': 453, 'host': 'host_value', 'scheme': 'scheme_value', 'http_headers': [{'name': 'name_value', 'value': 'value_value'}]}, 'grpc': {'port': 453, 'service': 'service_value'}, 'tcp_socket': {'port': 453, 'host': 'host_value'}, 'period_seconds': 1489, 'timeout_seconds': 1621, 'failure_threshold': 1812, 'success_threshold': 1829, 'initial_delay_seconds': 2214}, 'health_probe': {}, 'liveness_probe': {}}}, 'endpoint': 'endpoint_value', 'input_config': {'gcs_source': {'uris': ['uris_value1', 'uris_value2']}, 'bigquery_source': {'input_uri': 'input_uri_value'}, 'vertex_multimodal_dataset_source': {'dataset_name': 'dataset_name_value'}, 'instances_format': 'instances_format_value'}, 'instance_config': {'instance_type': 'instance_type_value', 'key_field': 'key_field_value', 'included_fields': ['included_fields_value1', 'included_fields_value2'], 'excluded_fields': ['excluded_fields_value1', 'excluded_fields_value2']}, 'model_parameters': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {'fields': {}}, 'list_value': {'values': {}}}, 'output_config': {'gcs_destination': {'output_uri_prefix': 'output_uri_prefix_value'}, 'bigquery_destination': {'output_uri': 'output_uri_value'}, 'vertex_multimodal_dataset_destination': {'bigquery_destination': {}, 'display_name': 'display_name_value'}, 'predictions_format': 'predictions_format_value'}, 'dedicated_resources': {'machine_spec': {'machine_type': 'machine_type_value', 'accelerator_type': 1, 'accelerator_count': 1805, 'gpu_partition_size': 'gpu_partition_size_value', 'tpu_topology': 'tpu_topology_value', 'multihost_gpu_node_count': 2593, 'reservation_affinity': {'reservation_affinity_type': 1, 'key': 'key_value', 'values': ['values_value1', 'values_value2']}, 'min_gpu_driver_version': 'min_gpu_driver_version_value'}, 'starting_replica_count': 2355, 'max_replica_count': 1805, 'flex_start': {'max_runtime_duration': {}}, 'spot': True}, 'service_account': 'service_account_value', 'manual_batch_tuning_parameters': {'batch_size': 1052}, 'generate_explanation': True, 'explanation_spec': {'parameters': {'sampled_shapley_attribution': {'path_count': 1077}, 'integrated_gradients_attribution': {'step_count': 1092, 'smooth_grad_config': {'noise_sigma': 0.11660000000000001, 'feature_noise_sigma': {'noise_sigma': [{'name': 'name_value', 'sigma': 0.529}]}, 'noisy_sample_count': 1947}, 'blur_baseline_config': {'max_blur_sigma': 0.1482}}, 'xrai_attribution': {'step_count': 1092, 'smooth_grad_config': {}, 'blur_baseline_config': {}}, 'examples': {'example_gcs_source': {'data_format': 1, 'gcs_source': {}}, 'nearest_neighbor_search_config': {}, 'presets': {'query': 1, 'modality': 1}, 'gcs_source': {}, 'neighbor_count': 1494}, 'top_k': 541, 'output_indices': {}}, 'metadata': {'inputs': {}, 'outputs': {}, 'feature_attributions_schema_uri': 'feature_attributions_schema_uri_value', 'latent_space_source': 'latent_space_source_value'}}, 'output_info': {'gcs_output_directory': 'gcs_output_directory_value', 'bigquery_output_dataset': 'bigquery_output_dataset_value', 'vertex_multimodal_dataset_name': 'vertex_multimodal_dataset_name_value', 'bigquery_output_table': 'bigquery_output_table_value'}, 'state': 1, 'error': {'code': 411, 'message': 'message_value', 'details': [{'type_url': 'type.googleapis.com/google.protobuf.Duration', 'value': b'\x08\x0c\x10\xdb\x07'}]}, 'partial_failures': {}, 'resources_consumed': {'replica_hours': 0.13920000000000002}, 'completion_stats': {'successful_count': 1736, 'failed_count': 1261, 'incomplete_count': 1720, 'successful_forecast_point_count': 3335}, 'create_time': {'seconds': 751, 'nanos': 543}, 'start_time': {}, 'end_time': {}, 'update_time': {}, 'labels': {}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}, 'model_monitoring_config': {'objective_configs': [{'training_dataset': {'dataset': 'dataset_value', 'gcs_source': {}, 'bigquery_source': {}, 'data_format': 'data_format_value', 'target_field': 'target_field_value', 'logging_sampling_strategy': {'random_sample_config': {'sample_rate': 0.1165}}}, 'training_prediction_skew_detection_config': {'skew_thresholds': {}, 'attribution_score_skew_thresholds': {}, 'default_skew_threshold': {'value': 0.541}}, 'prediction_drift_detection_config': {'drift_thresholds': {}, 'attribution_score_drift_thresholds': {}, 'default_drift_threshold': {}}, 'explanation_config': {'enable_feature_attributes': True, 'explanation_baseline': {'gcs': {}, 'bigquery': {}, 'prediction_format': 2}}}], 'alert_config': {'email_alert_config': {'user_emails': ['user_emails_value1', 'user_emails_value2']}, 'enable_logging': True, 'notification_channels': ['notification_channels_value1', 'notification_channels_value2']}, 'analysis_instance_schema_uri': 'analysis_instance_schema_uri_value', 'stats_anomalies_base_directory': {}}, 'model_monitoring_stats_anomalies': [{'objective': 1, 'deployed_model_id': 'deployed_model_id_value', 'anomaly_count': 1401, 'feature_stats': [{'feature_display_name': 'feature_display_name_value', 'threshold': {}, 'training_stats': {'score': 0.54, 'stats_uri': 'stats_uri_value', 'anomaly_uri': 'anomaly_uri_value', 'distribution_deviation': 0.23700000000000002, 'anomaly_detection_threshold': 0.28750000000000003, 'start_time': {}, 'end_time': {}}, 'prediction_stats': {}}]}], 'model_monitoring_status': {}, 'disable_container_logging': True, 'satisfies_pzs': True, 'satisfies_pzi': True} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 @@ -24852,6 +24862,7 @@ def get_message_fields(field): display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -24878,6 +24889,7 @@ def get_message_fields(field): assert response.display_name == 'display_name_value' assert response.model == 'model_value' assert response.model_version_id == 'model_version_id_value' + assert response.endpoint == 'endpoint_value' assert response.service_account == 'service_account_value' assert response.generate_explanation is True assert response.state == job_state.JobState.JOB_STATE_QUEUED @@ -24976,6 +24988,7 @@ def test_get_batch_prediction_job_rest_call_success(request_type): display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -25002,6 +25015,7 @@ def test_get_batch_prediction_job_rest_call_success(request_type): assert response.display_name == 'display_name_value' assert response.model == 'model_value' assert response.model_version_id == 'model_version_id_value' + assert response.endpoint == 'endpoint_value' assert response.service_account == 'service_account_value' assert response.generate_explanation is True assert response.state == job_state.JobState.JOB_STATE_QUEUED @@ -30415,7 +30429,7 @@ async def test_create_batch_prediction_job_rest_asyncio_call_success(request_typ # send a request that will satisfy transcoding request_init = {'parent': 'projects/sample1/locations/sample2'} - request_init["batch_prediction_job"] = {'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'model_version_id': 'model_version_id_value', 'unmanaged_container_model': {'artifact_uri': 'artifact_uri_value', 'predict_schemata': {'instance_schema_uri': 'instance_schema_uri_value', 'parameters_schema_uri': 'parameters_schema_uri_value', 'prediction_schema_uri': 'prediction_schema_uri_value'}, 'container_spec': {'image_uri': 'image_uri_value', 'command': ['command_value1', 'command_value2'], 'args': ['args_value1', 'args_value2'], 'env': [{'name': 'name_value', 'value': 'value_value'}], 'ports': [{'container_port': 1511}], 'predict_route': 'predict_route_value', 'health_route': 'health_route_value', 'invoke_route_prefix': 'invoke_route_prefix_value', 'grpc_ports': {}, 'deployment_timeout': {'seconds': 751, 'nanos': 543}, 'shared_memory_size_mb': 2231, 'startup_probe': {'exec_': {'command': ['command_value1', 'command_value2']}, 'http_get': {'path': 'path_value', 'port': 453, 'host': 'host_value', 'scheme': 'scheme_value', 'http_headers': [{'name': 'name_value', 'value': 'value_value'}]}, 'grpc': {'port': 453, 'service': 'service_value'}, 'tcp_socket': {'port': 453, 'host': 'host_value'}, 'period_seconds': 1489, 'timeout_seconds': 1621, 'failure_threshold': 1812, 'success_threshold': 1829, 'initial_delay_seconds': 2214}, 'health_probe': {}, 'liveness_probe': {}}}, 'input_config': {'gcs_source': {'uris': ['uris_value1', 'uris_value2']}, 'bigquery_source': {'input_uri': 'input_uri_value'}, 'vertex_multimodal_dataset_source': {'dataset_name': 'dataset_name_value'}, 'instances_format': 'instances_format_value'}, 'instance_config': {'instance_type': 'instance_type_value', 'key_field': 'key_field_value', 'included_fields': ['included_fields_value1', 'included_fields_value2'], 'excluded_fields': ['excluded_fields_value1', 'excluded_fields_value2']}, 'model_parameters': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {'fields': {}}, 'list_value': {'values': {}}}, 'output_config': {'gcs_destination': {'output_uri_prefix': 'output_uri_prefix_value'}, 'bigquery_destination': {'output_uri': 'output_uri_value'}, 'vertex_multimodal_dataset_destination': {'bigquery_destination': {}, 'display_name': 'display_name_value'}, 'predictions_format': 'predictions_format_value'}, 'dedicated_resources': {'machine_spec': {'machine_type': 'machine_type_value', 'accelerator_type': 1, 'accelerator_count': 1805, 'gpu_partition_size': 'gpu_partition_size_value', 'tpu_topology': 'tpu_topology_value', 'multihost_gpu_node_count': 2593, 'reservation_affinity': {'reservation_affinity_type': 1, 'key': 'key_value', 'values': ['values_value1', 'values_value2']}, 'min_gpu_driver_version': 'min_gpu_driver_version_value'}, 'starting_replica_count': 2355, 'max_replica_count': 1805, 'flex_start': {'max_runtime_duration': {}}, 'spot': True}, 'service_account': 'service_account_value', 'manual_batch_tuning_parameters': {'batch_size': 1052}, 'generate_explanation': True, 'explanation_spec': {'parameters': {'sampled_shapley_attribution': {'path_count': 1077}, 'integrated_gradients_attribution': {'step_count': 1092, 'smooth_grad_config': {'noise_sigma': 0.11660000000000001, 'feature_noise_sigma': {'noise_sigma': [{'name': 'name_value', 'sigma': 0.529}]}, 'noisy_sample_count': 1947}, 'blur_baseline_config': {'max_blur_sigma': 0.1482}}, 'xrai_attribution': {'step_count': 1092, 'smooth_grad_config': {}, 'blur_baseline_config': {}}, 'examples': {'example_gcs_source': {'data_format': 1, 'gcs_source': {}}, 'nearest_neighbor_search_config': {}, 'presets': {'query': 1, 'modality': 1}, 'gcs_source': {}, 'neighbor_count': 1494}, 'top_k': 541, 'output_indices': {}}, 'metadata': {'inputs': {}, 'outputs': {}, 'feature_attributions_schema_uri': 'feature_attributions_schema_uri_value', 'latent_space_source': 'latent_space_source_value'}}, 'output_info': {'gcs_output_directory': 'gcs_output_directory_value', 'bigquery_output_dataset': 'bigquery_output_dataset_value', 'vertex_multimodal_dataset_name': 'vertex_multimodal_dataset_name_value', 'bigquery_output_table': 'bigquery_output_table_value'}, 'state': 1, 'error': {'code': 411, 'message': 'message_value', 'details': [{'type_url': 'type.googleapis.com/google.protobuf.Duration', 'value': b'\x08\x0c\x10\xdb\x07'}]}, 'partial_failures': {}, 'resources_consumed': {'replica_hours': 0.13920000000000002}, 'completion_stats': {'successful_count': 1736, 'failed_count': 1261, 'incomplete_count': 1720, 'successful_forecast_point_count': 3335}, 'create_time': {'seconds': 751, 'nanos': 543}, 'start_time': {}, 'end_time': {}, 'update_time': {}, 'labels': {}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}, 'model_monitoring_config': {'objective_configs': [{'training_dataset': {'dataset': 'dataset_value', 'gcs_source': {}, 'bigquery_source': {}, 'data_format': 'data_format_value', 'target_field': 'target_field_value', 'logging_sampling_strategy': {'random_sample_config': {'sample_rate': 0.1165}}}, 'training_prediction_skew_detection_config': {'skew_thresholds': {}, 'attribution_score_skew_thresholds': {}, 'default_skew_threshold': {'value': 0.541}}, 'prediction_drift_detection_config': {'drift_thresholds': {}, 'attribution_score_drift_thresholds': {}, 'default_drift_threshold': {}}, 'explanation_config': {'enable_feature_attributes': True, 'explanation_baseline': {'gcs': {}, 'bigquery': {}, 'prediction_format': 2}}}], 'alert_config': {'email_alert_config': {'user_emails': ['user_emails_value1', 'user_emails_value2']}, 'enable_logging': True, 'notification_channels': ['notification_channels_value1', 'notification_channels_value2']}, 'analysis_instance_schema_uri': 'analysis_instance_schema_uri_value', 'stats_anomalies_base_directory': {}}, 'model_monitoring_stats_anomalies': [{'objective': 1, 'deployed_model_id': 'deployed_model_id_value', 'anomaly_count': 1401, 'feature_stats': [{'feature_display_name': 'feature_display_name_value', 'threshold': {}, 'training_stats': {'score': 0.54, 'stats_uri': 'stats_uri_value', 'anomaly_uri': 'anomaly_uri_value', 'distribution_deviation': 0.23700000000000002, 'anomaly_detection_threshold': 0.28750000000000003, 'start_time': {}, 'end_time': {}}, 'prediction_stats': {}}]}], 'model_monitoring_status': {}, 'disable_container_logging': True, 'satisfies_pzs': True, 'satisfies_pzi': True} + request_init["batch_prediction_job"] = {'name': 'name_value', 'display_name': 'display_name_value', 'model': 'model_value', 'model_version_id': 'model_version_id_value', 'unmanaged_container_model': {'artifact_uri': 'artifact_uri_value', 'predict_schemata': {'instance_schema_uri': 'instance_schema_uri_value', 'parameters_schema_uri': 'parameters_schema_uri_value', 'prediction_schema_uri': 'prediction_schema_uri_value'}, 'container_spec': {'image_uri': 'image_uri_value', 'command': ['command_value1', 'command_value2'], 'args': ['args_value1', 'args_value2'], 'env': [{'name': 'name_value', 'value': 'value_value'}], 'ports': [{'container_port': 1511}], 'predict_route': 'predict_route_value', 'health_route': 'health_route_value', 'invoke_route_prefix': 'invoke_route_prefix_value', 'grpc_ports': {}, 'deployment_timeout': {'seconds': 751, 'nanos': 543}, 'shared_memory_size_mb': 2231, 'startup_probe': {'exec_': {'command': ['command_value1', 'command_value2']}, 'http_get': {'path': 'path_value', 'port': 453, 'host': 'host_value', 'scheme': 'scheme_value', 'http_headers': [{'name': 'name_value', 'value': 'value_value'}]}, 'grpc': {'port': 453, 'service': 'service_value'}, 'tcp_socket': {'port': 453, 'host': 'host_value'}, 'period_seconds': 1489, 'timeout_seconds': 1621, 'failure_threshold': 1812, 'success_threshold': 1829, 'initial_delay_seconds': 2214}, 'health_probe': {}, 'liveness_probe': {}}}, 'endpoint': 'endpoint_value', 'input_config': {'gcs_source': {'uris': ['uris_value1', 'uris_value2']}, 'bigquery_source': {'input_uri': 'input_uri_value'}, 'vertex_multimodal_dataset_source': {'dataset_name': 'dataset_name_value'}, 'instances_format': 'instances_format_value'}, 'instance_config': {'instance_type': 'instance_type_value', 'key_field': 'key_field_value', 'included_fields': ['included_fields_value1', 'included_fields_value2'], 'excluded_fields': ['excluded_fields_value1', 'excluded_fields_value2']}, 'model_parameters': {'null_value': 0, 'number_value': 0.1285, 'string_value': 'string_value_value', 'bool_value': True, 'struct_value': {'fields': {}}, 'list_value': {'values': {}}}, 'output_config': {'gcs_destination': {'output_uri_prefix': 'output_uri_prefix_value'}, 'bigquery_destination': {'output_uri': 'output_uri_value'}, 'vertex_multimodal_dataset_destination': {'bigquery_destination': {}, 'display_name': 'display_name_value'}, 'predictions_format': 'predictions_format_value'}, 'dedicated_resources': {'machine_spec': {'machine_type': 'machine_type_value', 'accelerator_type': 1, 'accelerator_count': 1805, 'gpu_partition_size': 'gpu_partition_size_value', 'tpu_topology': 'tpu_topology_value', 'multihost_gpu_node_count': 2593, 'reservation_affinity': {'reservation_affinity_type': 1, 'key': 'key_value', 'values': ['values_value1', 'values_value2']}, 'min_gpu_driver_version': 'min_gpu_driver_version_value'}, 'starting_replica_count': 2355, 'max_replica_count': 1805, 'flex_start': {'max_runtime_duration': {}}, 'spot': True}, 'service_account': 'service_account_value', 'manual_batch_tuning_parameters': {'batch_size': 1052}, 'generate_explanation': True, 'explanation_spec': {'parameters': {'sampled_shapley_attribution': {'path_count': 1077}, 'integrated_gradients_attribution': {'step_count': 1092, 'smooth_grad_config': {'noise_sigma': 0.11660000000000001, 'feature_noise_sigma': {'noise_sigma': [{'name': 'name_value', 'sigma': 0.529}]}, 'noisy_sample_count': 1947}, 'blur_baseline_config': {'max_blur_sigma': 0.1482}}, 'xrai_attribution': {'step_count': 1092, 'smooth_grad_config': {}, 'blur_baseline_config': {}}, 'examples': {'example_gcs_source': {'data_format': 1, 'gcs_source': {}}, 'nearest_neighbor_search_config': {}, 'presets': {'query': 1, 'modality': 1}, 'gcs_source': {}, 'neighbor_count': 1494}, 'top_k': 541, 'output_indices': {}}, 'metadata': {'inputs': {}, 'outputs': {}, 'feature_attributions_schema_uri': 'feature_attributions_schema_uri_value', 'latent_space_source': 'latent_space_source_value'}}, 'output_info': {'gcs_output_directory': 'gcs_output_directory_value', 'bigquery_output_dataset': 'bigquery_output_dataset_value', 'vertex_multimodal_dataset_name': 'vertex_multimodal_dataset_name_value', 'bigquery_output_table': 'bigquery_output_table_value'}, 'state': 1, 'error': {'code': 411, 'message': 'message_value', 'details': [{'type_url': 'type.googleapis.com/google.protobuf.Duration', 'value': b'\x08\x0c\x10\xdb\x07'}]}, 'partial_failures': {}, 'resources_consumed': {'replica_hours': 0.13920000000000002}, 'completion_stats': {'successful_count': 1736, 'failed_count': 1261, 'incomplete_count': 1720, 'successful_forecast_point_count': 3335}, 'create_time': {'seconds': 751, 'nanos': 543}, 'start_time': {}, 'end_time': {}, 'update_time': {}, 'labels': {}, 'encryption_spec': {'kms_key_name': 'kms_key_name_value'}, 'model_monitoring_config': {'objective_configs': [{'training_dataset': {'dataset': 'dataset_value', 'gcs_source': {}, 'bigquery_source': {}, 'data_format': 'data_format_value', 'target_field': 'target_field_value', 'logging_sampling_strategy': {'random_sample_config': {'sample_rate': 0.1165}}}, 'training_prediction_skew_detection_config': {'skew_thresholds': {}, 'attribution_score_skew_thresholds': {}, 'default_skew_threshold': {'value': 0.541}}, 'prediction_drift_detection_config': {'drift_thresholds': {}, 'attribution_score_drift_thresholds': {}, 'default_drift_threshold': {}}, 'explanation_config': {'enable_feature_attributes': True, 'explanation_baseline': {'gcs': {}, 'bigquery': {}, 'prediction_format': 2}}}], 'alert_config': {'email_alert_config': {'user_emails': ['user_emails_value1', 'user_emails_value2']}, 'enable_logging': True, 'notification_channels': ['notification_channels_value1', 'notification_channels_value2']}, 'analysis_instance_schema_uri': 'analysis_instance_schema_uri_value', 'stats_anomalies_base_directory': {}}, 'model_monitoring_stats_anomalies': [{'objective': 1, 'deployed_model_id': 'deployed_model_id_value', 'anomaly_count': 1401, 'feature_stats': [{'feature_display_name': 'feature_display_name_value', 'threshold': {}, 'training_stats': {'score': 0.54, 'stats_uri': 'stats_uri_value', 'anomaly_uri': 'anomaly_uri_value', 'distribution_deviation': 0.23700000000000002, 'anomaly_detection_threshold': 0.28750000000000003, 'start_time': {}, 'end_time': {}}, 'prediction_stats': {}}]}], 'model_monitoring_status': {}, 'disable_container_logging': True, 'satisfies_pzs': True, 'satisfies_pzi': True} # The version of a generated dependency at test runtime may differ from the version used during generation. # Delete any fields which are not present in the current runtime dependency # See https://github.com/googleapis/gapic-generator-python/issues/1748 @@ -30489,6 +30503,7 @@ def get_message_fields(field): display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -30515,6 +30530,7 @@ def get_message_fields(field): assert response.display_name == 'display_name_value' assert response.model == 'model_value' assert response.model_version_id == 'model_version_id_value' + assert response.endpoint == 'endpoint_value' assert response.service_account == 'service_account_value' assert response.generate_explanation is True assert response.state == job_state.JobState.JOB_STATE_QUEUED @@ -30620,6 +30636,7 @@ async def test_get_batch_prediction_job_rest_asyncio_call_success(request_type): display_name='display_name_value', model='model_value', model_version_id='model_version_id_value', + endpoint='endpoint_value', service_account='service_account_value', generate_explanation=True, state=job_state.JobState.JOB_STATE_QUEUED, @@ -30646,6 +30663,7 @@ async def test_get_batch_prediction_job_rest_asyncio_call_success(request_type): assert response.display_name == 'display_name_value' assert response.model == 'model_value' assert response.model_version_id == 'model_version_id_value' + assert response.endpoint == 'endpoint_value' assert response.service_account == 'service_account_value' assert response.generate_explanation is True assert response.state == job_state.JobState.JOB_STATE_QUEUED