Browse Source

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

pull/52/head
JeshuaRen 6 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_" BlockChainJobCreatePrefix = "job_create_"


Finish = "Finish"
Complete = "Complete"


NodeTypeBinding = "binding" NodeTypeBinding = "binding"
NodeTypeUpload = "upload" NodeTypeUpload = "upload"


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

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


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


type CompleteJobInfo 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 JobInfoBase
Type string `json:"type"` Type string `json:"type"`
} }


Loading…
Cancel
Save