Browse Source

推理节点新增数据调度功能

pull/52/head
JeshuaRen 11 months ago
parent
commit
d01bb92ec9
2 changed files with 9 additions and 2 deletions
  1. +1
    -1
      sdks/pcmscheduler/models.go
  2. +8
    -1
      sdks/scheduler/models.go

+ 1
- 1
sdks/pcmscheduler/models.go View File

@@ -84,7 +84,7 @@ const (

BlockChainJobCreatePrefix = "job_create_"

Finish = "Finish"
Complete = "Complete"

NodeTypeBinding = "binding"
NodeTypeUpload = "upload"


+ 8
- 1
sdks/scheduler/models.go View File

@@ -71,6 +71,7 @@ var JobInfoTypeUnion = types.NewTypeUnion[JobInfo](
(*BindingJobInfo)(nil),
(*PCMInferenceJobInfo)(nil),
(*CompleteJobInfo)(nil),
(*StartJobInfo)(nil),
)
var _ = serder.UseTypeUnionInternallyTagged(&JobInfoTypeUnion, "type")

@@ -150,7 +151,13 @@ type AIJobInfo struct {
}

type CompleteJobInfo struct {
serder.Metadata `union:"Finish"`
serder.Metadata `union:"Complete"`
JobInfoBase
Type string `json:"type"`
}

type StartJobInfo struct {
serder.Metadata `union:"Start"`
JobInfoBase
Type string `json:"type"`
}


Loading…
Cancel
Save