From f497361ca433afb77764e2021dd0a4c910bb800d Mon Sep 17 00:00:00 2001
From: dongjiang <dongjiang1989@126.com>
Date: Tue, 24 Dec 2024 16:05:09 +0800
Subject: [PATCH] update modelx config yaml

Signed-off-by: dongjiang <dongjiang1989@126.com>
---
 cmd/modelx/model/config.go | 20 +++++++++++---------
 cmd/modelx/model/init.go   |  8 +++++---
 2 files changed, 16 insertions(+), 12 deletions(-)

diff --git a/cmd/modelx/model/config.go b/cmd/modelx/model/config.go
index 15e5db6e..961df893 100644
--- a/cmd/modelx/model/config.go
+++ b/cmd/modelx/model/config.go
@@ -17,13 +17,15 @@ limitations under the License.
 package model
 
 type ModelConfig struct {
-	Description string            `json:"description"`
-	FrameWork   string            `json:"framework"`
-	Task        string            `json:"task"`
-	Tags        []string          `json:"tags"`
-	Resources   map[string]any    `json:"resources"`
-	Mantainers  []string          `json:"maintainers"`
-	Annotations map[string]string `json:"annotations,omitempty"`
-	ModelFiles  []string          `json:"modelFiles"`
-	Config      any               `json:"config"`
+	Description  string            `json:"description"`
+	FrameWork    string            `json:"framework"`
+	Command      []string          `json:"command,omitempty"`
+	Args         []string          `json:"args,omitempty"`
+	Tags         []string          `json:"tags"`
+	Resources    map[string]any    `json:"resources"`
+	Mantainers   []string          `json:"maintainers"`
+	Annotations  map[string]string `json:"annotations,omitempty"`
+	ModelFiles   []string          `json:"modelFiles"`
+	Config       any               `json:"config"`
+	Dependencies []*ModelConfig    `json:"dependencies,omitempty"`
 }
diff --git a/cmd/modelx/model/init.go b/cmd/modelx/model/init.go
index 2955ea96..0811ac5f 100644
--- a/cmd/modelx/model/init.go
+++ b/cmd/modelx/model/init.go
@@ -74,8 +74,8 @@ func InitModelx(ctx context.Context, path string, force bool) error {
 		return fmt.Errorf("create modelx directory:%s %w", path, err)
 	}
 	config := ModelConfig{
-		Description: "This is a modelx model",
-		FrameWork:   "<some framework.Pytorch|TensorFlow|Caffe|MindSpore|etc>",
+		Description: "This is a modelx model.",
+		FrameWork:   "<Pytorch/TensorFlow/Caffe/MindSpore/etc>",
 		Config: map[string]interface{}{
 			"inputs":  map[string]interface{}{},
 			"outputs": map[string]interface{}{},
@@ -96,7 +96,9 @@ func InitModelx(ctx context.Context, path string, force bool) error {
 		Mantainers: []string{
 			"maintainer",
 		},
-		ModelFiles: []string{},
+		ModelFiles:   []string{},
+		Command:      []string{},
+		Dependencies: nil,
 	}
 	var configcontent bytes.Buffer
 	encoder := yaml.NewEncoder(&configcontent)