feat: add Gen AI logging public preview API

Google APIs · copybara-github · commit 3b446a9493c4 · 2025-04-02T11:45:33.000-07:00
PiperOrigin-RevId: 743220774
diff --git a/google/cloud/aiplatform/v1beta1/endpoint.proto b/google/cloud/aiplatform/v1beta1/endpoint.proto
@@ -359,6 +359,23 @@ message PredictRequestResponseLoggingConfig {
   // most special characters will become underscores). If no table name is
   // given, a new table will be created with name `request_response_logging`
   BigQueryDestination bigquery_destination = 3;
+
+  // Output only. The schema version used in creating the BigQuery table for the
+  // request response logging. The versions are "v1" and "v2". The current
+  // default version is "v1".
+  string request_response_logging_schema_version = 4
+      [(google.api.field_behavior) = OUTPUT_ONLY];
+
+  // This field is used for large models. If true, in addition to the
+  // original large model logs, logs will be converted in OTel schema format,
+  // and saved in otel_log column. Default value is false.
+  bool enable_otel_logging = 6;
+}
+
+// This message contains configs of a publisher model.
+message PublisherModelConfig {
+  // The prediction request/response logging config.
+  PredictRequestResponseLoggingConfig logging_config = 3;
 }
 
 // Configurations (e.g. inference timeout) that are applied on your endpoints.
diff --git a/google/cloud/aiplatform/v1beta1/endpoint_service.proto b/google/cloud/aiplatform/v1beta1/endpoint_service.proto
@@ -154,6 +154,30 @@ service EndpointService {
       metadata_type: "MutateDeployedModelOperationMetadata"
     };
   }
+
+  // Sets (creates or updates) configs of publisher models. For example, sets
+  // the request/response logging config.
+  rpc SetPublisherModelConfig(SetPublisherModelConfigRequest)
+      returns (google.longrunning.Operation) {
+    option (google.api.http) = {
+      post: "/v1beta1/{name=projects/*/locations/*/publishers/*/models/*}:setPublisherModelConfig"
+      body: "*"
+    };
+    option (google.api.method_signature) = "name,publisher_model_config";
+    option (google.longrunning.operation_info) = {
+      response_type: "PublisherModelConfig"
+      metadata_type: "SetPublisherModelConfigOperationMetadata"
+    };
+  }
+
+  // Fetches the configs of publisher models.
+  rpc FetchPublisherModelConfig(FetchPublisherModelConfigRequest)
+      returns (PublisherModelConfig) {
+    option (google.api.http) = {
+      get: "/v1beta1/{name=projects/*/locations/*/publishers/*/models/*}:fetchPublisherModelConfig"
+    };
+    option (google.api.method_signature) = "name";
+  }
 }
 
 // Request message for
@@ -393,6 +417,43 @@ message UndeployModelRequest {
 // [EndpointService.UndeployModel][google.cloud.aiplatform.v1beta1.EndpointService.UndeployModel].
 message UndeployModelResponse {}
 
+// Request message for
+// [EndpointService.SetPublisherModelConfig][google.cloud.aiplatform.v1beta1.EndpointService.SetPublisherModelConfig].
+message SetPublisherModelConfigRequest {
+  // Required. The name of the publisher model, in the format of
+  // `projects/{project}/locations/{location}/publishers/{publisher}/models/{model}`.
+  string name = 1 [
+    (google.api.field_behavior) = REQUIRED,
+    (google.api.resource_reference) = {
+      type: "aiplatform.googleapis.com/Endpoint"
+    }
+  ];
+
+  // Required. The publisher model config.
+  PublisherModelConfig publisher_model_config = 2
+      [(google.api.field_behavior) = REQUIRED];
+}
+
+// Runtime operation information for
+// [EndpointService.SetPublisherModelConfig][google.cloud.aiplatform.v1beta1.EndpointService.SetPublisherModelConfig].
+message SetPublisherModelConfigOperationMetadata {
+  // The operation generic information.
+  GenericOperationMetadata generic_metadata = 1;
+}
+
+// Request message for
+// [EndpointService.FetchPublisherModelConfig][google.cloud.aiplatform.v1beta1.EndpointService.FetchPublisherModelConfig].
+message FetchPublisherModelConfigRequest {
+  // Required. The name of the publisher model, in the format of
+  // `projects/{project}/locations/{location}/publishers/{publisher}/models/{model}`.
+  string name = 1 [
+    (google.api.field_behavior) = REQUIRED,
+    (google.api.resource_reference) = {
+      type: "aiplatform.googleapis.com/Endpoint"
+    }
+  ];
+}
+
 // Runtime operation information for
 // [EndpointService.UndeployModel][google.cloud.aiplatform.v1beta1.EndpointService.UndeployModel].
 message UndeployModelOperationMetadata {
