diff --git a/magefiles/generate/discovery/discovery.go b/magefiles/generate/discovery/discovery.go index 18a10f2d..09a456a9 100644 --- a/magefiles/generate/discovery/discovery.go +++ b/magefiles/generate/discovery/discovery.go @@ -35,13 +35,18 @@ var routeParameterRegex = regexp.MustCompile(`<[^>]+:([^>]+)>`) // Get the safe path to use in Fiber registration. func (e Endpoint) GetFiberPath() string { // e.Path cannot be trusted, it could be something like /mlflow-artifacts/artifacts/ - // Which would need to converted to /mlflow-artifacts/artifacts/:path + // which would need to be converted to /mlflow-artifacts/artifacts/:path path := routeParameterRegex.ReplaceAllStringFunc(e.Path, func(s string) string { parts := strings.Split(s, ":") return ":" + strings.Trim(parts[0], "< ") }) + // or it could be something like /mlflow/traces/{request_id}/tags + // which would need to be converted to /mlflow/traces/:request_id/tags + path = strings.ReplaceAll(path, "{", ":") + path = strings.ReplaceAll(path, "}", "") + return path } diff --git a/magefiles/generate/query_annotations.go b/magefiles/generate/query_annotations.go index c661e9b5..072315d8 100644 --- a/magefiles/generate/query_annotations.go +++ b/magefiles/generate/query_annotations.go @@ -47,11 +47,12 @@ func addQueryAnnotation(generatedGoFile string) error { tagValue := field.Tag.Value hasQuery := strings.Contains(tagValue, "query:") + hasParams := strings.Contains(tagValue, "params:") hasValidate := strings.Contains(tagValue, "validate:") validationKey := fmt.Sprintf("%s_%s", typeSpec.Name, field.Names[0]) validationRule, needsValidation := validations[validationKey] - if hasQuery && (!needsValidation || needsValidation && hasValidate) { + if hasParams && hasQuery && (!needsValidation || needsValidation && hasValidate) { continue } @@ -59,7 +60,7 @@ func addQueryAnnotation(generatedGoFile string) error { newTag := tagValue[0 : len(tagValue)-1] matches := jsonFieldTagRegexp.FindStringSubmatch(tagValue) - if len(matches) > 0 && !hasQuery { + if len(matches) > 0 && !hasQuery && !hasParams { // Modify the tag here // The json annotation could be something like `json:"key,omitempty"` // We only want the key part, the `omitempty` is not relevant for the query annotation @@ -68,7 +69,7 @@ func addQueryAnnotation(generatedGoFile string) error { key = strings.Split(key, ",")[0] } // Add query annotation - newTag += fmt.Sprintf(" query:\"%s\"", key) + newTag += fmt.Sprintf(" query:\"%s\" params:\"%s\"", key, key) } if needsValidation { diff --git a/pkg/contract/service/tracking.g.go b/pkg/contract/service/tracking.g.go index dccc25d1..67a73b6e 100644 --- a/pkg/contract/service/tracking.g.go +++ b/pkg/contract/service/tracking.g.go @@ -4,8 +4,8 @@ package service import ( "context" - "github.com/mlflow/mlflow-go/pkg/contract" "github.com/mlflow/mlflow-go/pkg/protos" + "github.com/mlflow/mlflow-go/pkg/contract" ) type TrackingService interface { diff --git a/pkg/lib/tracking.g.go b/pkg/lib/tracking.g.go index 2b4d78e8..ff6254e4 100644 --- a/pkg/lib/tracking.g.go +++ b/pkg/lib/tracking.g.go @@ -4,10 +4,9 @@ package main import "C" import ( - "github.com/mlflow/mlflow-go/pkg/protos" "unsafe" + "github.com/mlflow/mlflow-go/pkg/protos" ) - //export TrackingServiceGetExperimentByName func TrackingServiceGetExperimentByName(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -16,7 +15,6 @@ func TrackingServiceGetExperimentByName(serviceID int64, requestData unsafe.Poin } return invokeServiceMethod(service.GetExperimentByName, new(protos.GetExperimentByName), requestData, requestSize, responseSize) } - //export TrackingServiceCreateExperiment func TrackingServiceCreateExperiment(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -25,7 +23,6 @@ func TrackingServiceCreateExperiment(serviceID int64, requestData unsafe.Pointer } return invokeServiceMethod(service.CreateExperiment, new(protos.CreateExperiment), requestData, requestSize, responseSize) } - //export TrackingServiceGetExperiment func TrackingServiceGetExperiment(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -34,7 +31,6 @@ func TrackingServiceGetExperiment(serviceID int64, requestData unsafe.Pointer, r } return invokeServiceMethod(service.GetExperiment, new(protos.GetExperiment), requestData, requestSize, responseSize) } - //export TrackingServiceDeleteExperiment func TrackingServiceDeleteExperiment(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -43,7 +39,6 @@ func TrackingServiceDeleteExperiment(serviceID int64, requestData unsafe.Pointer } return invokeServiceMethod(service.DeleteExperiment, new(protos.DeleteExperiment), requestData, requestSize, responseSize) } - //export TrackingServiceRestoreExperiment func TrackingServiceRestoreExperiment(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -52,7 +47,6 @@ func TrackingServiceRestoreExperiment(serviceID int64, requestData unsafe.Pointe } return invokeServiceMethod(service.RestoreExperiment, new(protos.RestoreExperiment), requestData, requestSize, responseSize) } - //export TrackingServiceUpdateExperiment func TrackingServiceUpdateExperiment(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -61,7 +55,6 @@ func TrackingServiceUpdateExperiment(serviceID int64, requestData unsafe.Pointer } return invokeServiceMethod(service.UpdateExperiment, new(protos.UpdateExperiment), requestData, requestSize, responseSize) } - //export TrackingServiceCreateRun func TrackingServiceCreateRun(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -70,7 +63,6 @@ func TrackingServiceCreateRun(serviceID int64, requestData unsafe.Pointer, reque } return invokeServiceMethod(service.CreateRun, new(protos.CreateRun), requestData, requestSize, responseSize) } - //export TrackingServiceUpdateRun func TrackingServiceUpdateRun(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -79,7 +71,6 @@ func TrackingServiceUpdateRun(serviceID int64, requestData unsafe.Pointer, reque } return invokeServiceMethod(service.UpdateRun, new(protos.UpdateRun), requestData, requestSize, responseSize) } - //export TrackingServiceDeleteRun func TrackingServiceDeleteRun(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -88,7 +79,6 @@ func TrackingServiceDeleteRun(serviceID int64, requestData unsafe.Pointer, reque } return invokeServiceMethod(service.DeleteRun, new(protos.DeleteRun), requestData, requestSize, responseSize) } - //export TrackingServiceRestoreRun func TrackingServiceRestoreRun(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -97,7 +87,6 @@ func TrackingServiceRestoreRun(serviceID int64, requestData unsafe.Pointer, requ } return invokeServiceMethod(service.RestoreRun, new(protos.RestoreRun), requestData, requestSize, responseSize) } - //export TrackingServiceLogMetric func TrackingServiceLogMetric(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -106,7 +95,6 @@ func TrackingServiceLogMetric(serviceID int64, requestData unsafe.Pointer, reque } return invokeServiceMethod(service.LogMetric, new(protos.LogMetric), requestData, requestSize, responseSize) } - //export TrackingServiceLogParam func TrackingServiceLogParam(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -115,16 +103,14 @@ func TrackingServiceLogParam(serviceID int64, requestData unsafe.Pointer, reques } return invokeServiceMethod(service.LogParam, new(protos.LogParam), requestData, requestSize, responseSize) } - //export TrackingServiceSetTraceTag func TrackingServiceSetTraceTag(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { - service, err := trackingServices.Get(serviceID) + service, err := trackingServices.Get(serviceID) if err != nil { return makePointerFromError(err, responseSize) } return invokeServiceMethod(service.SetTraceTag, new(protos.SetTraceTag), requestData, requestSize, responseSize) } - //export TrackingServiceDeleteTag func TrackingServiceDeleteTag(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -133,7 +119,6 @@ func TrackingServiceDeleteTag(serviceID int64, requestData unsafe.Pointer, reque } return invokeServiceMethod(service.DeleteTag, new(protos.DeleteTag), requestData, requestSize, responseSize) } - //export TrackingServiceGetRun func TrackingServiceGetRun(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -142,7 +127,6 @@ func TrackingServiceGetRun(serviceID int64, requestData unsafe.Pointer, requestS } return invokeServiceMethod(service.GetRun, new(protos.GetRun), requestData, requestSize, responseSize) } - //export TrackingServiceSearchRuns func TrackingServiceSearchRuns(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) @@ -151,7 +135,6 @@ func TrackingServiceSearchRuns(serviceID int64, requestData unsafe.Pointer, requ } return invokeServiceMethod(service.SearchRuns, new(protos.SearchRuns), requestData, requestSize, responseSize) } - //export TrackingServiceLogBatch func TrackingServiceLogBatch(serviceID int64, requestData unsafe.Pointer, requestSize C.int, responseSize *C.int) unsafe.Pointer { service, err := trackingServices.Get(serviceID) diff --git a/pkg/protos/artifacts/mlflow_artifacts.pb.go b/pkg/protos/artifacts/mlflow_artifacts.pb.go index 96335222..96a5fae5 100644 --- a/pkg/protos/artifacts/mlflow_artifacts.pb.go +++ b/pkg/protos/artifacts/mlflow_artifacts.pb.go @@ -110,7 +110,7 @@ type ListArtifacts struct { unknownFields protoimpl.UnknownFields // Filter artifacts matching this path (a relative path from the root artifact directory). - Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path"` + Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path" params:"path"` } func (x *ListArtifacts) Reset() { @@ -196,11 +196,11 @@ type FileInfo struct { unknownFields protoimpl.UnknownFields // Path relative to the root artifact directory run. - Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path"` + Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path" params:"path"` // Whether the path is a directory. - IsDir *bool `protobuf:"varint,2,opt,name=is_dir,json=isDir" json:"is_dir,omitempty" query:"is_dir"` + IsDir *bool `protobuf:"varint,2,opt,name=is_dir,json=isDir" json:"is_dir,omitempty" query:"is_dir" params:"is_dir"` // Size in bytes. Unset for directories. - FileSize *int64 `protobuf:"varint,3,opt,name=file_size,json=fileSize" json:"file_size,omitempty" query:"file_size"` + FileSize *int64 `protobuf:"varint,3,opt,name=file_size,json=fileSize" json:"file_size,omitempty" query:"file_size" params:"file_size"` } func (x *FileInfo) Reset() { @@ -261,8 +261,8 @@ type CreateMultipartUpload struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path"` - NumParts *int64 `protobuf:"varint,2,opt,name=num_parts,json=numParts" json:"num_parts,omitempty" query:"num_parts"` + Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path" params:"path"` + NumParts *int64 `protobuf:"varint,2,opt,name=num_parts,json=numParts" json:"num_parts,omitempty" query:"num_parts" params:"num_parts"` } func (x *CreateMultipartUpload) Reset() { @@ -316,9 +316,9 @@ type CompleteMultipartUpload struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path"` - UploadId *string `protobuf:"bytes,2,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id"` - Parts []*MultipartUploadPart `protobuf:"bytes,3,rep,name=parts" json:"parts,omitempty" query:"parts"` + Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path" params:"path"` + UploadId *string `protobuf:"bytes,2,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id" params:"upload_id"` + Parts []*MultipartUploadPart `protobuf:"bytes,3,rep,name=parts" json:"parts,omitempty" query:"parts" params:"parts"` } func (x *CompleteMultipartUpload) Reset() { @@ -379,8 +379,8 @@ type AbortMultipartUpload struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path"` - UploadId *string `protobuf:"bytes,2,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id"` + Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path" params:"path"` + UploadId *string `protobuf:"bytes,2,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id" params:"upload_id"` } func (x *AbortMultipartUpload) Reset() { @@ -434,9 +434,9 @@ type MultipartUploadCredential struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - Url *string `protobuf:"bytes,1,opt,name=url" json:"url,omitempty" query:"url"` - PartNumber *int64 `protobuf:"varint,2,opt,name=part_number,json=partNumber" json:"part_number,omitempty" query:"part_number"` - Headers map[string]string `protobuf:"bytes,3,rep,name=headers" json:"headers,omitempty" protobuf_key:"bytes,1,opt,name=key" protobuf_val:"bytes,2,opt,name=value" query:"headers"` + Url *string `protobuf:"bytes,1,opt,name=url" json:"url,omitempty" query:"url" params:"url"` + PartNumber *int64 `protobuf:"varint,2,opt,name=part_number,json=partNumber" json:"part_number,omitempty" query:"part_number" params:"part_number"` + Headers map[string]string `protobuf:"bytes,3,rep,name=headers" json:"headers,omitempty" protobuf_key:"bytes,1,opt,name=key" protobuf_val:"bytes,2,opt,name=value" query:"headers" params:"headers"` } func (x *MultipartUploadCredential) Reset() { @@ -497,9 +497,9 @@ type MultipartUploadPart struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - PartNumber *int64 `protobuf:"varint,1,opt,name=part_number,json=partNumber" json:"part_number,omitempty" query:"part_number"` - Etag *string `protobuf:"bytes,2,opt,name=etag" json:"etag,omitempty" query:"etag"` - Url *string `protobuf:"bytes,3,opt,name=url" json:"url,omitempty" query:"url"` + PartNumber *int64 `protobuf:"varint,1,opt,name=part_number,json=partNumber" json:"part_number,omitempty" query:"part_number" params:"part_number"` + Etag *string `protobuf:"bytes,2,opt,name=etag" json:"etag,omitempty" query:"etag" params:"etag"` + Url *string `protobuf:"bytes,3,opt,name=url" json:"url,omitempty" query:"url" params:"url"` } func (x *MultipartUploadPart) Reset() { @@ -637,7 +637,7 @@ type ListArtifacts_Response struct { unknownFields protoimpl.UnknownFields // File location and metadata for artifacts. - Files []*FileInfo `protobuf:"bytes,1,rep,name=files" json:"files,omitempty" query:"files"` + Files []*FileInfo `protobuf:"bytes,1,rep,name=files" json:"files,omitempty" query:"files" params:"files"` } func (x *ListArtifacts_Response) Reset() { @@ -722,8 +722,8 @@ type CreateMultipartUpload_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - UploadId *string `protobuf:"bytes,1,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id"` - Credentials []*MultipartUploadCredential `protobuf:"bytes,2,rep,name=credentials" json:"credentials,omitempty" query:"credentials"` + UploadId *string `protobuf:"bytes,1,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id" params:"upload_id"` + Credentials []*MultipartUploadCredential `protobuf:"bytes,2,rep,name=credentials" json:"credentials,omitempty" query:"credentials" params:"credentials"` } func (x *CreateMultipartUpload_Response) Reset() { diff --git a/pkg/protos/databricks.pb.go b/pkg/protos/databricks.pb.go index 112dda97..ec8eaeee 100644 --- a/pkg/protos/databricks.pb.go +++ b/pkg/protos/databricks.pb.go @@ -548,17 +548,17 @@ type DatabricksRpcOptions struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - Endpoints []*HttpEndpoint `protobuf:"bytes,1,rep,name=endpoints" json:"endpoints,omitempty" query:"endpoints"` + Endpoints []*HttpEndpoint `protobuf:"bytes,1,rep,name=endpoints" json:"endpoints,omitempty" query:"endpoints" params:"endpoints"` // Indicates which users are allowed to initiate this RPC. - Visibility *Visibility `protobuf:"varint,2,opt,name=visibility,enum=mlflow.Visibility" json:"visibility,omitempty" query:"visibility"` + Visibility *Visibility `protobuf:"varint,2,opt,name=visibility,enum=mlflow.Visibility" json:"visibility,omitempty" query:"visibility" params:"visibility"` // Complete definition of all error codes (from a statically defined set) which this method // may return. - ErrorCodes []ErrorCode `protobuf:"varint,3,rep,name=error_codes,json=errorCodes,enum=mlflow.ErrorCode" json:"error_codes,omitempty" query:"error_codes"` + ErrorCodes []ErrorCode `protobuf:"varint,3,rep,name=error_codes,json=errorCodes,enum=mlflow.ErrorCode" json:"error_codes,omitempty" query:"error_codes" params:"error_codes"` // If defined, a rate limit will be applied to this RPC for all requests from the API proxy. - RateLimit *RateLimit `protobuf:"bytes,4,opt,name=rate_limit,json=rateLimit" json:"rate_limit,omitempty" query:"rate_limit"` + RateLimit *RateLimit `protobuf:"bytes,4,opt,name=rate_limit,json=rateLimit" json:"rate_limit,omitempty" query:"rate_limit" params:"rate_limit"` // If defined, overrides the default title used for in the API docs. See ProtobufDocGenerator // for more info. - RpcDocTitle *string `protobuf:"bytes,5,opt,name=rpc_doc_title,json=rpcDocTitle" json:"rpc_doc_title,omitempty" query:"rpc_doc_title"` + RpcDocTitle *string `protobuf:"bytes,5,opt,name=rpc_doc_title,json=rpcDocTitle" json:"rpc_doc_title,omitempty" query:"rpc_doc_title" params:"rpc_doc_title"` } func (x *DatabricksRpcOptions) Reset() { @@ -675,12 +675,12 @@ type HttpEndpoint struct { unknownFields protoimpl.UnknownFields // HTTP method like POST or GET. - Method *string `protobuf:"bytes,1,opt,name=method,def=POST" json:"method,omitempty" query:"method"` + Method *string `protobuf:"bytes,1,opt,name=method,def=POST" json:"method,omitempty" query:"method" params:"method"` // Conceptual path of the API, like "/clusters" or "/clusters/create". Should start with a slash. - Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path"` + Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path" params:"path"` // A version like 1.1 which is prepended to the URL (e.g., GET /1.1/clusters). // Breaking changes to an RPC must use a different version number. - Since *ApiVersion `protobuf:"bytes,3,opt,name=since" json:"since,omitempty" query:"since"` + Since *ApiVersion `protobuf:"bytes,3,opt,name=since" json:"since,omitempty" query:"since" params:"since"` } // Default values for HttpEndpoint fields. @@ -746,8 +746,8 @@ type ApiVersion struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - Major *int32 `protobuf:"varint,1,opt,name=major" json:"major,omitempty" query:"major"` - Minor *int32 `protobuf:"varint,2,opt,name=minor" json:"minor,omitempty" query:"minor"` + Major *int32 `protobuf:"varint,1,opt,name=major" json:"major,omitempty" query:"major" params:"major"` + Minor *int32 `protobuf:"varint,2,opt,name=minor" json:"minor,omitempty" query:"minor" params:"minor"` } func (x *ApiVersion) Reset() { @@ -805,10 +805,10 @@ type RateLimit struct { // The maximum burst of API requests allowed for a single endpoint. In the context of the // token bucket algorithm, this constant represents the total capacity of the token bucket. - MaxBurst *int64 `protobuf:"varint,1,opt,name=max_burst,json=maxBurst" json:"max_burst,omitempty" query:"max_burst"` + MaxBurst *int64 `protobuf:"varint,1,opt,name=max_burst,json=maxBurst" json:"max_burst,omitempty" query:"max_burst" params:"max_burst"` // The maximum sustained request per second limit for a single endpoint. In the context of the, // token bucket algorithm, this constant represents the rate at which the token bucket fills. - MaxSustainedPerSecond *int64 `protobuf:"varint,2,opt,name=max_sustained_per_second,json=maxSustainedPerSecond" json:"max_sustained_per_second,omitempty" query:"max_sustained_per_second"` + MaxSustainedPerSecond *int64 `protobuf:"varint,2,opt,name=max_sustained_per_second,json=maxSustainedPerSecond" json:"max_sustained_per_second,omitempty" query:"max_sustained_per_second" params:"max_sustained_per_second"` } func (x *RateLimit) Reset() { @@ -864,23 +864,23 @@ type DocumentationMetadata struct { unknownFields protoimpl.UnknownFields // The string of documentation attached to this particular item. - Docstring *string `protobuf:"bytes,1,opt,name=docstring" json:"docstring,omitempty" query:"docstring"` + Docstring *string `protobuf:"bytes,1,opt,name=docstring" json:"docstring,omitempty" query:"docstring" params:"docstring"` // The string of documentation that is *before* this item. This only makes sense for top-level // items such as (top-level) messages, (top-level) enumerations, or services. In all other // cases, this string is empty. - LeadDoc *string `protobuf:"bytes,2,opt,name=lead_doc,json=leadDoc" json:"lead_doc,omitempty" query:"lead_doc"` + LeadDoc *string `protobuf:"bytes,2,opt,name=lead_doc,json=leadDoc" json:"lead_doc,omitempty" query:"lead_doc" params:"lead_doc"` // The visibility level when the docstring was generated. // The documentation extractor builds multiple versions of the documentation, one for each // visibility level. The documentation is then generated for each visibility level. - Visibility *Visibility `protobuf:"varint,3,opt,name=visibility,enum=mlflow.Visibility" json:"visibility,omitempty" query:"visibility"` + Visibility *Visibility `protobuf:"varint,3,opt,name=visibility,enum=mlflow.Visibility" json:"visibility,omitempty" query:"visibility" params:"visibility"` // The original proto path in the internal representation. This is useful when performing field // flattening to figure out what the original field was. // One example is ["jobs","Run","original_attempt_run_id"] for jobs. // This path is unique. - OriginalProtoPath []string `protobuf:"bytes,4,rep,name=original_proto_path,json=originalProtoPath" json:"original_proto_path,omitempty" query:"original_proto_path"` + OriginalProtoPath []string `protobuf:"bytes,4,rep,name=original_proto_path,json=originalProtoPath" json:"original_proto_path,omitempty" query:"original_proto_path" params:"original_proto_path"` // The location (line number) of the start of the documentation. This is required to keep the // pieces of documentation sorted. - Position *int32 `protobuf:"varint,5,opt,name=position" json:"position,omitempty" query:"position"` + Position *int32 `protobuf:"varint,5,opt,name=position" json:"position,omitempty" query:"position" params:"position"` } func (x *DocumentationMetadata) Reset() { diff --git a/pkg/protos/databricks_artifacts.pb.go b/pkg/protos/databricks_artifacts.pb.go index beddd2bd..0079a438 100644 --- a/pkg/protos/databricks_artifacts.pb.go +++ b/pkg/protos/databricks_artifacts.pb.go @@ -102,18 +102,18 @@ type ArtifactCredentialInfo struct { // The ID of the MLflow Run containing the artifact that can be accessed // with the credential - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // The path, relative to the Run's artifact root location, of the artifact // that can be accessed with the credential - Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path"` + Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path" params:"path"` // The signed URI credential that provides access to the artifact - SignedUri *string `protobuf:"bytes,3,opt,name=signed_uri,json=signedUri" json:"signed_uri,omitempty" query:"signed_uri"` + SignedUri *string `protobuf:"bytes,3,opt,name=signed_uri,json=signedUri" json:"signed_uri,omitempty" query:"signed_uri" params:"signed_uri"` // A collection of HTTP headers that should be specified when uploading to // or downloading from the specified `signed_uri` - Headers []*ArtifactCredentialInfo_HttpHeader `protobuf:"bytes,4,rep,name=headers" json:"headers,omitempty" query:"headers"` + Headers []*ArtifactCredentialInfo_HttpHeader `protobuf:"bytes,4,rep,name=headers" json:"headers,omitempty" query:"headers" params:"headers"` // The type of the signed credential URI (e.g., an AWS presigned URL // or an Azure Shared Access Signature URI) - Type *ArtifactCredentialType `protobuf:"varint,5,opt,name=type,enum=mlflow.ArtifactCredentialType" json:"type,omitempty" query:"type"` + Type *ArtifactCredentialType `protobuf:"varint,5,opt,name=type,enum=mlflow.ArtifactCredentialType" json:"type,omitempty" query:"type" params:"type"` } func (x *ArtifactCredentialInfo) Reset() { @@ -189,12 +189,12 @@ type GetCredentialsForRead struct { unknownFields protoimpl.UnknownFields // The ID of the MLflow Run for which to fetch artifact read credentials - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // The artifact paths, relative to the Run's artifact root location, for which to // fetch artifact read credentials. Must not be empty. - Path []string `protobuf:"bytes,2,rep,name=path" json:"path,omitempty" query:"path"` + Path []string `protobuf:"bytes,2,rep,name=path" json:"path,omitempty" query:"path" params:"path"` // Token specifying the page of credentials to fetch for large requests that require pagination - PageToken *string `protobuf:"bytes,3,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token"` + PageToken *string `protobuf:"bytes,3,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token" params:"page_token"` } func (x *GetCredentialsForRead) Reset() { @@ -256,12 +256,12 @@ type GetCredentialsForWrite struct { unknownFields protoimpl.UnknownFields // The ID of the MLflow Run for which to fetch artifact write credentials - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // The artifact paths, relative to the Run's artifact root location, for which to // fetch artifact write credentials. Must not be empty. - Path []string `protobuf:"bytes,2,rep,name=path" json:"path,omitempty" query:"path"` + Path []string `protobuf:"bytes,2,rep,name=path" json:"path,omitempty" query:"path" params:"path"` // Token specifying the page of credentials to fetch for large requests that require pagination - PageToken *string `protobuf:"bytes,3,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token"` + PageToken *string `protobuf:"bytes,3,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token" params:"page_token"` } func (x *GetCredentialsForWrite) Reset() { @@ -323,11 +323,11 @@ type CreateMultipartUpload struct { unknownFields protoimpl.UnknownFields // Run ID - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // Artifact path, relative to the Run's artifact root location (e.g. "path/to/file") - Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path"` + Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path" params:"path"` // Number of file parts (chunks of data) to upload in the initiated multipart upload - NumParts *int64 `protobuf:"varint,3,opt,name=num_parts,json=numParts" json:"num_parts,omitempty" query:"num_parts"` + NumParts *int64 `protobuf:"varint,3,opt,name=num_parts,json=numParts" json:"num_parts,omitempty" query:"num_parts" params:"num_parts"` } func (x *CreateMultipartUpload) Reset() { @@ -388,8 +388,8 @@ type PartEtag struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - PartNumber *int64 `protobuf:"varint,1,opt,name=part_number,json=partNumber" json:"part_number,omitempty" query:"part_number"` - Etag *string `protobuf:"bytes,2,opt,name=etag" json:"etag,omitempty" query:"etag"` + PartNumber *int64 `protobuf:"varint,1,opt,name=part_number,json=partNumber" json:"part_number,omitempty" query:"part_number" params:"part_number"` + Etag *string `protobuf:"bytes,2,opt,name=etag" json:"etag,omitempty" query:"etag" params:"etag"` } func (x *PartEtag) Reset() { @@ -444,13 +444,13 @@ type CompleteMultipartUpload struct { unknownFields protoimpl.UnknownFields // Run ID - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // Artifact path, relative to the Run's artifact root location (e.g. "path/to/file") - Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path"` + Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path" params:"path"` // ID identifying the multipart upload to complete - UploadId *string `protobuf:"bytes,3,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id"` + UploadId *string `protobuf:"bytes,3,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id" params:"upload_id"` // A list of file parts uploaded in the multipart upload to complete - PartEtags []*PartEtag `protobuf:"bytes,4,rep,name=part_etags,json=partEtags" json:"part_etags,omitempty" query:"part_etags"` + PartEtags []*PartEtag `protobuf:"bytes,4,rep,name=part_etags,json=partEtags" json:"part_etags,omitempty" query:"part_etags" params:"part_etags"` } func (x *CompleteMultipartUpload) Reset() { @@ -519,13 +519,13 @@ type GetPresignedUploadPartUrl struct { unknownFields protoimpl.UnknownFields // Run ID - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // Atifact path, relative to the Run's artifact root location (e.g. "path/to/file") - Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path"` + Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path" params:"path"` // ID identifying the multipart upload in which the part is uploaded - UploadId *string `protobuf:"bytes,3,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id"` + UploadId *string `protobuf:"bytes,3,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id" params:"upload_id"` // Part number - PartNumber *int64 `protobuf:"varint,4,opt,name=part_number,json=partNumber" json:"part_number,omitempty" query:"part_number"` + PartNumber *int64 `protobuf:"varint,4,opt,name=part_number,json=partNumber" json:"part_number,omitempty" query:"part_number" params:"part_number"` } func (x *GetPresignedUploadPartUrl) Reset() { @@ -593,7 +593,7 @@ type GetCredentialsForTraceDataDownload struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id"` + RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id" params:"request_id"` } func (x *GetCredentialsForTraceDataDownload) Reset() { @@ -640,7 +640,7 @@ type GetCredentialsForTraceDataUpload struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id"` + RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id" params:"request_id"` } func (x *GetCredentialsForTraceDataUpload) Reset() { @@ -688,9 +688,9 @@ type ArtifactCredentialInfo_HttpHeader struct { unknownFields protoimpl.UnknownFields // The HTTP header name - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // The HTTP header value - Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *ArtifactCredentialInfo_HttpHeader) Reset() { @@ -745,9 +745,9 @@ type GetCredentialsForRead_Response struct { unknownFields protoimpl.UnknownFields // Credentials for reading from the specified artifact locations - CredentialInfos []*ArtifactCredentialInfo `protobuf:"bytes,2,rep,name=credential_infos,json=credentialInfos" json:"credential_infos,omitempty" query:"credential_infos"` + CredentialInfos []*ArtifactCredentialInfo `protobuf:"bytes,2,rep,name=credential_infos,json=credentialInfos" json:"credential_infos,omitempty" query:"credential_infos" params:"credential_infos"` // Token used to fetch the next page of credentials for large requests that require pagination - NextPageToken *string `protobuf:"bytes,3,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token"` + NextPageToken *string `protobuf:"bytes,3,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token" params:"next_page_token"` } func (x *GetCredentialsForRead_Response) Reset() { @@ -802,9 +802,9 @@ type GetCredentialsForWrite_Response struct { unknownFields protoimpl.UnknownFields // Credentials for writing to the specified artifact locations - CredentialInfos []*ArtifactCredentialInfo `protobuf:"bytes,2,rep,name=credential_infos,json=credentialInfos" json:"credential_infos,omitempty" query:"credential_infos"` + CredentialInfos []*ArtifactCredentialInfo `protobuf:"bytes,2,rep,name=credential_infos,json=credentialInfos" json:"credential_infos,omitempty" query:"credential_infos" params:"credential_infos"` // Token used to fetch the next page of credentials for large requests that require pagination - NextPageToken *string `protobuf:"bytes,3,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token"` + NextPageToken *string `protobuf:"bytes,3,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token" params:"next_page_token"` } func (x *GetCredentialsForWrite_Response) Reset() { @@ -859,11 +859,11 @@ type CreateMultipartUpload_Response struct { unknownFields protoimpl.UnknownFields // ID identifying the initiated multipart upload - UploadId *string `protobuf:"bytes,1,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id"` + UploadId *string `protobuf:"bytes,1,opt,name=upload_id,json=uploadId" json:"upload_id,omitempty" query:"upload_id" params:"upload_id"` // Credentials for uploading file parts in the initiated multipart upload - UploadCredentialInfos []*ArtifactCredentialInfo `protobuf:"bytes,2,rep,name=upload_credential_infos,json=uploadCredentialInfos" json:"upload_credential_infos,omitempty" query:"upload_credential_infos"` + UploadCredentialInfos []*ArtifactCredentialInfo `protobuf:"bytes,2,rep,name=upload_credential_infos,json=uploadCredentialInfos" json:"upload_credential_infos,omitempty" query:"upload_credential_infos" params:"upload_credential_infos"` // Credential for aborting the initiated multipart upload - AbortCredentialInfo *ArtifactCredentialInfo `protobuf:"bytes,3,opt,name=abort_credential_info,json=abortCredentialInfo" json:"abort_credential_info,omitempty" query:"abort_credential_info"` + AbortCredentialInfo *ArtifactCredentialInfo `protobuf:"bytes,3,opt,name=abort_credential_info,json=abortCredentialInfo" json:"abort_credential_info,omitempty" query:"abort_credential_info" params:"abort_credential_info"` } func (x *CreateMultipartUpload_Response) Reset() { @@ -963,7 +963,7 @@ type GetPresignedUploadPartUrl_Response struct { unknownFields protoimpl.UnknownFields // Credential for uploading the part - UploadCredentialInfo *ArtifactCredentialInfo `protobuf:"bytes,1,opt,name=upload_credential_info,json=uploadCredentialInfo" json:"upload_credential_info,omitempty" query:"upload_credential_info"` + UploadCredentialInfo *ArtifactCredentialInfo `protobuf:"bytes,1,opt,name=upload_credential_info,json=uploadCredentialInfo" json:"upload_credential_info,omitempty" query:"upload_credential_info" params:"upload_credential_info"` } func (x *GetPresignedUploadPartUrl_Response) Reset() { @@ -1010,7 +1010,7 @@ type GetCredentialsForTraceDataDownload_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - CredentialInfo *ArtifactCredentialInfo `protobuf:"bytes,1,opt,name=credential_info,json=credentialInfo" json:"credential_info,omitempty" query:"credential_info"` + CredentialInfo *ArtifactCredentialInfo `protobuf:"bytes,1,opt,name=credential_info,json=credentialInfo" json:"credential_info,omitempty" query:"credential_info" params:"credential_info"` } func (x *GetCredentialsForTraceDataDownload_Response) Reset() { @@ -1057,7 +1057,7 @@ type GetCredentialsForTraceDataUpload_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - CredentialInfo *ArtifactCredentialInfo `protobuf:"bytes,1,opt,name=credential_info,json=credentialInfo" json:"credential_info,omitempty" query:"credential_info"` + CredentialInfo *ArtifactCredentialInfo `protobuf:"bytes,1,opt,name=credential_info,json=credentialInfo" json:"credential_info,omitempty" query:"credential_info" params:"credential_info"` } func (x *GetCredentialsForTraceDataUpload_Response) Reset() { diff --git a/pkg/protos/model_registry.pb.go b/pkg/protos/model_registry.pb.go index a6aec3f3..36514bfd 100644 --- a/pkg/protos/model_registry.pb.go +++ b/pkg/protos/model_registry.pb.go @@ -89,23 +89,23 @@ type RegisteredModel struct { unknownFields protoimpl.UnknownFields // Unique name for the model. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Timestamp recorded when this “registered_model“ was created. - CreationTimestamp *int64 `protobuf:"varint,2,opt,name=creation_timestamp,json=creationTimestamp" json:"creation_timestamp,omitempty" query:"creation_timestamp"` + CreationTimestamp *int64 `protobuf:"varint,2,opt,name=creation_timestamp,json=creationTimestamp" json:"creation_timestamp,omitempty" query:"creation_timestamp" params:"creation_timestamp"` // Timestamp recorded when metadata for this “registered_model“ was last updated. - LastUpdatedTimestamp *int64 `protobuf:"varint,3,opt,name=last_updated_timestamp,json=lastUpdatedTimestamp" json:"last_updated_timestamp,omitempty" query:"last_updated_timestamp"` + LastUpdatedTimestamp *int64 `protobuf:"varint,3,opt,name=last_updated_timestamp,json=lastUpdatedTimestamp" json:"last_updated_timestamp,omitempty" query:"last_updated_timestamp" params:"last_updated_timestamp"` // User that created this “registered_model“ // NOTE: this field is not currently returned. - UserId *string `protobuf:"bytes,4,opt,name=user_id,json=userId" json:"user_id,omitempty" query:"user_id"` + UserId *string `protobuf:"bytes,4,opt,name=user_id,json=userId" json:"user_id,omitempty" query:"user_id" params:"user_id"` // Description of this “registered_model“. - Description *string `protobuf:"bytes,5,opt,name=description" json:"description,omitempty" query:"description"` + Description *string `protobuf:"bytes,5,opt,name=description" json:"description,omitempty" query:"description" params:"description"` // Collection of latest model versions for each stage. // Only contains models with current “READY“ status. - LatestVersions []*ModelVersion `protobuf:"bytes,6,rep,name=latest_versions,json=latestVersions" json:"latest_versions,omitempty" query:"latest_versions"` + LatestVersions []*ModelVersion `protobuf:"bytes,6,rep,name=latest_versions,json=latestVersions" json:"latest_versions,omitempty" query:"latest_versions" params:"latest_versions"` // Tags: Additional metadata key-value pairs for this “registered_model“. - Tags []*RegisteredModelTag `protobuf:"bytes,7,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*RegisteredModelTag `protobuf:"bytes,7,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` // Aliases pointing to model versions associated with this “registered_model“. - Aliases []*RegisteredModelAlias `protobuf:"bytes,8,rep,name=aliases" json:"aliases,omitempty" query:"aliases"` + Aliases []*RegisteredModelAlias `protobuf:"bytes,8,rep,name=aliases" json:"aliases,omitempty" query:"aliases" params:"aliases"` } func (x *RegisteredModel) Reset() { @@ -202,35 +202,35 @@ type ModelVersion struct { unknownFields protoimpl.UnknownFields // Unique name of the model - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Model's version number. - Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version" params:"version"` // Timestamp recorded when this “model_version“ was created. - CreationTimestamp *int64 `protobuf:"varint,3,opt,name=creation_timestamp,json=creationTimestamp" json:"creation_timestamp,omitempty" query:"creation_timestamp"` + CreationTimestamp *int64 `protobuf:"varint,3,opt,name=creation_timestamp,json=creationTimestamp" json:"creation_timestamp,omitempty" query:"creation_timestamp" params:"creation_timestamp"` // Timestamp recorded when metadata for this “model_version“ was last updated. - LastUpdatedTimestamp *int64 `protobuf:"varint,4,opt,name=last_updated_timestamp,json=lastUpdatedTimestamp" json:"last_updated_timestamp,omitempty" query:"last_updated_timestamp"` + LastUpdatedTimestamp *int64 `protobuf:"varint,4,opt,name=last_updated_timestamp,json=lastUpdatedTimestamp" json:"last_updated_timestamp,omitempty" query:"last_updated_timestamp" params:"last_updated_timestamp"` // User that created this “model_version“. - UserId *string `protobuf:"bytes,5,opt,name=user_id,json=userId" json:"user_id,omitempty" query:"user_id"` + UserId *string `protobuf:"bytes,5,opt,name=user_id,json=userId" json:"user_id,omitempty" query:"user_id" params:"user_id"` // Current stage for this “model_version“. - CurrentStage *string `protobuf:"bytes,6,opt,name=current_stage,json=currentStage" json:"current_stage,omitempty" query:"current_stage"` + CurrentStage *string `protobuf:"bytes,6,opt,name=current_stage,json=currentStage" json:"current_stage,omitempty" query:"current_stage" params:"current_stage"` // Description of this “model_version“. - Description *string `protobuf:"bytes,7,opt,name=description" json:"description,omitempty" query:"description"` + Description *string `protobuf:"bytes,7,opt,name=description" json:"description,omitempty" query:"description" params:"description"` // URI indicating the location of the source model artifacts, used when creating “model_version“ - Source *string `protobuf:"bytes,8,opt,name=source" json:"source,omitempty" query:"source"` + Source *string `protobuf:"bytes,8,opt,name=source" json:"source,omitempty" query:"source" params:"source"` // MLflow run ID used when creating “model_version“, if “source“ was generated by an // experiment run stored in MLflow tracking server. - RunId *string `protobuf:"bytes,9,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,9,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // Current status of “model_version“ - Status *ModelVersionStatus `protobuf:"varint,10,opt,name=status,enum=mlflow.ModelVersionStatus" json:"status,omitempty" query:"status"` + Status *ModelVersionStatus `protobuf:"varint,10,opt,name=status,enum=mlflow.ModelVersionStatus" json:"status,omitempty" query:"status" params:"status"` // Details on current “status“, if it is pending or failed. - StatusMessage *string `protobuf:"bytes,11,opt,name=status_message,json=statusMessage" json:"status_message,omitempty" query:"status_message"` + StatusMessage *string `protobuf:"bytes,11,opt,name=status_message,json=statusMessage" json:"status_message,omitempty" query:"status_message" params:"status_message"` // Tags: Additional metadata key-value pairs for this “model_version“. - Tags []*ModelVersionTag `protobuf:"bytes,12,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*ModelVersionTag `protobuf:"bytes,12,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` // Run Link: Direct link to the run that generated this version. This field is set at model version creation time // only for model versions whose source run is from a tracking server that is different from the registry server. - RunLink *string `protobuf:"bytes,13,opt,name=run_link,json=runLink" json:"run_link,omitempty" query:"run_link"` + RunLink *string `protobuf:"bytes,13,opt,name=run_link,json=runLink" json:"run_link,omitempty" query:"run_link" params:"run_link"` // Aliases pointing to this “model_version“. - Aliases []string `protobuf:"bytes,14,rep,name=aliases" json:"aliases,omitempty" query:"aliases"` + Aliases []string `protobuf:"bytes,14,rep,name=aliases" json:"aliases,omitempty" query:"aliases" params:"aliases"` } func (x *ModelVersion) Reset() { @@ -369,11 +369,11 @@ type CreateRegisteredModel struct { unknownFields protoimpl.UnknownFields // Register models under this name - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Additional metadata for registered model. - Tags []*RegisteredModelTag `protobuf:"bytes,2,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*RegisteredModelTag `protobuf:"bytes,2,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` // Optional description for registered model. - Description *string `protobuf:"bytes,3,opt,name=description" json:"description,omitempty" query:"description"` + Description *string `protobuf:"bytes,3,opt,name=description" json:"description,omitempty" query:"description" params:"description"` } func (x *CreateRegisteredModel) Reset() { @@ -435,9 +435,9 @@ type RenameRegisteredModel struct { unknownFields protoimpl.UnknownFields // Registered model unique name identifier. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // If provided, updates the name for this “registered_model“. - NewName *string `protobuf:"bytes,2,opt,name=new_name,json=newName" json:"new_name,omitempty" query:"new_name"` + NewName *string `protobuf:"bytes,2,opt,name=new_name,json=newName" json:"new_name,omitempty" query:"new_name" params:"new_name"` } func (x *RenameRegisteredModel) Reset() { @@ -492,9 +492,9 @@ type UpdateRegisteredModel struct { unknownFields protoimpl.UnknownFields // Registered model unique name identifier. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // If provided, updates the description for this “registered_model“. - Description *string `protobuf:"bytes,2,opt,name=description" json:"description,omitempty" query:"description"` + Description *string `protobuf:"bytes,2,opt,name=description" json:"description,omitempty" query:"description" params:"description"` } func (x *UpdateRegisteredModel) Reset() { @@ -549,7 +549,7 @@ type DeleteRegisteredModel struct { unknownFields protoimpl.UnknownFields // Registered model unique name identifier. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` } func (x *DeleteRegisteredModel) Reset() { @@ -597,7 +597,7 @@ type GetRegisteredModel struct { unknownFields protoimpl.UnknownFields // Registered model unique name identifier. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` } func (x *GetRegisteredModel) Reset() { @@ -647,15 +647,15 @@ type SearchRegisteredModels struct { // String filter condition, like "name LIKE 'my-model-name'". // Interpreted in the backend automatically as "name LIKE '%my-model-name%'". // Single boolean condition, with string values wrapped in single quotes. - Filter *string `protobuf:"bytes,1,opt,name=filter" json:"filter,omitempty" query:"filter"` + Filter *string `protobuf:"bytes,1,opt,name=filter" json:"filter,omitempty" query:"filter" params:"filter"` // Maximum number of models desired. Default is 100. Max threshold is 1000. - MaxResults *int64 `protobuf:"varint,2,opt,name=max_results,json=maxResults,def=100" json:"max_results,omitempty" query:"max_results"` + MaxResults *int64 `protobuf:"varint,2,opt,name=max_results,json=maxResults,def=100" json:"max_results,omitempty" query:"max_results" params:"max_results"` // List of columns for ordering search results, which can include model name and last updated // timestamp with an optional "DESC" or "ASC" annotation, where "ASC" is the default. // Tiebreaks are done by model name ASC. - OrderBy []string `protobuf:"bytes,3,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by"` + OrderBy []string `protobuf:"bytes,3,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by" params:"order_by"` // Pagination token to go to the next page based on a previous search query. - PageToken *string `protobuf:"bytes,4,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token"` + PageToken *string `protobuf:"bytes,4,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token" params:"page_token"` } // Default values for SearchRegisteredModels fields. @@ -729,9 +729,9 @@ type GetLatestVersions struct { unknownFields protoimpl.UnknownFields // Registered model unique name identifier. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" validate:"required"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name" validate:"required"` // List of stages. - Stages []string `protobuf:"bytes,2,rep,name=stages" json:"stages,omitempty" query:"stages"` + Stages []string `protobuf:"bytes,2,rep,name=stages" json:"stages,omitempty" query:"stages" params:"stages"` } func (x *GetLatestVersions) Reset() { @@ -786,19 +786,19 @@ type CreateModelVersion struct { unknownFields protoimpl.UnknownFields // Register model under this name - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // URI indicating the location of the model artifacts. - Source *string `protobuf:"bytes,2,opt,name=source" json:"source,omitempty" query:"source"` + Source *string `protobuf:"bytes,2,opt,name=source" json:"source,omitempty" query:"source" params:"source"` // MLflow run ID for correlation, if “source“ was generated by an experiment run in // MLflow tracking server - RunId *string `protobuf:"bytes,3,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,3,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // Additional metadata for model version. - Tags []*ModelVersionTag `protobuf:"bytes,4,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*ModelVersionTag `protobuf:"bytes,4,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` // MLflow run link - this is the exact link of the run that generated this model version, // potentially hosted at another instance of MLflow. - RunLink *string `protobuf:"bytes,5,opt,name=run_link,json=runLink" json:"run_link,omitempty" query:"run_link"` + RunLink *string `protobuf:"bytes,5,opt,name=run_link,json=runLink" json:"run_link,omitempty" query:"run_link" params:"run_link"` // Optional description for model version. - Description *string `protobuf:"bytes,6,opt,name=description" json:"description,omitempty" query:"description"` + Description *string `protobuf:"bytes,6,opt,name=description" json:"description,omitempty" query:"description" params:"description"` } func (x *CreateModelVersion) Reset() { @@ -881,11 +881,11 @@ type UpdateModelVersion struct { unknownFields protoimpl.UnknownFields // Name of the registered model - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Model version number - Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version" params:"version"` // If provided, updates the description for this “registered_model“. - Description *string `protobuf:"bytes,3,opt,name=description" json:"description,omitempty" query:"description"` + Description *string `protobuf:"bytes,3,opt,name=description" json:"description,omitempty" query:"description" params:"description"` } func (x *UpdateModelVersion) Reset() { @@ -947,16 +947,16 @@ type TransitionModelVersionStage struct { unknownFields protoimpl.UnknownFields // Name of the registered model - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Model version number - Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version" params:"version"` // Transition `model_version` to new stage. - Stage *string `protobuf:"bytes,3,opt,name=stage" json:"stage,omitempty" query:"stage"` + Stage *string `protobuf:"bytes,3,opt,name=stage" json:"stage,omitempty" query:"stage" params:"stage"` // When transitioning a model version to a particular stage, this flag dictates whether all // existing model versions in that stage should be atomically moved to the "archived" stage. // This ensures that at-most-one model version exists in the target stage. // This field is *required* when transitioning a model versions's stage - ArchiveExistingVersions *bool `protobuf:"varint,4,opt,name=archive_existing_versions,json=archiveExistingVersions" json:"archive_existing_versions,omitempty" query:"archive_existing_versions"` + ArchiveExistingVersions *bool `protobuf:"varint,4,opt,name=archive_existing_versions,json=archiveExistingVersions" json:"archive_existing_versions,omitempty" query:"archive_existing_versions" params:"archive_existing_versions"` } func (x *TransitionModelVersionStage) Reset() { @@ -1025,9 +1025,9 @@ type DeleteModelVersion struct { unknownFields protoimpl.UnknownFields // Name of the registered model - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Model version number - Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version" params:"version"` } func (x *DeleteModelVersion) Reset() { @@ -1082,9 +1082,9 @@ type GetModelVersion struct { unknownFields protoimpl.UnknownFields // Name of the registered model - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Model version number - Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version" params:"version"` } func (x *GetModelVersion) Reset() { @@ -1140,17 +1140,17 @@ type SearchModelVersions struct { // String filter condition, like "name='my-model-name'". Must be a single boolean condition, // with string values wrapped in single quotes. - Filter *string `protobuf:"bytes,1,opt,name=filter" json:"filter,omitempty" query:"filter"` + Filter *string `protobuf:"bytes,1,opt,name=filter" json:"filter,omitempty" query:"filter" params:"filter"` // Maximum number of models desired. Max threshold is 200K. Backends may choose a lower default // value and maximum threshold. - MaxResults *int64 `protobuf:"varint,2,opt,name=max_results,json=maxResults,def=200000" json:"max_results,omitempty" query:"max_results"` + MaxResults *int64 `protobuf:"varint,2,opt,name=max_results,json=maxResults,def=200000" json:"max_results,omitempty" query:"max_results" params:"max_results"` // List of columns to be ordered by including model name, version, stage with an // optional "DESC" or "ASC" annotation, where "ASC" is the default. // Tiebreaks are done by latest stage transition timestamp, followed by name ASC, followed by // version DESC. - OrderBy []string `protobuf:"bytes,3,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by"` + OrderBy []string `protobuf:"bytes,3,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by" params:"order_by"` // Pagination token to go to next page based on previous search query. - PageToken *string `protobuf:"bytes,4,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token"` + PageToken *string `protobuf:"bytes,4,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token" params:"page_token"` } // Default values for SearchModelVersions fields. @@ -1224,9 +1224,9 @@ type GetModelVersionDownloadUri struct { unknownFields protoimpl.UnknownFields // Name of the registered model - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Model version number - Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version" params:"version"` } func (x *GetModelVersionDownloadUri) Reset() { @@ -1282,9 +1282,9 @@ type ModelVersionTag struct { unknownFields protoimpl.UnknownFields // The tag key. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // The tag value. - Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *ModelVersionTag) Reset() { @@ -1340,9 +1340,9 @@ type RegisteredModelTag struct { unknownFields protoimpl.UnknownFields // The tag key. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // The tag value. - Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *RegisteredModelTag) Reset() { @@ -1397,13 +1397,13 @@ type SetRegisteredModelTag struct { unknownFields protoimpl.UnknownFields // Unique name of the model. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Name of the tag. Maximum size depends on storage backend. // If a tag with this name already exists, its preexisting value will be replaced by the specified `value`. // All storage backends are guaranteed to support key values up to 250 bytes in size. - Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // String value of the tag being logged. Maximum size depends on storage backend. - Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *SetRegisteredModelTag) Reset() { @@ -1465,15 +1465,15 @@ type SetModelVersionTag struct { unknownFields protoimpl.UnknownFields // Unique name of the model. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Model version number. - Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version" params:"version"` // Name of the tag. Maximum size depends on storage backend. // If a tag with this name already exists, its preexisting value will be replaced by the specified `value`. // All storage backends are guaranteed to support key values up to 250 bytes in size. - Key *string `protobuf:"bytes,3,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,3,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // String value of the tag being logged. Maximum size depends on storage backend. - Value *string `protobuf:"bytes,4,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,4,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *SetModelVersionTag) Reset() { @@ -1542,9 +1542,9 @@ type DeleteRegisteredModelTag struct { unknownFields protoimpl.UnknownFields // Name of the registered model that the tag was logged under. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Name of the tag. The name must be an exact match; wild-card deletion is not supported. Maximum size is 250 bytes. - Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" params:"key"` } func (x *DeleteRegisteredModelTag) Reset() { @@ -1599,11 +1599,11 @@ type DeleteModelVersionTag struct { unknownFields protoimpl.UnknownFields // Name of the registered model that the tag was logged under. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Model version number that the tag was logged under. - Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version" params:"version"` // Name of the tag. The name must be an exact match; wild-card deletion is not supported. Maximum size is 250 bytes. - Key *string `protobuf:"bytes,3,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,3,opt,name=key" json:"key,omitempty" query:"key" params:"key"` } func (x *DeleteModelVersionTag) Reset() { @@ -1666,9 +1666,9 @@ type RegisteredModelAlias struct { unknownFields protoimpl.UnknownFields // The name of the alias. - Alias *string `protobuf:"bytes,1,opt,name=alias" json:"alias,omitempty" query:"alias"` + Alias *string `protobuf:"bytes,1,opt,name=alias" json:"alias,omitempty" query:"alias" params:"alias"` // The model version number that the alias points to. - Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,2,opt,name=version" json:"version,omitempty" query:"version" params:"version"` } func (x *RegisteredModelAlias) Reset() { @@ -1723,13 +1723,13 @@ type SetRegisteredModelAlias struct { unknownFields protoimpl.UnknownFields // Name of the registered model. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Name of the alias. Maximum size depends on storage backend. // If an alias with this name already exists, its preexisting value will be replaced by the specified `version`. // All storage backends are guaranteed to support alias name values up to 256 bytes in size. - Alias *string `protobuf:"bytes,2,opt,name=alias" json:"alias,omitempty" query:"alias"` + Alias *string `protobuf:"bytes,2,opt,name=alias" json:"alias,omitempty" query:"alias" params:"alias"` // Model version number. - Version *string `protobuf:"bytes,3,opt,name=version" json:"version,omitempty" query:"version"` + Version *string `protobuf:"bytes,3,opt,name=version" json:"version,omitempty" query:"version" params:"version"` } func (x *SetRegisteredModelAlias) Reset() { @@ -1791,9 +1791,9 @@ type DeleteRegisteredModelAlias struct { unknownFields protoimpl.UnknownFields // Name of the registered model. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Name of the alias. The name must be an exact match; wild-card deletion is not supported. Maximum size is 256 bytes. - Alias *string `protobuf:"bytes,2,opt,name=alias" json:"alias,omitempty" query:"alias"` + Alias *string `protobuf:"bytes,2,opt,name=alias" json:"alias,omitempty" query:"alias" params:"alias"` } func (x *DeleteRegisteredModelAlias) Reset() { @@ -1848,9 +1848,9 @@ type GetModelVersionByAlias struct { unknownFields protoimpl.UnknownFields // Name of the registered model. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Name of the alias. Maximum size is 256 bytes. - Alias *string `protobuf:"bytes,2,opt,name=alias" json:"alias,omitempty" query:"alias"` + Alias *string `protobuf:"bytes,2,opt,name=alias" json:"alias,omitempty" query:"alias" params:"alias"` } func (x *GetModelVersionByAlias) Reset() { @@ -1904,7 +1904,7 @@ type CreateRegisteredModel_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - RegisteredModel *RegisteredModel `protobuf:"bytes,1,opt,name=registered_model,json=registeredModel" json:"registered_model,omitempty" query:"registered_model"` + RegisteredModel *RegisteredModel `protobuf:"bytes,1,opt,name=registered_model,json=registeredModel" json:"registered_model,omitempty" query:"registered_model" params:"registered_model"` } func (x *CreateRegisteredModel_Response) Reset() { @@ -1951,7 +1951,7 @@ type RenameRegisteredModel_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - RegisteredModel *RegisteredModel `protobuf:"bytes,1,opt,name=registered_model,json=registeredModel" json:"registered_model,omitempty" query:"registered_model"` + RegisteredModel *RegisteredModel `protobuf:"bytes,1,opt,name=registered_model,json=registeredModel" json:"registered_model,omitempty" query:"registered_model" params:"registered_model"` } func (x *RenameRegisteredModel_Response) Reset() { @@ -1998,7 +1998,7 @@ type UpdateRegisteredModel_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - RegisteredModel *RegisteredModel `protobuf:"bytes,1,opt,name=registered_model,json=registeredModel" json:"registered_model,omitempty" query:"registered_model"` + RegisteredModel *RegisteredModel `protobuf:"bytes,1,opt,name=registered_model,json=registeredModel" json:"registered_model,omitempty" query:"registered_model" params:"registered_model"` } func (x *UpdateRegisteredModel_Response) Reset() { @@ -2083,7 +2083,7 @@ type GetRegisteredModel_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - RegisteredModel *RegisteredModel `protobuf:"bytes,1,opt,name=registered_model,json=registeredModel" json:"registered_model,omitempty" query:"registered_model"` + RegisteredModel *RegisteredModel `protobuf:"bytes,1,opt,name=registered_model,json=registeredModel" json:"registered_model,omitempty" query:"registered_model" params:"registered_model"` } func (x *GetRegisteredModel_Response) Reset() { @@ -2131,9 +2131,9 @@ type SearchRegisteredModels_Response struct { unknownFields protoimpl.UnknownFields // Registered Models that match the search criteria. - RegisteredModels []*RegisteredModel `protobuf:"bytes,1,rep,name=registered_models,json=registeredModels" json:"registered_models,omitempty" query:"registered_models"` + RegisteredModels []*RegisteredModel `protobuf:"bytes,1,rep,name=registered_models,json=registeredModels" json:"registered_models,omitempty" query:"registered_models" params:"registered_models"` // Pagination token to request the next page of models. - NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token"` + NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token" params:"next_page_token"` } func (x *SearchRegisteredModels_Response) Reset() { @@ -2189,7 +2189,7 @@ type GetLatestVersions_Response struct { // Latest version models for each requests stage. Only return models with current “READY“ status. // If no “stages“ provided, returns the latest version for each stage, including “"None"“. - ModelVersions []*ModelVersion `protobuf:"bytes,1,rep,name=model_versions,json=modelVersions" json:"model_versions,omitempty" query:"model_versions"` + ModelVersions []*ModelVersion `protobuf:"bytes,1,rep,name=model_versions,json=modelVersions" json:"model_versions,omitempty" query:"model_versions" params:"model_versions"` } func (x *GetLatestVersions_Response) Reset() { @@ -2237,7 +2237,7 @@ type CreateModelVersion_Response struct { unknownFields protoimpl.UnknownFields // Return new version number generated for this model in registry. - ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version"` + ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version" params:"model_version"` } func (x *CreateModelVersion_Response) Reset() { @@ -2285,7 +2285,7 @@ type UpdateModelVersion_Response struct { unknownFields protoimpl.UnknownFields // Return new version number generated for this model in registry. - ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version"` + ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version" params:"model_version"` } func (x *UpdateModelVersion_Response) Reset() { @@ -2333,7 +2333,7 @@ type TransitionModelVersionStage_Response struct { unknownFields protoimpl.UnknownFields // Updated model version - ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version"` + ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version" params:"model_version"` } func (x *TransitionModelVersionStage_Response) Reset() { @@ -2418,7 +2418,7 @@ type GetModelVersion_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version"` + ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version" params:"model_version"` } func (x *GetModelVersion_Response) Reset() { @@ -2466,9 +2466,9 @@ type SearchModelVersions_Response struct { unknownFields protoimpl.UnknownFields // Models that match the search criteria - ModelVersions []*ModelVersion `protobuf:"bytes,1,rep,name=model_versions,json=modelVersions" json:"model_versions,omitempty" query:"model_versions"` + ModelVersions []*ModelVersion `protobuf:"bytes,1,rep,name=model_versions,json=modelVersions" json:"model_versions,omitempty" query:"model_versions" params:"model_versions"` // Pagination token to request next page of models for the same search query. - NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token"` + NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token" params:"next_page_token"` } func (x *SearchModelVersions_Response) Reset() { @@ -2523,7 +2523,7 @@ type GetModelVersionDownloadUri_Response struct { unknownFields protoimpl.UnknownFields // URI corresponding to where artifacts for this model version are stored. - ArtifactUri *string `protobuf:"bytes,1,opt,name=artifact_uri,json=artifactUri" json:"artifact_uri,omitempty" query:"artifact_uri"` + ArtifactUri *string `protobuf:"bytes,1,opt,name=artifact_uri,json=artifactUri" json:"artifact_uri,omitempty" query:"artifact_uri" params:"artifact_uri"` } func (x *GetModelVersionDownloadUri_Response) Reset() { @@ -2798,7 +2798,7 @@ type GetModelVersionByAlias_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version"` + ModelVersion *ModelVersion `protobuf:"bytes,1,opt,name=model_version,json=modelVersion" json:"model_version,omitempty" query:"model_version" params:"model_version"` } func (x *GetModelVersionByAlias_Response) Reset() { diff --git a/pkg/protos/scalapb/scalapb.pb.go b/pkg/protos/scalapb/scalapb.pb.go index 3b4f0902..c75bba60 100644 --- a/pkg/protos/scalapb/scalapb.pb.go +++ b/pkg/protos/scalapb/scalapb.pb.go @@ -27,15 +27,15 @@ type ScalaPbOptions struct { unknownFields protoimpl.UnknownFields // If set then it overrides the java_package and package. - PackageName *string `protobuf:"bytes,1,opt,name=package_name,json=packageName" json:"package_name,omitempty" query:"package_name"` + PackageName *string `protobuf:"bytes,1,opt,name=package_name,json=packageName" json:"package_name,omitempty" query:"package_name" params:"package_name"` // If true, the compiler does not append the proto base file name // into the generated package name. If false (the default), the // generated scala package name is the package_name.basename where // basename is the proto file name without the .proto extension. - FlatPackage *bool `protobuf:"varint,2,opt,name=flat_package,json=flatPackage" json:"flat_package,omitempty" query:"flat_package"` + FlatPackage *bool `protobuf:"varint,2,opt,name=flat_package,json=flatPackage" json:"flat_package,omitempty" query:"flat_package" params:"flat_package"` // Adds the following imports at the top of the file (this is meant // to provide implicit TypeMappers) - Import []string `protobuf:"bytes,3,rep,name=import" json:"import,omitempty" query:"import"` + Import []string `protobuf:"bytes,3,rep,name=import" json:"import,omitempty" query:"import" params:"import"` } func (x *ScalaPbOptions) Reset() { @@ -97,7 +97,7 @@ type MessageOptions struct { unknownFields protoimpl.UnknownFields // additional classes and traits to mix in to the case class. - Extends []string `protobuf:"bytes,1,rep,name=extends" json:"extends,omitempty" query:"extends"` + Extends []string `protobuf:"bytes,1,rep,name=extends" json:"extends,omitempty" query:"extends" params:"extends"` } func (x *MessageOptions) Reset() { @@ -144,7 +144,7 @@ type FieldOptions struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - Type *string `protobuf:"bytes,1,opt,name=type" json:"type,omitempty" query:"type"` + Type *string `protobuf:"bytes,1,opt,name=type" json:"type,omitempty" query:"type" params:"type"` } func (x *FieldOptions) Reset() { diff --git a/pkg/protos/service.pb.go b/pkg/protos/service.pb.go index 824e4b69..090a0326 100644 --- a/pkg/protos/service.pb.go +++ b/pkg/protos/service.pb.go @@ -298,13 +298,13 @@ type Metric struct { unknownFields protoimpl.UnknownFields // Key identifying this metric. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` // Value associated with this metric. - Value *float64 `protobuf:"fixed64,2,opt,name=value" json:"value,omitempty" query:"value" validate:"required"` + Value *float64 `protobuf:"fixed64,2,opt,name=value" json:"value,omitempty" query:"value" params:"value" validate:"required"` // The timestamp at which this metric was recorded. - Timestamp *int64 `protobuf:"varint,3,opt,name=timestamp" json:"timestamp,omitempty" query:"timestamp" validate:"required"` + Timestamp *int64 `protobuf:"varint,3,opt,name=timestamp" json:"timestamp,omitempty" query:"timestamp" params:"timestamp" validate:"required"` // Step at which to log the metric. - Step *int64 `protobuf:"varint,4,opt,name=step,def=0" json:"step,omitempty" query:"step"` + Step *int64 `protobuf:"varint,4,opt,name=step,def=0" json:"step,omitempty" query:"step" params:"step"` } // Default values for Metric fields. @@ -379,9 +379,9 @@ type Param struct { unknownFields protoimpl.UnknownFields // Key identifying this param. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` // Value associated with this param. - Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" validate:"omitempty,truncate=6000"` + Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" params:"value" validate:"omitempty,truncate=6000"` } func (x *Param) Reset() { @@ -437,11 +437,11 @@ type Run struct { unknownFields protoimpl.UnknownFields // Run metadata. - Info *RunInfo `protobuf:"bytes,1,opt,name=info" json:"info,omitempty" query:"info"` + Info *RunInfo `protobuf:"bytes,1,opt,name=info" json:"info,omitempty" query:"info" params:"info"` // Run data. - Data *RunData `protobuf:"bytes,2,opt,name=data" json:"data,omitempty" query:"data"` + Data *RunData `protobuf:"bytes,2,opt,name=data" json:"data,omitempty" query:"data" params:"data"` // Run inputs. - Inputs *RunInputs `protobuf:"bytes,3,opt,name=inputs" json:"inputs,omitempty" query:"inputs"` + Inputs *RunInputs `protobuf:"bytes,3,opt,name=inputs" json:"inputs,omitempty" query:"inputs" params:"inputs"` } func (x *Run) Reset() { @@ -504,11 +504,11 @@ type RunData struct { unknownFields protoimpl.UnknownFields // Run metrics. - Metrics []*Metric `protobuf:"bytes,1,rep,name=metrics" json:"metrics,omitempty" query:"metrics"` + Metrics []*Metric `protobuf:"bytes,1,rep,name=metrics" json:"metrics,omitempty" query:"metrics" params:"metrics"` // Run parameters. - Params []*Param `protobuf:"bytes,2,rep,name=params" json:"params,omitempty" query:"params"` + Params []*Param `protobuf:"bytes,2,rep,name=params" json:"params,omitempty" query:"params" params:"params"` // Additional metadata key-value pairs. - Tags []*RunTag `protobuf:"bytes,3,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*RunTag `protobuf:"bytes,3,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` } func (x *RunData) Reset() { @@ -571,7 +571,7 @@ type RunInputs struct { unknownFields protoimpl.UnknownFields // Dataset inputs to the Run. - DatasetInputs []*DatasetInput `protobuf:"bytes,1,rep,name=dataset_inputs,json=datasetInputs" json:"dataset_inputs,omitempty" query:"dataset_inputs"` + DatasetInputs []*DatasetInput `protobuf:"bytes,1,rep,name=dataset_inputs,json=datasetInputs" json:"dataset_inputs,omitempty" query:"dataset_inputs" params:"dataset_inputs"` } func (x *RunInputs) Reset() { @@ -620,9 +620,9 @@ type RunTag struct { unknownFields protoimpl.UnknownFields // The tag key. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` // The tag value. - Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" validate:"omitempty,max=5000"` + Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" params:"value" validate:"omitempty,max=5000"` } func (x *RunTag) Reset() { @@ -678,9 +678,9 @@ type ExperimentTag struct { unknownFields protoimpl.UnknownFields // The tag key. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // The tag value. - Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *ExperimentTag) Reset() { @@ -736,31 +736,31 @@ type RunInfo struct { unknownFields protoimpl.UnknownFields // Unique identifier for the run. - RunId *string `protobuf:"bytes,15,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,15,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // [Deprecated, use run_id instead] Unique identifier for the run. This field will // be removed in a future MLflow version. - RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid"` + RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid" params:"run_uuid"` // The name of the run. - RunName *string `protobuf:"bytes,3,opt,name=run_name,json=runName" json:"run_name,omitempty" query:"run_name"` + RunName *string `protobuf:"bytes,3,opt,name=run_name,json=runName" json:"run_name,omitempty" query:"run_name" params:"run_name"` // The experiment ID. - ExperimentId *string `protobuf:"bytes,2,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,2,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` // User who initiated the run. // This field is deprecated as of MLflow 1.0, and will be removed in a future // MLflow release. Use 'mlflow.user' tag instead. - UserId *string `protobuf:"bytes,6,opt,name=user_id,json=userId" json:"user_id,omitempty" query:"user_id"` + UserId *string `protobuf:"bytes,6,opt,name=user_id,json=userId" json:"user_id,omitempty" query:"user_id" params:"user_id"` // Current status of the run. - Status *RunStatus `protobuf:"varint,7,opt,name=status,enum=mlflow.RunStatus" json:"status,omitempty" query:"status"` + Status *RunStatus `protobuf:"varint,7,opt,name=status,enum=mlflow.RunStatus" json:"status,omitempty" query:"status" params:"status"` // Unix timestamp of when the run started in milliseconds. - StartTime *int64 `protobuf:"varint,8,opt,name=start_time,json=startTime" json:"start_time,omitempty" query:"start_time"` + StartTime *int64 `protobuf:"varint,8,opt,name=start_time,json=startTime" json:"start_time,omitempty" query:"start_time" params:"start_time"` // Unix timestamp of when the run ended in milliseconds. - EndTime *int64 `protobuf:"varint,9,opt,name=end_time,json=endTime" json:"end_time,omitempty" query:"end_time"` + EndTime *int64 `protobuf:"varint,9,opt,name=end_time,json=endTime" json:"end_time,omitempty" query:"end_time" params:"end_time"` // URI of the directory where artifacts should be uploaded. // This can be a local path (starting with "/"), or a distributed file system (DFS) // path, like “s3://bucket/directory“ or “dbfs:/my/directory“. // If not set, the local “./mlruns“ directory is chosen. - ArtifactUri *string `protobuf:"bytes,13,opt,name=artifact_uri,json=artifactUri" json:"artifact_uri,omitempty" query:"artifact_uri"` + ArtifactUri *string `protobuf:"bytes,13,opt,name=artifact_uri,json=artifactUri" json:"artifact_uri,omitempty" query:"artifact_uri" params:"artifact_uri"` // Current life cycle stage of the experiment : OneOf("active", "deleted") - LifecycleStage *string `protobuf:"bytes,14,opt,name=lifecycle_stage,json=lifecycleStage" json:"lifecycle_stage,omitempty" query:"lifecycle_stage"` + LifecycleStage *string `protobuf:"bytes,14,opt,name=lifecycle_stage,json=lifecycleStage" json:"lifecycle_stage,omitempty" query:"lifecycle_stage" params:"lifecycle_stage"` } func (x *RunInfo) Reset() { @@ -872,20 +872,20 @@ type Experiment struct { unknownFields protoimpl.UnknownFields // Unique identifier for the experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` // Human readable name that identifies the experiment. - Name *string `protobuf:"bytes,2,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,2,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Location where artifacts for the experiment are stored. - ArtifactLocation *string `protobuf:"bytes,3,opt,name=artifact_location,json=artifactLocation" json:"artifact_location,omitempty" query:"artifact_location"` + ArtifactLocation *string `protobuf:"bytes,3,opt,name=artifact_location,json=artifactLocation" json:"artifact_location,omitempty" query:"artifact_location" params:"artifact_location"` // Current life cycle stage of the experiment: "active" or "deleted". // Deleted experiments are not returned by APIs. - LifecycleStage *string `protobuf:"bytes,4,opt,name=lifecycle_stage,json=lifecycleStage" json:"lifecycle_stage,omitempty" query:"lifecycle_stage"` + LifecycleStage *string `protobuf:"bytes,4,opt,name=lifecycle_stage,json=lifecycleStage" json:"lifecycle_stage,omitempty" query:"lifecycle_stage" params:"lifecycle_stage"` // Last update time - LastUpdateTime *int64 `protobuf:"varint,5,opt,name=last_update_time,json=lastUpdateTime" json:"last_update_time,omitempty" query:"last_update_time"` + LastUpdateTime *int64 `protobuf:"varint,5,opt,name=last_update_time,json=lastUpdateTime" json:"last_update_time,omitempty" query:"last_update_time" params:"last_update_time"` // Creation time - CreationTime *int64 `protobuf:"varint,6,opt,name=creation_time,json=creationTime" json:"creation_time,omitempty" query:"creation_time"` + CreationTime *int64 `protobuf:"varint,6,opt,name=creation_time,json=creationTime" json:"creation_time,omitempty" query:"creation_time" params:"creation_time"` // Tags: Additional metadata key-value pairs. - Tags []*ExperimentTag `protobuf:"bytes,7,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*ExperimentTag `protobuf:"bytes,7,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` } func (x *Experiment) Reset() { @@ -976,9 +976,9 @@ type DatasetInput struct { unknownFields protoimpl.UnknownFields // A list of tags for the dataset input, e.g. a “context” tag with value “training” - Tags []*InputTag `protobuf:"bytes,1,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*InputTag `protobuf:"bytes,1,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` // The dataset being used as a Run input. - Dataset *Dataset `protobuf:"bytes,2,opt,name=dataset" json:"dataset,omitempty" query:"dataset"` + Dataset *Dataset `protobuf:"bytes,2,opt,name=dataset" json:"dataset,omitempty" query:"dataset" params:"dataset"` } func (x *DatasetInput) Reset() { @@ -1034,9 +1034,9 @@ type InputTag struct { unknownFields protoimpl.UnknownFields // The tag key. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // The tag value. - Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *InputTag) Reset() { @@ -1093,22 +1093,22 @@ type Dataset struct { unknownFields protoimpl.UnknownFields // The name of the dataset. E.g. “my.uc.table@2” “nyc-taxi-dataset”, “fantastic-elk-3” - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Dataset digest, e.g. an md5 hash of the dataset that uniquely identifies it // within datasets of the same name. - Digest *string `protobuf:"bytes,2,opt,name=digest" json:"digest,omitempty" query:"digest"` + Digest *string `protobuf:"bytes,2,opt,name=digest" json:"digest,omitempty" query:"digest" params:"digest"` // Source information for the dataset. Note that the source may not exactly reproduce the // dataset if it was transformed / modified before use with MLflow. - SourceType *string `protobuf:"bytes,3,opt,name=source_type,json=sourceType" json:"source_type,omitempty" query:"source_type"` + SourceType *string `protobuf:"bytes,3,opt,name=source_type,json=sourceType" json:"source_type,omitempty" query:"source_type" params:"source_type"` // The type of the dataset source, e.g. ‘databricks-uc-table’, ‘DBFS’, ‘S3’, ... - Source *string `protobuf:"bytes,4,opt,name=source" json:"source,omitempty" query:"source"` + Source *string `protobuf:"bytes,4,opt,name=source" json:"source,omitempty" query:"source" params:"source"` // The schema of the dataset. E.g., MLflow ColSpec JSON for a dataframe, MLflow TensorSpec JSON // for an ndarray, or another schema format. - Schema *string `protobuf:"bytes,5,opt,name=schema" json:"schema,omitempty" query:"schema"` + Schema *string `protobuf:"bytes,5,opt,name=schema" json:"schema,omitempty" query:"schema" params:"schema"` // The profile of the dataset. Summary statistics for the dataset, such as the number of rows // in a table, the mean / std / mode of each column in a table, or the number of elements // in an array. - Profile *string `protobuf:"bytes,6,opt,name=profile" json:"profile,omitempty" query:"profile"` + Profile *string `protobuf:"bytes,6,opt,name=profile" json:"profile,omitempty" query:"profile" params:"profile"` } func (x *Dataset) Reset() { @@ -1191,15 +1191,15 @@ type CreateExperiment struct { unknownFields protoimpl.UnknownFields // Experiment name. - Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" validate:"required,max=500"` + Name *string `protobuf:"bytes,1,opt,name=name" json:"name,omitempty" query:"name" params:"name" validate:"required,max=500"` // Location where all artifacts for the experiment are stored. // If not provided, the remote server will select an appropriate default. - ArtifactLocation *string `protobuf:"bytes,2,opt,name=artifact_location,json=artifactLocation" json:"artifact_location,omitempty" query:"artifact_location" validate:"omitempty,uriWithoutFragmentsOrParamsOrDotDotInQuery"` + ArtifactLocation *string `protobuf:"bytes,2,opt,name=artifact_location,json=artifactLocation" json:"artifact_location,omitempty" query:"artifact_location" params:"artifact_location" validate:"omitempty,uriWithoutFragmentsOrParamsOrDotDotInQuery"` // A collection of tags to set on the experiment. Maximum tag size and number of tags per request // depends on the storage backend. All storage backends are guaranteed to support tag keys up // to 250 bytes in size and tag values up to 5000 bytes in size. All storage backends are also // guaranteed to support up to 20 tags per request. - Tags []*ExperimentTag `protobuf:"bytes,3,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*ExperimentTag `protobuf:"bytes,3,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` } func (x *CreateExperiment) Reset() { @@ -1265,9 +1265,9 @@ type SearchExperiments struct { // guaranteed to support a `max_results` threshold of at least 1,000 but may // support more. Callers of this endpoint are encouraged to pass max_results // explicitly and leverage page_token to iterate through experiments. - MaxResults *int64 `protobuf:"varint,1,opt,name=max_results,json=maxResults" json:"max_results,omitempty" query:"max_results"` + MaxResults *int64 `protobuf:"varint,1,opt,name=max_results,json=maxResults" json:"max_results,omitempty" query:"max_results" params:"max_results"` // Token indicating the page of experiments to fetch - PageToken *string `protobuf:"bytes,2,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token"` + PageToken *string `protobuf:"bytes,2,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token" params:"page_token"` // A filter expression over experiment attributes and tags that allows returning a subset of // experiments. The syntax is a subset of SQL that supports ANDing together binary operations // between an attribute or tag, and a constant. @@ -1280,14 +1280,14 @@ type SearchExperiments struct { // Example: “tags."extra-key" = 'value'“ or “tags.`extra-key` = 'value'“ // // Supported operators are “=“, “!=“, “LIKE“, and “ILIKE“. - Filter *string `protobuf:"bytes,3,opt,name=filter" json:"filter,omitempty" query:"filter"` + Filter *string `protobuf:"bytes,3,opt,name=filter" json:"filter,omitempty" query:"filter" params:"filter"` // List of columns for ordering search results, which can include experiment name and id // with an optional "DESC" or "ASC" annotation, where "ASC" is the default. // Tiebreaks are done by experiment id DESC. - OrderBy []string `protobuf:"bytes,4,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by"` + OrderBy []string `protobuf:"bytes,4,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by" params:"order_by"` // Qualifier for type of experiments to be returned. // If unspecified, return only active experiments. - ViewType *ViewType `protobuf:"varint,5,opt,name=view_type,json=viewType,enum=mlflow.ViewType" json:"view_type,omitempty" query:"view_type"` + ViewType *ViewType `protobuf:"varint,5,opt,name=view_type,json=viewType,enum=mlflow.ViewType" json:"view_type,omitempty" query:"view_type" params:"view_type"` } func (x *SearchExperiments) Reset() { @@ -1363,7 +1363,7 @@ type GetExperiment struct { unknownFields protoimpl.UnknownFields // ID of the associated experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" validate:"required,stringAsPositiveInteger"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id" validate:"required,stringAsPositiveInteger"` } func (x *GetExperiment) Reset() { @@ -1411,7 +1411,7 @@ type DeleteExperiment struct { unknownFields protoimpl.UnknownFields // ID of the associated experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" validate:"required,stringAsPositiveInteger"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id" validate:"required,stringAsPositiveInteger"` } func (x *DeleteExperiment) Reset() { @@ -1459,7 +1459,7 @@ type RestoreExperiment struct { unknownFields protoimpl.UnknownFields // ID of the associated experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` } func (x *RestoreExperiment) Reset() { @@ -1507,9 +1507,9 @@ type UpdateExperiment struct { unknownFields protoimpl.UnknownFields // ID of the associated experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` // If provided, the experiment's name is changed to the new name. The new name must be unique. - NewName *string `protobuf:"bytes,2,opt,name=new_name,json=newName" json:"new_name,omitempty" query:"new_name"` + NewName *string `protobuf:"bytes,2,opt,name=new_name,json=newName" json:"new_name,omitempty" query:"new_name" params:"new_name"` } func (x *UpdateExperiment) Reset() { @@ -1564,17 +1564,17 @@ type CreateRun struct { unknownFields protoimpl.UnknownFields // ID of the associated experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" validate:"required,stringAsPositiveInteger"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id" validate:"required,stringAsPositiveInteger"` // ID of the user executing the run. // This field is deprecated as of MLflow 1.0, and will be removed in a future // MLflow release. Use 'mlflow.user' tag instead. - UserId *string `protobuf:"bytes,2,opt,name=user_id,json=userId" json:"user_id,omitempty" query:"user_id"` + UserId *string `protobuf:"bytes,2,opt,name=user_id,json=userId" json:"user_id,omitempty" query:"user_id" params:"user_id"` // Name of the run. - RunName *string `protobuf:"bytes,3,opt,name=run_name,json=runName" json:"run_name,omitempty" query:"run_name"` + RunName *string `protobuf:"bytes,3,opt,name=run_name,json=runName" json:"run_name,omitempty" query:"run_name" params:"run_name"` // Unix timestamp in milliseconds of when the run started. - StartTime *int64 `protobuf:"varint,7,opt,name=start_time,json=startTime" json:"start_time,omitempty" query:"start_time"` + StartTime *int64 `protobuf:"varint,7,opt,name=start_time,json=startTime" json:"start_time,omitempty" query:"start_time" params:"start_time"` // Additional metadata for run. - Tags []*RunTag `protobuf:"bytes,9,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*RunTag `protobuf:"bytes,9,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` } func (x *CreateRun) Reset() { @@ -1650,16 +1650,16 @@ type UpdateRun struct { unknownFields protoimpl.UnknownFields // ID of the run to update. Must be provided. - RunId *string `protobuf:"bytes,4,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,4,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // [Deprecated, use run_id instead] ID of the run to update.. This field will // be removed in a future MLflow version. - RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid"` + RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid" params:"run_uuid"` // Updated status of the run. - Status *RunStatus `protobuf:"varint,2,opt,name=status,enum=mlflow.RunStatus" json:"status,omitempty" query:"status"` + Status *RunStatus `protobuf:"varint,2,opt,name=status,enum=mlflow.RunStatus" json:"status,omitempty" query:"status" params:"status"` // Unix timestamp in milliseconds of when the run ended. - EndTime *int64 `protobuf:"varint,3,opt,name=end_time,json=endTime" json:"end_time,omitempty" query:"end_time"` + EndTime *int64 `protobuf:"varint,3,opt,name=end_time,json=endTime" json:"end_time,omitempty" query:"end_time" params:"end_time"` // Updated name of the run. - RunName *string `protobuf:"bytes,5,opt,name=run_name,json=runName" json:"run_name,omitempty" query:"run_name"` + RunName *string `protobuf:"bytes,5,opt,name=run_name,json=runName" json:"run_name,omitempty" query:"run_name" params:"run_name"` } func (x *UpdateRun) Reset() { @@ -1735,7 +1735,7 @@ type DeleteRun struct { unknownFields protoimpl.UnknownFields // ID of the run to delete. - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` } func (x *DeleteRun) Reset() { @@ -1783,7 +1783,7 @@ type RestoreRun struct { unknownFields protoimpl.UnknownFields // ID of the run to restore. - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` } func (x *RestoreRun) Reset() { @@ -1831,18 +1831,18 @@ type LogMetric struct { unknownFields protoimpl.UnknownFields // ID of the run under which to log the metric. Must be provided. - RunId *string `protobuf:"bytes,6,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" validate:"required"` + RunId *string `protobuf:"bytes,6,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id" validate:"required"` // [Deprecated, use run_id instead] ID of the run under which to log the metric. This field will // be removed in a future MLflow version. - RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid"` + RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid" params:"run_uuid"` // Name of the metric. - Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" validate:"required"` + Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" params:"key" validate:"required"` // Double value of the metric being logged. - Value *float64 `protobuf:"fixed64,3,opt,name=value" json:"value,omitempty" query:"value" validate:"required"` + Value *float64 `protobuf:"fixed64,3,opt,name=value" json:"value,omitempty" query:"value" params:"value" validate:"required"` // Unix timestamp in milliseconds at the time metric was logged. - Timestamp *int64 `protobuf:"varint,4,opt,name=timestamp" json:"timestamp,omitempty" query:"timestamp" validate:"required"` + Timestamp *int64 `protobuf:"varint,4,opt,name=timestamp" json:"timestamp,omitempty" query:"timestamp" params:"timestamp" validate:"required"` // Step at which to log the metric - Step *int64 `protobuf:"varint,5,opt,name=step,def=0" json:"step,omitempty" query:"step"` + Step *int64 `protobuf:"varint,5,opt,name=step,def=0" json:"step,omitempty" query:"step" params:"step"` } // Default values for LogMetric fields. @@ -1930,14 +1930,14 @@ type LogParam struct { unknownFields protoimpl.UnknownFields // ID of the run under which to log the param. Must be provided. - RunId *string `protobuf:"bytes,4,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,4,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // [Deprecated, use run_id instead] ID of the run under which to log the param. This field will // be removed in a future MLflow version. - RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid"` + RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid" params:"run_uuid"` // Name of the param. Maximum size is 255 bytes. - Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` + Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" params:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` // String value of the param being logged. Maximum size is 6000 bytes. - Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value" validate:"omitempty,truncate=6000"` + Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value" params:"value" validate:"omitempty,truncate=6000"` } func (x *LogParam) Reset() { @@ -2006,13 +2006,13 @@ type SetExperimentTag struct { unknownFields protoimpl.UnknownFields // ID of the experiment under which to log the tag. Must be provided. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` // Name of the tag. Maximum size depends on storage backend. // All storage backends are guaranteed to support key values up to 250 bytes in size. - Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // String value of the tag being logged. Maximum size depends on storage backend. // All storage backends are guaranteed to support key values up to 5000 bytes in size. - Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *SetExperimentTag) Reset() { @@ -2074,16 +2074,16 @@ type SetTag struct { unknownFields protoimpl.UnknownFields // ID of the run under which to log the tag. Must be provided. - RunId *string `protobuf:"bytes,4,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,4,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // [Deprecated, use run_id instead] ID of the run under which to log the tag. This field will // be removed in a future MLflow version. - RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid"` + RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid" params:"run_uuid"` // Name of the tag. Maximum size depends on storage backend. // All storage backends are guaranteed to support key values up to 250 bytes in size. - Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // String value of the tag being logged. Maximum size depends on storage backend. // All storage backends are guaranteed to support key values up to 5000 bytes in size. - Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *SetTag) Reset() { @@ -2152,9 +2152,9 @@ type DeleteTag struct { unknownFields protoimpl.UnknownFields // ID of the run that the tag was logged under. Must be provided. - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" validate:"required"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id" validate:"required"` // Name of the tag. Maximum size is 255 bytes. Must be provided. - Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" validate:"required"` + Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" params:"key" validate:"required"` } func (x *DeleteTag) Reset() { @@ -2209,10 +2209,10 @@ type GetRun struct { unknownFields protoimpl.UnknownFields // ID of the run to fetch. Must be provided. - RunId *string `protobuf:"bytes,2,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,2,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // [Deprecated, use run_id instead] ID of the run to fetch. This field will // be removed in a future MLflow version. - RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid"` + RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid" params:"run_uuid"` } func (x *GetRun) Reset() { @@ -2267,7 +2267,7 @@ type SearchRuns struct { unknownFields protoimpl.UnknownFields // List of experiment IDs to search over. - ExperimentIds []string `protobuf:"bytes,1,rep,name=experiment_ids,json=experimentIds" json:"experiment_ids,omitempty" query:"experiment_ids"` + ExperimentIds []string `protobuf:"bytes,1,rep,name=experiment_ids,json=experimentIds" json:"experiment_ids,omitempty" query:"experiment_ids" params:"experiment_ids"` // A filter expression over params, metrics, and tags, that allows returning a subset of // runs. The syntax is a subset of SQL that supports ANDing together binary operations // between a param, metric, or tag and a constant. @@ -2278,22 +2278,22 @@ type SearchRuns struct { // “metrics."model class" = 'LinearRegression' and tags."user-name" = 'Tomas'“ // // Supported operators are “=“, “!=“, “>“, “>=“, “<“, and “<=“. - Filter *string `protobuf:"bytes,4,opt,name=filter" json:"filter,omitempty" query:"filter"` + Filter *string `protobuf:"bytes,4,opt,name=filter" json:"filter,omitempty" query:"filter" params:"filter"` // Whether to display only active, only deleted, or all runs. // Defaults to only active runs. - RunViewType *ViewType `protobuf:"varint,3,opt,name=run_view_type,json=runViewType,enum=mlflow.ViewType,def=1" json:"run_view_type,omitempty" query:"run_view_type" validate:"omitempty"` + RunViewType *ViewType `protobuf:"varint,3,opt,name=run_view_type,json=runViewType,enum=mlflow.ViewType,def=1" json:"run_view_type,omitempty" query:"run_view_type" params:"run_view_type" validate:"omitempty"` // Maximum number of runs desired. If unspecified, defaults to 1000. // All servers are guaranteed to support a `max_results` threshold of at least 50,000 // but may support more. Callers of this endpoint are encouraged to pass max_results // explicitly and leverage page_token to iterate through experiments. - MaxResults *int32 `protobuf:"varint,5,opt,name=max_results,json=maxResults,def=1000" json:"max_results,omitempty" query:"max_results" validate:"gt=0,max=50000"` + MaxResults *int32 `protobuf:"varint,5,opt,name=max_results,json=maxResults,def=1000" json:"max_results,omitempty" query:"max_results" params:"max_results" validate:"gt=0,max=50000"` // List of columns to be ordered by, including attributes, params, metrics, and tags with an // optional "DESC" or "ASC" annotation, where "ASC" is the default. // Example: ["params.input DESC", "metrics.alpha ASC", "metrics.rmse"] // Tiebreaks are done by start_time DESC followed by run_id for runs with the same start time // (and this is the default ordering criterion if order_by is not provided). - OrderBy []string `protobuf:"bytes,6,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by"` - PageToken *string `protobuf:"bytes,7,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token"` + OrderBy []string `protobuf:"bytes,6,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by" params:"order_by"` + PageToken *string `protobuf:"bytes,7,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token" params:"page_token"` } // Default values for SearchRuns fields. @@ -2382,14 +2382,14 @@ type ListArtifacts struct { unknownFields protoimpl.UnknownFields // ID of the run whose artifacts to list. Must be provided. - RunId *string `protobuf:"bytes,3,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,3,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // [Deprecated, use run_id instead] ID of the run whose artifacts to list. This field will // be removed in a future MLflow version. - RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid"` + RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid" params:"run_uuid"` // Filter artifacts matching this path (a relative path from the root artifact directory). - Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path"` + Path *string `protobuf:"bytes,2,opt,name=path" json:"path,omitempty" query:"path" params:"path"` // Token indicating the page of artifact results to fetch - PageToken *string `protobuf:"bytes,4,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token"` + PageToken *string `protobuf:"bytes,4,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token" params:"page_token"` } func (x *ListArtifacts) Reset() { @@ -2459,11 +2459,11 @@ type FileInfo struct { unknownFields protoimpl.UnknownFields // Path relative to the root artifact directory run. - Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path"` + Path *string `protobuf:"bytes,1,opt,name=path" json:"path,omitempty" query:"path" params:"path"` // Whether the path is a directory. - IsDir *bool `protobuf:"varint,2,opt,name=is_dir,json=isDir" json:"is_dir,omitempty" query:"is_dir"` + IsDir *bool `protobuf:"varint,2,opt,name=is_dir,json=isDir" json:"is_dir,omitempty" query:"is_dir" params:"is_dir"` // Size in bytes. Unset for directories. - FileSize *int64 `protobuf:"varint,3,opt,name=file_size,json=fileSize" json:"file_size,omitempty" query:"file_size"` + FileSize *int64 `protobuf:"varint,3,opt,name=file_size,json=fileSize" json:"file_size,omitempty" query:"file_size" params:"file_size"` } func (x *FileInfo) Reset() { @@ -2525,19 +2525,19 @@ type GetMetricHistory struct { unknownFields protoimpl.UnknownFields // ID of the run from which to fetch metric values. Must be provided. - RunId *string `protobuf:"bytes,3,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,3,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // [Deprecated, use run_id instead] ID of the run from which to fetch metric values. This field // will be removed in a future MLflow version. - RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid"` + RunUuid *string `protobuf:"bytes,1,opt,name=run_uuid,json=runUuid" json:"run_uuid,omitempty" query:"run_uuid" params:"run_uuid"` // Name of the metric. - MetricKey *string `protobuf:"bytes,2,opt,name=metric_key,json=metricKey" json:"metric_key,omitempty" query:"metric_key"` + MetricKey *string `protobuf:"bytes,2,opt,name=metric_key,json=metricKey" json:"metric_key,omitempty" query:"metric_key" params:"metric_key"` // Token indicating the page of metric history to fetch - PageToken *string `protobuf:"bytes,4,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token"` + PageToken *string `protobuf:"bytes,4,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token" params:"page_token"` // Maximum number of logged instances of a metric for a run to return per call. // Backend servers may restrict the value of `max_results` depending on performance requirements. // Requests that do not specify this value will behave as non-paginated queries where all // metric history values for a given metric within a run are returned in a single response. - MaxResults *int32 `protobuf:"varint,5,opt,name=max_results,json=maxResults" json:"max_results,omitempty" query:"max_results"` + MaxResults *int32 `protobuf:"varint,5,opt,name=max_results,json=maxResults" json:"max_results,omitempty" query:"max_results" params:"max_results"` } func (x *GetMetricHistory) Reset() { @@ -2613,15 +2613,15 @@ type MetricWithRunId struct { unknownFields protoimpl.UnknownFields // Key identifying this metric. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // Value associated with this metric. - Value *float64 `protobuf:"fixed64,2,opt,name=value" json:"value,omitempty" query:"value"` + Value *float64 `protobuf:"fixed64,2,opt,name=value" json:"value,omitempty" query:"value" params:"value"` // The timestamp at which this metric was recorded. - Timestamp *int64 `protobuf:"varint,3,opt,name=timestamp" json:"timestamp,omitempty" query:"timestamp"` + Timestamp *int64 `protobuf:"varint,3,opt,name=timestamp" json:"timestamp,omitempty" query:"timestamp" params:"timestamp"` // Step at which to log the metric. - Step *int64 `protobuf:"varint,4,opt,name=step,def=0" json:"step,omitempty" query:"step"` + Step *int64 `protobuf:"varint,4,opt,name=step,def=0" json:"step,omitempty" query:"step" params:"step"` // The ID of the run containing the metric - RunId *string `protobuf:"bytes,5,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,5,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` } // Default values for MetricWithRunId fields. @@ -2702,19 +2702,19 @@ type GetMetricHistoryBulkInterval struct { unknownFields protoimpl.UnknownFields // ID(s) of the run(s) from which to fetch metric values. Must be provided. - RunIds []string `protobuf:"bytes,1,rep,name=run_ids,json=runIds" json:"run_ids,omitempty" query:"run_ids"` + RunIds []string `protobuf:"bytes,1,rep,name=run_ids,json=runIds" json:"run_ids,omitempty" query:"run_ids" params:"run_ids"` // Name of the metric. - MetricKey *string `protobuf:"bytes,2,opt,name=metric_key,json=metricKey" json:"metric_key,omitempty" query:"metric_key"` + MetricKey *string `protobuf:"bytes,2,opt,name=metric_key,json=metricKey" json:"metric_key,omitempty" query:"metric_key" params:"metric_key"` // Optional start step to only fetch metrics after the specified step. Must be defined if // end_step is defined. - StartStep *int32 `protobuf:"varint,3,opt,name=start_step,json=startStep" json:"start_step,omitempty" query:"start_step"` + StartStep *int32 `protobuf:"varint,3,opt,name=start_step,json=startStep" json:"start_step,omitempty" query:"start_step" params:"start_step"` // Optional end step to only fetch metrics before the specified step. Must be defined if // start_step is defined. - EndStep *int32 `protobuf:"varint,4,opt,name=end_step,json=endStep" json:"end_step,omitempty" query:"end_step"` + EndStep *int32 `protobuf:"varint,4,opt,name=end_step,json=endStep" json:"end_step,omitempty" query:"end_step" params:"end_step"` // Maximum number of results to fetch per run specified. Must be set to a positive number. // Note, in reality, the API returns at most (max_results + # of run IDs) x (# run IDs) metric // data points. - MaxResults *int32 `protobuf:"varint,5,opt,name=max_results,json=maxResults" json:"max_results,omitempty" query:"max_results"` + MaxResults *int32 `protobuf:"varint,5,opt,name=max_results,json=maxResults" json:"max_results,omitempty" query:"max_results" params:"max_results"` } func (x *GetMetricHistoryBulkInterval) Reset() { @@ -2790,16 +2790,16 @@ type LogBatch struct { unknownFields protoimpl.UnknownFields // ID of the run to log under - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" validate:"required,runId"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id" validate:"required,runId"` // Metrics to log. A single request can contain up to 1000 metrics, and up to 1000 // metrics, params, and tags in total. - Metrics []*Metric `protobuf:"bytes,2,rep,name=metrics" json:"metrics,omitempty" query:"metrics" validate:"max=1000,dive"` + Metrics []*Metric `protobuf:"bytes,2,rep,name=metrics" json:"metrics,omitempty" query:"metrics" params:"metrics" validate:"max=1000,dive"` // Params to log. A single request can contain up to 100 params, and up to 1000 // metrics, params, and tags in total. - Params []*Param `protobuf:"bytes,3,rep,name=params" json:"params,omitempty" query:"params" validate:"omitempty,uniqueParams,max=100,dive"` + Params []*Param `protobuf:"bytes,3,rep,name=params" json:"params,omitempty" query:"params" params:"params" validate:"omitempty,uniqueParams,max=100,dive"` // Tags to log. A single request can contain up to 100 tags, and up to 1000 // metrics, params, and tags in total. - Tags []*RunTag `protobuf:"bytes,4,rep,name=tags" json:"tags,omitempty" query:"tags" validate:"max=100"` + Tags []*RunTag `protobuf:"bytes,4,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags" validate:"max=100"` } func (x *LogBatch) Reset() { @@ -2868,9 +2868,9 @@ type LogModel struct { unknownFields protoimpl.UnknownFields // ID of the run to log under - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // MLmodel file in json format. - ModelJson *string `protobuf:"bytes,2,opt,name=model_json,json=modelJson" json:"model_json,omitempty" query:"model_json"` + ModelJson *string `protobuf:"bytes,2,opt,name=model_json,json=modelJson" json:"model_json,omitempty" query:"model_json" params:"model_json"` } func (x *LogModel) Reset() { @@ -2925,9 +2925,9 @@ type LogInputs struct { unknownFields protoimpl.UnknownFields // ID of the run to log under - RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id"` + RunId *string `protobuf:"bytes,1,opt,name=run_id,json=runId" json:"run_id,omitempty" query:"run_id" params:"run_id"` // Dataset inputs - Datasets []*DatasetInput `protobuf:"bytes,2,rep,name=datasets" json:"datasets,omitempty" query:"datasets"` + Datasets []*DatasetInput `protobuf:"bytes,2,rep,name=datasets" json:"datasets,omitempty" query:"datasets" params:"datasets"` } func (x *LogInputs) Reset() { @@ -2982,7 +2982,7 @@ type GetExperimentByName struct { unknownFields protoimpl.UnknownFields // Name of the associated experiment. - ExperimentName *string `protobuf:"bytes,1,opt,name=experiment_name,json=experimentName" json:"experiment_name,omitempty" query:"experiment_name" validate:"required"` + ExperimentName *string `protobuf:"bytes,1,opt,name=experiment_name,json=experimentName" json:"experiment_name,omitempty" query:"experiment_name" params:"experiment_name" validate:"required"` } func (x *GetExperimentByName) Reset() { @@ -3031,19 +3031,19 @@ type TraceInfo struct { unknownFields protoimpl.UnknownFields // Unique identifier for the trace. - RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id"` + RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id" params:"request_id"` // The ID of the experiment that contains the trace. - ExperimentId *string `protobuf:"bytes,2,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,2,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` // Unix timestamp of when the trace started in milliseconds. - TimestampMs *int64 `protobuf:"varint,3,opt,name=timestamp_ms,json=timestampMs" json:"timestamp_ms,omitempty" query:"timestamp_ms"` + TimestampMs *int64 `protobuf:"varint,3,opt,name=timestamp_ms,json=timestampMs" json:"timestamp_ms,omitempty" query:"timestamp_ms" params:"timestamp_ms"` // Unix timestamp of the duration of the trace in milliseconds. - ExecutionTimeMs *int64 `protobuf:"varint,4,opt,name=execution_time_ms,json=executionTimeMs" json:"execution_time_ms,omitempty" query:"execution_time_ms"` + ExecutionTimeMs *int64 `protobuf:"varint,4,opt,name=execution_time_ms,json=executionTimeMs" json:"execution_time_ms,omitempty" query:"execution_time_ms" params:"execution_time_ms"` // Overall status of the operation being traced (OK, error, etc.). - Status *TraceStatus `protobuf:"varint,5,opt,name=status,enum=mlflow.TraceStatus" json:"status,omitempty" query:"status"` + Status *TraceStatus `protobuf:"varint,5,opt,name=status,enum=mlflow.TraceStatus" json:"status,omitempty" query:"status" params:"status"` // Other trace metadata. - RequestMetadata []*TraceRequestMetadata `protobuf:"bytes,6,rep,name=request_metadata,json=requestMetadata" json:"request_metadata,omitempty" query:"request_metadata"` + RequestMetadata []*TraceRequestMetadata `protobuf:"bytes,6,rep,name=request_metadata,json=requestMetadata" json:"request_metadata,omitempty" query:"request_metadata" params:"request_metadata"` // Tags for the trace. - Tags []*TraceTag `protobuf:"bytes,7,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*TraceTag `protobuf:"bytes,7,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` } func (x *TraceInfo) Reset() { @@ -3133,9 +3133,9 @@ type TraceRequestMetadata struct { unknownFields protoimpl.UnknownFields // Key identifying this metadata. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // Value identifying this metadata. - Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *TraceRequestMetadata) Reset() { @@ -3190,9 +3190,9 @@ type TraceTag struct { unknownFields protoimpl.UnknownFields // Key identifying this trace tag. - Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,1,opt,name=key" json:"key,omitempty" query:"key" params:"key"` // Value associated with this trace tag. - Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value"` + Value *string `protobuf:"bytes,2,opt,name=value" json:"value,omitempty" query:"value" params:"value"` } func (x *TraceTag) Reset() { @@ -3247,13 +3247,13 @@ type StartTrace struct { unknownFields protoimpl.UnknownFields // ID of the associated experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` // Unix timestamp of when the trace started in milliseconds. - TimestampMs *int64 `protobuf:"varint,2,opt,name=timestamp_ms,json=timestampMs" json:"timestamp_ms,omitempty" query:"timestamp_ms"` + TimestampMs *int64 `protobuf:"varint,2,opt,name=timestamp_ms,json=timestampMs" json:"timestamp_ms,omitempty" query:"timestamp_ms" params:"timestamp_ms"` // Metadata about the request that initiated the trace. - RequestMetadata []*TraceRequestMetadata `protobuf:"bytes,3,rep,name=request_metadata,json=requestMetadata" json:"request_metadata,omitempty" query:"request_metadata"` + RequestMetadata []*TraceRequestMetadata `protobuf:"bytes,3,rep,name=request_metadata,json=requestMetadata" json:"request_metadata,omitempty" query:"request_metadata" params:"request_metadata"` // Tags for the trace. - Tags []*TraceTag `protobuf:"bytes,4,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*TraceTag `protobuf:"bytes,4,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` } func (x *StartTrace) Reset() { @@ -3322,15 +3322,15 @@ type EndTrace struct { unknownFields protoimpl.UnknownFields // ID of the trace to end. - RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id"` + RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id" params:"request_id"` // Unix timestamp of when the trace ended in milliseconds. - TimestampMs *int64 `protobuf:"varint,2,opt,name=timestamp_ms,json=timestampMs" json:"timestamp_ms,omitempty" query:"timestamp_ms"` + TimestampMs *int64 `protobuf:"varint,2,opt,name=timestamp_ms,json=timestampMs" json:"timestamp_ms,omitempty" query:"timestamp_ms" params:"timestamp_ms"` // Overall status of the operation being traced (OK, error, etc). - Status *TraceStatus `protobuf:"varint,3,opt,name=status,enum=mlflow.TraceStatus" json:"status,omitempty" query:"status"` + Status *TraceStatus `protobuf:"varint,3,opt,name=status,enum=mlflow.TraceStatus" json:"status,omitempty" query:"status" params:"status"` // Additional metadata about the operation being traced. - RequestMetadata []*TraceRequestMetadata `protobuf:"bytes,4,rep,name=request_metadata,json=requestMetadata" json:"request_metadata,omitempty" query:"request_metadata"` + RequestMetadata []*TraceRequestMetadata `protobuf:"bytes,4,rep,name=request_metadata,json=requestMetadata" json:"request_metadata,omitempty" query:"request_metadata" params:"request_metadata"` // Additional tags to add to the trace. - Tags []*TraceTag `protobuf:"bytes,5,rep,name=tags" json:"tags,omitempty" query:"tags"` + Tags []*TraceTag `protobuf:"bytes,5,rep,name=tags" json:"tags,omitempty" query:"tags" params:"tags"` } func (x *EndTrace) Reset() { @@ -3406,7 +3406,7 @@ type GetTraceInfo struct { unknownFields protoimpl.UnknownFields // ID of the trace to fetch. Must be provided. - RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id"` + RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id" params:"request_id"` } func (x *GetTraceInfo) Reset() { @@ -3454,17 +3454,17 @@ type SearchTraces struct { unknownFields protoimpl.UnknownFields // List of experiment IDs to search over. - ExperimentIds []string `protobuf:"bytes,1,rep,name=experiment_ids,json=experimentIds" json:"experiment_ids,omitempty" query:"experiment_ids"` + ExperimentIds []string `protobuf:"bytes,1,rep,name=experiment_ids,json=experimentIds" json:"experiment_ids,omitempty" query:"experiment_ids" params:"experiment_ids"` // A filter expression over trace attributes and tags that allows returning a subset of // traces. The syntax is a subset of SQL that supports ANDing together binary operations // Example: “trace.status = 'OK' and trace.timestamp_ms > 1711089570679“. - Filter *string `protobuf:"bytes,2,opt,name=filter" json:"filter,omitempty" query:"filter"` + Filter *string `protobuf:"bytes,2,opt,name=filter" json:"filter,omitempty" query:"filter" params:"filter"` // Maximum number of traces desired. Max threshold is 500. - MaxResults *int32 `protobuf:"varint,3,opt,name=max_results,json=maxResults,def=100" json:"max_results,omitempty" query:"max_results"` + MaxResults *int32 `protobuf:"varint,3,opt,name=max_results,json=maxResults,def=100" json:"max_results,omitempty" query:"max_results" params:"max_results"` // List of columns for ordering the results, e.g. “["timestamp_ms DESC"]“. - OrderBy []string `protobuf:"bytes,4,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by"` + OrderBy []string `protobuf:"bytes,4,rep,name=order_by,json=orderBy" json:"order_by,omitempty" query:"order_by" params:"order_by"` // Token indicating the page of traces to fetch. - PageToken *string `protobuf:"bytes,5,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token"` + PageToken *string `protobuf:"bytes,5,opt,name=page_token,json=pageToken" json:"page_token,omitempty" query:"page_token" params:"page_token"` } // Default values for SearchTraces fields. @@ -3545,15 +3545,15 @@ type DeleteTraces struct { unknownFields protoimpl.UnknownFields // ID of the associated experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` // Case 1: max_timestamp_millis and max_traces must be specified for time-based deletion // The maximum timestamp in milliseconds since the UNIX epoch for deleting traces. - MaxTimestampMillis *int64 `protobuf:"varint,2,opt,name=max_timestamp_millis,json=maxTimestampMillis" json:"max_timestamp_millis,omitempty" query:"max_timestamp_millis"` + MaxTimestampMillis *int64 `protobuf:"varint,2,opt,name=max_timestamp_millis,json=maxTimestampMillis" json:"max_timestamp_millis,omitempty" query:"max_timestamp_millis" params:"max_timestamp_millis"` // The maximum number of traces to delete. - MaxTraces *int32 `protobuf:"varint,3,opt,name=max_traces,json=maxTraces" json:"max_traces,omitempty" query:"max_traces"` + MaxTraces *int32 `protobuf:"varint,3,opt,name=max_traces,json=maxTraces" json:"max_traces,omitempty" query:"max_traces" params:"max_traces"` // Case 2: request_ids must be specified for ID-based deletion // A set of request IDs to delete - RequestIds []string `protobuf:"bytes,4,rep,name=request_ids,json=requestIds" json:"request_ids,omitempty" query:"request_ids"` + RequestIds []string `protobuf:"bytes,4,rep,name=request_ids,json=requestIds" json:"request_ids,omitempty" query:"request_ids" params:"request_ids"` } func (x *DeleteTraces) Reset() { @@ -3622,13 +3622,13 @@ type SetTraceTag struct { unknownFields protoimpl.UnknownFields // ID of the trace on which to set a tag. - RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id"` + RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id" params:"request_id"` // Name of the tag. Maximum size depends on storage backend. // All storage backends are guaranteed to support key values up to 250 bytes in size. - Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` + Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" params:"key" validate:"required,max=250,validMetricParamOrTagName,pathIsUnique"` // String value of the tag being logged. Maximum size depends on storage backend. // All storage backends are guaranteed to support key values up to 250 bytes in size. - Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value" validate:"omitempty,truncate=8000"` + Value *string `protobuf:"bytes,3,opt,name=value" json:"value,omitempty" query:"value" params:"value" validate:"omitempty,truncate=8000"` } func (x *SetTraceTag) Reset() { @@ -3690,9 +3690,9 @@ type DeleteTraceTag struct { unknownFields protoimpl.UnknownFields // ID of the trace from which to delete the tag. - RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id"` + RequestId *string `protobuf:"bytes,1,opt,name=request_id,json=requestId" json:"request_id,omitempty" query:"request_id" params:"request_id"` // Name of the tag to delete. - Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key"` + Key *string `protobuf:"bytes,2,opt,name=key" json:"key,omitempty" query:"key" params:"key"` } func (x *DeleteTraceTag) Reset() { @@ -3748,14 +3748,14 @@ type DatasetSummary struct { unknownFields protoimpl.UnknownFields // Unique identifier for the experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` // The name of the dataset. E.g. “my.uc.table@2” “nyc-taxi-dataset”, “fantastic-elk-3” - Name *string `protobuf:"bytes,2,opt,name=name" json:"name,omitempty" query:"name"` + Name *string `protobuf:"bytes,2,opt,name=name" json:"name,omitempty" query:"name" params:"name"` // Dataset digest, e.g. an md5 hash of the dataset that uniquely identifies it // within datasets of the same name. - Digest *string `protobuf:"bytes,3,opt,name=digest" json:"digest,omitempty" query:"digest"` + Digest *string `protobuf:"bytes,3,opt,name=digest" json:"digest,omitempty" query:"digest" params:"digest"` // Value of "context" tag if set for the given dataset. - Context *string `protobuf:"bytes,4,opt,name=context" json:"context,omitempty" query:"context"` + Context *string `protobuf:"bytes,4,opt,name=context" json:"context,omitempty" query:"context" params:"context"` } func (x *DatasetSummary) Reset() { @@ -3824,7 +3824,7 @@ type SearchDatasets struct { unknownFields protoimpl.UnknownFields // List of experiment IDs to search over. - ExperimentIds []string `protobuf:"bytes,1,rep,name=experiment_ids,json=experimentIds" json:"experiment_ids,omitempty" query:"experiment_ids"` + ExperimentIds []string `protobuf:"bytes,1,rep,name=experiment_ids,json=experimentIds" json:"experiment_ids,omitempty" query:"experiment_ids" params:"experiment_ids"` } func (x *SearchDatasets) Reset() { @@ -3872,7 +3872,7 @@ type CreateExperiment_Response struct { unknownFields protoimpl.UnknownFields // Unique identifier for the experiment. - ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id"` + ExperimentId *string `protobuf:"bytes,1,opt,name=experiment_id,json=experimentId" json:"experiment_id,omitempty" query:"experiment_id" params:"experiment_id"` } func (x *CreateExperiment_Response) Reset() { @@ -3920,10 +3920,10 @@ type SearchExperiments_Response struct { unknownFields protoimpl.UnknownFields // Experiments that match the search criteria - Experiments []*Experiment `protobuf:"bytes,1,rep,name=experiments" json:"experiments,omitempty" query:"experiments"` + Experiments []*Experiment `protobuf:"bytes,1,rep,name=experiments" json:"experiments,omitempty" query:"experiments" params:"experiments"` // Token that can be used to retrieve the next page of experiments. // An empty token means that no more experiments are available for retrieval. - NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token"` + NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token" params:"next_page_token"` } func (x *SearchExperiments_Response) Reset() { @@ -3978,7 +3978,7 @@ type GetExperiment_Response struct { unknownFields protoimpl.UnknownFields // Experiment details. - Experiment *Experiment `protobuf:"bytes,1,opt,name=experiment" json:"experiment,omitempty" query:"experiment"` + Experiment *Experiment `protobuf:"bytes,1,opt,name=experiment" json:"experiment,omitempty" query:"experiment" params:"experiment"` } func (x *GetExperiment_Response) Reset() { @@ -4140,7 +4140,7 @@ type CreateRun_Response struct { unknownFields protoimpl.UnknownFields // The newly created run. - Run *Run `protobuf:"bytes,1,opt,name=run" json:"run,omitempty" query:"run"` + Run *Run `protobuf:"bytes,1,opt,name=run" json:"run,omitempty" query:"run" params:"run"` } func (x *CreateRun_Response) Reset() { @@ -4188,7 +4188,7 @@ type UpdateRun_Response struct { unknownFields protoimpl.UnknownFields // Updated metadata of the run. - RunInfo *RunInfo `protobuf:"bytes,1,opt,name=run_info,json=runInfo" json:"run_info,omitempty" query:"run_info"` + RunInfo *RunInfo `protobuf:"bytes,1,opt,name=run_info,json=runInfo" json:"run_info,omitempty" query:"run_info" params:"run_info"` } func (x *UpdateRun_Response) Reset() { @@ -4502,7 +4502,7 @@ type GetRun_Response struct { unknownFields protoimpl.UnknownFields // Run metadata (name, start time, etc) and data (metrics, params, and tags). - Run *Run `protobuf:"bytes,1,opt,name=run" json:"run,omitempty" query:"run"` + Run *Run `protobuf:"bytes,1,opt,name=run" json:"run,omitempty" query:"run" params:"run"` } func (x *GetRun_Response) Reset() { @@ -4550,8 +4550,8 @@ type SearchRuns_Response struct { unknownFields protoimpl.UnknownFields // Runs that match the search criteria. - Runs []*Run `protobuf:"bytes,1,rep,name=runs" json:"runs,omitempty" query:"runs"` - NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token"` + Runs []*Run `protobuf:"bytes,1,rep,name=runs" json:"runs,omitempty" query:"runs" params:"runs"` + NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token" params:"next_page_token"` } func (x *SearchRuns_Response) Reset() { @@ -4606,11 +4606,11 @@ type ListArtifacts_Response struct { unknownFields protoimpl.UnknownFields // Root artifact directory for the run. - RootUri *string `protobuf:"bytes,1,opt,name=root_uri,json=rootUri" json:"root_uri,omitempty" query:"root_uri"` + RootUri *string `protobuf:"bytes,1,opt,name=root_uri,json=rootUri" json:"root_uri,omitempty" query:"root_uri" params:"root_uri"` // File location and metadata for artifacts. - Files []*FileInfo `protobuf:"bytes,2,rep,name=files" json:"files,omitempty" query:"files"` + Files []*FileInfo `protobuf:"bytes,2,rep,name=files" json:"files,omitempty" query:"files" params:"files"` // Token that can be used to retrieve the next page of artifact results - NextPageToken *string `protobuf:"bytes,3,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token"` + NextPageToken *string `protobuf:"bytes,3,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token" params:"next_page_token"` } func (x *ListArtifacts_Response) Reset() { @@ -4672,10 +4672,10 @@ type GetMetricHistory_Response struct { unknownFields protoimpl.UnknownFields // All logged values for this metric. - Metrics []*Metric `protobuf:"bytes,1,rep,name=metrics" json:"metrics,omitempty" query:"metrics"` + Metrics []*Metric `protobuf:"bytes,1,rep,name=metrics" json:"metrics,omitempty" query:"metrics" params:"metrics"` // Token that can be used to issue a query for the next page of metric history values. // A missing token indicates that no additional metrics are available to fetch. - NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token"` + NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token" params:"next_page_token"` } func (x *GetMetricHistory_Response) Reset() { @@ -4730,7 +4730,7 @@ type GetMetricHistoryBulkInterval_Response struct { unknownFields protoimpl.UnknownFields // List of metrics representing history of values and metadata. - Metrics []*MetricWithRunId `protobuf:"bytes,1,rep,name=metrics" json:"metrics,omitempty" query:"metrics"` + Metrics []*MetricWithRunId `protobuf:"bytes,1,rep,name=metrics" json:"metrics,omitempty" query:"metrics" params:"metrics"` } func (x *GetMetricHistoryBulkInterval_Response) Reset() { @@ -4892,7 +4892,7 @@ type GetExperimentByName_Response struct { unknownFields protoimpl.UnknownFields // Experiment details. - Experiment *Experiment `protobuf:"bytes,1,opt,name=experiment" json:"experiment,omitempty" query:"experiment"` + Experiment *Experiment `protobuf:"bytes,1,opt,name=experiment" json:"experiment,omitempty" query:"experiment" params:"experiment"` } func (x *GetExperimentByName_Response) Reset() { @@ -4940,7 +4940,7 @@ type StartTrace_Response struct { unknownFields protoimpl.UnknownFields // The newly created trace. - TraceInfo *TraceInfo `protobuf:"bytes,1,opt,name=trace_info,json=traceInfo" json:"trace_info,omitempty" query:"trace_info"` + TraceInfo *TraceInfo `protobuf:"bytes,1,opt,name=trace_info,json=traceInfo" json:"trace_info,omitempty" query:"trace_info" params:"trace_info"` } func (x *StartTrace_Response) Reset() { @@ -4988,7 +4988,7 @@ type EndTrace_Response struct { unknownFields protoimpl.UnknownFields // The updated trace. - TraceInfo *TraceInfo `protobuf:"bytes,1,opt,name=trace_info,json=traceInfo" json:"trace_info,omitempty" query:"trace_info"` + TraceInfo *TraceInfo `protobuf:"bytes,1,opt,name=trace_info,json=traceInfo" json:"trace_info,omitempty" query:"trace_info" params:"trace_info"` } func (x *EndTrace_Response) Reset() { @@ -5036,7 +5036,7 @@ type GetTraceInfo_Response struct { unknownFields protoimpl.UnknownFields // Metadata of the requested trace. - TraceInfo *TraceInfo `protobuf:"bytes,1,opt,name=trace_info,json=traceInfo" json:"trace_info,omitempty" query:"trace_info"` + TraceInfo *TraceInfo `protobuf:"bytes,1,opt,name=trace_info,json=traceInfo" json:"trace_info,omitempty" query:"trace_info" params:"trace_info"` } func (x *GetTraceInfo_Response) Reset() { @@ -5084,8 +5084,8 @@ type SearchTraces_Response struct { unknownFields protoimpl.UnknownFields // Information about traces that match the search criteria. - Traces []*TraceInfo `protobuf:"bytes,1,rep,name=traces" json:"traces,omitempty" query:"traces"` - NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token"` + Traces []*TraceInfo `protobuf:"bytes,1,rep,name=traces" json:"traces,omitempty" query:"traces" params:"traces"` + NextPageToken *string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken" json:"next_page_token,omitempty" query:"next_page_token" params:"next_page_token"` } func (x *SearchTraces_Response) Reset() { @@ -5139,7 +5139,7 @@ type DeleteTraces_Response struct { sizeCache protoimpl.SizeCache unknownFields protoimpl.UnknownFields - TracesDeleted *int32 `protobuf:"varint,1,opt,name=traces_deleted,json=tracesDeleted" json:"traces_deleted,omitempty" query:"traces_deleted"` + TracesDeleted *int32 `protobuf:"varint,1,opt,name=traces_deleted,json=tracesDeleted" json:"traces_deleted,omitempty" query:"traces_deleted" params:"traces_deleted"` } func (x *DeleteTraces_Response) Reset() { @@ -5263,7 +5263,7 @@ type SearchDatasets_Response struct { unknownFields protoimpl.UnknownFields // Return the summary for most recently created N datasets, as configured in backend - DatasetSummaries []*DatasetSummary `protobuf:"bytes,1,rep,name=dataset_summaries,json=datasetSummaries" json:"dataset_summaries,omitempty" query:"dataset_summaries"` + DatasetSummaries []*DatasetSummary `protobuf:"bytes,1,rep,name=dataset_summaries,json=datasetSummaries" json:"dataset_summaries,omitempty" query:"dataset_summaries" params:"dataset_summaries"` } func (x *SearchDatasets_Response) Reset() { diff --git a/pkg/server/parser/http_request_parser.go b/pkg/server/parser/http_request_parser.go index 9b9db530..e9b6fd80 100644 --- a/pkg/server/parser/http_request_parser.go +++ b/pkg/server/parser/http_request_parser.go @@ -52,6 +52,11 @@ func (p *HTTPRequestParser) ParseBody(ctx *fiber.Ctx, input proto.Message) *cont return contract.NewError(protos.ErrorCode_BAD_REQUEST, protojsonErr.Error()) } + // try to parse all the parameters from query url to an internal proto object. + if err := ctx.ParamsParser(input); err != nil { + return contract.NewError(protos.ErrorCode_BAD_REQUEST, err.Error()) + } + if err := p.validator.Struct(input); err != nil { return validation.NewErrorFromValidationError(err) } diff --git a/pkg/server/routes/tracking.g.go b/pkg/server/routes/tracking.g.go index 568cd2b4..0d3b596b 100644 --- a/pkg/server/routes/tracking.g.go +++ b/pkg/server/routes/tracking.g.go @@ -4,10 +4,10 @@ package routes import ( "github.com/gofiber/fiber/v2" - "github.com/mlflow/mlflow-go/pkg/contract/service" - "github.com/mlflow/mlflow-go/pkg/protos" "github.com/mlflow/mlflow-go/pkg/server/parser" + "github.com/mlflow/mlflow-go/pkg/contract/service" "github.com/mlflow/mlflow-go/pkg/utils" + "github.com/mlflow/mlflow-go/pkg/protos" ) func RegisterTrackingServiceRoutes(service service.TrackingService, parser *parser.HTTPRequestParser, app *fiber.App) { @@ -143,20 +143,18 @@ func RegisterTrackingServiceRoutes(service service.TrackingService, parser *pars } return ctx.JSON(output) }) - - app.Patch("/mlflow/traces/{request_id}/tags", func(ctx *fiber.Ctx) error { + app.Patch("/mlflow/traces/:request_id/tags", func(ctx *fiber.Ctx) error { input := &protos.SetTraceTag{} if err := parser.ParseBody(ctx, input); err != nil { return err } output, err := service.SetTraceTag(utils.NewContextWithLoggerFromFiberContext(ctx), input) - if err != nil { + if err != nil { return err } return ctx.JSON(output) - }) - - app.Post("/mlflow/runs/delete-tag", func(ctx *fiber.Ctx) error { + }) + app.Post("/mlflow/runs/delete-tag", func(ctx *fiber.Ctx) error { input := &protos.DeleteTag{} if err := parser.ParseBody(ctx, input); err != nil { return err @@ -167,7 +165,6 @@ func RegisterTrackingServiceRoutes(service service.TrackingService, parser *pars } return ctx.JSON(output) }) - app.Get("/mlflow/runs/get", func(ctx *fiber.Ctx) error { input := &protos.GetRun{} if err := parser.ParseQuery(ctx, input); err != nil {