Add tracker to get command output

Author: deliahu

cli/cmd/get.go

@@ -267,7 +267,7 @@ func describeAPI(name string, resourcesRes *schema.GetResourcesResponse, flagVer
 	out += "\n\n" + describeModelInput(groupStatus, apiEndpoint)
 
 	if api != nil {
-		out += "\n" + titleStr("configuration") + api.UserConfigStr()
+		out += "\n" + titleStr("configuration") + strings.TrimSpace(api.UserConfigStr())
 	}
 
 	return out, nil

pkg/operator/api/userconfig/apis.go

@@ -193,6 +193,19 @@ func (api *API) UserConfigStr() string {
 		sb.WriteString(fmt.Sprintf("%s:\n", ComputeKey))
 		sb.WriteString(s.Indent(api.Compute.UserConfigStr(), "  "))
 	}
+	if api.Tracker != nil {
+		sb.WriteString(fmt.Sprintf("%s:\n", TrackerKey))
+		sb.WriteString(s.Indent(api.Tracker.UserConfigStr(), "  "))
+	}
+	return sb.String()
+}
+
+func (tracker *Tracker) UserConfigStr() string {
+	var sb strings.Builder
+	sb.WriteString(fmt.Sprintf("%s: %s\n", ModelTypeKey, tracker.ModelType.String()))
+	if tracker.Key != nil {
+		sb.WriteString(fmt.Sprintf("%s: %s\n", KeyKey, *tracker.Key))
+	}
 	return sb.String()
 }
 

pkg/operator/api/userconfig/compute.go

@@ -124,7 +124,7 @@ func (ac *APICompute) UserConfigStr() string {
 	if ac.Mem != nil {
 		sb.WriteString(fmt.Sprintf("%s: %s\n", MemKey, ac.Mem.UserString))
 	}
-	return strings.TrimSpace(sb.String())
+	return sb.String()
 }
 
 func (ac *APICompute) Validate() error {

pkg/operator/api/userconfig/config_key.go

@@ -28,6 +28,9 @@ const (
 	ModelFormatKey    = "model_format"
 	RequestHandlerKey = "request_handler"
 	TFSignatureKeyKey = "tf_signature_key"
+	TrackerKey        = "tracker"
+	ModelTypeKey      = "model_type"
+	KeyKey            = "key"
 
 	// compute
 	ComputeKey              = "compute"