cloud.google.com/go/aiplatform@v1.106.0/apiv1/aiplatformpb/vizier_service_grpc.pb.go (about)

     1  // Copyright 2025 Google LLC
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  
    15  // Code generated by protoc-gen-go-grpc. DO NOT EDIT.
    16  // versions:
    17  // - protoc-gen-go-grpc v1.3.0
    18  // - protoc             v4.25.7
    19  // source: google/cloud/aiplatform/v1/vizier_service.proto
    20  
    21  package aiplatformpb
    22  
    23  import (
    24  	longrunningpb "cloud.google.com/go/longrunning/autogen/longrunningpb"
    25  	context "context"
    26  	grpc "google.golang.org/grpc"
    27  	codes "google.golang.org/grpc/codes"
    28  	status "google.golang.org/grpc/status"
    29  	emptypb "google.golang.org/protobuf/types/known/emptypb"
    30  )
    31  
    32  // This is a compile-time assertion to ensure that this generated file
    33  // is compatible with the grpc package it is being compiled against.
    34  // Requires gRPC-Go v1.32.0 or later.
    35  const _ = grpc.SupportPackageIsVersion7
    36  
    37  const (
    38  	VizierService_CreateStudy_FullMethodName                  = "/google.cloud.aiplatform.v1.VizierService/CreateStudy"
    39  	VizierService_GetStudy_FullMethodName                     = "/google.cloud.aiplatform.v1.VizierService/GetStudy"
    40  	VizierService_ListStudies_FullMethodName                  = "/google.cloud.aiplatform.v1.VizierService/ListStudies"
    41  	VizierService_DeleteStudy_FullMethodName                  = "/google.cloud.aiplatform.v1.VizierService/DeleteStudy"
    42  	VizierService_LookupStudy_FullMethodName                  = "/google.cloud.aiplatform.v1.VizierService/LookupStudy"
    43  	VizierService_SuggestTrials_FullMethodName                = "/google.cloud.aiplatform.v1.VizierService/SuggestTrials"
    44  	VizierService_CreateTrial_FullMethodName                  = "/google.cloud.aiplatform.v1.VizierService/CreateTrial"
    45  	VizierService_GetTrial_FullMethodName                     = "/google.cloud.aiplatform.v1.VizierService/GetTrial"
    46  	VizierService_ListTrials_FullMethodName                   = "/google.cloud.aiplatform.v1.VizierService/ListTrials"
    47  	VizierService_AddTrialMeasurement_FullMethodName          = "/google.cloud.aiplatform.v1.VizierService/AddTrialMeasurement"
    48  	VizierService_CompleteTrial_FullMethodName                = "/google.cloud.aiplatform.v1.VizierService/CompleteTrial"
    49  	VizierService_DeleteTrial_FullMethodName                  = "/google.cloud.aiplatform.v1.VizierService/DeleteTrial"
    50  	VizierService_CheckTrialEarlyStoppingState_FullMethodName = "/google.cloud.aiplatform.v1.VizierService/CheckTrialEarlyStoppingState"
    51  	VizierService_StopTrial_FullMethodName                    = "/google.cloud.aiplatform.v1.VizierService/StopTrial"
    52  	VizierService_ListOptimalTrials_FullMethodName            = "/google.cloud.aiplatform.v1.VizierService/ListOptimalTrials"
    53  )
    54  
    55  // VizierServiceClient is the client API for VizierService service.
    56  //
    57  // For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
    58  type VizierServiceClient interface {
    59  	// Creates a Study. A resource name will be generated after creation of the
    60  	// Study.
    61  	CreateStudy(ctx context.Context, in *CreateStudyRequest, opts ...grpc.CallOption) (*Study, error)
    62  	// Gets a Study by name.
    63  	GetStudy(ctx context.Context, in *GetStudyRequest, opts ...grpc.CallOption) (*Study, error)
    64  	// Lists all the studies in a region for an associated project.
    65  	ListStudies(ctx context.Context, in *ListStudiesRequest, opts ...grpc.CallOption) (*ListStudiesResponse, error)
    66  	// Deletes a Study.
    67  	DeleteStudy(ctx context.Context, in *DeleteStudyRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)
    68  	// Looks a study up using the user-defined display_name field instead of the
    69  	// fully qualified resource name.
    70  	LookupStudy(ctx context.Context, in *LookupStudyRequest, opts ...grpc.CallOption) (*Study, error)
    71  	// Adds one or more Trials to a Study, with parameter values
    72  	// suggested by Vertex AI Vizier. Returns a long-running
    73  	// operation associated with the generation of Trial suggestions.
    74  	// When this long-running operation succeeds, it will contain
    75  	// a
    76  	// [SuggestTrialsResponse][google.cloud.aiplatform.v1.SuggestTrialsResponse].
    77  	SuggestTrials(ctx context.Context, in *SuggestTrialsRequest, opts ...grpc.CallOption) (*longrunningpb.Operation, error)
    78  	// Adds a user provided Trial to a Study.
    79  	CreateTrial(ctx context.Context, in *CreateTrialRequest, opts ...grpc.CallOption) (*Trial, error)
    80  	// Gets a Trial.
    81  	GetTrial(ctx context.Context, in *GetTrialRequest, opts ...grpc.CallOption) (*Trial, error)
    82  	// Lists the Trials associated with a Study.
    83  	ListTrials(ctx context.Context, in *ListTrialsRequest, opts ...grpc.CallOption) (*ListTrialsResponse, error)
    84  	// Adds a measurement of the objective metrics to a Trial. This measurement
    85  	// is assumed to have been taken before the Trial is complete.
    86  	AddTrialMeasurement(ctx context.Context, in *AddTrialMeasurementRequest, opts ...grpc.CallOption) (*Trial, error)
    87  	// Marks a Trial as complete.
    88  	CompleteTrial(ctx context.Context, in *CompleteTrialRequest, opts ...grpc.CallOption) (*Trial, error)
    89  	// Deletes a Trial.
    90  	DeleteTrial(ctx context.Context, in *DeleteTrialRequest, opts ...grpc.CallOption) (*emptypb.Empty, error)
    91  	// Checks  whether a Trial should stop or not. Returns a
    92  	// long-running operation. When the operation is successful,
    93  	// it will contain a
    94  	// [CheckTrialEarlyStoppingStateResponse][google.cloud.aiplatform.v1.CheckTrialEarlyStoppingStateResponse].
    95  	CheckTrialEarlyStoppingState(ctx context.Context, in *CheckTrialEarlyStoppingStateRequest, opts ...grpc.CallOption) (*longrunningpb.Operation, error)
    96  	// Stops a Trial.
    97  	StopTrial(ctx context.Context, in *StopTrialRequest, opts ...grpc.CallOption) (*Trial, error)
    98  	// Lists the pareto-optimal Trials for multi-objective Study or the
    99  	// optimal Trials for single-objective Study. The definition of
   100  	// pareto-optimal can be checked in wiki page.
   101  	// https://en.wikipedia.org/wiki/Pareto_efficiency
   102  	ListOptimalTrials(ctx context.Context, in *ListOptimalTrialsRequest, opts ...grpc.CallOption) (*ListOptimalTrialsResponse, error)
   103  }
   104  
   105  type vizierServiceClient struct {
   106  	cc grpc.ClientConnInterface
   107  }
   108  
   109  func NewVizierServiceClient(cc grpc.ClientConnInterface) VizierServiceClient {
   110  	return &vizierServiceClient{cc}
   111  }
   112  
   113  func (c *vizierServiceClient) CreateStudy(ctx context.Context, in *CreateStudyRequest, opts ...grpc.CallOption) (*Study, error) {
   114  	out := new(Study)
   115  	err := c.cc.Invoke(ctx, VizierService_CreateStudy_FullMethodName, in, out, opts...)
   116  	if err != nil {
   117  		return nil, err
   118  	}
   119  	return out, nil
   120  }
   121  
   122  func (c *vizierServiceClient) GetStudy(ctx context.Context, in *GetStudyRequest, opts ...grpc.CallOption) (*Study, error) {
   123  	out := new(Study)
   124  	err := c.cc.Invoke(ctx, VizierService_GetStudy_FullMethodName, in, out, opts...)
   125  	if err != nil {
   126  		return nil, err
   127  	}
   128  	return out, nil
   129  }
   130  
   131  func (c *vizierServiceClient) ListStudies(ctx context.Context, in *ListStudiesRequest, opts ...grpc.CallOption) (*ListStudiesResponse, error) {
   132  	out := new(ListStudiesResponse)
   133  	err := c.cc.Invoke(ctx, VizierService_ListStudies_FullMethodName, in, out, opts...)
   134  	if err != nil {
   135  		return nil, err
   136  	}
   137  	return out, nil
   138  }
   139  
   140  func (c *vizierServiceClient) DeleteStudy(ctx context.Context, in *DeleteStudyRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) {
   141  	out := new(emptypb.Empty)
   142  	err := c.cc.Invoke(ctx, VizierService_DeleteStudy_FullMethodName, in, out, opts...)
   143  	if err != nil {
   144  		return nil, err
   145  	}
   146  	return out, nil
   147  }
   148  
   149  func (c *vizierServiceClient) LookupStudy(ctx context.Context, in *LookupStudyRequest, opts ...grpc.CallOption) (*Study, error) {
   150  	out := new(Study)
   151  	err := c.cc.Invoke(ctx, VizierService_LookupStudy_FullMethodName, in, out, opts...)
   152  	if err != nil {
   153  		return nil, err
   154  	}
   155  	return out, nil
   156  }
   157  
   158  func (c *vizierServiceClient) SuggestTrials(ctx context.Context, in *SuggestTrialsRequest, opts ...grpc.CallOption) (*longrunningpb.Operation, error) {
   159  	out := new(longrunningpb.Operation)
   160  	err := c.cc.Invoke(ctx, VizierService_SuggestTrials_FullMethodName, in, out, opts...)
   161  	if err != nil {
   162  		return nil, err
   163  	}
   164  	return out, nil
   165  }
   166  
   167  func (c *vizierServiceClient) CreateTrial(ctx context.Context, in *CreateTrialRequest, opts ...grpc.CallOption) (*Trial, error) {
   168  	out := new(Trial)
   169  	err := c.cc.Invoke(ctx, VizierService_CreateTrial_FullMethodName, in, out, opts...)
   170  	if err != nil {
   171  		return nil, err
   172  	}
   173  	return out, nil
   174  }
   175  
   176  func (c *vizierServiceClient) GetTrial(ctx context.Context, in *GetTrialRequest, opts ...grpc.CallOption) (*Trial, error) {
   177  	out := new(Trial)
   178  	err := c.cc.Invoke(ctx, VizierService_GetTrial_FullMethodName, in, out, opts...)
   179  	if err != nil {
   180  		return nil, err
   181  	}
   182  	return out, nil
   183  }
   184  
   185  func (c *vizierServiceClient) ListTrials(ctx context.Context, in *ListTrialsRequest, opts ...grpc.CallOption) (*ListTrialsResponse, error) {
   186  	out := new(ListTrialsResponse)
   187  	err := c.cc.Invoke(ctx, VizierService_ListTrials_FullMethodName, in, out, opts...)
   188  	if err != nil {
   189  		return nil, err
   190  	}
   191  	return out, nil
   192  }
   193  
   194  func (c *vizierServiceClient) AddTrialMeasurement(ctx context.Context, in *AddTrialMeasurementRequest, opts ...grpc.CallOption) (*Trial, error) {
   195  	out := new(Trial)
   196  	err := c.cc.Invoke(ctx, VizierService_AddTrialMeasurement_FullMethodName, in, out, opts...)
   197  	if err != nil {
   198  		return nil, err
   199  	}
   200  	return out, nil
   201  }
   202  
   203  func (c *vizierServiceClient) CompleteTrial(ctx context.Context, in *CompleteTrialRequest, opts ...grpc.CallOption) (*Trial, error) {
   204  	out := new(Trial)
   205  	err := c.cc.Invoke(ctx, VizierService_CompleteTrial_FullMethodName, in, out, opts...)
   206  	if err != nil {
   207  		return nil, err
   208  	}
   209  	return out, nil
   210  }
   211  
   212  func (c *vizierServiceClient) DeleteTrial(ctx context.Context, in *DeleteTrialRequest, opts ...grpc.CallOption) (*emptypb.Empty, error) {
   213  	out := new(emptypb.Empty)
   214  	err := c.cc.Invoke(ctx, VizierService_DeleteTrial_FullMethodName, in, out, opts...)
   215  	if err != nil {
   216  		return nil, err
   217  	}
   218  	return out, nil
   219  }
   220  
   221  func (c *vizierServiceClient) CheckTrialEarlyStoppingState(ctx context.Context, in *CheckTrialEarlyStoppingStateRequest, opts ...grpc.CallOption) (*longrunningpb.Operation, error) {
   222  	out := new(longrunningpb.Operation)
   223  	err := c.cc.Invoke(ctx, VizierService_CheckTrialEarlyStoppingState_FullMethodName, in, out, opts...)
   224  	if err != nil {
   225  		return nil, err
   226  	}
   227  	return out, nil
   228  }
   229  
   230  func (c *vizierServiceClient) StopTrial(ctx context.Context, in *StopTrialRequest, opts ...grpc.CallOption) (*Trial, error) {
   231  	out := new(Trial)
   232  	err := c.cc.Invoke(ctx, VizierService_StopTrial_FullMethodName, in, out, opts...)
   233  	if err != nil {
   234  		return nil, err
   235  	}
   236  	return out, nil
   237  }
   238  
   239  func (c *vizierServiceClient) ListOptimalTrials(ctx context.Context, in *ListOptimalTrialsRequest, opts ...grpc.CallOption) (*ListOptimalTrialsResponse, error) {
   240  	out := new(ListOptimalTrialsResponse)
   241  	err := c.cc.Invoke(ctx, VizierService_ListOptimalTrials_FullMethodName, in, out, opts...)
   242  	if err != nil {
   243  		return nil, err
   244  	}
   245  	return out, nil
   246  }
   247  
   248  // VizierServiceServer is the server API for VizierService service.
   249  // All implementations should embed UnimplementedVizierServiceServer
   250  // for forward compatibility
   251  type VizierServiceServer interface {
   252  	// Creates a Study. A resource name will be generated after creation of the
   253  	// Study.
   254  	CreateStudy(context.Context, *CreateStudyRequest) (*Study, error)
   255  	// Gets a Study by name.
   256  	GetStudy(context.Context, *GetStudyRequest) (*Study, error)
   257  	// Lists all the studies in a region for an associated project.
   258  	ListStudies(context.Context, *ListStudiesRequest) (*ListStudiesResponse, error)
   259  	// Deletes a Study.
   260  	DeleteStudy(context.Context, *DeleteStudyRequest) (*emptypb.Empty, error)
   261  	// Looks a study up using the user-defined display_name field instead of the
   262  	// fully qualified resource name.
   263  	LookupStudy(context.Context, *LookupStudyRequest) (*Study, error)
   264  	// Adds one or more Trials to a Study, with parameter values
   265  	// suggested by Vertex AI Vizier. Returns a long-running
   266  	// operation associated with the generation of Trial suggestions.
   267  	// When this long-running operation succeeds, it will contain
   268  	// a
   269  	// [SuggestTrialsResponse][google.cloud.aiplatform.v1.SuggestTrialsResponse].
   270  	SuggestTrials(context.Context, *SuggestTrialsRequest) (*longrunningpb.Operation, error)
   271  	// Adds a user provided Trial to a Study.
   272  	CreateTrial(context.Context, *CreateTrialRequest) (*Trial, error)
   273  	// Gets a Trial.
   274  	GetTrial(context.Context, *GetTrialRequest) (*Trial, error)
   275  	// Lists the Trials associated with a Study.
   276  	ListTrials(context.Context, *ListTrialsRequest) (*ListTrialsResponse, error)
   277  	// Adds a measurement of the objective metrics to a Trial. This measurement
   278  	// is assumed to have been taken before the Trial is complete.
   279  	AddTrialMeasurement(context.Context, *AddTrialMeasurementRequest) (*Trial, error)
   280  	// Marks a Trial as complete.
   281  	CompleteTrial(context.Context, *CompleteTrialRequest) (*Trial, error)
   282  	// Deletes a Trial.
   283  	DeleteTrial(context.Context, *DeleteTrialRequest) (*emptypb.Empty, error)
   284  	// Checks  whether a Trial should stop or not. Returns a
   285  	// long-running operation. When the operation is successful,
   286  	// it will contain a
   287  	// [CheckTrialEarlyStoppingStateResponse][google.cloud.aiplatform.v1.CheckTrialEarlyStoppingStateResponse].
   288  	CheckTrialEarlyStoppingState(context.Context, *CheckTrialEarlyStoppingStateRequest) (*longrunningpb.Operation, error)
   289  	// Stops a Trial.
   290  	StopTrial(context.Context, *StopTrialRequest) (*Trial, error)
   291  	// Lists the pareto-optimal Trials for multi-objective Study or the
   292  	// optimal Trials for single-objective Study. The definition of
   293  	// pareto-optimal can be checked in wiki page.
   294  	// https://en.wikipedia.org/wiki/Pareto_efficiency
   295  	ListOptimalTrials(context.Context, *ListOptimalTrialsRequest) (*ListOptimalTrialsResponse, error)
   296  }
   297  
   298  // UnimplementedVizierServiceServer should be embedded to have forward compatible implementations.
   299  type UnimplementedVizierServiceServer struct {
   300  }
   301  
   302  func (UnimplementedVizierServiceServer) CreateStudy(context.Context, *CreateStudyRequest) (*Study, error) {
   303  	return nil, status.Errorf(codes.Unimplemented, "method CreateStudy not implemented")
   304  }
   305  func (UnimplementedVizierServiceServer) GetStudy(context.Context, *GetStudyRequest) (*Study, error) {
   306  	return nil, status.Errorf(codes.Unimplemented, "method GetStudy not implemented")
   307  }
   308  func (UnimplementedVizierServiceServer) ListStudies(context.Context, *ListStudiesRequest) (*ListStudiesResponse, error) {
   309  	return nil, status.Errorf(codes.Unimplemented, "method ListStudies not implemented")
   310  }
   311  func (UnimplementedVizierServiceServer) DeleteStudy(context.Context, *DeleteStudyRequest) (*emptypb.Empty, error) {
   312  	return nil, status.Errorf(codes.Unimplemented, "method DeleteStudy not implemented")
   313  }
   314  func (UnimplementedVizierServiceServer) LookupStudy(context.Context, *LookupStudyRequest) (*Study, error) {
   315  	return nil, status.Errorf(codes.Unimplemented, "method LookupStudy not implemented")
   316  }
   317  func (UnimplementedVizierServiceServer) SuggestTrials(context.Context, *SuggestTrialsRequest) (*longrunningpb.Operation, error) {
   318  	return nil, status.Errorf(codes.Unimplemented, "method SuggestTrials not implemented")
   319  }
   320  func (UnimplementedVizierServiceServer) CreateTrial(context.Context, *CreateTrialRequest) (*Trial, error) {
   321  	return nil, status.Errorf(codes.Unimplemented, "method CreateTrial not implemented")
   322  }
   323  func (UnimplementedVizierServiceServer) GetTrial(context.Context, *GetTrialRequest) (*Trial, error) {
   324  	return nil, status.Errorf(codes.Unimplemented, "method GetTrial not implemented")
   325  }
   326  func (UnimplementedVizierServiceServer) ListTrials(context.Context, *ListTrialsRequest) (*ListTrialsResponse, error) {
   327  	return nil, status.Errorf(codes.Unimplemented, "method ListTrials not implemented")
   328  }
   329  func (UnimplementedVizierServiceServer) AddTrialMeasurement(context.Context, *AddTrialMeasurementRequest) (*Trial, error) {
   330  	return nil, status.Errorf(codes.Unimplemented, "method AddTrialMeasurement not implemented")
   331  }
   332  func (UnimplementedVizierServiceServer) CompleteTrial(context.Context, *CompleteTrialRequest) (*Trial, error) {
   333  	return nil, status.Errorf(codes.Unimplemented, "method CompleteTrial not implemented")
   334  }
   335  func (UnimplementedVizierServiceServer) DeleteTrial(context.Context, *DeleteTrialRequest) (*emptypb.Empty, error) {
   336  	return nil, status.Errorf(codes.Unimplemented, "method DeleteTrial not implemented")
   337  }
   338  func (UnimplementedVizierServiceServer) CheckTrialEarlyStoppingState(context.Context, *CheckTrialEarlyStoppingStateRequest) (*longrunningpb.Operation, error) {
   339  	return nil, status.Errorf(codes.Unimplemented, "method CheckTrialEarlyStoppingState not implemented")
   340  }
   341  func (UnimplementedVizierServiceServer) StopTrial(context.Context, *StopTrialRequest) (*Trial, error) {
   342  	return nil, status.Errorf(codes.Unimplemented, "method StopTrial not implemented")
   343  }
   344  func (UnimplementedVizierServiceServer) ListOptimalTrials(context.Context, *ListOptimalTrialsRequest) (*ListOptimalTrialsResponse, error) {
   345  	return nil, status.Errorf(codes.Unimplemented, "method ListOptimalTrials not implemented")
   346  }
   347  
   348  // UnsafeVizierServiceServer may be embedded to opt out of forward compatibility for this service.
   349  // Use of this interface is not recommended, as added methods to VizierServiceServer will
   350  // result in compilation errors.
   351  type UnsafeVizierServiceServer interface {
   352  	mustEmbedUnimplementedVizierServiceServer()
   353  }
   354  
   355  func RegisterVizierServiceServer(s grpc.ServiceRegistrar, srv VizierServiceServer) {
   356  	s.RegisterService(&VizierService_ServiceDesc, srv)
   357  }
   358  
   359  func _VizierService_CreateStudy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   360  	in := new(CreateStudyRequest)
   361  	if err := dec(in); err != nil {
   362  		return nil, err
   363  	}
   364  	if interceptor == nil {
   365  		return srv.(VizierServiceServer).CreateStudy(ctx, in)
   366  	}
   367  	info := &grpc.UnaryServerInfo{
   368  		Server:     srv,
   369  		FullMethod: VizierService_CreateStudy_FullMethodName,
   370  	}
   371  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   372  		return srv.(VizierServiceServer).CreateStudy(ctx, req.(*CreateStudyRequest))
   373  	}
   374  	return interceptor(ctx, in, info, handler)
   375  }
   376  
   377  func _VizierService_GetStudy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   378  	in := new(GetStudyRequest)
   379  	if err := dec(in); err != nil {
   380  		return nil, err
   381  	}
   382  	if interceptor == nil {
   383  		return srv.(VizierServiceServer).GetStudy(ctx, in)
   384  	}
   385  	info := &grpc.UnaryServerInfo{
   386  		Server:     srv,
   387  		FullMethod: VizierService_GetStudy_FullMethodName,
   388  	}
   389  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   390  		return srv.(VizierServiceServer).GetStudy(ctx, req.(*GetStudyRequest))
   391  	}
   392  	return interceptor(ctx, in, info, handler)
   393  }
   394  
   395  func _VizierService_ListStudies_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   396  	in := new(ListStudiesRequest)
   397  	if err := dec(in); err != nil {
   398  		return nil, err
   399  	}
   400  	if interceptor == nil {
   401  		return srv.(VizierServiceServer).ListStudies(ctx, in)
   402  	}
   403  	info := &grpc.UnaryServerInfo{
   404  		Server:     srv,
   405  		FullMethod: VizierService_ListStudies_FullMethodName,
   406  	}
   407  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   408  		return srv.(VizierServiceServer).ListStudies(ctx, req.(*ListStudiesRequest))
   409  	}
   410  	return interceptor(ctx, in, info, handler)
   411  }
   412  
   413  func _VizierService_DeleteStudy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   414  	in := new(DeleteStudyRequest)
   415  	if err := dec(in); err != nil {
   416  		return nil, err
   417  	}
   418  	if interceptor == nil {
   419  		return srv.(VizierServiceServer).DeleteStudy(ctx, in)
   420  	}
   421  	info := &grpc.UnaryServerInfo{
   422  		Server:     srv,
   423  		FullMethod: VizierService_DeleteStudy_FullMethodName,
   424  	}
   425  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   426  		return srv.(VizierServiceServer).DeleteStudy(ctx, req.(*DeleteStudyRequest))
   427  	}
   428  	return interceptor(ctx, in, info, handler)
   429  }
   430  
   431  func _VizierService_LookupStudy_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   432  	in := new(LookupStudyRequest)
   433  	if err := dec(in); err != nil {
   434  		return nil, err
   435  	}
   436  	if interceptor == nil {
   437  		return srv.(VizierServiceServer).LookupStudy(ctx, in)
   438  	}
   439  	info := &grpc.UnaryServerInfo{
   440  		Server:     srv,
   441  		FullMethod: VizierService_LookupStudy_FullMethodName,
   442  	}
   443  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   444  		return srv.(VizierServiceServer).LookupStudy(ctx, req.(*LookupStudyRequest))
   445  	}
   446  	return interceptor(ctx, in, info, handler)
   447  }
   448  
   449  func _VizierService_SuggestTrials_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   450  	in := new(SuggestTrialsRequest)
   451  	if err := dec(in); err != nil {
   452  		return nil, err
   453  	}
   454  	if interceptor == nil {
   455  		return srv.(VizierServiceServer).SuggestTrials(ctx, in)
   456  	}
   457  	info := &grpc.UnaryServerInfo{
   458  		Server:     srv,
   459  		FullMethod: VizierService_SuggestTrials_FullMethodName,
   460  	}
   461  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   462  		return srv.(VizierServiceServer).SuggestTrials(ctx, req.(*SuggestTrialsRequest))
   463  	}
   464  	return interceptor(ctx, in, info, handler)
   465  }
   466  
   467  func _VizierService_CreateTrial_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   468  	in := new(CreateTrialRequest)
   469  	if err := dec(in); err != nil {
   470  		return nil, err
   471  	}
   472  	if interceptor == nil {
   473  		return srv.(VizierServiceServer).CreateTrial(ctx, in)
   474  	}
   475  	info := &grpc.UnaryServerInfo{
   476  		Server:     srv,
   477  		FullMethod: VizierService_CreateTrial_FullMethodName,
   478  	}
   479  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   480  		return srv.(VizierServiceServer).CreateTrial(ctx, req.(*CreateTrialRequest))
   481  	}
   482  	return interceptor(ctx, in, info, handler)
   483  }
   484  
   485  func _VizierService_GetTrial_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   486  	in := new(GetTrialRequest)
   487  	if err := dec(in); err != nil {
   488  		return nil, err
   489  	}
   490  	if interceptor == nil {
   491  		return srv.(VizierServiceServer).GetTrial(ctx, in)
   492  	}
   493  	info := &grpc.UnaryServerInfo{
   494  		Server:     srv,
   495  		FullMethod: VizierService_GetTrial_FullMethodName,
   496  	}
   497  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   498  		return srv.(VizierServiceServer).GetTrial(ctx, req.(*GetTrialRequest))
   499  	}
   500  	return interceptor(ctx, in, info, handler)
   501  }
   502  
   503  func _VizierService_ListTrials_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   504  	in := new(ListTrialsRequest)
   505  	if err := dec(in); err != nil {
   506  		return nil, err
   507  	}
   508  	if interceptor == nil {
   509  		return srv.(VizierServiceServer).ListTrials(ctx, in)
   510  	}
   511  	info := &grpc.UnaryServerInfo{
   512  		Server:     srv,
   513  		FullMethod: VizierService_ListTrials_FullMethodName,
   514  	}
   515  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   516  		return srv.(VizierServiceServer).ListTrials(ctx, req.(*ListTrialsRequest))
   517  	}
   518  	return interceptor(ctx, in, info, handler)
   519  }
   520  
   521  func _VizierService_AddTrialMeasurement_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   522  	in := new(AddTrialMeasurementRequest)
   523  	if err := dec(in); err != nil {
   524  		return nil, err
   525  	}
   526  	if interceptor == nil {
   527  		return srv.(VizierServiceServer).AddTrialMeasurement(ctx, in)
   528  	}
   529  	info := &grpc.UnaryServerInfo{
   530  		Server:     srv,
   531  		FullMethod: VizierService_AddTrialMeasurement_FullMethodName,
   532  	}
   533  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   534  		return srv.(VizierServiceServer).AddTrialMeasurement(ctx, req.(*AddTrialMeasurementRequest))
   535  	}
   536  	return interceptor(ctx, in, info, handler)
   537  }
   538  
   539  func _VizierService_CompleteTrial_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   540  	in := new(CompleteTrialRequest)
   541  	if err := dec(in); err != nil {
   542  		return nil, err
   543  	}
   544  	if interceptor == nil {
   545  		return srv.(VizierServiceServer).CompleteTrial(ctx, in)
   546  	}
   547  	info := &grpc.UnaryServerInfo{
   548  		Server:     srv,
   549  		FullMethod: VizierService_CompleteTrial_FullMethodName,
   550  	}
   551  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   552  		return srv.(VizierServiceServer).CompleteTrial(ctx, req.(*CompleteTrialRequest))
   553  	}
   554  	return interceptor(ctx, in, info, handler)
   555  }
   556  
   557  func _VizierService_DeleteTrial_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   558  	in := new(DeleteTrialRequest)
   559  	if err := dec(in); err != nil {
   560  		return nil, err
   561  	}
   562  	if interceptor == nil {
   563  		return srv.(VizierServiceServer).DeleteTrial(ctx, in)
   564  	}
   565  	info := &grpc.UnaryServerInfo{
   566  		Server:     srv,
   567  		FullMethod: VizierService_DeleteTrial_FullMethodName,
   568  	}
   569  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   570  		return srv.(VizierServiceServer).DeleteTrial(ctx, req.(*DeleteTrialRequest))
   571  	}
   572  	return interceptor(ctx, in, info, handler)
   573  }
   574  
   575  func _VizierService_CheckTrialEarlyStoppingState_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   576  	in := new(CheckTrialEarlyStoppingStateRequest)
   577  	if err := dec(in); err != nil {
   578  		return nil, err
   579  	}
   580  	if interceptor == nil {
   581  		return srv.(VizierServiceServer).CheckTrialEarlyStoppingState(ctx, in)
   582  	}
   583  	info := &grpc.UnaryServerInfo{
   584  		Server:     srv,
   585  		FullMethod: VizierService_CheckTrialEarlyStoppingState_FullMethodName,
   586  	}
   587  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   588  		return srv.(VizierServiceServer).CheckTrialEarlyStoppingState(ctx, req.(*CheckTrialEarlyStoppingStateRequest))
   589  	}
   590  	return interceptor(ctx, in, info, handler)
   591  }
   592  
   593  func _VizierService_StopTrial_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   594  	in := new(StopTrialRequest)
   595  	if err := dec(in); err != nil {
   596  		return nil, err
   597  	}
   598  	if interceptor == nil {
   599  		return srv.(VizierServiceServer).StopTrial(ctx, in)
   600  	}
   601  	info := &grpc.UnaryServerInfo{
   602  		Server:     srv,
   603  		FullMethod: VizierService_StopTrial_FullMethodName,
   604  	}
   605  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   606  		return srv.(VizierServiceServer).StopTrial(ctx, req.(*StopTrialRequest))
   607  	}
   608  	return interceptor(ctx, in, info, handler)
   609  }
   610  
   611  func _VizierService_ListOptimalTrials_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
   612  	in := new(ListOptimalTrialsRequest)
   613  	if err := dec(in); err != nil {
   614  		return nil, err
   615  	}
   616  	if interceptor == nil {
   617  		return srv.(VizierServiceServer).ListOptimalTrials(ctx, in)
   618  	}
   619  	info := &grpc.UnaryServerInfo{
   620  		Server:     srv,
   621  		FullMethod: VizierService_ListOptimalTrials_FullMethodName,
   622  	}
   623  	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
   624  		return srv.(VizierServiceServer).ListOptimalTrials(ctx, req.(*ListOptimalTrialsRequest))
   625  	}
   626  	return interceptor(ctx, in, info, handler)
   627  }
   628  
   629  // VizierService_ServiceDesc is the grpc.ServiceDesc for VizierService service.
   630  // It's only intended for direct use with grpc.RegisterService,
   631  // and not to be introspected or modified (even as a copy)
   632  var VizierService_ServiceDesc = grpc.ServiceDesc{
   633  	ServiceName: "google.cloud.aiplatform.v1.VizierService",
   634  	HandlerType: (*VizierServiceServer)(nil),
   635  	Methods: []grpc.MethodDesc{
   636  		{
   637  			MethodName: "CreateStudy",
   638  			Handler:    _VizierService_CreateStudy_Handler,
   639  		},
   640  		{
   641  			MethodName: "GetStudy",
   642  			Handler:    _VizierService_GetStudy_Handler,
   643  		},
   644  		{
   645  			MethodName: "ListStudies",
   646  			Handler:    _VizierService_ListStudies_Handler,
   647  		},
   648  		{
   649  			MethodName: "DeleteStudy",
   650  			Handler:    _VizierService_DeleteStudy_Handler,
   651  		},
   652  		{
   653  			MethodName: "LookupStudy",
   654  			Handler:    _VizierService_LookupStudy_Handler,
   655  		},
   656  		{
   657  			MethodName: "SuggestTrials",
   658  			Handler:    _VizierService_SuggestTrials_Handler,
   659  		},
   660  		{
   661  			MethodName: "CreateTrial",
   662  			Handler:    _VizierService_CreateTrial_Handler,
   663  		},
   664  		{
   665  			MethodName: "GetTrial",
   666  			Handler:    _VizierService_GetTrial_Handler,
   667  		},
   668  		{
   669  			MethodName: "ListTrials",
   670  			Handler:    _VizierService_ListTrials_Handler,
   671  		},
   672  		{
   673  			MethodName: "AddTrialMeasurement",
   674  			Handler:    _VizierService_AddTrialMeasurement_Handler,
   675  		},
   676  		{
   677  			MethodName: "CompleteTrial",
   678  			Handler:    _VizierService_CompleteTrial_Handler,
   679  		},
   680  		{
   681  			MethodName: "DeleteTrial",
   682  			Handler:    _VizierService_DeleteTrial_Handler,
   683  		},
   684  		{
   685  			MethodName: "CheckTrialEarlyStoppingState",
   686  			Handler:    _VizierService_CheckTrialEarlyStoppingState_Handler,
   687  		},
   688  		{
   689  			MethodName: "StopTrial",
   690  			Handler:    _VizierService_StopTrial_Handler,
   691  		},
   692  		{
   693  			MethodName: "ListOptimalTrials",
   694  			Handler:    _VizierService_ListOptimalTrials_Handler,
   695  		},
   696  	},
   697  	Streams:  []grpc.StreamDesc{},
   698  	Metadata: "google/cloud/aiplatform/v1/vizier_service.proto",
   699  }