google-cloud-ai_platform-v1 0.58.0 → 0.59.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/lib/google/cloud/ai_platform/v1/dataset_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/dataset_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/deployment_resource_pool_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/deployment_resource_pool_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/endpoint_service/client.rb +103 -0
- data/lib/google/cloud/ai_platform/v1/endpoint_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/endpoint_service/rest/client.rb +96 -0
- data/lib/google/cloud/ai_platform/v1/endpoint_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/endpoint_service/rest/service_stub.rb +60 -0
- data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/feature_online_store_admin_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/feature_registry_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/feature_registry_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/featurestore_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/featurestore_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/gen_ai_tuning_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/gen_ai_tuning_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/index_endpoint_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/index_endpoint_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/index_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/index_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/job_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/job_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/llm_utility_service/paths.rb +19 -0
- data/lib/google/cloud/ai_platform/v1/metadata_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/metadata_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/migration_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/migration_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/model_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/model_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/notebook_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/notebook_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/persistent_resource_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/persistent_resource_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/pipeline_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/pipeline_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/prediction_service/paths.rb +19 -0
- data/lib/google/cloud/ai_platform/v1/rest.rb +2 -0
- data/lib/google/cloud/ai_platform/v1/schedule_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/schedule_service/paths.rb +38 -0
- data/lib/google/cloud/ai_platform/v1/schedule_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/specialist_pool_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/specialist_pool_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/tensorboard_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/tensorboard_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1/version.rb +1 -1
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/client.rb +1405 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/credentials.rb +47 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/operations.rb +810 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/paths.rb +173 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/client.rb +1318 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/operations.rb +4244 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest/service_stub.rb +663 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service/rest.rb +54 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_data_service.rb +56 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_service/client.rb +672 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_service/credentials.rb +47 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_service/paths.rb +69 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_service/rest/client.rb +634 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_service/rest/service_stub.rb +249 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_service/rest.rb +53 -0
- data/lib/google/cloud/ai_platform/v1/vertex_rag_service.rb +55 -0
- data/lib/google/cloud/ai_platform/v1/vizier_service/operations.rb +3 -10
- data/lib/google/cloud/ai_platform/v1/vizier_service/rest/operations.rb +73 -10
- data/lib/google/cloud/ai_platform/v1.rb +2 -0
- data/lib/google/cloud/aiplatform/v1/api_auth_pb.rb +46 -0
- data/lib/google/cloud/aiplatform/v1/endpoint_pb.rb +5 -1
- data/lib/google/cloud/aiplatform/v1/endpoint_service_pb.rb +3 -1
- data/lib/google/cloud/aiplatform/v1/endpoint_service_services_pb.rb +2 -0
- data/lib/google/cloud/aiplatform/v1/io_pb.rb +16 -1
- data/lib/google/cloud/aiplatform/v1/machine_resources_pb.rb +1 -1
- data/lib/google/cloud/aiplatform/v1/notebook_execution_job_pb.rb +6 -1
- data/lib/google/cloud/aiplatform/v1/tool_pb.rb +5 -1
- data/lib/google/cloud/aiplatform/v1/vertex_rag_data_pb.rb +67 -0
- data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_pb.rb +71 -0
- data/lib/google/cloud/aiplatform/v1/vertex_rag_data_service_services_pb.rb +63 -0
- data/lib/google/cloud/aiplatform/v1/vertex_rag_service_pb.rb +65 -0
- data/lib/google/cloud/aiplatform/v1/vertex_rag_service_services_pb.rb +52 -0
- data/proto_docs/google/cloud/aiplatform/v1/api_auth.rb +45 -0
- data/proto_docs/google/cloud/aiplatform/v1/endpoint.rb +30 -0
- data/proto_docs/google/cloud/aiplatform/v1/endpoint_service.rb +22 -0
- data/proto_docs/google/cloud/aiplatform/v1/io.rb +158 -0
- data/proto_docs/google/cloud/aiplatform/v1/machine_resources.rb +8 -0
- data/proto_docs/google/cloud/aiplatform/v1/notebook_execution_job.rb +18 -0
- data/proto_docs/google/cloud/aiplatform/v1/tool.rb +72 -0
- data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_data.rb +334 -0
- data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_data_service.rb +293 -0
- data/proto_docs/google/cloud/aiplatform/v1/vertex_rag_service.rb +279 -0
- data/proto_docs/google/longrunning/operations.rb +19 -14
- metadata +28 -2
|
@@ -15,7 +15,7 @@ require 'google/protobuf/empty_pb'
|
|
|
15
15
|
require 'google/protobuf/field_mask_pb'
|
|
16
16
|
|
|
17
17
|
|
|
18
|
-
descriptor_data = "\n1google/cloud/aiplatform/v1/endpoint_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a)google/cloud/aiplatform/v1/endpoint.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a google/protobuf/field_mask.proto\"\xa9\x01\n\x15\x43reateEndpointRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12;\n\x08\x65ndpoint\x18\x02 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\x12\x18\n\x0b\x65ndpoint_id\x18\x04 \x01(\tB\x03\xe0\x41\x05\"q\n\x1f\x43reateEndpointOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"N\n\x12GetEndpointRequest\x12\x38\n\x04name\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xdd\x01\n\x14ListEndpointsRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x13\n\x06\x66ilter\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12\x32\n\tread_mask\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x01\x12\x10\n\x08order_by\x18\x06 \x01(\t\"i\n\x15ListEndpointsResponse\x12\x37\n\tendpoints\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.Endpoint\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"\x8a\x01\n\x15UpdateEndpointRequest\x12;\n\x08\x65ndpoint\x18\x01 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\x12\x34\n\x0bupdate_mask\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x02\"Q\n\x15\x44\x65leteEndpointRequest\x12\x38\n\x04name\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xa8\x02\n\x12\x44\x65ployModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x46\n\x0e\x64\x65ployed_model\x18\x02 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModelB\x03\xe0\x41\x02\x12W\n\rtraffic_split\x18\x03 \x03(\x0b\x32@.google.cloud.aiplatform.v1.DeployModelRequest.TrafficSplitEntry\x1a\x33\n\x11TrafficSplitEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"X\n\x13\x44\x65ployModelResponse\x12\x41\n\x0e\x64\x65ployed_model\x18\x01 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModel\"n\n\x1c\x44\x65ployModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\x84\x02\n\x14UndeployModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x1e\n\x11\x64\x65ployed_model_id\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12Y\n\rtraffic_split\x18\x03 \x03(\x0b\x32\x42.google.cloud.aiplatform.v1.UndeployModelRequest.TrafficSplitEntry\x1a\x33\n\x11TrafficSplitEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"\x17\n\x15UndeployModelResponse\"p\n\x1eUndeployModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\xd8\x01\n\x1aMutateDeployedModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x46\n\x0e\x64\x65ployed_model\x18\x02 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModelB\x03\xe0\x41\x02\x12\x34\n\x0bupdate_mask\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x02\"`\n\x1bMutateDeployedModelResponse\x12\x41\n\x0e\x64\x65ployed_model\x18\x01 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModel\"v\n$MutateDeployedModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata2\
|
|
18
|
+
descriptor_data = "\n1google/cloud/aiplatform/v1/endpoint_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a)google/cloud/aiplatform/v1/endpoint.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a google/protobuf/field_mask.proto\"\xa9\x01\n\x15\x43reateEndpointRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12;\n\x08\x65ndpoint\x18\x02 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\x12\x18\n\x0b\x65ndpoint_id\x18\x04 \x01(\tB\x03\xe0\x41\x05\"q\n\x1f\x43reateEndpointOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"N\n\x12GetEndpointRequest\x12\x38\n\x04name\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xdd\x01\n\x14ListEndpointsRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x13\n\x06\x66ilter\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12\x16\n\tpage_size\x18\x03 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x04 \x01(\tB\x03\xe0\x41\x01\x12\x32\n\tread_mask\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x01\x12\x10\n\x08order_by\x18\x06 \x01(\t\"i\n\x15ListEndpointsResponse\x12\x37\n\tendpoints\x18\x01 \x03(\x0b\x32$.google.cloud.aiplatform.v1.Endpoint\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"\x8a\x01\n\x15UpdateEndpointRequest\x12;\n\x08\x65ndpoint\x18\x01 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\x12\x34\n\x0bupdate_mask\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x02\"_\n UpdateEndpointLongRunningRequest\x12;\n\x08\x65ndpoint\x18\x01 \x01(\x0b\x32$.google.cloud.aiplatform.v1.EndpointB\x03\xe0\x41\x02\"q\n\x1fUpdateEndpointOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"Q\n\x15\x44\x65leteEndpointRequest\x12\x38\n\x04name\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\"\xa8\x02\n\x12\x44\x65ployModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x46\n\x0e\x64\x65ployed_model\x18\x02 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModelB\x03\xe0\x41\x02\x12W\n\rtraffic_split\x18\x03 \x03(\x0b\x32@.google.cloud.aiplatform.v1.DeployModelRequest.TrafficSplitEntry\x1a\x33\n\x11TrafficSplitEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"X\n\x13\x44\x65ployModelResponse\x12\x41\n\x0e\x64\x65ployed_model\x18\x01 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModel\"n\n\x1c\x44\x65ployModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\x84\x02\n\x14UndeployModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x1e\n\x11\x64\x65ployed_model_id\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12Y\n\rtraffic_split\x18\x03 \x03(\x0b\x32\x42.google.cloud.aiplatform.v1.UndeployModelRequest.TrafficSplitEntry\x1a\x33\n\x11TrafficSplitEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"\x17\n\x15UndeployModelResponse\"p\n\x1eUndeployModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\xd8\x01\n\x1aMutateDeployedModelRequest\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x46\n\x0e\x64\x65ployed_model\x18\x02 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModelB\x03\xe0\x41\x02\x12\x34\n\x0bupdate_mask\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskB\x03\xe0\x41\x02\"`\n\x1bMutateDeployedModelResponse\x12\x41\n\x0e\x64\x65ployed_model\x18\x01 \x01(\x0b\x32).google.cloud.aiplatform.v1.DeployedModel\"v\n$MutateDeployedModelOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata2\xad\x11\n\x0f\x45ndpointService\x12\x82\x02\n\x0e\x43reateEndpoint\x12\x31.google.cloud.aiplatform.v1.CreateEndpointRequest\x1a\x1d.google.longrunning.Operation\"\x9d\x01\xca\x41+\n\x08\x45ndpoint\x12\x1f\x43reateEndpointOperationMetadata\xda\x41\x0fparent,endpoint\xda\x41\x1bparent,endpoint,endpoint_id\x82\xd3\xe4\x93\x02\x39\"-/v1/{parent=projects/*/locations/*}/endpoints:\x08\x65ndpoint\x12\xa1\x01\n\x0bGetEndpoint\x12..google.cloud.aiplatform.v1.GetEndpointRequest\x1a$.google.cloud.aiplatform.v1.Endpoint\"<\xda\x41\x04name\x82\xd3\xe4\x93\x02/\x12-/v1/{name=projects/*/locations/*/endpoints/*}\x12\xb4\x01\n\rListEndpoints\x12\x30.google.cloud.aiplatform.v1.ListEndpointsRequest\x1a\x31.google.cloud.aiplatform.v1.ListEndpointsResponse\">\xda\x41\x06parent\x82\xd3\xe4\x93\x02/\x12-/v1/{parent=projects/*/locations/*}/endpoints\x12\xca\x01\n\x0eUpdateEndpoint\x12\x31.google.cloud.aiplatform.v1.UpdateEndpointRequest\x1a$.google.cloud.aiplatform.v1.Endpoint\"_\xda\x41\x14\x65ndpoint,update_mask\x82\xd3\xe4\x93\x02\x42\x32\x36/v1/{endpoint.name=projects/*/locations/*/endpoints/*}:\x08\x65ndpoint\x12\xfc\x01\n\x19UpdateEndpointLongRunning\x12<.google.cloud.aiplatform.v1.UpdateEndpointLongRunningRequest\x1a\x1d.google.longrunning.Operation\"\x81\x01\xca\x41+\n\x08\x45ndpoint\x12\x1fUpdateEndpointOperationMetadata\xda\x41\x08\x65ndpoint\x82\xd3\xe4\x93\x02\x42\"=/v1/{endpoint.name=projects/*/locations/*/endpoints/*}:update:\x01*\x12\xd3\x01\n\x0e\x44\x65leteEndpoint\x12\x31.google.cloud.aiplatform.v1.DeleteEndpointRequest\x1a\x1d.google.longrunning.Operation\"o\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02/*-/v1/{name=projects/*/locations/*/endpoints/*}\x12\x85\x02\n\x0b\x44\x65ployModel\x12..google.cloud.aiplatform.v1.DeployModelRequest\x1a\x1d.google.longrunning.Operation\"\xa6\x01\xca\x41\x33\n\x13\x44\x65ployModelResponse\x12\x1c\x44\x65ployModelOperationMetadata\xda\x41%endpoint,deployed_model,traffic_split\x82\xd3\xe4\x93\x02\x42\"=/v1/{endpoint=projects/*/locations/*/endpoints/*}:deployModel:\x01*\x12\x92\x02\n\rUndeployModel\x12\x30.google.cloud.aiplatform.v1.UndeployModelRequest\x1a\x1d.google.longrunning.Operation\"\xaf\x01\xca\x41\x37\n\x15UndeployModelResponse\x12\x1eUndeployModelOperationMetadata\xda\x41(endpoint,deployed_model_id,traffic_split\x82\xd3\xe4\x93\x02\x44\"?/v1/{endpoint=projects/*/locations/*/endpoints/*}:undeployModel:\x01*\x12\xab\x02\n\x13MutateDeployedModel\x12\x36.google.cloud.aiplatform.v1.MutateDeployedModelRequest\x1a\x1d.google.longrunning.Operation\"\xbc\x01\xca\x41\x43\n\x1bMutateDeployedModelResponse\x12$MutateDeployedModelOperationMetadata\xda\x41#endpoint,deployed_model,update_mask\x82\xd3\xe4\x93\x02J\"E/v1/{endpoint=projects/*/locations/*/endpoints/*}:mutateDeployedModel:\x01*\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd2\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x14\x45ndpointServiceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
|
|
19
19
|
|
|
20
20
|
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
21
21
|
|
|
@@ -54,6 +54,8 @@ module Google
|
|
|
54
54
|
ListEndpointsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListEndpointsRequest").msgclass
|
|
55
55
|
ListEndpointsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListEndpointsResponse").msgclass
|
|
56
56
|
UpdateEndpointRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateEndpointRequest").msgclass
|
|
57
|
+
UpdateEndpointLongRunningRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateEndpointLongRunningRequest").msgclass
|
|
58
|
+
UpdateEndpointOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateEndpointOperationMetadata").msgclass
|
|
57
59
|
DeleteEndpointRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeleteEndpointRequest").msgclass
|
|
58
60
|
DeployModelRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeployModelRequest").msgclass
|
|
59
61
|
DeployModelResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeployModelResponse").msgclass
|
|
@@ -41,6 +41,8 @@ module Google
|
|
|
41
41
|
rpc :ListEndpoints, ::Google::Cloud::AIPlatform::V1::ListEndpointsRequest, ::Google::Cloud::AIPlatform::V1::ListEndpointsResponse
|
|
42
42
|
# Updates an Endpoint.
|
|
43
43
|
rpc :UpdateEndpoint, ::Google::Cloud::AIPlatform::V1::UpdateEndpointRequest, ::Google::Cloud::AIPlatform::V1::Endpoint
|
|
44
|
+
# Updates an Endpoint with a long running operation.
|
|
45
|
+
rpc :UpdateEndpointLongRunning, ::Google::Cloud::AIPlatform::V1::UpdateEndpointLongRunningRequest, ::Google::Longrunning::Operation
|
|
44
46
|
# Deletes an Endpoint.
|
|
45
47
|
rpc :DeleteEndpoint, ::Google::Cloud::AIPlatform::V1::DeleteEndpointRequest, ::Google::Longrunning::Operation
|
|
46
48
|
# Deploys a Model into this Endpoint, creating a DeployedModel within it.
|
|
@@ -5,9 +5,11 @@
|
|
|
5
5
|
require 'google/protobuf'
|
|
6
6
|
|
|
7
7
|
require 'google/api/field_behavior_pb'
|
|
8
|
+
require 'google/cloud/aiplatform/v1/api_auth_pb'
|
|
9
|
+
require 'google/protobuf/timestamp_pb'
|
|
8
10
|
|
|
9
11
|
|
|
10
|
-
descriptor_data = "\n#google/cloud/aiplatform/v1/io.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\"L\n\nAvroSource\x12>\n\ngcs_source\x18\x01 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceB\x03\xe0\x41\x02\"K\n\tCsvSource\x12>\n\ngcs_source\x18\x01 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceB\x03\xe0\x41\x02\"\x1e\n\tGcsSource\x12\x11\n\x04uris\x18\x01 \x03(\tB\x03\xe0\x41\x02\"0\n\x0eGcsDestination\x12\x1e\n\x11output_uri_prefix\x18\x01 \x01(\tB\x03\xe0\x41\x02\"(\n\x0e\x42igQuerySource\x12\x16\n\tinput_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\".\n\x13\x42igQueryDestination\x12\x17\n\noutput_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\"Z\n\x0e\x43svDestination\x12H\n\x0fgcs_destination\x18\x01 \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestinationB\x03\xe0\x41\x02\"_\n\x13TFRecordDestination\x12H\n\x0fgcs_destination\x18\x01 \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestinationB\x03\xe0\x41\x02\"7\n\x1c\x43ontainerRegistryDestination\x12\x17\n\noutput_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\x42\xc5\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x07IoProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
|
|
12
|
+
descriptor_data = "\n#google/cloud/aiplatform/v1/io.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a)google/cloud/aiplatform/v1/api_auth.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"L\n\nAvroSource\x12>\n\ngcs_source\x18\x01 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceB\x03\xe0\x41\x02\"K\n\tCsvSource\x12>\n\ngcs_source\x18\x01 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceB\x03\xe0\x41\x02\"\x1e\n\tGcsSource\x12\x11\n\x04uris\x18\x01 \x03(\tB\x03\xe0\x41\x02\"0\n\x0eGcsDestination\x12\x1e\n\x11output_uri_prefix\x18\x01 \x01(\tB\x03\xe0\x41\x02\"(\n\x0e\x42igQuerySource\x12\x16\n\tinput_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\".\n\x13\x42igQueryDestination\x12\x17\n\noutput_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\"Z\n\x0e\x43svDestination\x12H\n\x0fgcs_destination\x18\x01 \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestinationB\x03\xe0\x41\x02\"_\n\x13TFRecordDestination\x12H\n\x0fgcs_destination\x18\x01 \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestinationB\x03\xe0\x41\x02\"7\n\x1c\x43ontainerRegistryDestination\x12\x17\n\noutput_uri\x18\x01 \x01(\tB\x03\xe0\x41\x02\"\xd5\x02\n\x11GoogleDriveSource\x12S\n\x0cresource_ids\x18\x01 \x03(\x0b\x32\x38.google.cloud.aiplatform.v1.GoogleDriveSource.ResourceIdB\x03\xe0\x41\x02\x1a\xea\x01\n\nResourceId\x12\x61\n\rresource_type\x18\x01 \x01(\x0e\x32\x45.google.cloud.aiplatform.v1.GoogleDriveSource.ResourceId.ResourceTypeB\x03\xe0\x41\x02\x12\x18\n\x0bresource_id\x18\x02 \x01(\tB\x03\xe0\x41\x02\"_\n\x0cResourceType\x12\x1d\n\x19RESOURCE_TYPE_UNSPECIFIED\x10\x00\x12\x16\n\x12RESOURCE_TYPE_FILE\x10\x01\x12\x18\n\x14RESOURCE_TYPE_FOLDER\x10\x02\"\x14\n\x12\x44irectUploadSource\"\xa9\x03\n\x0bSlackSource\x12L\n\x08\x63hannels\x18\x01 \x03(\x0b\x32\x35.google.cloud.aiplatform.v1.SlackSource.SlackChannelsB\x03\xe0\x41\x02\x1a\xcb\x02\n\rSlackChannels\x12Y\n\x08\x63hannels\x18\x01 \x03(\x0b\x32\x42.google.cloud.aiplatform.v1.SlackSource.SlackChannels.SlackChannelB\x03\xe0\x41\x02\x12M\n\x0e\x61pi_key_config\x18\x03 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ApiAuth.ApiKeyConfigB\x03\xe0\x41\x02\x1a\x8f\x01\n\x0cSlackChannel\x12\x17\n\nchannel_id\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x33\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x01\x12\x31\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x01\"\x91\x02\n\nJiraSource\x12M\n\x0cjira_queries\x18\x01 \x03(\x0b\x32\x32.google.cloud.aiplatform.v1.JiraSource.JiraQueriesB\x03\xe0\x41\x02\x1a\xb3\x01\n\x0bJiraQueries\x12\x10\n\x08projects\x18\x03 \x03(\t\x12\x16\n\x0e\x63ustom_queries\x18\x04 \x03(\t\x12\x12\n\x05\x65mail\x18\x05 \x01(\tB\x03\xe0\x41\x02\x12\x17\n\nserver_uri\x18\x06 \x01(\tB\x03\xe0\x41\x02\x12M\n\x0e\x61pi_key_config\x18\x07 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ApiAuth.ApiKeyConfigB\x03\xe0\x41\x02\"\xb5\x03\n\x11SharePointSources\x12[\n\x13share_point_sources\x18\x01 \x03(\x0b\x32>.google.cloud.aiplatform.v1.SharePointSources.SharePointSource\x1a\xc2\x02\n\x10SharePointSource\x12 \n\x16sharepoint_folder_path\x18\x05 \x01(\tH\x00\x12\x1e\n\x14sharepoint_folder_id\x18\x06 \x01(\tH\x00\x12\x14\n\ndrive_name\x18\x07 \x01(\tH\x01\x12\x12\n\x08\x64rive_id\x18\x08 \x01(\tH\x01\x12\x11\n\tclient_id\x18\x01 \x01(\t\x12G\n\rclient_secret\x18\x02 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ApiAuth.ApiKeyConfig\x12\x11\n\ttenant_id\x18\x03 \x01(\t\x12\x1c\n\x14sharepoint_site_name\x18\x04 \x01(\t\x12\x14\n\x07\x66ile_id\x18\t \x01(\tB\x03\xe0\x41\x03\x42\x0f\n\rfolder_sourceB\x0e\n\x0c\x64rive_sourceB\xc5\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x07IoProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
|
|
11
13
|
|
|
12
14
|
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
13
15
|
|
|
@@ -22,6 +24,8 @@ rescue TypeError
|
|
|
22
24
|
file = pool.add_serialized_file(serialized)
|
|
23
25
|
warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
|
|
24
26
|
imports = [
|
|
27
|
+
["google.cloud.aiplatform.v1.ApiAuth.ApiKeyConfig", "google/cloud/aiplatform/v1/api_auth.proto"],
|
|
28
|
+
["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
|
|
25
29
|
]
|
|
26
30
|
imports.each do |type_name, expected_filename|
|
|
27
31
|
import_file = pool.lookup(type_name).file_descriptor
|
|
@@ -46,6 +50,17 @@ module Google
|
|
|
46
50
|
CsvDestination = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CsvDestination").msgclass
|
|
47
51
|
TFRecordDestination = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.TFRecordDestination").msgclass
|
|
48
52
|
ContainerRegistryDestination = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ContainerRegistryDestination").msgclass
|
|
53
|
+
GoogleDriveSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GoogleDriveSource").msgclass
|
|
54
|
+
GoogleDriveSource::ResourceId = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GoogleDriveSource.ResourceId").msgclass
|
|
55
|
+
GoogleDriveSource::ResourceId::ResourceType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GoogleDriveSource.ResourceId.ResourceType").enummodule
|
|
56
|
+
DirectUploadSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DirectUploadSource").msgclass
|
|
57
|
+
SlackSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SlackSource").msgclass
|
|
58
|
+
SlackSource::SlackChannels = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SlackSource.SlackChannels").msgclass
|
|
59
|
+
SlackSource::SlackChannels::SlackChannel = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SlackSource.SlackChannels.SlackChannel").msgclass
|
|
60
|
+
JiraSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.JiraSource").msgclass
|
|
61
|
+
JiraSource::JiraQueries = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.JiraSource.JiraQueries").msgclass
|
|
62
|
+
SharePointSources = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SharePointSources").msgclass
|
|
63
|
+
SharePointSources::SharePointSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.SharePointSources.SharePointSource").msgclass
|
|
49
64
|
end
|
|
50
65
|
end
|
|
51
66
|
end
|
|
@@ -9,7 +9,7 @@ require 'google/cloud/aiplatform/v1/accelerator_type_pb'
|
|
|
9
9
|
require 'google/cloud/aiplatform/v1/reservation_affinity_pb'
|
|
10
10
|
|
|
11
11
|
|
|
12
|
-
descriptor_data = "\n2google/cloud/aiplatform/v1/machine_resources.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x31google/cloud/aiplatform/v1/accelerator_type.proto\x1a\x35google/cloud/aiplatform/v1/reservation_affinity.proto\"\x81\x02\n\x0bMachineSpec\x12\x19\n\x0cmachine_type\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12J\n\x10\x61\x63\x63\x65lerator_type\x18\x02 \x01(\x0e\x32+.google.cloud.aiplatform.v1.AcceleratorTypeB\x03\xe0\x41\x05\x12\x19\n\x11\x61\x63\x63\x65lerator_count\x18\x03 \x01(\x05\x12\x19\n\x0ctpu_topology\x18\x04 \x01(\tB\x03\xe0\x41\x05\x12U\n\x14reservation_affinity\x18\x05 \x01(\x0b\x32/.google.cloud.aiplatform.v1.ReservationAffinityB\x06\xe0\x41\x05\xe0\x41\x01\"\
|
|
12
|
+
descriptor_data = "\n2google/cloud/aiplatform/v1/machine_resources.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x31google/cloud/aiplatform/v1/accelerator_type.proto\x1a\x35google/cloud/aiplatform/v1/reservation_affinity.proto\"\x81\x02\n\x0bMachineSpec\x12\x19\n\x0cmachine_type\x18\x01 \x01(\tB\x03\xe0\x41\x05\x12J\n\x10\x61\x63\x63\x65lerator_type\x18\x02 \x01(\x0e\x32+.google.cloud.aiplatform.v1.AcceleratorTypeB\x03\xe0\x41\x05\x12\x19\n\x11\x61\x63\x63\x65lerator_count\x18\x03 \x01(\x05\x12\x19\n\x0ctpu_topology\x18\x04 \x01(\tB\x03\xe0\x41\x05\x12U\n\x14reservation_affinity\x18\x05 \x01(\x0b\x32/.google.cloud.aiplatform.v1.ReservationAffinityB\x06\xe0\x41\x05\xe0\x41\x01\"\xb0\x02\n\x12\x44\x65\x64icatedResources\x12\x45\n\x0cmachine_spec\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x02\xe0\x41\x05\x12!\n\x11min_replica_count\x18\x02 \x01(\x05\x42\x06\xe0\x41\x02\xe0\x41\x05\x12\x1e\n\x11max_replica_count\x18\x03 \x01(\x05\x42\x03\xe0\x41\x05\x12#\n\x16required_replica_count\x18\t \x01(\x05\x42\x03\xe0\x41\x01\x12X\n\x18\x61utoscaling_metric_specs\x18\x04 \x03(\x0b\x32\x31.google.cloud.aiplatform.v1.AutoscalingMetricSpecB\x03\xe0\x41\x05\x12\x11\n\x04spot\x18\x05 \x01(\x08\x42\x03\xe0\x41\x01\"T\n\x12\x41utomaticResources\x12\x1e\n\x11min_replica_count\x18\x01 \x01(\x05\x42\x03\xe0\x41\x05\x12\x1e\n\x11max_replica_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x05\"\xa5\x01\n\x17\x42\x61tchDedicatedResources\x12\x45\n\x0cmachine_spec\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpecB\x06\xe0\x41\x02\xe0\x41\x05\x12#\n\x16starting_replica_count\x18\x02 \x01(\x05\x42\x03\xe0\x41\x05\x12\x1e\n\x11max_replica_count\x18\x03 \x01(\x05\x42\x03\xe0\x41\x05\"/\n\x11ResourcesConsumed\x12\x1a\n\rreplica_hours\x18\x01 \x01(\x01\x42\x03\xe0\x41\x03\"=\n\x08\x44iskSpec\x12\x16\n\x0e\x62oot_disk_type\x18\x01 \x01(\t\x12\x19\n\x11\x62oot_disk_size_gb\x18\x02 \x01(\x05\"=\n\x12PersistentDiskSpec\x12\x11\n\tdisk_type\x18\x01 \x01(\t\x12\x14\n\x0c\x64isk_size_gb\x18\x02 \x01(\x03\"L\n\x08NfsMount\x12\x13\n\x06server\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x11\n\x04path\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0bmount_point\x18\x03 \x01(\tB\x03\xe0\x41\x02\"A\n\x15\x41utoscalingMetricSpec\x12\x18\n\x0bmetric_name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x0e\n\x06target\x18\x02 \x01(\x05\".\n\x10ShieldedVmConfig\x12\x1a\n\x12\x65nable_secure_boot\x18\x01 \x01(\x08\x42\xd3\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x15MachineResourcesProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
|
|
13
13
|
|
|
14
14
|
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
15
15
|
|
|
@@ -8,12 +8,14 @@ require 'google/api/field_behavior_pb'
|
|
|
8
8
|
require 'google/api/resource_pb'
|
|
9
9
|
require 'google/cloud/aiplatform/v1/encryption_spec_pb'
|
|
10
10
|
require 'google/cloud/aiplatform/v1/job_state_pb'
|
|
11
|
+
require 'google/cloud/aiplatform/v1/machine_resources_pb'
|
|
12
|
+
require 'google/cloud/aiplatform/v1/network_spec_pb'
|
|
11
13
|
require 'google/protobuf/duration_pb'
|
|
12
14
|
require 'google/protobuf/timestamp_pb'
|
|
13
15
|
require 'google/rpc/status_pb'
|
|
14
16
|
|
|
15
17
|
|
|
16
|
-
descriptor_data = "\n7google/cloud/aiplatform/v1/notebook_execution_job.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a*google/cloud/aiplatform/v1/job_state.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\
|
|
18
|
+
descriptor_data = "\n7google/cloud/aiplatform/v1/notebook_execution_job.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a\x30google/cloud/aiplatform/v1/encryption_spec.proto\x1a*google/cloud/aiplatform/v1/job_state.proto\x1a\x32google/cloud/aiplatform/v1/machine_resources.proto\x1a-google/cloud/aiplatform/v1/network_spec.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\xe3\x0e\n\x14NotebookExecutionJob\x12o\n\x1a\x64\x61taform_repository_source\x18\x03 \x01(\x0b\x32I.google.cloud.aiplatform.v1.NotebookExecutionJob.DataformRepositorySourceH\x00\x12\x61\n\x13gcs_notebook_source\x18\x04 \x01(\x0b\x32\x42.google.cloud.aiplatform.v1.NotebookExecutionJob.GcsNotebookSourceH\x00\x12g\n\x16\x64irect_notebook_source\x18\x11 \x01(\x0b\x32\x45.google.cloud.aiplatform.v1.NotebookExecutionJob.DirectNotebookSourceH\x00\x12i\n\'notebook_runtime_template_resource_name\x18\x0e \x01(\tB6\xfa\x41\x33\n1aiplatform.googleapis.com/NotebookRuntimeTemplateH\x01\x12i\n\x17\x63ustom_environment_spec\x18\x10 \x01(\x0b\x32\x46.google.cloud.aiplatform.v1.NotebookExecutionJob.CustomEnvironmentSpecH\x01\x12\x18\n\x0egcs_output_uri\x18\x08 \x01(\tH\x02\x12\x18\n\x0e\x65xecution_user\x18\t \x01(\tH\x03\x12\x19\n\x0fservice_account\x18\x12 \x01(\tH\x03\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x14\n\x0c\x64isplay_name\x18\x02 \x01(\t\x12\x34\n\x11\x65xecution_timeout\x18\x05 \x01(\x0b\x32\x19.google.protobuf.Duration\x12J\n\x16schedule_resource_name\x18\x06 \x01(\tB*\xe0\x41\x03\xfa\x41$\n\"aiplatform.googleapis.com/Schedule\x12<\n\tjob_state\x18\n \x01(\x0e\x32$.google.cloud.aiplatform.v1.JobStateB\x03\xe0\x41\x03\x12\'\n\x06status\x18\x0b \x01(\x0b\x32\x12.google.rpc.StatusB\x03\xe0\x41\x03\x12\x34\n\x0b\x63reate_time\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12L\n\x06labels\x18\x13 \x03(\x0b\x32<.google.cloud.aiplatform.v1.NotebookExecutionJob.LabelsEntry\x12\x43\n\x0f\x65ncryption_spec\x18\x16 \x01(\x0b\x32*.google.cloud.aiplatform.v1.EncryptionSpec\x1aY\n\x18\x44\x61taformRepositorySource\x12)\n!dataform_repository_resource_name\x18\x01 \x01(\t\x12\x12\n\ncommit_sha\x18\x02 \x01(\t\x1a\x34\n\x11GcsNotebookSource\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x12\n\ngeneration\x18\x02 \x01(\t\x1a\'\n\x14\x44irectNotebookSource\x12\x0f\n\x07\x63ontent\x18\x01 \x01(\x0c\x1a\xe3\x01\n\x15\x43ustomEnvironmentSpec\x12=\n\x0cmachine_spec\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.MachineSpec\x12L\n\x14persistent_disk_spec\x18\x02 \x01(\x0b\x32..google.cloud.aiplatform.v1.PersistentDiskSpec\x12=\n\x0cnetwork_spec\x18\x03 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.NetworkSpec\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01:\xb9\x01\xea\x41\xb5\x01\n.aiplatform.googleapis.com/NotebookExecutionJob\x12Vprojects/{project}/locations/{location}/notebookExecutionJobs/{notebook_execution_job}*\x15notebookExecutionJobs2\x14notebookExecutionJobB\x11\n\x0fnotebook_sourceB\x12\n\x10\x65nvironment_specB\x10\n\x0e\x65xecution_sinkB\x14\n\x12\x65xecution_identityB\xd7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x19NotebookExecutionJobProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
|
|
17
19
|
|
|
18
20
|
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
19
21
|
|
|
@@ -32,6 +34,8 @@ rescue TypeError
|
|
|
32
34
|
["google.rpc.Status", "google/rpc/status.proto"],
|
|
33
35
|
["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
|
|
34
36
|
["google.cloud.aiplatform.v1.EncryptionSpec", "google/cloud/aiplatform/v1/encryption_spec.proto"],
|
|
37
|
+
["google.cloud.aiplatform.v1.MachineSpec", "google/cloud/aiplatform/v1/machine_resources.proto"],
|
|
38
|
+
["google.cloud.aiplatform.v1.NetworkSpec", "google/cloud/aiplatform/v1/network_spec.proto"],
|
|
35
39
|
]
|
|
36
40
|
imports.each do |type_name, expected_filename|
|
|
37
41
|
import_file = pool.lookup(type_name).file_descriptor
|
|
@@ -51,6 +55,7 @@ module Google
|
|
|
51
55
|
NotebookExecutionJob::DataformRepositorySource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.NotebookExecutionJob.DataformRepositorySource").msgclass
|
|
52
56
|
NotebookExecutionJob::GcsNotebookSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.NotebookExecutionJob.GcsNotebookSource").msgclass
|
|
53
57
|
NotebookExecutionJob::DirectNotebookSource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.NotebookExecutionJob.DirectNotebookSource").msgclass
|
|
58
|
+
NotebookExecutionJob::CustomEnvironmentSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.NotebookExecutionJob.CustomEnvironmentSpec").msgclass
|
|
54
59
|
end
|
|
55
60
|
end
|
|
56
61
|
end
|
|
@@ -10,7 +10,7 @@ require 'google/cloud/aiplatform/v1/openapi_pb'
|
|
|
10
10
|
require 'google/protobuf/struct_pb'
|
|
11
11
|
|
|
12
12
|
|
|
13
|
-
descriptor_data = "\n%google/cloud/aiplatform/v1/tool.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xf3\x01\n\x04Tool\x12S\n\x15\x66unction_declarations\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.FunctionDeclarationB\x03\xe0\x41\x01\x12=\n\tretrieval\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RetrievalB\x03\xe0\x41\x01\x12W\n\x17google_search_retrieval\x18\x03 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.GoogleSearchRetrievalB\x03\xe0\x41\x01\"\xba\x01\n\x13\x46unctionDeclaration\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x65scription\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12;\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\x12\x39\n\x08response\x18\x04 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\"M\n\x0c\x46unctionCall\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12*\n\x04\x61rgs\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\"U\n\x10\x46unctionResponse\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12.\n\x08response\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x02\"\
|
|
13
|
+
descriptor_data = "\n%google/cloud/aiplatform/v1/tool.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/openapi.proto\x1a\x1cgoogle/protobuf/struct.proto\"\xf3\x01\n\x04Tool\x12S\n\x15\x66unction_declarations\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.FunctionDeclarationB\x03\xe0\x41\x01\x12=\n\tretrieval\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RetrievalB\x03\xe0\x41\x01\x12W\n\x17google_search_retrieval\x18\x03 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.GoogleSearchRetrievalB\x03\xe0\x41\x01\"\xba\x01\n\x13\x46unctionDeclaration\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x65scription\x18\x02 \x01(\tB\x03\xe0\x41\x01\x12;\n\nparameters\x18\x03 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\x12\x39\n\x08response\x18\x04 \x01(\x0b\x32\".google.cloud.aiplatform.v1.SchemaB\x03\xe0\x41\x01\"M\n\x0c\x46unctionCall\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12*\n\x04\x61rgs\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x01\"U\n\x10\x46unctionResponse\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x02\x12.\n\x08response\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructB\x03\xe0\x41\x02\"\xc9\x01\n\tRetrieval\x12\x46\n\x10vertex_ai_search\x18\x02 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexAISearchH\x00\x12\x46\n\x10vertex_rag_store\x18\x04 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexRagStoreH\x00\x12\"\n\x13\x64isable_attribution\x18\x03 \x01(\x08\x42\x05\x18\x01\xe0\x41\x01\x42\x08\n\x06source\"\xaa\x03\n\x0eVertexRagStore\x12R\n\rrag_resources\x18\x04 \x03(\x0b\x32\x36.google.cloud.aiplatform.v1.VertexRagStore.RagResourceB\x03\xe0\x41\x01\x12$\n\x10similarity_top_k\x18\x02 \x01(\x05\x42\x05\x18\x01\xe0\x41\x01H\x00\x88\x01\x01\x12-\n\x19vector_distance_threshold\x18\x03 \x01(\x01\x42\x05\x18\x01\xe0\x41\x01H\x01\x88\x01\x01\x12Q\n\x14rag_retrieval_config\x18\x06 \x01(\x0b\x32..google.cloud.aiplatform.v1.RagRetrievalConfigB\x03\xe0\x41\x01\x1ai\n\x0bRagResource\x12?\n\nrag_corpus\x18\x01 \x01(\tB+\xe0\x41\x01\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x19\n\x0crag_file_ids\x18\x02 \x03(\tB\x03\xe0\x41\x01\x42\x13\n\x11_similarity_top_kB\x1c\n\x1a_vector_distance_threshold\"(\n\x0eVertexAISearch\x12\x16\n\tdatastore\x18\x01 \x01(\tB\x03\xe0\x41\x02\"m\n\x15GoogleSearchRetrieval\x12T\n\x18\x64ynamic_retrieval_config\x18\x02 \x01(\x0b\x32\x32.google.cloud.aiplatform.v1.DynamicRetrievalConfig\"\xca\x01\n\x16\x44ynamicRetrievalConfig\x12\x45\n\x04mode\x18\x01 \x01(\x0e\x32\x37.google.cloud.aiplatform.v1.DynamicRetrievalConfig.Mode\x12#\n\x11\x64ynamic_threshold\x18\x02 \x01(\x02\x42\x03\xe0\x41\x01H\x00\x88\x01\x01\".\n\x04Mode\x12\x14\n\x10MODE_UNSPECIFIED\x10\x00\x12\x10\n\x0cMODE_DYNAMIC\x10\x01\x42\x14\n\x12_dynamic_threshold\"e\n\nToolConfig\x12W\n\x17\x66unction_calling_config\x18\x01 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.FunctionCallingConfigB\x03\xe0\x41\x01\"\xc2\x01\n\x15\x46unctionCallingConfig\x12I\n\x04mode\x18\x01 \x01(\x0e\x32\x36.google.cloud.aiplatform.v1.FunctionCallingConfig.ModeB\x03\xe0\x41\x01\x12#\n\x16\x61llowed_function_names\x18\x02 \x03(\tB\x03\xe0\x41\x01\"9\n\x04Mode\x12\x14\n\x10MODE_UNSPECIFIED\x10\x00\x12\x08\n\x04\x41UTO\x10\x01\x12\x07\n\x03\x41NY\x10\x02\x12\x08\n\x04NONE\x10\x03\"\x8a\x02\n\x12RagRetrievalConfig\x12\x12\n\x05top_k\x18\x01 \x01(\x05\x42\x03\xe0\x41\x01\x12J\n\x06\x66ilter\x18\x03 \x01(\x0b\x32\x35.google.cloud.aiplatform.v1.RagRetrievalConfig.FilterB\x03\xe0\x41\x01\x1a\x93\x01\n\x06\x46ilter\x12(\n\x19vector_distance_threshold\x18\x03 \x01(\x01\x42\x03\xe0\x41\x01H\x00\x12*\n\x1bvector_similarity_threshold\x18\x04 \x01(\x01\x42\x03\xe0\x41\x01H\x00\x12\x1c\n\x0fmetadata_filter\x18\x02 \x01(\tB\x03\xe0\x41\x01\x42\x15\n\x13vector_db_thresholdB\xc7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\tToolProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
|
|
14
14
|
|
|
15
15
|
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
16
16
|
|
|
@@ -47,6 +47,8 @@ module Google
|
|
|
47
47
|
FunctionCall = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCall").msgclass
|
|
48
48
|
FunctionResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionResponse").msgclass
|
|
49
49
|
Retrieval = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Retrieval").msgclass
|
|
50
|
+
VertexRagStore = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.VertexRagStore").msgclass
|
|
51
|
+
VertexRagStore::RagResource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.VertexRagStore.RagResource").msgclass
|
|
50
52
|
VertexAISearch = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.VertexAISearch").msgclass
|
|
51
53
|
GoogleSearchRetrieval = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GoogleSearchRetrieval").msgclass
|
|
52
54
|
DynamicRetrievalConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DynamicRetrievalConfig").msgclass
|
|
@@ -54,6 +56,8 @@ module Google
|
|
|
54
56
|
ToolConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ToolConfig").msgclass
|
|
55
57
|
FunctionCallingConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCallingConfig").msgclass
|
|
56
58
|
FunctionCallingConfig::Mode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FunctionCallingConfig.Mode").enummodule
|
|
59
|
+
RagRetrievalConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagRetrievalConfig").msgclass
|
|
60
|
+
RagRetrievalConfig::Filter = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagRetrievalConfig.Filter").msgclass
|
|
57
61
|
end
|
|
58
62
|
end
|
|
59
63
|
end
|
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
# frozen_string_literal: true
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: google/cloud/aiplatform/v1/vertex_rag_data.proto
|
|
4
|
+
|
|
5
|
+
require 'google/protobuf'
|
|
6
|
+
|
|
7
|
+
require 'google/api/field_behavior_pb'
|
|
8
|
+
require 'google/api/resource_pb'
|
|
9
|
+
require 'google/cloud/aiplatform/v1/api_auth_pb'
|
|
10
|
+
require 'google/cloud/aiplatform/v1/io_pb'
|
|
11
|
+
require 'google/protobuf/timestamp_pb'
|
|
12
|
+
|
|
13
|
+
|
|
14
|
+
descriptor_data = "\n0google/cloud/aiplatform/v1/vertex_rag_data.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a)google/cloud/aiplatform/v1/api_auth.proto\x1a#google/cloud/aiplatform/v1/io.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xcf\x02\n\x17RagEmbeddingModelConfig\x12r\n\x1avertex_prediction_endpoint\x18\x01 \x01(\x0b\x32L.google.cloud.aiplatform.v1.RagEmbeddingModelConfig.VertexPredictionEndpointH\x00\x1a\xaf\x01\n\x18VertexPredictionEndpoint\x12<\n\x08\x65ndpoint\x18\x01 \x01(\tB*\xe0\x41\x02\xfa\x41$\n\"aiplatform.googleapis.com/Endpoint\x12\x36\n\x05model\x18\x02 \x01(\tB\'\xe0\x41\x03\xfa\x41!\n\x1f\x61iplatform.googleapis.com/Model\x12\x1d\n\x10model_version_id\x18\x03 \x01(\tB\x03\xe0\x41\x03\x42\x0e\n\x0cmodel_config\"\xa9\x04\n\x11RagVectorDbConfig\x12T\n\x0erag_managed_db\x18\x01 \x01(\x0b\x32:.google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDbH\x00\x12J\n\x08pinecone\x18\x03 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.RagVectorDbConfig.PineconeH\x00\x12`\n\x14vertex_vector_search\x18\x06 \x01(\x0b\x32@.google.cloud.aiplatform.v1.RagVectorDbConfig.VertexVectorSearchH\x00\x12\x35\n\x08\x61pi_auth\x18\x05 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ApiAuth\x12_\n\x1arag_embedding_model_config\x18\x07 \x01(\x0b\x32\x33.google.cloud.aiplatform.v1.RagEmbeddingModelConfigB\x06\xe0\x41\x01\xe0\x41\x05\x1a\x0e\n\x0cRagManagedDb\x1a\x1e\n\x08Pinecone\x12\x12\n\nindex_name\x18\x01 \x01(\t\x1a;\n\x12VertexVectorSearch\x12\x16\n\x0eindex_endpoint\x18\x01 \x01(\t\x12\r\n\x05index\x18\x02 \x01(\tB\x0b\n\tvector_db\"\xa0\x01\n\nFileStatus\x12@\n\x05state\x18\x01 \x01(\x0e\x32,.google.cloud.aiplatform.v1.FileStatus.StateB\x03\xe0\x41\x03\x12\x19\n\x0c\x65rror_status\x18\x02 \x01(\tB\x03\xe0\x41\x03\"5\n\x05State\x12\x15\n\x11STATE_UNSPECIFIED\x10\x00\x12\n\n\x06\x41\x43TIVE\x10\x01\x12\t\n\x05\x45RROR\x10\x02\"\xab\x01\n\x0c\x43orpusStatus\x12\x42\n\x05state\x18\x01 \x01(\x0e\x32..google.cloud.aiplatform.v1.CorpusStatus.StateB\x03\xe0\x41\x03\x12\x19\n\x0c\x65rror_status\x18\x02 \x01(\tB\x03\xe0\x41\x03\"<\n\x05State\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x0f\n\x0bINITIALIZED\x10\x01\x12\n\n\x06\x41\x43TIVE\x10\x02\x12\t\n\x05\x45RROR\x10\x03\"\xed\x03\n\tRagCorpus\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x0b\x63reate_time\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x44\n\rcorpus_status\x18\x08 \x01(\x0b\x32(.google.cloud.aiplatform.v1.CorpusStatusB\x03\xe0\x41\x03\x12Q\n\x10vector_db_config\x18\t \x01(\x0b\x32-.google.cloud.aiplatform.v1.RagVectorDbConfigB\x06\xe0\x41\x01\xe0\x41\x05H\x00:\x80\x01\xea\x41}\n#aiplatform.googleapis.com/RagCorpus\x12?projects/{project}/locations/{location}/ragCorpora/{rag_corpus}*\nragCorpora2\tragCorpusB\x10\n\x0e\x62\x61\x63kend_config\"\xdc\x06\n\x07RagFile\x12@\n\ngcs_source\x18\x08 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceB\x03\xe0\x41\x03H\x00\x12Q\n\x13google_drive_source\x18\t \x01(\x0b\x32-.google.cloud.aiplatform.v1.GoogleDriveSourceB\x03\xe0\x41\x03H\x00\x12S\n\x14\x64irect_upload_source\x18\n \x01(\x0b\x32..google.cloud.aiplatform.v1.DirectUploadSourceB\x03\xe0\x41\x03H\x00\x12?\n\x0cslack_source\x18\x0b \x01(\x0b\x32\'.google.cloud.aiplatform.v1.SlackSourceH\x00\x12=\n\x0bjira_source\x18\x0c \x01(\x0b\x32&.google.cloud.aiplatform.v1.JiraSourceH\x00\x12L\n\x13share_point_sources\x18\x0e \x01(\x0b\x32-.google.cloud.aiplatform.v1.SharePointSourcesH\x00\x12\x11\n\x04name\x18\x01 \x01(\tB\x03\xe0\x41\x03\x12\x19\n\x0c\x64isplay_name\x18\x02 \x01(\tB\x03\xe0\x41\x02\x12\x18\n\x0b\x64\x65scription\x18\x03 \x01(\tB\x03\xe0\x41\x01\x12\x34\n\x0b\x63reate_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12\x34\n\x0bupdate_time\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x03\xe0\x41\x03\x12@\n\x0b\x66ile_status\x18\r \x01(\x0b\x32&.google.cloud.aiplatform.v1.FileStatusB\x03\xe0\x41\x03:\x8f\x01\xea\x41\x8b\x01\n!aiplatform.googleapis.com/RagFile\x12Sprojects/{project}/locations/{location}/ragCorpora/{rag_corpus}/ragFiles/{rag_file}*\x08ragFiles2\x07ragFileB\x11\n\x0frag_file_source\"\xd4\x01\n\x15RagFileChunkingConfig\x12\x66\n\x15\x66ixed_length_chunking\x18\x03 \x01(\x0b\x32\x45.google.cloud.aiplatform.v1.RagFileChunkingConfig.FixedLengthChunkingH\x00\x1a@\n\x13\x46ixedLengthChunking\x12\x12\n\nchunk_size\x18\x01 \x01(\x05\x12\x15\n\rchunk_overlap\x18\x02 \x01(\x05\x42\x11\n\x0f\x63hunking_config\"r\n\x1bRagFileTransformationConfig\x12S\n\x18rag_file_chunking_config\x18\x01 \x01(\x0b\x32\x31.google.cloud.aiplatform.v1.RagFileChunkingConfig\"v\n\x13UploadRagFileConfig\x12_\n\x1erag_file_transformation_config\x18\x03 \x01(\x0b\x32\x37.google.cloud.aiplatform.v1.RagFileTransformationConfig\"\xd8\x05\n\x14ImportRagFilesConfig\x12;\n\ngcs_source\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.GcsSourceH\x00\x12L\n\x13google_drive_source\x18\x03 \x01(\x0b\x32-.google.cloud.aiplatform.v1.GoogleDriveSourceH\x00\x12?\n\x0cslack_source\x18\x06 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.SlackSourceH\x00\x12=\n\x0bjira_source\x18\x07 \x01(\x0b\x32&.google.cloud.aiplatform.v1.JiraSourceH\x00\x12L\n\x13share_point_sources\x18\r \x01(\x0b\x32-.google.cloud.aiplatform.v1.SharePointSourcesH\x00\x12R\n\x18partial_failure_gcs_sink\x18\x0b \x01(\x0b\x32*.google.cloud.aiplatform.v1.GcsDestinationB\x02\x18\x01H\x01\x12\\\n\x1dpartial_failure_bigquery_sink\x18\x0c \x01(\x0b\x32/.google.cloud.aiplatform.v1.BigQueryDestinationB\x02\x18\x01H\x01\x12_\n\x1erag_file_transformation_config\x18\x10 \x01(\x0b\x32\x37.google.cloud.aiplatform.v1.RagFileTransformationConfig\x12+\n\x1emax_embedding_requests_per_min\x18\x05 \x01(\x05\x42\x03\xe0\x41\x01\x42\x0f\n\rimport_sourceB\x16\n\x14partial_failure_sinkB\xd0\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x12VertexRagDataProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
|
|
15
|
+
|
|
16
|
+
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
17
|
+
|
|
18
|
+
begin
|
|
19
|
+
pool.add_serialized_file(descriptor_data)
|
|
20
|
+
rescue TypeError
|
|
21
|
+
# Compatibility code: will be removed in the next major version.
|
|
22
|
+
require 'google/protobuf/descriptor_pb'
|
|
23
|
+
parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
|
|
24
|
+
parsed.clear_dependency
|
|
25
|
+
serialized = parsed.class.encode(parsed)
|
|
26
|
+
file = pool.add_serialized_file(serialized)
|
|
27
|
+
warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
|
|
28
|
+
imports = [
|
|
29
|
+
["google.cloud.aiplatform.v1.ApiAuth", "google/cloud/aiplatform/v1/api_auth.proto"],
|
|
30
|
+
["google.protobuf.Timestamp", "google/protobuf/timestamp.proto"],
|
|
31
|
+
["google.cloud.aiplatform.v1.GcsSource", "google/cloud/aiplatform/v1/io.proto"],
|
|
32
|
+
]
|
|
33
|
+
imports.each do |type_name, expected_filename|
|
|
34
|
+
import_file = pool.lookup(type_name).file_descriptor
|
|
35
|
+
if import_file.name != expected_filename
|
|
36
|
+
warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
|
|
37
|
+
end
|
|
38
|
+
end
|
|
39
|
+
warn "Each proto file must use a consistent fully-qualified name."
|
|
40
|
+
warn "This will become an error in the next major version."
|
|
41
|
+
end
|
|
42
|
+
|
|
43
|
+
module Google
|
|
44
|
+
module Cloud
|
|
45
|
+
module AIPlatform
|
|
46
|
+
module V1
|
|
47
|
+
RagEmbeddingModelConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagEmbeddingModelConfig").msgclass
|
|
48
|
+
RagEmbeddingModelConfig::VertexPredictionEndpoint = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagEmbeddingModelConfig.VertexPredictionEndpoint").msgclass
|
|
49
|
+
RagVectorDbConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagVectorDbConfig").msgclass
|
|
50
|
+
RagVectorDbConfig::RagManagedDb = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagVectorDbConfig.RagManagedDb").msgclass
|
|
51
|
+
RagVectorDbConfig::Pinecone = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagVectorDbConfig.Pinecone").msgclass
|
|
52
|
+
RagVectorDbConfig::VertexVectorSearch = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagVectorDbConfig.VertexVectorSearch").msgclass
|
|
53
|
+
FileStatus = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FileStatus").msgclass
|
|
54
|
+
FileStatus::State = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.FileStatus.State").enummodule
|
|
55
|
+
CorpusStatus = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CorpusStatus").msgclass
|
|
56
|
+
CorpusStatus::State = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CorpusStatus.State").enummodule
|
|
57
|
+
RagCorpus = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagCorpus").msgclass
|
|
58
|
+
RagFile = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagFile").msgclass
|
|
59
|
+
RagFileChunkingConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagFileChunkingConfig").msgclass
|
|
60
|
+
RagFileChunkingConfig::FixedLengthChunking = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagFileChunkingConfig.FixedLengthChunking").msgclass
|
|
61
|
+
RagFileTransformationConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagFileTransformationConfig").msgclass
|
|
62
|
+
UploadRagFileConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UploadRagFileConfig").msgclass
|
|
63
|
+
ImportRagFilesConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportRagFilesConfig").msgclass
|
|
64
|
+
end
|
|
65
|
+
end
|
|
66
|
+
end
|
|
67
|
+
end
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
# frozen_string_literal: true
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: google/cloud/aiplatform/v1/vertex_rag_data_service.proto
|
|
4
|
+
|
|
5
|
+
require 'google/protobuf'
|
|
6
|
+
|
|
7
|
+
require 'google/api/annotations_pb'
|
|
8
|
+
require 'google/api/client_pb'
|
|
9
|
+
require 'google/api/field_behavior_pb'
|
|
10
|
+
require 'google/api/resource_pb'
|
|
11
|
+
require 'google/cloud/aiplatform/v1/operation_pb'
|
|
12
|
+
require 'google/cloud/aiplatform/v1/vertex_rag_data_pb'
|
|
13
|
+
require 'google/longrunning/operations_pb'
|
|
14
|
+
require 'google/protobuf/empty_pb'
|
|
15
|
+
require 'google/rpc/status_pb'
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
descriptor_data = "\n8google/cloud/aiplatform/v1/vertex_rag_data_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a*google/cloud/aiplatform/v1/operation.proto\x1a\x30google/cloud/aiplatform/v1/vertex_rag_data.proto\x1a#google/longrunning/operations.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a\x17google/rpc/status.proto\"\x93\x01\n\x16\x43reateRagCorpusRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12>\n\nrag_corpus\x18\x02 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpusB\x03\xe0\x41\x02\"P\n\x13GetRagCorpusRequest\x12\x39\n\x04name\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\"\x83\x01\n\x15ListRagCorporaRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x16\n\tpage_size\x18\x02 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x03 \x01(\tB\x03\xe0\x41\x01\"m\n\x16ListRagCorporaResponse\x12:\n\x0brag_corpora\x18\x01 \x03(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpus\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"g\n\x16\x44\x65leteRagCorpusRequest\x12\x39\n\x04name\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x12\n\x05\x66orce\x18\x02 \x01(\x08\x42\x03\xe0\x41\x01\"\xe5\x01\n\x14UploadRagFileRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12:\n\x08rag_file\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RagFileB\x03\xe0\x41\x02\x12T\n\x16upload_rag_file_config\x18\x05 \x01(\x0b\x32/.google.cloud.aiplatform.v1.UploadRagFileConfigB\x03\xe0\x41\x02\"\x7f\n\x15UploadRagFileResponse\x12\x37\n\x08rag_file\x18\x01 \x01(\x0b\x32#.google.cloud.aiplatform.v1.RagFileH\x00\x12#\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x12.google.rpc.StatusH\x00\x42\x08\n\x06result\"\xac\x01\n\x15ImportRagFilesRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12V\n\x17import_rag_files_config\x18\x02 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ImportRagFilesConfigB\x03\xe0\x41\x02\"\xe3\x01\n\x16ImportRagFilesResponse\x12#\n\x19partial_failures_gcs_path\x18\x04 \x01(\tH\x00\x12)\n\x1fpartial_failures_bigquery_table\x18\x05 \x01(\tH\x00\x12 \n\x18imported_rag_files_count\x18\x01 \x01(\x03\x12\x1e\n\x16\x66\x61iled_rag_files_count\x18\x02 \x01(\x03\x12\x1f\n\x17skipped_rag_files_count\x18\x03 \x01(\x03\x42\x16\n\x14partial_failure_sink\"L\n\x11GetRagFileRequest\x12\x37\n\x04name\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!aiplatform.googleapis.com/RagFile\"\x83\x01\n\x13ListRagFilesRequest\x12;\n\x06parent\x18\x01 \x01(\tB+\xe0\x41\x02\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x16\n\tpage_size\x18\x02 \x01(\x05\x42\x03\xe0\x41\x01\x12\x17\n\npage_token\x18\x03 \x01(\tB\x03\xe0\x41\x01\"g\n\x14ListRagFilesResponse\x12\x36\n\trag_files\x18\x01 \x03(\x0b\x32#.google.cloud.aiplatform.v1.RagFile\x12\x17\n\x0fnext_page_token\x18\x02 \x01(\t\"O\n\x14\x44\x65leteRagFileRequest\x12\x37\n\x04name\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!aiplatform.googleapis.com/RagFile\"r\n CreateRagCorpusOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"X\n\x16UpdateRagCorpusRequest\x12>\n\nrag_corpus\x18\x01 \x01(\x0b\x32%.google.cloud.aiplatform.v1.RagCorpusB\x03\xe0\x41\x02\"r\n UpdateRagCorpusOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\"\xfd\x01\n\x1fImportRagFilesOperationMetadata\x12N\n\x10generic_metadata\x18\x01 \x01(\x0b\x32\x34.google.cloud.aiplatform.v1.GenericOperationMetadata\x12\x15\n\rrag_corpus_id\x18\x02 \x01(\x03\x12V\n\x17import_rag_files_config\x18\x03 \x01(\x0b\x32\x30.google.cloud.aiplatform.v1.ImportRagFilesConfigB\x03\xe0\x41\x03\x12\x1b\n\x13progress_percentage\x18\x04 \x01(\x05\x32\xcf\x11\n\x14VertexRagDataService\x12\xed\x01\n\x0f\x43reateRagCorpus\x12\x32.google.cloud.aiplatform.v1.CreateRagCorpusRequest\x1a\x1d.google.longrunning.Operation\"\x86\x01\xca\x41-\n\tRagCorpus\x12 CreateRagCorpusOperationMetadata\xda\x41\x11parent,rag_corpus\x82\xd3\xe4\x93\x02<\"./v1/{parent=projects/*/locations/*}/ragCorpora:\nrag_corpus\x12\xf1\x01\n\x0fUpdateRagCorpus\x12\x32.google.cloud.aiplatform.v1.UpdateRagCorpusRequest\x1a\x1d.google.longrunning.Operation\"\x8a\x01\xca\x41-\n\tRagCorpus\x12 UpdateRagCorpusOperationMetadata\xda\x41\nrag_corpus\x82\xd3\xe4\x93\x02G29/v1/{rag_corpus.name=projects/*/locations/*/ragCorpora/*}:\nrag_corpus\x12\xa5\x01\n\x0cGetRagCorpus\x12/.google.cloud.aiplatform.v1.GetRagCorpusRequest\x1a%.google.cloud.aiplatform.v1.RagCorpus\"=\xda\x41\x04name\x82\xd3\xe4\x93\x02\x30\x12./v1/{name=projects/*/locations/*/ragCorpora/*}\x12\xb8\x01\n\x0eListRagCorpora\x12\x31.google.cloud.aiplatform.v1.ListRagCorporaRequest\x1a\x32.google.cloud.aiplatform.v1.ListRagCorporaResponse\"?\xda\x41\x06parent\x82\xd3\xe4\x93\x02\x30\x12./v1/{parent=projects/*/locations/*}/ragCorpora\x12\xd6\x01\n\x0f\x44\x65leteRagCorpus\x12\x32.google.cloud.aiplatform.v1.DeleteRagCorpusRequest\x1a\x1d.google.longrunning.Operation\"p\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02\x30*./v1/{name=projects/*/locations/*/ragCorpora/*}\x12\xea\x01\n\rUploadRagFile\x12\x30.google.cloud.aiplatform.v1.UploadRagFileRequest\x1a\x31.google.cloud.aiplatform.v1.UploadRagFileResponse\"t\xda\x41&parent,rag_file,upload_rag_file_config\x82\xd3\xe4\x93\x02\x45\"@/v1/{parent=projects/*/locations/*/ragCorpora/*}/ragFiles:upload:\x01*\x12\x8d\x02\n\x0eImportRagFiles\x12\x31.google.cloud.aiplatform.v1.ImportRagFilesRequest\x1a\x1d.google.longrunning.Operation\"\xa8\x01\xca\x41\x39\n\x16ImportRagFilesResponse\x12\x1fImportRagFilesOperationMetadata\xda\x41\x1eparent,import_rag_files_config\x82\xd3\xe4\x93\x02\x45\"@/v1/{parent=projects/*/locations/*/ragCorpora/*}/ragFiles:import:\x01*\x12\xaa\x01\n\nGetRagFile\x12-.google.cloud.aiplatform.v1.GetRagFileRequest\x1a#.google.cloud.aiplatform.v1.RagFile\"H\xda\x41\x04name\x82\xd3\xe4\x93\x02;\x12\x39/v1/{name=projects/*/locations/*/ragCorpora/*/ragFiles/*}\x12\xbd\x01\n\x0cListRagFiles\x12/.google.cloud.aiplatform.v1.ListRagFilesRequest\x1a\x30.google.cloud.aiplatform.v1.ListRagFilesResponse\"J\xda\x41\x06parent\x82\xd3\xe4\x93\x02;\x12\x39/v1/{parent=projects/*/locations/*/ragCorpora/*}/ragFiles\x12\xdd\x01\n\rDeleteRagFile\x12\x30.google.cloud.aiplatform.v1.DeleteRagFileRequest\x1a\x1d.google.longrunning.Operation\"{\xca\x41\x30\n\x15google.protobuf.Empty\x12\x17\x44\x65leteOperationMetadata\xda\x41\x04name\x82\xd3\xe4\x93\x02;*9/v1/{name=projects/*/locations/*/ragCorpora/*/ragFiles/*}\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd7\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x19VertexRagDataServiceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
|
|
19
|
+
|
|
20
|
+
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
21
|
+
|
|
22
|
+
begin
|
|
23
|
+
pool.add_serialized_file(descriptor_data)
|
|
24
|
+
rescue TypeError
|
|
25
|
+
# Compatibility code: will be removed in the next major version.
|
|
26
|
+
require 'google/protobuf/descriptor_pb'
|
|
27
|
+
parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
|
|
28
|
+
parsed.clear_dependency
|
|
29
|
+
serialized = parsed.class.encode(parsed)
|
|
30
|
+
file = pool.add_serialized_file(serialized)
|
|
31
|
+
warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
|
|
32
|
+
imports = [
|
|
33
|
+
["google.cloud.aiplatform.v1.RagCorpus", "google/cloud/aiplatform/v1/vertex_rag_data.proto"],
|
|
34
|
+
["google.rpc.Status", "google/rpc/status.proto"],
|
|
35
|
+
["google.cloud.aiplatform.v1.GenericOperationMetadata", "google/cloud/aiplatform/v1/operation.proto"],
|
|
36
|
+
]
|
|
37
|
+
imports.each do |type_name, expected_filename|
|
|
38
|
+
import_file = pool.lookup(type_name).file_descriptor
|
|
39
|
+
if import_file.name != expected_filename
|
|
40
|
+
warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
|
|
41
|
+
end
|
|
42
|
+
end
|
|
43
|
+
warn "Each proto file must use a consistent fully-qualified name."
|
|
44
|
+
warn "This will become an error in the next major version."
|
|
45
|
+
end
|
|
46
|
+
|
|
47
|
+
module Google
|
|
48
|
+
module Cloud
|
|
49
|
+
module AIPlatform
|
|
50
|
+
module V1
|
|
51
|
+
CreateRagCorpusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CreateRagCorpusRequest").msgclass
|
|
52
|
+
GetRagCorpusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetRagCorpusRequest").msgclass
|
|
53
|
+
ListRagCorporaRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListRagCorporaRequest").msgclass
|
|
54
|
+
ListRagCorporaResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListRagCorporaResponse").msgclass
|
|
55
|
+
DeleteRagCorpusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeleteRagCorpusRequest").msgclass
|
|
56
|
+
UploadRagFileRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UploadRagFileRequest").msgclass
|
|
57
|
+
UploadRagFileResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UploadRagFileResponse").msgclass
|
|
58
|
+
ImportRagFilesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportRagFilesRequest").msgclass
|
|
59
|
+
ImportRagFilesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportRagFilesResponse").msgclass
|
|
60
|
+
GetRagFileRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.GetRagFileRequest").msgclass
|
|
61
|
+
ListRagFilesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListRagFilesRequest").msgclass
|
|
62
|
+
ListRagFilesResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ListRagFilesResponse").msgclass
|
|
63
|
+
DeleteRagFileRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.DeleteRagFileRequest").msgclass
|
|
64
|
+
CreateRagCorpusOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CreateRagCorpusOperationMetadata").msgclass
|
|
65
|
+
UpdateRagCorpusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateRagCorpusRequest").msgclass
|
|
66
|
+
UpdateRagCorpusOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.UpdateRagCorpusOperationMetadata").msgclass
|
|
67
|
+
ImportRagFilesOperationMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ImportRagFilesOperationMetadata").msgclass
|
|
68
|
+
end
|
|
69
|
+
end
|
|
70
|
+
end
|
|
71
|
+
end
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
2
|
+
# Source: google/cloud/aiplatform/v1/vertex_rag_data_service.proto for package 'Google.Cloud.AIPlatform.V1'
|
|
3
|
+
# Original file comments:
|
|
4
|
+
# Copyright 2024 Google LLC
|
|
5
|
+
#
|
|
6
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
7
|
+
# you may not use this file except in compliance with the License.
|
|
8
|
+
# You may obtain a copy of the License at
|
|
9
|
+
#
|
|
10
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
11
|
+
#
|
|
12
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
13
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
14
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
15
|
+
# See the License for the specific language governing permissions and
|
|
16
|
+
# limitations under the License.
|
|
17
|
+
#
|
|
18
|
+
|
|
19
|
+
require 'grpc'
|
|
20
|
+
require 'google/cloud/aiplatform/v1/vertex_rag_data_service_pb'
|
|
21
|
+
|
|
22
|
+
module Google
|
|
23
|
+
module Cloud
|
|
24
|
+
module AIPlatform
|
|
25
|
+
module V1
|
|
26
|
+
module VertexRagDataService
|
|
27
|
+
# A service for managing user data for RAG.
|
|
28
|
+
class Service
|
|
29
|
+
|
|
30
|
+
include ::GRPC::GenericService
|
|
31
|
+
|
|
32
|
+
self.marshal_class_method = :encode
|
|
33
|
+
self.unmarshal_class_method = :decode
|
|
34
|
+
self.service_name = 'google.cloud.aiplatform.v1.VertexRagDataService'
|
|
35
|
+
|
|
36
|
+
# Creates a RagCorpus.
|
|
37
|
+
rpc :CreateRagCorpus, ::Google::Cloud::AIPlatform::V1::CreateRagCorpusRequest, ::Google::Longrunning::Operation
|
|
38
|
+
# Updates a RagCorpus.
|
|
39
|
+
rpc :UpdateRagCorpus, ::Google::Cloud::AIPlatform::V1::UpdateRagCorpusRequest, ::Google::Longrunning::Operation
|
|
40
|
+
# Gets a RagCorpus.
|
|
41
|
+
rpc :GetRagCorpus, ::Google::Cloud::AIPlatform::V1::GetRagCorpusRequest, ::Google::Cloud::AIPlatform::V1::RagCorpus
|
|
42
|
+
# Lists RagCorpora in a Location.
|
|
43
|
+
rpc :ListRagCorpora, ::Google::Cloud::AIPlatform::V1::ListRagCorporaRequest, ::Google::Cloud::AIPlatform::V1::ListRagCorporaResponse
|
|
44
|
+
# Deletes a RagCorpus.
|
|
45
|
+
rpc :DeleteRagCorpus, ::Google::Cloud::AIPlatform::V1::DeleteRagCorpusRequest, ::Google::Longrunning::Operation
|
|
46
|
+
# Upload a file into a RagCorpus.
|
|
47
|
+
rpc :UploadRagFile, ::Google::Cloud::AIPlatform::V1::UploadRagFileRequest, ::Google::Cloud::AIPlatform::V1::UploadRagFileResponse
|
|
48
|
+
# Import files from Google Cloud Storage or Google Drive into a RagCorpus.
|
|
49
|
+
rpc :ImportRagFiles, ::Google::Cloud::AIPlatform::V1::ImportRagFilesRequest, ::Google::Longrunning::Operation
|
|
50
|
+
# Gets a RagFile.
|
|
51
|
+
rpc :GetRagFile, ::Google::Cloud::AIPlatform::V1::GetRagFileRequest, ::Google::Cloud::AIPlatform::V1::RagFile
|
|
52
|
+
# Lists RagFiles in a RagCorpus.
|
|
53
|
+
rpc :ListRagFiles, ::Google::Cloud::AIPlatform::V1::ListRagFilesRequest, ::Google::Cloud::AIPlatform::V1::ListRagFilesResponse
|
|
54
|
+
# Deletes a RagFile.
|
|
55
|
+
rpc :DeleteRagFile, ::Google::Cloud::AIPlatform::V1::DeleteRagFileRequest, ::Google::Longrunning::Operation
|
|
56
|
+
end
|
|
57
|
+
|
|
58
|
+
Stub = Service.rpc_stub_class
|
|
59
|
+
end
|
|
60
|
+
end
|
|
61
|
+
end
|
|
62
|
+
end
|
|
63
|
+
end
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
# frozen_string_literal: true
|
|
2
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
3
|
+
# source: google/cloud/aiplatform/v1/vertex_rag_service.proto
|
|
4
|
+
|
|
5
|
+
require 'google/protobuf'
|
|
6
|
+
|
|
7
|
+
require 'google/api/annotations_pb'
|
|
8
|
+
require 'google/api/client_pb'
|
|
9
|
+
require 'google/api/field_behavior_pb'
|
|
10
|
+
require 'google/api/resource_pb'
|
|
11
|
+
require 'google/cloud/aiplatform/v1/content_pb'
|
|
12
|
+
require 'google/cloud/aiplatform/v1/tool_pb'
|
|
13
|
+
|
|
14
|
+
|
|
15
|
+
descriptor_data = "\n3google/cloud/aiplatform/v1/vertex_rag_service.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x17google/api/client.proto\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a(google/cloud/aiplatform/v1/content.proto\x1a%google/cloud/aiplatform/v1/tool.proto\"{\n\x08RagQuery\x12\x13\n\x04text\x18\x01 \x01(\tB\x03\xe0\x41\x01H\x00\x12Q\n\x14rag_retrieval_config\x18\x06 \x01(\x0b\x32..google.cloud.aiplatform.v1.RagRetrievalConfigB\x03\xe0\x41\x01\x42\x07\n\x05query\"\xb4\x04\n\x17RetrieveContextsRequest\x12^\n\x10vertex_rag_store\x18\x02 \x01(\x0b\x32\x42.google.cloud.aiplatform.v1.RetrieveContextsRequest.VertexRagStoreH\x00\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12\x38\n\x05query\x18\x03 \x01(\x0b\x32$.google.cloud.aiplatform.v1.RagQueryB\x03\xe0\x41\x02\x1a\xb4\x02\n\x0eVertexRagStore\x12j\n\rrag_resources\x18\x03 \x03(\x0b\x32N.google.cloud.aiplatform.v1.RetrieveContextsRequest.VertexRagStore.RagResourceB\x03\xe0\x41\x01\x12-\n\x19vector_distance_threshold\x18\x02 \x01(\x01\x42\x05\x18\x01\xe0\x41\x01H\x00\x88\x01\x01\x1ai\n\x0bRagResource\x12?\n\nrag_corpus\x18\x01 \x01(\tB+\xe0\x41\x01\xfa\x41%\n#aiplatform.googleapis.com/RagCorpus\x12\x19\n\x0crag_file_ids\x18\x02 \x03(\tB\x03\xe0\x41\x01\x42\x1c\n\x1a_vector_distance_thresholdB\r\n\x0b\x64\x61ta_source\"\xb8\x01\n\x0bRagContexts\x12\x41\n\x08\x63ontexts\x18\x01 \x03(\x0b\x32/.google.cloud.aiplatform.v1.RagContexts.Context\x1a\x66\n\x07\x43ontext\x12\x12\n\nsource_uri\x18\x01 \x01(\t\x12\x1b\n\x13source_display_name\x18\x05 \x01(\t\x12\x0c\n\x04text\x18\x02 \x01(\t\x12\x12\n\x05score\x18\x06 \x01(\x01H\x00\x88\x01\x01\x42\x08\n\x06_score\"U\n\x18RetrieveContextsResponse\x12\x39\n\x08\x63ontexts\x18\x01 \x01(\x0b\x32\'.google.cloud.aiplatform.v1.RagContexts\"\xee\x02\n\x14\x41ugmentPromptRequest\x12K\n\x10vertex_rag_store\x18\x04 \x01(\x0b\x32*.google.cloud.aiplatform.v1.VertexRagStoreB\x03\xe0\x41\x01H\x00\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12:\n\x08\x63ontents\x18\x02 \x03(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x01\x12J\n\x05model\x18\x03 \x01(\x0b\x32\x36.google.cloud.aiplatform.v1.AugmentPromptRequest.ModelB\x03\xe0\x41\x01\x1a\x37\n\x05Model\x12\x12\n\x05model\x18\x01 \x01(\tB\x03\xe0\x41\x01\x12\x1a\n\rmodel_version\x18\x02 \x01(\tB\x03\xe0\x41\x01\x42\r\n\x0b\x64\x61ta_source\"\x87\x01\n\x15\x41ugmentPromptResponse\x12=\n\x10\x61ugmented_prompt\x18\x01 \x03(\x0b\x32#.google.cloud.aiplatform.v1.Content\x12/\n\x05\x66\x61\x63ts\x18\x02 \x03(\x0b\x32 .google.cloud.aiplatform.v1.Fact\"\xe2\x02\n\x19\x43orroborateContentRequest\x12\x39\n\x06parent\x18\x01 \x01(\tB)\xe0\x41\x02\xfa\x41#\n!locations.googleapis.com/Location\x12>\n\x07\x63ontent\x18\x02 \x01(\x0b\x32#.google.cloud.aiplatform.v1.ContentB\x03\xe0\x41\x01H\x00\x88\x01\x01\x12\x34\n\x05\x66\x61\x63ts\x18\x03 \x03(\x0b\x32 .google.cloud.aiplatform.v1.FactB\x03\xe0\x41\x01\x12Y\n\nparameters\x18\x04 \x01(\x0b\x32@.google.cloud.aiplatform.v1.CorroborateContentRequest.ParametersB\x03\xe0\x41\x01\x1a-\n\nParameters\x12\x1f\n\x12\x63itation_threshold\x18\x01 \x01(\x01\x42\x03\xe0\x41\x01\x42\n\n\x08_content\"\x89\x01\n\x1a\x43orroborateContentResponse\x12 \n\x13\x63orroboration_score\x18\x01 \x01(\x02H\x00\x88\x01\x01\x12\x31\n\x06\x63laims\x18\x02 \x03(\x0b\x32!.google.cloud.aiplatform.v1.ClaimB\x16\n\x14_corroboration_score\"\xd2\x01\n\x04\x46\x61\x63t\x12\x12\n\x05query\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05title\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x10\n\x03uri\x18\x03 \x01(\tH\x02\x88\x01\x01\x12\x14\n\x07summary\x18\x04 \x01(\tH\x03\x88\x01\x01\x12 \n\x0fvector_distance\x18\x05 \x01(\x01\x42\x02\x18\x01H\x04\x88\x01\x01\x12\x12\n\x05score\x18\x06 \x01(\x01H\x05\x88\x01\x01\x42\x08\n\x06_queryB\x08\n\x06_titleB\x06\n\x04_uriB\n\n\x08_summaryB\x12\n\x10_vector_distanceB\x08\n\x06_score\"\x8b\x01\n\x05\x43laim\x12\x18\n\x0bstart_index\x18\x01 \x01(\x05H\x00\x88\x01\x01\x12\x16\n\tend_index\x18\x02 \x01(\x05H\x01\x88\x01\x01\x12\x14\n\x0c\x66\x61\x63t_indexes\x18\x03 \x03(\x05\x12\x12\n\x05score\x18\x04 \x01(\x02H\x02\x88\x01\x01\x42\x0e\n\x0c_start_indexB\x0c\n\n_end_indexB\x08\n\x06_score2\xe6\x05\n\x10VertexRagService\x12\xcd\x01\n\x10RetrieveContexts\x12\x33.google.cloud.aiplatform.v1.RetrieveContextsRequest\x1a\x34.google.cloud.aiplatform.v1.RetrieveContextsResponse\"N\xda\x41\x0cparent,query\x82\xd3\xe4\x93\x02\x39\"4/v1/{parent=projects/*/locations/*}:retrieveContexts:\x01*\x12\xd2\x01\n\rAugmentPrompt\x12\x30.google.cloud.aiplatform.v1.AugmentPromptRequest\x1a\x31.google.cloud.aiplatform.v1.AugmentPromptResponse\"\\\xda\x41\x1dparent,model,vertex_rag_store\x82\xd3\xe4\x93\x02\x36\"1/v1/{parent=projects/*/locations/*}:augmentPrompt:\x01*\x12\xdd\x01\n\x12\x43orroborateContent\x12\x35.google.cloud.aiplatform.v1.CorroborateContentRequest\x1a\x36.google.cloud.aiplatform.v1.CorroborateContentResponse\"X\xda\x41\x14parent,content,facts\x82\xd3\xe4\x93\x02;\"6/v1/{parent=projects/*/locations/*}:corroborateContent:\x01*\x1aM\xca\x41\x19\x61iplatform.googleapis.com\xd2\x41.https://www.googleapis.com/auth/cloud-platformB\xd3\x01\n\x1e\x63om.google.cloud.aiplatform.v1B\x15VertexRagServiceProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"
|
|
16
|
+
|
|
17
|
+
pool = Google::Protobuf::DescriptorPool.generated_pool
|
|
18
|
+
|
|
19
|
+
begin
|
|
20
|
+
pool.add_serialized_file(descriptor_data)
|
|
21
|
+
rescue TypeError
|
|
22
|
+
# Compatibility code: will be removed in the next major version.
|
|
23
|
+
require 'google/protobuf/descriptor_pb'
|
|
24
|
+
parsed = Google::Protobuf::FileDescriptorProto.decode(descriptor_data)
|
|
25
|
+
parsed.clear_dependency
|
|
26
|
+
serialized = parsed.class.encode(parsed)
|
|
27
|
+
file = pool.add_serialized_file(serialized)
|
|
28
|
+
warn "Warning: Protobuf detected an import path issue while loading generated file #{__FILE__}"
|
|
29
|
+
imports = [
|
|
30
|
+
["google.cloud.aiplatform.v1.RagRetrievalConfig", "google/cloud/aiplatform/v1/tool.proto"],
|
|
31
|
+
["google.cloud.aiplatform.v1.Content", "google/cloud/aiplatform/v1/content.proto"],
|
|
32
|
+
]
|
|
33
|
+
imports.each do |type_name, expected_filename|
|
|
34
|
+
import_file = pool.lookup(type_name).file_descriptor
|
|
35
|
+
if import_file.name != expected_filename
|
|
36
|
+
warn "- #{file.name} imports #{expected_filename}, but that import was loaded as #{import_file.name}"
|
|
37
|
+
end
|
|
38
|
+
end
|
|
39
|
+
warn "Each proto file must use a consistent fully-qualified name."
|
|
40
|
+
warn "This will become an error in the next major version."
|
|
41
|
+
end
|
|
42
|
+
|
|
43
|
+
module Google
|
|
44
|
+
module Cloud
|
|
45
|
+
module AIPlatform
|
|
46
|
+
module V1
|
|
47
|
+
RagQuery = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagQuery").msgclass
|
|
48
|
+
RetrieveContextsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrieveContextsRequest").msgclass
|
|
49
|
+
RetrieveContextsRequest::VertexRagStore = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrieveContextsRequest.VertexRagStore").msgclass
|
|
50
|
+
RetrieveContextsRequest::VertexRagStore::RagResource = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrieveContextsRequest.VertexRagStore.RagResource").msgclass
|
|
51
|
+
RagContexts = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagContexts").msgclass
|
|
52
|
+
RagContexts::Context = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RagContexts.Context").msgclass
|
|
53
|
+
RetrieveContextsResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.RetrieveContextsResponse").msgclass
|
|
54
|
+
AugmentPromptRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.AugmentPromptRequest").msgclass
|
|
55
|
+
AugmentPromptRequest::Model = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.AugmentPromptRequest.Model").msgclass
|
|
56
|
+
AugmentPromptResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.AugmentPromptResponse").msgclass
|
|
57
|
+
CorroborateContentRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CorroborateContentRequest").msgclass
|
|
58
|
+
CorroborateContentRequest::Parameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CorroborateContentRequest.Parameters").msgclass
|
|
59
|
+
CorroborateContentResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CorroborateContentResponse").msgclass
|
|
60
|
+
Fact = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Fact").msgclass
|
|
61
|
+
Claim = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Claim").msgclass
|
|
62
|
+
end
|
|
63
|
+
end
|
|
64
|
+
end
|
|
65
|
+
end
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
# Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
2
|
+
# Source: google/cloud/aiplatform/v1/vertex_rag_service.proto for package 'Google.Cloud.AIPlatform.V1'
|
|
3
|
+
# Original file comments:
|
|
4
|
+
# Copyright 2024 Google LLC
|
|
5
|
+
#
|
|
6
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
7
|
+
# you may not use this file except in compliance with the License.
|
|
8
|
+
# You may obtain a copy of the License at
|
|
9
|
+
#
|
|
10
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
11
|
+
#
|
|
12
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
13
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
14
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
15
|
+
# See the License for the specific language governing permissions and
|
|
16
|
+
# limitations under the License.
|
|
17
|
+
#
|
|
18
|
+
|
|
19
|
+
require 'grpc'
|
|
20
|
+
require 'google/cloud/aiplatform/v1/vertex_rag_service_pb'
|
|
21
|
+
|
|
22
|
+
module Google
|
|
23
|
+
module Cloud
|
|
24
|
+
module AIPlatform
|
|
25
|
+
module V1
|
|
26
|
+
module VertexRagService
|
|
27
|
+
# A service for retrieving relevant contexts.
|
|
28
|
+
class Service
|
|
29
|
+
|
|
30
|
+
include ::GRPC::GenericService
|
|
31
|
+
|
|
32
|
+
self.marshal_class_method = :encode
|
|
33
|
+
self.unmarshal_class_method = :decode
|
|
34
|
+
self.service_name = 'google.cloud.aiplatform.v1.VertexRagService'
|
|
35
|
+
|
|
36
|
+
# Retrieves relevant contexts for a query.
|
|
37
|
+
rpc :RetrieveContexts, ::Google::Cloud::AIPlatform::V1::RetrieveContextsRequest, ::Google::Cloud::AIPlatform::V1::RetrieveContextsResponse
|
|
38
|
+
# Given an input prompt, it returns augmented prompt from vertex rag store
|
|
39
|
+
# to guide LLM towards generating grounded responses.
|
|
40
|
+
rpc :AugmentPrompt, ::Google::Cloud::AIPlatform::V1::AugmentPromptRequest, ::Google::Cloud::AIPlatform::V1::AugmentPromptResponse
|
|
41
|
+
# Given an input text, it returns a score that evaluates the factuality of
|
|
42
|
+
# the text. It also extracts and returns claims from the text and provides
|
|
43
|
+
# supporting facts.
|
|
44
|
+
rpc :CorroborateContent, ::Google::Cloud::AIPlatform::V1::CorroborateContentRequest, ::Google::Cloud::AIPlatform::V1::CorroborateContentResponse
|
|
45
|
+
end
|
|
46
|
+
|
|
47
|
+
Stub = Service.rpc_stub_class
|
|
48
|
+
end
|
|
49
|
+
end
|
|
50
|
+
end
|
|
51
|
+
end
|
|
52
|
+
end
|