From a8b5daeac9d366144e138eccebaffd50d13f0022 Mon Sep 17 00:00:00 2001 From: JeshuaRen <270813223@qq.com> Date: Wed, 11 Sep 2024 13:56:55 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E6=9B=99=E5=85=89=E8=8A=82?= =?UTF-8?q?=E7=82=B9=E5=88=9B=E5=BB=BA=E7=AD=89=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- sdks/scheduler/modeljob.go | 3 +++ sdks/scheduler/models.go | 6 ++++-- utils/http/http.go | 21 +++++++++++++++++++++ 3 files changed, 28 insertions(+), 2 deletions(-) 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 {