diff --git a/sdks/scheduler/modeljob.go b/sdks/scheduler/modeljob.go index b82915e..f9afeaf 100644 --- a/sdks/scheduler/modeljob.go +++ b/sdks/scheduler/modeljob.go @@ -86,6 +86,9 @@ const ( PauseECS = "pause" DestroyECS = "destroy" OperateServer = "operate" + + RcloneMount = "rclone" + Mounted = "mounted" ) type QueryRunningModelsReq struct { diff --git a/sdks/scheduler/models.go b/sdks/scheduler/models.go index ccb2ec6..35f37b6 100644 --- a/sdks/scheduler/models.go +++ b/sdks/scheduler/models.go @@ -257,8 +257,10 @@ func (b *NoEnvBootstrap) GetBootstrapType() string { } const ( - JobDataInEnv = "SCH_DATA_IN" - JobDataOutEnv = "SCH_DATA_OUT" + JobDataInEnv = "SCH_DATA_IN" + JobDataOutEnv = "SCH_DATA_OUT" + FinetuningOutEnv = "FINETUNING_OUT" + AccessPath = "ACCESS_PATH" ) type Rclone struct { diff --git a/utils/http/http.go b/utils/http/http.go index e8ffbbc..a3871f1 100644 --- a/utils/http/http.go +++ b/utils/http/http.go @@ -56,6 +56,27 @@ func GetJSON(url string, param RequestParam) (*http.Response, error) { return defaultClient.Do(req) } +func DeleteJSON(url string, param RequestParam) (*http.Response, error) { + req, err := http.NewRequest(http.MethodDelete, url, nil) + if err != nil { + return nil, err + } + + if err = prepareQuery(req, param.Query); err != nil { + return nil, err + } + + if err = prepareHeader(req, param.Header); err != nil { + return nil, err + } + + if err = prepareJSONBody(req, param.Body); err != nil { + return nil, err + } + + return defaultClient.Do(req) +} + func GetForm(url string, param RequestParam) (*http.Response, error) { req, err := http.NewRequest(http.MethodGet, url, nil) if err != nil {