Browse Source

新增曙光节点创建等功能

gitlink
JeshuaRen 1 year ago
parent
commit
a8b5daeac9
3 changed files with 28 additions and 2 deletions
  1. +3
    -0
      sdks/scheduler/modeljob.go
  2. +4
    -2
      sdks/scheduler/models.go
  3. +21
    -0
      utils/http/http.go

+ 3
- 0
sdks/scheduler/modeljob.go View File

@@ -86,6 +86,9 @@ const (
PauseECS = "pause"
DestroyECS = "destroy"
OperateServer = "operate"

RcloneMount = "rclone"
Mounted = "mounted"
)

type QueryRunningModelsReq struct {


+ 4
- 2
sdks/scheduler/models.go View File

@@ -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 {


+ 21
- 0
utils/http/http.go View File

@@ -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 {


Loading…
Cancel
Save